[BEAM-6205] Setup gradle task ro run fnapi worker test with use_execu…

…table_stage_bundle_execution
apache · Dec 11, 2018 · 5e506bf · 5e506bf
1 parent b0aae58
commit 5e506bf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/runners/google-cloud-dataflow-java/build.gradle b/runners/google-cloud-dataflow-java/build.gradle
@@ -115,12 +115,14 @@ def dataflowFnApiWorkerJar = project.findProperty('dataflowWorkerJar') ?: projec
 def dockerImageRoot = project.findProperty('dockerImageRoot') ?: "us.gcr.io/${dataflowProject}/java-postcommit-it"
 def dockerImageContainer = "${dockerImageRoot}/java"
 def dockerTag = new Date().format('yyyyMMddHHmmss')
+// If -PuseExecutableStage is set, the use_executable_stage_bundle_execution wil be enabled.
+def fnapiExperiments = project.findProperty('useExecutableStage') ? 'beam_fn_api,use_executable_stage_bundle_execution' : "beam_fn_api"
 ext.dockerImageName = "${dockerImageContainer}:${dockerTag}"
 
 def fnApiPipelineOptions = [
       "--dataflowWorkerJar=${dataflowFnApiWorkerJar}",
       "--workerHarnessContainerImage=${dockerImageContainer}:${dockerTag}",
-      "--experiments=beam_fn_api",
+      "--experiments=${fnapiExperiments}",
 ]
 
 def commonExcludeCategories = [

diff --git a/runners/google-cloud-dataflow-java/examples/build.gradle b/runners/google-cloud-dataflow-java/examples/build.gradle
@@ -39,6 +39,8 @@ dependencies {
 def gcpProject = project.findProperty('gcpProject') ?: 'apache-beam-testing'
 def gcsTempRoot = project.findProperty('gcsTempRoot') ?: 'gs://temp-storage-for-end-to-end-tests/'
 def dockerImageName = project(':beam-runners-google-cloud-dataflow-java').ext.dockerImageName
+// If -PuseExecutableStage is set, the use_executable_stage_bundle_execution wil be enabled.
+def fnapiExperiments = project.hasProperty('useExecutableStage') ? 'beam_fn_api,use_executable_stage_bundle_execution' : "beam_fn_api"
 
 task preCommitLegacyWorker(type: Test) {
   dependsOn ":beam-runners-google-cloud-dataflow-java-legacy-worker:shadowJar"
@@ -72,7 +74,7 @@ task preCommitFnApiWorker(type: Test) {
           "--runner=TestDataflowRunner",
           "--dataflowWorkerJar=${dataflowWorkerJar}",
           "--workerHarnessContainerImage=${dockerImageName}",
-          "--experiments=beam_fn_api",
+          "--experiments=${fnapiExperiments}",
   ]
   testClassesDirs = files(project(":beam-examples-java").sourceSets.test.output.classesDirs)
   include "**/WordCountIT.class"