Merge pull request #7782 from markflyhigh/py3-bq-it

[BEAM-6621] Basic Bigquery IO tests on Python 3 PostCommit
apache · Feb 8, 2019 · 381ab55 · 381ab55
2 parents af8c400 + 7f6d201
commit 381ab55
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/sdks/python/precommit/dataflow/py3/build.gradle b/sdks/python/precommit/dataflow/py3/build.gradle
@@ -26,8 +26,12 @@ def runScriptsDir = "${project.rootDir}/sdks/python/scripts"
 
 task preCommitIT(dependsOn: ['sdist', 'installGcpTest']) {
   doLast {
+    def tests = [
+        "apache_beam.examples.wordcount_it_test:WordCountIT.test_wordcount_it",
+        "apache_beam.examples.cookbook.bigquery_tornadoes_it_test:BigqueryTornadoesIT.test_bigquery_tornadoes_it",
+    ]
     def testOpts = [
-        "--tests=apache_beam.examples.wordcount_it_test:WordCountIT.test_wordcount_it",
+        "--tests=${tests.join(',')}",
         "--nocapture",    // Print stdout instantly
     ]
     def cmdArgs = project.mapToArgString(["test_opts": testOpts])

diff --git a/sdks/python/precommit/direct/py3/build.gradle b/sdks/python/precommit/direct/py3/build.gradle
@@ -27,8 +27,12 @@ def runScriptsDir = "${project.rootDir}/sdks/python/scripts"
 task preCommitIT(dependsOn: ['sdist', 'installGcpTest']) {
   // Run IT tests with TestDirectRunner in batch in Python 3.
   doLast {
+    def batchTests = [
+        "apache_beam.examples.wordcount_it_test:WordCountIT.test_wordcount_it",
+        "apache_beam.examples.cookbook.bigquery_tornadoes_it_test:BigqueryTornadoesIT.test_bigquery_tornadoes_it",
+    ]
     def testOpts = [
-        "--tests=apache_beam.examples.wordcount_it_test:WordCountIT.test_wordcount_it",
+        "--tests=${batchTests.join(',')}",
         "--nocapture",    // Print stdout instantly
     ]
     def argMap = ["runner": "TestDirectRunner",