adds ConsulAgent interface to allow for consul testing

Author: ncode

pkg/tagit/tagit.go

@@ -25,7 +25,13 @@ type TagIt struct {
 
 // ConsulClient is an interface for the Consul client.
 type ConsulClient interface {
-	Agent() *api.Agent
+	Agent() ConsulAgent
+}
+
+// ConsulClientAgent is an interface for the Consul agent.
+type ConsulAgent interface {
+	Services() (map[string]*api.AgentService, error)
+	ServiceRegister(*api.AgentServiceRegistration) error
 }
 
 // CommandExecutor is an interface for running commands.

pkg/tagit/tagit_test.go

@@ -290,13 +290,26 @@ func TestRunScript(t *testing.T) {
 	}
 }
 
-// MockConsulClient is a mock implementation of the ConsulClient interface
-type MockConsulClient struct{}
+// MockConsulClient implements the ConsulClient interface for testing.
+type MockConsulClient struct {
+	MockAgent *MockAgent
+}
+
+func (m *MockConsulClient) Agent() ConsulAgent {
+	return m.MockAgent
+}
+
+// MockAgent simulates the Agent part of the Consul client.
+type MockAgent struct {
+	ServicesFunc func() (map[string]*api.AgentService, error)
+}
+
+func (m *MockAgent) Services() (map[string]*api.AgentService, error) {
+	return m.ServicesFunc()
+}
 
-// TODO: Implement the Agent method properly so that we can test the rest of the methods
-func (m *MockConsulClient) Agent() *api.Agent {
-	// Return a mock *api.Agent if needed
-	return &api.Agent{}
+func (m *MockAgent) ServiceRegister(reg *api.AgentServiceRegistration) error {
+	return nil
 }
 
 func TestNew(t *testing.T) {