From 3c98aafd6f08ba5e00200cb86dbe0a02ae4cdfec Mon Sep 17 00:00:00 2001
From: Mihir Gandhi <mihir20121997@gmail.com>
Date: Thu, 15 Feb 2024 12:51:23 +0530
Subject: [PATCH] chore: add docker resource for transformer (#334)

---
 .../resource/transformer/transformer.go       | 100 ++++++++++++++++++
 .../resource/transformer/transformer_test.go  |  60 +++++++++++
 2 files changed, 160 insertions(+)
 create mode 100644 testhelper/docker/resource/transformer/transformer.go
 create mode 100644 testhelper/docker/resource/transformer/transformer_test.go

diff --git a/testhelper/docker/resource/transformer/transformer.go b/testhelper/docker/resource/transformer/transformer.go
new file mode 100644
index 00000000..e5a78471
--- /dev/null
+++ b/testhelper/docker/resource/transformer/transformer.go
@@ -0,0 +1,100 @@
+package transformer
+
+import (
+	"errors"
+	"fmt"
+	"net/http"
+
+	"github.com/ory/dockertest/v3"
+	"github.com/ory/dockertest/v3/docker"
+
+	"github.com/rudderlabs/rudder-go-kit/httputil"
+	"github.com/rudderlabs/rudder-go-kit/testhelper/docker/resource"
+)
+
+type Resource struct {
+	TransformerURL string
+	Port           string
+}
+
+type config struct {
+	repository   string
+	tag          string
+	exposedPorts []string
+	env          []string
+}
+
+func WithConfigBackendURL(url string) func(*config) {
+	return func(conf *config) {
+		conf.env = []string{fmt.Sprintf("CONFIG_BACKEND_URL=%s", url)}
+	}
+}
+
+func WithDockerImageTag(tag string) func(*config) {
+	return func(conf *config) {
+		conf.tag = tag
+	}
+}
+
+func Setup(pool *dockertest.Pool, d resource.Cleaner, opts ...func(conf *config)) (*Resource, error) {
+	// Set Rudder Transformer
+	// pulls an image first to make sure we don't have an old cached version locally,
+	// then it creates a container based on it and runs it
+	conf := &config{
+		repository:   "rudderstack/rudder-transformer",
+		tag:          "latest",
+		exposedPorts: []string{"9090"},
+		env: []string{
+			"CONFIG_BACKEND_URL=https://api.rudderstack.com",
+		},
+	}
+
+	for _, opt := range opts {
+		opt(conf)
+	}
+
+	err := pool.Client.PullImage(docker.PullImageOptions{
+		Repository: conf.repository,
+		Tag:        conf.tag,
+	}, docker.AuthConfiguration{})
+	if err != nil {
+		return nil, fmt.Errorf("failed to pull image: %w", err)
+	}
+	transformerContainer, err := pool.RunWithOptions(&dockertest.RunOptions{
+		Repository:   conf.repository,
+		Tag:          conf.tag,
+		ExposedPorts: conf.exposedPorts,
+		Env:          conf.env,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	d.Cleanup(func() {
+		if err := pool.Purge(transformerContainer); err != nil {
+			d.Log("Could not purge resource:", err)
+		}
+	})
+
+	transformerResource := &Resource{
+		TransformerURL: fmt.Sprintf("http://localhost:%s", transformerContainer.GetPort("9090/tcp")),
+		Port:           transformerContainer.GetPort("9090/tcp"),
+	}
+
+	err = pool.Retry(func() (err error) {
+		resp, err := http.Get(transformerResource.TransformerURL + "/health")
+		if err != nil {
+			return err
+		}
+		defer func() { httputil.CloseResponse(resp) }()
+		if resp.StatusCode != http.StatusOK {
+			return errors.New(resp.Status)
+		}
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return transformerResource, nil
+}
diff --git a/testhelper/docker/resource/transformer/transformer_test.go b/testhelper/docker/resource/transformer/transformer_test.go
new file mode 100644
index 00000000..bd9d1b74
--- /dev/null
+++ b/testhelper/docker/resource/transformer/transformer_test.go
@@ -0,0 +1,60 @@
+package transformer_test
+
+import (
+	"net/http"
+	"net/url"
+	"testing"
+
+	"github.com/ory/dockertest/v3"
+	"github.com/stretchr/testify/require"
+
+	"github.com/rudderlabs/rudder-go-kit/testhelper/docker/resource/transformer"
+)
+
+func TestSetup(t *testing.T) {
+	pool, err := dockertest.NewPool("")
+	require.NoError(t, err)
+
+	t.Run("check get endpoints", func(t *testing.T) {
+		tests := []struct {
+			name    string
+			tag     string
+			urlPath string
+		}{
+			{
+				name:    "transformer - health",
+				tag:     "latest",
+				urlPath: "health",
+			},
+			{
+				name:    "user transformer - health",
+				tag:     "ut-latest",
+				urlPath: "health",
+			},
+			{
+				name:    "transformer - features",
+				tag:     "latest",
+				urlPath: "features",
+			},
+			{
+				name:    "user transformer - features",
+				tag:     "ut-latest",
+				urlPath: "features",
+			},
+		}
+		for _, tt := range tests {
+			t.Run(tt.name, func(t *testing.T) {
+				transformerContainer, err := transformer.Setup(pool, t,
+					transformer.WithConfigBackendURL("random-url"),
+					transformer.WithDockerImageTag(tt.tag))
+				require.NoError(t, err)
+				endpoint, err := url.JoinPath(transformerContainer.TransformerURL, tt.urlPath)
+				require.NoError(t, err)
+				resp, err := http.Get(endpoint)
+				require.NoError(t, err)
+				defer func() { _ = resp.Body.Close() }()
+				require.Equal(t, resp.StatusCode, http.StatusOK)
+			})
+		}
+	})
+}