forked from etcd-io/etcd
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mock_store.go
67 lines (53 loc) · 2.22 KB
/
mock_store.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package mock
import (
"time"
"github.com/coreos/etcd/store"
"github.com/stretchr/testify/mock"
)
// A mock Store object used for testing.
type Store struct {
mock.Mock
}
func NewStore() *Store {
return &Store{}
}
func (s *Store) Get(nodePath string, recursive, sorted bool, index uint64, term uint64) (*store.Event, error) {
args := s.Called(nodePath, recursive, sorted, index, term)
return args.Get(0).(*store.Event), args.Error(1)
}
func (s *Store) Set(nodePath string, value string, expireTime time.Time, index uint64, term uint64) (*store.Event, error) {
args := s.Called(nodePath, value, expireTime, index, term)
return args.Get(0).(*store.Event), args.Error(1)
}
func (s *Store) Update(nodePath string, newValue string, expireTime time.Time, index uint64, term uint64) (*store.Event, error) {
args := s.Called(nodePath, newValue, expireTime, index, term)
return args.Get(0).(*store.Event), args.Error(1)
}
func (s *Store) Create(nodePath string, value string, incrementalSuffix bool, expireTime time.Time, index uint64, term uint64) (*store.Event, error) {
args := s.Called(nodePath, value, incrementalSuffix, expireTime, index, term)
return args.Get(0).(*store.Event), args.Error(1)
}
func (s *Store) CompareAndSwap(nodePath string, prevValue string, prevIndex uint64, value string, expireTime time.Time, index uint64, term uint64) (*store.Event, error) {
args := s.Called(nodePath, prevValue, prevIndex, value, expireTime, index, term)
return args.Get(0).(*store.Event), args.Error(1)
}
func (s *Store) Delete(nodePath string, recursive bool, index uint64, term uint64) (*store.Event, error) {
args := s.Called(nodePath, recursive, index, term)
return args.Get(0).(*store.Event), args.Error(1)
}
func (s *Store) Watch(prefix string, recursive bool, sinceIndex uint64, index uint64, term uint64) (<-chan *store.Event, error) {
args := s.Called(prefix, recursive, sinceIndex, index, term)
return args.Get(0).(<-chan *store.Event), args.Error(1)
}
func (s *Store) Save() ([]byte, error) {
args := s.Called()
return args.Get(0).([]byte), args.Error(1)
}
func (s *Store) Recovery(b []byte) error {
args := s.Called(b)
return args.Error(1)
}
func (s *Store) JsonStats() []byte {
args := s.Called()
return args.Get(0).([]byte)
}