1/*
2Copyright 2016 The Kubernetes Authors.
3
4Licensed under the Apache License, Version 2.0 (the "License");
5you may not use this file except in compliance with the License.
6You may obtain a copy of the License at
7
8    http://www.apache.org/licenses/LICENSE-2.0
9
10Unless required by applicable law or agreed to in writing, software
11distributed under the License is distributed on an "AS IS" BASIS,
12WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13See the License for the specific language governing permissions and
14limitations under the License.
15*/
16
17package cache
18
19// FakeStore lets you define custom functions for store operations
20type FakeCustomStore struct {
21	AddFunc      func(obj interface{}) error
22	UpdateFunc   func(obj interface{}) error
23	DeleteFunc   func(obj interface{}) error
24	ListFunc     func() []interface{}
25	ListKeysFunc func() []string
26	GetFunc      func(obj interface{}) (item interface{}, exists bool, err error)
27	GetByKeyFunc func(key string) (item interface{}, exists bool, err error)
28	ReplaceFunc  func(list []interface{}, resourceVerion string) error
29	ResyncFunc   func() error
30}
31
32// Add calls the custom Add function if defined
33func (f *FakeCustomStore) Add(obj interface{}) error {
34	if f.AddFunc != nil {
35		return f.AddFunc(obj)
36	}
37	return nil
38}
39
40// Update calls the custom Update function if defined
41func (f *FakeCustomStore) Update(obj interface{}) error {
42	if f.UpdateFunc != nil {
43		return f.UpdateFunc(obj)
44	}
45	return nil
46}
47
48// Delete calls the custom Delete function if defined
49func (f *FakeCustomStore) Delete(obj interface{}) error {
50	if f.DeleteFunc != nil {
51		return f.DeleteFunc(obj)
52	}
53	return nil
54}
55
56// List calls the custom List function if defined
57func (f *FakeCustomStore) List() []interface{} {
58	if f.ListFunc != nil {
59		return f.ListFunc()
60	}
61	return nil
62}
63
64// ListKeys calls the custom ListKeys function if defined
65func (f *FakeCustomStore) ListKeys() []string {
66	if f.ListKeysFunc != nil {
67		return f.ListKeysFunc()
68	}
69	return nil
70}
71
72// Get calls the custom Get function if defined
73func (f *FakeCustomStore) Get(obj interface{}) (item interface{}, exists bool, err error) {
74	if f.GetFunc != nil {
75		return f.GetFunc(obj)
76	}
77	return nil, false, nil
78}
79
80// GetByKey calls the custom GetByKey function if defined
81func (f *FakeCustomStore) GetByKey(key string) (item interface{}, exists bool, err error) {
82	if f.GetByKeyFunc != nil {
83		return f.GetByKeyFunc(key)
84	}
85	return nil, false, nil
86}
87
88// Replace calls the custom Replace function if defined
89func (f *FakeCustomStore) Replace(list []interface{}, resourceVersion string) error {
90	if f.ReplaceFunc != nil {
91		return f.ReplaceFunc(list, resourceVersion)
92	}
93	return nil
94}
95
96// Resync calls the custom Resync function if defined
97func (f *FakeCustomStore) Resync() error {
98	if f.ResyncFunc != nil {
99		return f.ResyncFunc()
100	}
101	return nil
102}
103