forked from sensu/sensu-go
/
mock_cluster_client.go
33 lines (27 loc) · 1019 Bytes
/
mock_cluster_client.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
package testing
import "github.com/coreos/etcd/clientv3"
// MemberList ...
func (c *MockClient) MemberList() (*clientv3.MemberListResponse, error) {
args := c.Called()
return args.Get(0).(*clientv3.MemberListResponse), args.Error(1)
}
// MemberAdd ...
func (c *MockClient) MemberAdd(peerAddrs []string) (*clientv3.MemberAddResponse, error) {
args := c.Called(peerAddrs)
return args.Get(0).(*clientv3.MemberAddResponse), args.Error(1)
}
// MemberUpdate ...
func (c *MockClient) MemberUpdate(id uint64, peerAddrs []string) (*clientv3.MemberUpdateResponse, error) {
args := c.Called(id, peerAddrs)
return args.Get(0).(*clientv3.MemberUpdateResponse), args.Error(1)
}
// MemberRemove ...
func (c *MockClient) MemberRemove(id uint64) (*clientv3.MemberRemoveResponse, error) {
args := c.Called(id)
return args.Get(0).(*clientv3.MemberRemoveResponse), args.Error(1)
}
// FetchClusterID ...
func (c *MockClient) FetchClusterID() (string, error) {
args := c.Called()
return args.Get(0).(string), args.Error(1)
}