package agent import ( "io" "github.com/hashicorp/serf/serf" "github.com/stretchr/testify/mock" "github.com/hashicorp/consul/acl" "github.com/hashicorp/consul/agent/consul" "github.com/hashicorp/consul/agent/structs" "github.com/hashicorp/consul/lib" ) type delegateMock struct { mock.Mock } func (m *delegateMock) GetLANCoordinate() (lib.CoordinateSet, error) { ret := m.Called() return ret.Get(0).(lib.CoordinateSet), ret.Error(1) } func (m *delegateMock) Leave() error { return m.Called().Error(0) } func (m *delegateMock) LANMembers() []serf.Member { return m.Called().Get(0).([]serf.Member) } func (m *delegateMock) LANMembersAllSegments() ([]serf.Member, error) { ret := m.Called() return ret.Get(0).([]serf.Member), ret.Error(1) } func (m *delegateMock) LANSegmentMembers(segment string) ([]serf.Member, error) { ret := m.Called() return ret.Get(0).([]serf.Member), ret.Error(1) } func (m *delegateMock) LocalMember() serf.Member { return m.Called().Get(0).(serf.Member) } func (m *delegateMock) JoinLAN(addrs []string) (n int, err error) { ret := m.Called(addrs) return ret.Int(0), ret.Error(1) } func (m *delegateMock) RemoveFailedNode(node string, prune bool) error { return m.Called(node, prune).Error(0) } func (m *delegateMock) ResolveTokenToIdentity(token string) (structs.ACLIdentity, error) { ret := m.Called(token) return ret.Get(0).(structs.ACLIdentity), ret.Error(1) } func (m *delegateMock) ResolveTokenAndDefaultMeta(token string, entMeta *structs.EnterpriseMeta, authzContext *acl.AuthorizerContext) (acl.Authorizer, error) { ret := m.Called(token, entMeta, authzContext) return ret.Get(0).(acl.Authorizer), ret.Error(1) } func (m *delegateMock) RPC(method string, args interface{}, reply interface{}) error { return m.Called(method, args, reply).Error(0) } func (m *delegateMock) SnapshotRPC(args *structs.SnapshotRequest, in io.Reader, out io.Writer, replyFn structs.SnapshotReplyFn) error { return m.Called(args, in, out, replyFn).Error(0) } func (m *delegateMock) Shutdown() error { return m.Called().Error(0) } func (m *delegateMock) Stats() map[string]map[string]string { return m.Called().Get(0).(map[string]map[string]string) } func (m *delegateMock) ReloadConfig(config consul.ReloadableConfig) error { return m.Called(config).Error(0) }