adds ConsulAPIWrapper

Author: ncode

cmd/run.go

@@ -70,7 +70,7 @@ example: tagit run -s my-super-service -x '/tmp/tag-role.sh'
 			os.Exit(1)
 		}
 
-		t := tagit.New(consulClient, &tagit.CmdExecutor{}, serviceID, script, validInterval, tagPrefix)
+		t := tagit.New(tagit.NewConsulAPIWrapper(consulClient), &tagit.CmdExecutor{}, serviceID, script, validInterval, tagPrefix)
 		t.Run()
 	},
 }

pkg/tagit/tagit.go

@@ -34,6 +34,21 @@ type ConsulAgent interface {
 	ServiceRegister(*api.AgentServiceRegistration) error
 }
 
+// ConsulAPIWrapper wraps the Consul API client to conform to the ConsulClient interface.
+type ConsulAPIWrapper struct {
+	client *api.Client
+}
+
+// NewConsulAPIWrapper creates a new instance of ConsulAPIWrapper.
+func NewConsulAPIWrapper(client *api.Client) *ConsulAPIWrapper {
+	return &ConsulAPIWrapper{client: client}
+}
+
+// Agent returns an object that conforms to the ConsulAgent interface.
+func (w *ConsulAPIWrapper) Agent() ConsulAgent {
+	return w.client.Agent()
+}
+
 // CommandExecutor is an interface for running commands.
 type CommandExecutor interface {
 	Execute(command string) ([]byte, error)