Add CWService interface and MockCWService for testing

aws/cwatch/mock/mock.go

@@ -0,0 +1,45 @@
+package mock
+
+import (
+	"sync"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/service/cloudwatch/types"
+	"github.com/nyaruka/gocommon/aws/cwatch"
+)
+
+type MockCWService struct {
+	namespace  string
+	deployment types.Dimension
+	Batcher    []types.MetricDatum
+
+	Stopped bool
+}
+
+func NewMockCWService(accessKey, secretKey, region, namespace, deployment string) (*MockCWService, error) {
+	mockCW := MockCWService{
+		namespace:  namespace,
+		deployment: types.Dimension{Name: aws.String("Deployment"), Value: aws.String(deployment)},
+		Batcher:    nil,
+	}
+
+	return &mockCW, nil
+}
+
+func (s *MockCWService) Queue(d types.MetricDatum) {
+	if s.Stopped {
+		return
+	}
+	s.Batcher = append(s.Batcher, d)
+}
+
+func (s *MockCWService) StartQueue(wg *sync.WaitGroup) {
+	s.Batcher = []types.MetricDatum{}
+	s.Stopped = false
+}
+
+func (s *MockCWService) StopQueue() {
+	s.Stopped = true
+}
+
+var _ cwatch.CWService = (*MockCWService)(nil)

aws/cwatch/service.go

@@ -71,3 +71,11 @@ func (s *Service) processBatch(batch []types.MetricDatum) {
 		slog.Error("error sending metrics to cloudwatch", "error", err, "count", len(batch))
 	}
 }
+
+type CWService interface {
+	Queue(d types.MetricDatum)
+	StartQueue(wg *sync.WaitGroup)
+	StopQueue()
+}
+
+var _ CWService = (*Service)(nil)