fix: use enterprise token from env as fallback (#2667)

rudderlabs · Nov 11, 2022 · 2d8f4ba · 2d8f4ba
1 parent 17be4fa
commit 2d8f4ba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/integration_test/docker_test/docker_test.go b/integration_test/docker_test/docker_test.go
@@ -475,7 +475,7 @@ func setupMainFlow(svcCtx context.Context, t *testing.T) <-chan struct{} {
 
 	svcDone := make(chan struct{})
 	go func() {
-		r := runner.New(runner.ReleaseInfo{})
+		r := runner.New(runner.ReleaseInfo{EnterpriseToken: os.Getenv("ENTERPRISE_TOKEN")})
 		_ = r.Run(svcCtx, []string{"docker-test-rudder-server"})
 		close(svcDone)
 	}()

diff --git a/main.go b/main.go
@@ -8,6 +8,7 @@ import (
 
 	_ "go.uber.org/automaxprocs"
 
+	"github.com/rudderlabs/rudder-server/config"
 	"github.com/rudderlabs/rudder-server/runner"
 )
 
@@ -27,7 +28,7 @@ func main() {
 		BuildDate:       buildDate,
 		BuiltBy:         builtBy,
 		GitURL:          gitURL,
-		EnterpriseToken: enterpriseToken,
+		EnterpriseToken: config.GetString("ENTERPRISE_TOKEN", enterpriseToken),
 	})
 	exitCode := r.Run(ctx, os.Args)
 	cancel()