add clear env

useragent/cicd_test.go

@@ -1,6 +1,8 @@
 package useragent
 
 import (
+	"os"
+	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -26,7 +28,27 @@ func TestCiCdProviderDetect(t *testing.T) {
 	assert.True(t, cicdProvider.detect(), "should detect when all env vars are set")
 }
 
+func cleanupEnv(t *testing.T) {
+	// We use set dummy env here to ensure this function is not called from a
+	// test that runs in parallel.
+	t.Setenv("FOO_BAR", "")
+
+	environ := os.Environ()
+	t.Cleanup(func() {
+		// Restore original environment once test is done.
+		for _, kv := range environ {
+			kvs := strings.SplitN(kv, "=", 2)
+			os.Setenv(kvs[0], kvs[1])
+		}
+	})
+
+	// Clear out all existing environment variables.
+	os.Clearenv()
+}
+
 func TestCiCdProviderGithubActions(t *testing.T) {
+	cleanupEnv(t)
+
 	// No provider detected.
 	assert.Equal(t, "", CiCdProvider())
 
@@ -36,42 +58,56 @@ func TestCiCdProviderGithubActions(t *testing.T) {
 }
 
 func TestCiCdProviderGitlab(t *testing.T) {
+	cleanupEnv(t)
+
 	// Gitlab detected.
 	t.Setenv("GITLAB_CI", "true")
 	assert.Equal(t, "gitlab", CiCdProvider())
 }
 
 func TestCiCdProviderJenkins(t *testing.T) {
+	cleanupEnv(t)
+
 	// Jenkins detected.
 	t.Setenv("JENKINS_URL", "https://jenkins.example.com")
 	assert.Equal(t, "jenkins", CiCdProvider())
 }
 
 func TestCiCdProviderAzureDevops(t *testing.T) {
+	cleanupEnv(t)
+
 	// Azure Devops detected.
 	t.Setenv("TF_BUILD", "True")
 	assert.Equal(t, "azure-devops", CiCdProvider())
 }
 
 func TestCiCdProviderCircle(t *testing.T) {
+	cleanupEnv(t)
+
 	// Circle detected.
 	t.Setenv("CIRCLECI", "true")
 	assert.Equal(t, "circle", CiCdProvider())
 }
 
 func TestCiCdProviderTravis(t *testing.T) {
+	cleanupEnv(t)
+
 	// Travis detected.
 	t.Setenv("TRAVIS", "true")
 	assert.Equal(t, "travis", CiCdProvider())
 }
 
 func TestCiCdProviderBitbucket(t *testing.T) {
+	cleanupEnv(t)
+
 	// Bitbucket detected.
 	t.Setenv("BITBUCKET_BUILD_NUMBER", "123")
 	assert.Equal(t, "bitbucket", CiCdProvider())
 }
 
 func TestCiCdProviderGoogleCloudBuild(t *testing.T) {
+	cleanupEnv(t)
+
 	// Google Cloud Build detected.
 	t.Setenv("PROJECT_ID", "foo")
 	t.Setenv("BUILD_ID", "bar")
@@ -81,18 +117,24 @@ func TestCiCdProviderGoogleCloudBuild(t *testing.T) {
 }
 
 func TestCiCdProviderAwsCodeBuild(t *testing.T) {
+	cleanupEnv(t)
+
 	// AWS Code Build detected.
 	t.Setenv("CODEBUILD_BUILD_ARN", "arn:aws:codebuild:us-east-1:123456789012:build/my-demo-project:b1e6deae-e4f2-4151-be79-3cc4e82a0bf0")
 	assert.Equal(t, "aws-code-build", CiCdProvider())
 }
 
 func TestCiCdProviderTfCloud(t *testing.T) {
+	cleanupEnv(t)
+
 	// Terraform Cloud detected.
 	t.Setenv("TFC_RUN_ID", "run-123")
 	assert.Equal(t, "tf-cloud", CiCdProvider())
 }
 
 func TestCiCdProviderMultiple(t *testing.T) {
+	cleanupEnv(t)
+
 	// Multiple providers detected. The first one detected is set.
 	t.Setenv("GITHUB_ACTIONS", "true")
 	t.Setenv("GITLAB_CI", "true")