Add GOOGLE_VMWAREENGINE_PROJECT env var for vmwareengine tests (#12707) (#20855)

[upstream:415b35bae8e1e92e6949d472b987105f369be650]

Signed-off-by: Modular Magician <magic-modules@google.com>

Author: modular-magician

.changelog/12707.txt

@@ -0,0 +1,3 @@
+```release-note:none
+
+```

.teamcity/components/builds/build_parameters.kt

@@ -61,6 +61,11 @@ class AllContextParameters(
     val chronicleInstanceIdBeta: String,
     val chronicleInstanceIdVcr: String,
 
+    // GOOGLE_VMWAREENGINE_PROJECT
+    val vmwareengineProjectGa: String,
+    val vmwareengineProjectBeta: String,
+    val vmwareengineProjectVcr: String,
+
     // Values that are the same across GA, Beta, and VCR testing environments
     val billingAccount: String,   // GOOGLE_BILLING_ACCOUNT
     val billingAccount2: String,  // GOOGLE_BILLING_ACCOUNT_2
@@ -94,6 +99,7 @@ class AccTestConfiguration(
     val projectNumber: String,
     val region: String,
     val serviceAccount: String,
+    val vmwareengineProject: String,
     val zone: String,
 
     // VCR specific
@@ -120,6 +126,7 @@ fun getGaAcceptanceTestConfig(allConfig: AllContextParameters): AccTestConfigura
         allConfig.projectNumberGa,
         allConfig.region,
         allConfig.serviceAccountGa,
+        allConfig.vmwareengineProjectGa,
         allConfig.zone,
         allConfig.infraProject,
         allConfig.vcrBucketName,
@@ -143,6 +150,7 @@ fun getBetaAcceptanceTestConfig(allConfig: AllContextParameters): AccTestConfigu
         allConfig.projectNumberBeta,
         allConfig.region,
         allConfig.serviceAccountBeta,
+        allConfig.vmwareengineProjectBeta,
         allConfig.zone,
         allConfig.infraProject,
         allConfig.vcrBucketName,
@@ -166,6 +174,7 @@ fun getVcrAcceptanceTestConfig(allConfig: AllContextParameters): AccTestConfigur
         allConfig.projectNumberVcr,
         allConfig.region,
         allConfig.serviceAccountVcr,
+        allConfig.vmwareengineProjectVcr,
         allConfig.zone,
         allConfig.infraProject,
         allConfig.vcrBucketName,
@@ -190,6 +199,7 @@ fun ParametrizedWithType.configureGoogleSpecificTestParameters(config: AccTestCo
     hiddenVariable("env.GOOGLE_ZONE", config.zone, "The google zone to use")
     hiddenVariable("env.GOOGLE_IDENTITY_USER", config.identityUser, "The user for the identity platform")
     hiddenVariable("env.GOOGLE_CHRONICLE_INSTANCE_ID", config.chronicleInstanceId, "The id of the Chronicle instance")
+    hiddenVariable("env.GOOGLE_VMWAREENGINE_PROJECT", config.vmwareengineProject, "The project used for vmwareengine tests")
     hiddenPasswordVariable("env.GOOGLE_CREDENTIALS", config.credentials, "The Google credentials for this test runner")
 }
 

.teamcity/settings.kts

@@ -49,6 +49,10 @@ val org2Vcr  = DslContext.getParameter("org2Vcr", "")
 val chronicleInstanceIdGa   = DslContext.getParameter("chronicleInstanceIdGa", "")
 val chronicleInstanceIdBeta = DslContext.getParameter("chronicleInstanceIdBeta", "")
 val chronicleInstanceIdVcr  = DslContext.getParameter("chronicleInstanceIdVcr", "")
+// GOOGLE_VMWAREENGINE_PROJECT
+val vmwareengineProjectGa   = DslContext.getParameter("vmwareengineProjectGa", "")
+val vmwareengineProjectBeta = DslContext.getParameter("vmwareengineProjectBeta", "")
+val vmwareengineProjectVcr  = DslContext.getParameter("vmwareengineProjectVcr", "")
 
 // Values that are the same across GA, Beta, and VCR testing environments
 val billingAccount  = DslContext.getParameter("billingAccount", "")   // GOOGLE_BILLING_ACCOUNT
@@ -91,6 +95,9 @@ var allContextParams = AllContextParameters(
     chronicleInstanceIdGa,
     chronicleInstanceIdBeta,
     chronicleInstanceIdVcr,
+    vmwareengineProjectGa,
+    vmwareengineProjectBeta,
+    vmwareengineProjectVcr,
     billingAccount,
     billingAccount2,
     custId,

.teamcity/tests/test_utils.kt

@@ -44,6 +44,9 @@ fun testContextParameters(): AllContextParameters {
         "chronicleInstanceIdGa",
         "chronicleInstanceIdBeta",
         "chronicleInstanceIdVcr",
+        "vmwareengineProjectGa",
+        "vmwareengineProjectBeta",
+        "vmwareengineProjectVcr",
         "billingAccount",
         "billingAccount2",
         "custId",

google/envvar/envvar_utils.go

@@ -117,6 +117,12 @@ var ImpersonateServiceAccountEnvVars = []string{
 	"GOOGLE_IMPERSONATE_SERVICE_ACCOUNT",
 }
 
+// This value is the project used for vmwareengine tests. A separate project is needed
+// due to the limited quota allocated to each testing project
+var vmwareengineProjectEnvVars = []string{
+	"GOOGLE_VMWAREENGINE_PROJECT",
+}
+
 // AccTestPreCheck ensures at least one of the project env variables is set.
 func GetTestProjectNumberFromEnv() string {
 	return transport_tpg.MultiEnvSearch(ProjectNumberEnvVars)
@@ -217,6 +223,11 @@ func GetTestChronicleInstanceIdFromEnv(t *testing.T) string {
 	return transport_tpg.MultiEnvSearch(ChronicleInstanceIdEnvVars)
 }
 
+func GetTestVmwareengineProjectFromEnv(t *testing.T) string {
+	SkipIfEnvNotSet(t, vmwareengineProjectEnvVars...)
+	return transport_tpg.MultiEnvSearch(vmwareengineProjectEnvVars)
+}
+
 func SkipIfEnvNotSet(t *testing.T, envs ...string) {
 	if t == nil {
 		log.Printf("[DEBUG] Not running inside of test - skip skipping")