Add a mock Agent delegate to ease/improve some types of testing
This commit is contained in:
parent
320b20c708
commit
7c1ef8f515
|
@ -0,0 +1,86 @@
|
|||
package agent
|
||||
|
||||
import (
|
||||
"io"
|
||||
|
||||
"github.com/hashicorp/consul/acl"
|
||||
"github.com/hashicorp/consul/agent/consul"
|
||||
"github.com/hashicorp/consul/agent/structs"
|
||||
"github.com/hashicorp/consul/lib"
|
||||
"github.com/hashicorp/serf/serf"
|
||||
"github.com/stretchr/testify/mock"
|
||||
)
|
||||
|
||||
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) UseLegacyACLs() bool {
|
||||
return m.Called().Bool(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)
|
||||
}
|
Loading…
Reference in New Issue