open-consul/agent/delegate_mock_test.go

76 lines
2.1 KiB
Go

package agent
import (
"context"
"io"
"github.com/hashicorp/serf/serf"
"github.com/stretchr/testify/mock"
"github.com/hashicorp/consul/acl"
"github.com/hashicorp/consul/acl/resolver"
"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) LANMembersInAgentPartition() []serf.Member {
return m.Called().Get(0).([]serf.Member)
}
func (m *delegateMock) LANMembers(f consul.LANMemberFilter) ([]serf.Member, error) {
ret := m.Called(f)
return ret.Get(0).([]serf.Member), ret.Error(1)
}
func (m *delegateMock) AgentLocalMember() serf.Member {
return m.Called().Get(0).(serf.Member)
}
func (m *delegateMock) JoinLAN(addrs []string, entMeta *acl.EnterpriseMeta) (n int, err error) {
ret := m.Called(addrs, entMeta)
return ret.Int(0), ret.Error(1)
}
func (m *delegateMock) RemoveFailedNode(node string, prune bool, entMeta *acl.EnterpriseMeta) error {
return m.Called(node, prune, entMeta).Error(0)
}
func (m *delegateMock) ResolveTokenAndDefaultMeta(token string, entMeta *acl.EnterpriseMeta, authzContext *acl.AuthorizerContext) (resolver.Result, error) {
ret := m.Called(token, entMeta, authzContext)
return ret.Get(0).(resolver.Result), ret.Error(1)
}
func (m *delegateMock) RPC(ctx context.Context, 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)
}