From 7c1ef8f5159495dc7bfc4ad418a9b2aa62025c1a Mon Sep 17 00:00:00 2001
From: Matt Keeler <mjkeeler7@gmail.com>
Date: Wed, 22 Sep 2021 09:24:30 -0400
Subject: [PATCH] Add a mock Agent delegate to ease/improve some types of
 testing

---
 agent/delegate_mock_test.go | 86 +++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 agent/delegate_mock_test.go

diff --git a/agent/delegate_mock_test.go b/agent/delegate_mock_test.go
new file mode 100644
index 000000000..208313aed
--- /dev/null
+++ b/agent/delegate_mock_test.go
@@ -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)
+}