From 4febe03c3261e6b5c8cad29cdef698cf73f2d5c0 Mon Sep 17 00:00:00 2001 From: achettyiitr Date: Thu, 17 Oct 2024 06:30:10 +0530 Subject: [PATCH] chore: add support for withenv for traksformer resource --- .../docker/resource/transformer/transformer.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/testhelper/docker/resource/transformer/transformer.go b/testhelper/docker/resource/transformer/transformer.go index e4e27af3..4f19a493 100644 --- a/testhelper/docker/resource/transformer/transformer.go +++ b/testhelper/docker/resource/transformer/transformer.go @@ -95,6 +95,18 @@ func WithDockerNetwork(network *docker.Network) func(*config) { } } +func WithEnv(env string) func(*config) { + return func(conf *config) { + conf.envs = append(conf.envs, env) + } +} + +func WithRepository(repository string) func(*config) { + return func(conf *config) { + conf.repository = repository + } +} + 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,