1
//go:build !no_mocks
2
 
3
package mocks
4
 
5
import (
6
	"context"
7
	"time"
8
 
9
	"go.miragespace.co/specter/spec/chord"
10
	"go.miragespace.co/specter/spec/protocol"
11
 
12
	"github.com/stretchr/testify/mock"
13
)
14
 
15
type VNode struct {
16
	mock.Mock
17
}
18
 
19
var _ chord.VNode = (*VNode)(nil)
20
 
21
func (n *VNode) Put(ctx context.Context, key []byte, value []byte) error {
22
	args := n.Called(ctx, key, value)
23
	return args.Error(0)
24
}
25
 
26
func (n *VNode) Get(ctx context.Context, key []byte) (value []byte, err error) {
27
	args := n.Called(ctx, key)
28
	v := args.Get(0)
29
	e := args.Error(1)
30
	if v == nil {
31
		return nil, e
32
	}
33
	return v.([]byte), e
34
}
35
 
36
func (n *VNode) Delete(ctx context.Context, key []byte) error {
37
	args := n.Called(ctx, key)
38
	return args.Error(0)
39
}
40
 
41
func (n *VNode) PrefixAppend(ctx context.Context, prefix []byte, child []byte) error {
42
	args := n.Called(ctx, prefix, child)
43
	e := args.Error(0)
44
	return e
45
}
46
 
47
func (n *VNode) PrefixList(ctx context.Context, prefix []byte) ([][]byte, error) {
48
	args := n.Called(ctx, prefix)
49
	v := args.Get(0)
50
	e := args.Error(1)
51
	return v.([][]byte), e
52
}
53
 
54
func (n *VNode) PrefixContains(ctx context.Context, prefix []byte, child []byte) (bool, error) {
55
	args := n.Called(ctx, prefix, child)
56
	v := args.Bool(0)
57
	e := args.Error(1)
58
	return v, e
59
}
60
 
61
func (n *VNode) PrefixRemove(ctx context.Context, prefix []byte, child []byte) error {
62
	args := n.Called(ctx, prefix, child)
63
	e := args.Error(0)
64
	return e
65
}
66
 
67
func (n *VNode) Acquire(ctx context.Context, lease []byte, ttl time.Duration) (token uint64, err error) {
68
	args := n.Called(ctx, lease, ttl)
69
	t := args.Get(0)
70
	e := args.Error(1)
71
	return t.(uint64), e
72
}
73
 
74
func (n *VNode) Renew(ctx context.Context, lease []byte, ttl time.Duration, prevToken uint64) (newToken uint64, err error) {
75
	args := n.Called(ctx, lease, ttl, prevToken)
76
	t := args.Get(0)
77
	e := args.Error(1)
78
	return t.(uint64), e
79
}
80
 
81
func (n *VNode) Release(ctx context.Context, lease []byte, token uint64) error {
82
	args := n.Called(ctx, lease, token)
83
	e := args.Error(0)
84
	return e
85
}
86
 
87
func (n *VNode) Import(ctx context.Context, keys [][]byte, values []*protocol.KVTransfer) error {
88
	args := n.Called(ctx, keys, values)
89
	e := args.Error(0)
90
	return e
91
}
92
 
93
func (n *VNode) ListKeys(ctx context.Context, prefix []byte) ([]*protocol.KeyComposite, error) {
94
	args := n.Called(ctx, prefix)
95
	t := args.Get(0)
96
	e := args.Error(1)
97
	return t.([]*protocol.KeyComposite), e
98
}
99
 
100
func (n *VNode) ID() uint64 {
101
	args := n.Called()
102
	v := args.Get(0)
103
	return v.(uint64)
104
}
105
 
106
func (n *VNode) Identity() *protocol.Node {
107
	args := n.Called()
108
	v := args.Get(0)
109
	if v == nil {
110
		return nil
111
	}
112
	return v.(*protocol.Node)
113
}
114
 
115
func (n *VNode) Ping() error {
116
	args := n.Called()
117
	e := args.Error(0)
118
	return e
119
}
120
 
121
func (n *VNode) Notify(predecessor chord.VNode) error {
122
	args := n.Called(predecessor)
123
	return args.Error(0)
124
}
125
 
126
func (n *VNode) FindSuccessor(key uint64) (chord.VNode, error) {
127
	args := n.Called(key)
128
	v := args.Get(0)
129
	e := args.Error(1)
130
	if v == nil {
131
		return nil, e
132
	}
133
	return v.(chord.VNode), e
134
}
135
 
136
func (n *VNode) GetSuccessors() ([]chord.VNode, error) {
137
	args := n.Called()
138
	v := args.Get(0)
139
	e := args.Error(1)
140
	if v == nil {
141
		return nil, e
142
	}
143
	return v.([]chord.VNode), e
144
}
145
 
146
func (n *VNode) GetPredecessor() (chord.VNode, error) {
147
	args := n.Called()
148
	v := args.Get(0)
149
	e := args.Error(1)
150
	if v == nil {
151
		return nil, e
152
	}
153
	return v.(chord.VNode), e
154
}
155
 
156
func (n *VNode) RequestToJoin(joiner chord.VNode) (chord.VNode, []chord.VNode, error) {
157
	args := n.Called(joiner)
158
	p := args.Get(0)
159
	s := args.Get(1)
160
	e := args.Error(2)
161
	if e != nil {
162
		return nil, nil, e
163
	}
164
	return p.(chord.VNode), s.([]chord.VNode), e
165
}
166
 
167
func (n *VNode) FinishJoin(stabilize bool, release bool) error {
168
	args := n.Called()
169
	e := args.Error(0)
170
	return e
171
}
172
 
173
func (n *VNode) RequestToLeave(leaver chord.VNode) error {
174
	args := n.Called(leaver)
175
	e := args.Error(0)
176
	return e
177
}
178
 
179
func (n *VNode) FinishLeave(stabilize bool, release bool) error {
180
	args := n.Called()
181
	e := args.Error(0)
182
	return e
183
}