diff --git a/main.go b/main.go index b472780a..aef0d002 100644 --- a/main.go +++ b/main.go @@ -255,6 +255,7 @@ func validateParams() { // Allow environment variables to overwrite args awsAccountIDEnv := os.Getenv("awsaccount") awsRegionEnv := os.Getenv("awsregion") + gcrURLEnv := os.Getenv("gcrurl") if len(awsRegionEnv) > 0 { argAWSRegion = &awsRegionEnv @@ -263,6 +264,10 @@ func validateParams() { if len(awsAccountIDEnv) > 0 { awsAccountID = awsAccountIDEnv } + + if len(gcrURLEnv) > 0 { + argGCRURL = &gcrURLEnv + } } func handler(c *controller, ns *v1.Namespace) error { diff --git a/main_test.go b/main_test.go index 3085d7fa..a6481771 100644 --- a/main_test.go +++ b/main_test.go @@ -633,6 +633,15 @@ func TestAwsRegionFromEnv(t *testing.T) { assert.Equal(t, expectedRegion, *argAWSRegion) } +func TestGcrURLFromEnv(t *testing.T) { + expectedURL := "http://test.me" + + os.Setenv("gcrurl", "http://test.me") + validateParams() + + assert.Equal(t, expectedURL, *argGCRURL) +} + func TestFailingGcrPassingEcrStillSucceeds(t *testing.T) { util := newKubeUtil() ecrClient := newFakeEcrClient()