fix

Author: vandyliu

internal/provider/provider_test.go

@@ -77,7 +77,8 @@ func explicitHostConfig() string {
 provider "astronomer" {
 organization_id = "%v"
 host = "%v"
-}`, os.Getenv("HYBRID_ORGANIZATION_ID"), os.Getenv("ASTRO_API_HOST")) + dataSourceConfig()
+token = "%v"
+}`, os.Getenv("HYBRID_ORGANIZATION_ID"), os.Getenv("ASTRO_API_HOST"), os.Getenv("HYBRID_ORGANIZATION_API_TOKEN")) + dataSourceConfig()
 }
 
 func missingOrganizationIdConfig() string {

internal/provider/provider_test_utils.go

@@ -22,24 +22,24 @@ func TestAccPreCheck(t *testing.T) {
 	// You can add code here to run prior to any test case execution, for example assertions
 	// about the appropriate environment variables being set are common to see in a pre-check
 	// function.
-	var errMsgs []string
+	var missingEnvVars []string
 	if hostedToken := os.Getenv("HOSTED_ORGANIZATION_API_TOKEN"); len(hostedToken) == 0 {
-		errMsgs = append(errMsgs, "HOSTED_ORGANIZATION_API_TOKEN must be set for acceptance tests")
+		missingEnvVars = append(missingEnvVars, "HOSTED_ORGANIZATION_API_TOKEN")
 	}
 	if hostedOrgId := os.Getenv("HOSTED_ORGANIZATION_ID"); len(hostedOrgId) == 0 {
-		errMsgs = append(errMsgs, "HOSTED_ORGANIZATION_ID must be set for acceptance tests")
+		missingEnvVars = append(missingEnvVars, "HOSTED_ORGANIZATION_ID")
 	}
 	if hybridToken := os.Getenv("HYBRID_ORGANIZATION_API_TOKEN"); len(hybridToken) == 0 {
-		errMsgs = append(errMsgs, "HYBRID_ORGANIZATION_API_TOKEN must be set for acceptance tests")
+		missingEnvVars = append(missingEnvVars, "HYBRID_ORGANIZATION_API_TOKEN")
 	}
 	if hybridOrgId := os.Getenv("HYBRID_ORGANIZATION_ID"); len(hybridOrgId) == 0 {
-		errMsgs = append(errMsgs, "HYBRID_ORGANIZATION_ID must be set for acceptance tests")
+		missingEnvVars = append(missingEnvVars, "HYBRID_ORGANIZATION_ID")
 	}
 	if host := os.Getenv("ASTRO_API_HOST"); len(host) == 0 {
-		errMsgs = append(errMsgs, "ASTRO_API_HOST must be set for acceptance tests")
+		missingEnvVars = append(missingEnvVars, "ASTRO_API_HOST")
 	}
-	if len(errMsgs) > 0 {
-		t.Fatalf("Pre-check failed: %+v", strings.Join(errMsgs, ", "))
+	if len(missingEnvVars) > 0 {
+		t.Fatalf("Pre-check failed: %+v must be set for acceptance tests", strings.Join(missingEnvVars, ", "))
 	}
 }
 

internal/provider/resources/resource_standard_deployment_test.go

@@ -308,14 +308,14 @@ func deleteDeploymentOutsideOfTerraform(t *testing.T, name string) {
 	assert.NoError(t, err)
 
 	ctx := context.Background()
-	resp, err := client.ListDeploymentsWithResponse(ctx, os.Getenv("ASTRO_HOSTED_ORGANIZATION_ID"), &platform.ListDeploymentsParams{
+	resp, err := client.ListDeploymentsWithResponse(ctx, os.Getenv("HOSTED_ORGANIZATION_ID"), &platform.ListDeploymentsParams{
 		Names: &[]string{name},
 	})
 	if err != nil {
 		assert.NoError(t, err)
 	}
 	assert.True(t, len(resp.JSON200.Deployments) >= 1, "deployment should exist but list deployments did not find it")
-	_, err = client.DeleteDeploymentWithResponse(ctx, os.Getenv("ASTRO_HOSTED_ORGANIZATION_ID"), resp.JSON200.Deployments[0].Id)
+	_, err = client.DeleteDeploymentWithResponse(ctx, os.Getenv("HOSTED_ORGANIZATION_ID"), resp.JSON200.Deployments[0].Id)
 	assert.NoError(t, err)
 }
 
@@ -326,7 +326,7 @@ func testAccCheckDeploymentExistence(t *testing.T, name string, shouldExist bool
 		assert.NoError(t, err)
 
 		ctx := context.Background()
-		resp, err := client.ListDeploymentsWithResponse(ctx, os.Getenv("ASTRO_HOSTED_ORGANIZATION_ID"), &platform.ListDeploymentsParams{
+		resp, err := client.ListDeploymentsWithResponse(ctx, os.Getenv("HOSTED_ORGANIZATION_ID"), &platform.ListDeploymentsParams{
 			Names: &[]string{name},
 		})
 		if err != nil {