Add to params instead of env

Author: manitgupta

v2/spanner-to-sourcedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToSourceDbITBase.java

@@ -231,6 +231,7 @@ public PipelineLauncher.LaunchInfo launchDataflowJob(
             put("maxNumWorkers", "1");
             put("numWorkers", "1");
             put("sourceType", sourceType);
+            put("workerMachineType", "n2-standard-4");
           }
         };
     if (jobParameters != null) {
@@ -263,7 +264,6 @@ public PipelineLauncher.LaunchInfo launchDataflowJob(
     options.setParameters(params);
     options.addEnvironment("additionalExperiments", Collections.singletonList("use_runner_v2"));
     options.addEnvironment("ipConfiguration", "WORKER_IP_PRIVATE");
-    options.addEnvironment("workerMachineType", "n2-standard-4");
     // Run
     PipelineLauncher.LaunchInfo jobInfo = launchTemplate(options, false);
     assertThatPipeline(jobInfo).isRunning();

v2/spanner-to-sourcedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToSourceDbLTBase.java

@@ -216,6 +216,7 @@ public PipelineLauncher.LaunchInfo launchDataflowJob(
             put("deadLetterQueueDirectory", getGcsPath(artifactBucket, "dlq", gcsResourceManager));
             put("maxShardConnections", "100");
             put("sourceType", sourceType);
+            put("workerMachineType", "n2-standard-4");
           }
         };
 
@@ -231,7 +232,6 @@ public PipelineLauncher.LaunchInfo launchDataflowJob(
     options
         .addEnvironment("maxWorkers", maxWorkers)
         .addEnvironment("numWorkers", numWorkers)
-        .addEnvironment("workerMachineType", "n2-standard-4")
         .addEnvironment("additionalExperiments", Collections.singletonList("use_runner_v2"));
 
     options.setParameters(params);