Updating Unit Test Count for Airflow Multipartition

devops/tasks/tests-unit.yml

@@ -42,6 +42,7 @@ steps:
       TF_VAR_principal_password: $(TF_VAR_principal_password)
       TF_VAR_principal_objectId: $(TF_VAR_principal_objectId)
       TF_VAR_gitops_path: $(TF_VAR_gitops_path)
+      TF_VAR_deploy_dp_airflow: $(TF_VAR_deploy_dp_airflow)
       TF_VAR_istio_int_load_balancer_ip: $(TF_VAR_istio_int_load_balancer_ip)
       TF_VAR_aks_dns_host: $(TF_VAR_aks_dns_host)
       ${{ if ne(variables['TF_VAR_ssl_challenge_required'], '') }}:
@@ -56,6 +57,10 @@ steps:
         #!/usr/bin/env bash
         set -euo pipefail
 
+        if [ $TF_VAR_deploy_dp_airflow != "true" ] && [ $TF_VAR_deploy_dp_airflow != "false" ]; then
+          export TF_VAR_deploy_dp_airflow=false
+        fi
+
         export ARM_TENANT_ID=$tenantId
         export ARM_CLIENT_SECRET=$servicePrincipalKey
         export ARM_CLIENT_ID=$servicePrincipalId

infra/templates/osdu-r3-mvp/data_partition/tests/unit/unit_test.go

@@ -35,7 +35,12 @@ var tfOptions = &terraform.Options{
 	},
 }
 
+var deploy_dp_airflow = os.Getenv("TF_VAR_deploy_dp_airflow")
+var airflow_dp_resource_count = 87
+var expected_resource_count = 139
+
 func TestTemplate(t *testing.T) {
+
 	expectedAppDevResourceGroup := asMap(t, `{
 		"location": "`+region+`"
 	}`)
@@ -44,12 +49,16 @@ func TestTemplate(t *testing.T) {
 		"azurerm_resource_group.main": expectedAppDevResourceGroup,
 	}
 
+	if deploy_dp_airflow == "true" {
+		expected_resource_count = expected_resource_count + airflow_dp_resource_count
+	}
+
 	testFixture := infratests.UnitTestFixture{
 		GoTest:                          t,
 		TfOptions:                       tfOptions,
 		Workspace:                       workspace,
 		PlanAssertions:                  nil,
-		ExpectedResourceCount:           139,
+		ExpectedResourceCount:           expected_resource_count,
 		ExpectedResourceAttributeValues: resourceDescription,
 	}