fix(ci): restore or remove env var from tests

Author: tembleking

sysdig/resource_sysdig_secure_accept_vulnerability_risk_test.go

@@ -12,18 +12,28 @@ import (
 )
 
 func TestAcceptSecureVulnerabilityRisk(t *testing.T) {
-	resource.ParallelTest(t, resource.TestCase{
+	// Temporal hack to validate the resource in the CI.
+	// The endpoint https://secure.sysdig.com cannot be used for this, and is the default if the env var is not provided.
+	// It works in all other regions but US1, in US1 you need to specify https://api.us1.sysdig.com. The fix is on it's way.
+	// We only need this for the tests to pass. A workaround for this bug can be applied by users by specifying a new alias
+	// provider and specifying the concrete url for this alias provider.
+	url, urlSet := os.LookupEnv("SYSDIG_SECURE_URL")
+	if url == "https://secure.sysdig.com" || url == "" {
+		os.Setenv("SYSDIG_SECURE_URL", "https://api.us1.sysdig.com")
+	}
+
+	defer func() {
+		if urlSet {
+			os.Setenv("SYSDIG_SECURE_URL", url)
+		} else {
+			os.Unsetenv("SYSDIG_SECURE_URL")
+		}
+	}()
+
+	resource.Test(t, resource.TestCase{
 		PreCheck: preCheckAnyEnv(t, SysdigSecureApiTokenEnv),
 		ProviderFactories: map[string]func() (*schema.Provider, error){
 			"sysdig": func() (*schema.Provider, error) {
-				// Temporal hack to validate the resource in the CI.
-				// The endpoint https://secure.sysdig.com cannot be used for this, and is the default if the env var is not provided.
-				// It works in all other regions but US1, in US1 you need to specify https://api.us1.sysdig.com. The fix is on it's way.
-				// We only need this for the tests to pass. A workaround for this bug can be applied by users by specifying a new alias
-				// provider and specifying the concrete url for this alias provider.
-				if url := os.Getenv("SYSDIG_SECURE_URL"); url == "https://secure.sysdig.com" || url == "" {
-					os.Setenv("SYSDIG_SECURE_URL", "https://api.us1.sysdig.com")
-				}
 				return sysdig.Provider(), nil
 			},
 		},