Pin a mocking test to DirectRunner (#35883)

damccorm · web-flow · commit 7ecbb2a7037e · 2025-08-15T16:34:53.000-04:00
* Pin a mocking test to DirectRunner

* Update bigquery_test.py
diff --git a/sdks/python/apache_beam/io/gcp/bigquery_test.py b/sdks/python/apache_beam/io/gcp/bigquery_test.py
@@ -1567,7 +1567,9 @@ def test_insert_rows_json_intermittent_retriable_exception(
         exception_type(error_message), exception_type(error_message), []
     ]
 
-    with beam.Pipeline() as p:
+    # This relies on DirectRunner-specific mocking behavior which can be
+    # inconsistent on Prism
+    with beam.Pipeline('FnApiRunner') as p:
       _ = (
           p
           | beam.Create([{

Original file line number	Diff line number	Diff line change
`@@ -1567,7 +1567,9 @@ def test_insert_rows_json_intermittent_retriable_exception(`
`1567`	`1567`	`exception_type(error_message), exception_type(error_message), []`
`1568`	`1568`	`]`
`1569`	`1569`
`1570`		`- with beam.Pipeline() as p:`
	`1570`	`+ # This relies on DirectRunner-specific mocking behavior which can be`
	`1571`	`+ # inconsistent on Prism`
	`1572`	`+ with beam.Pipeline('FnApiRunner') as p:`
`1571`	`1573`	`_ = (`
`1572`	`1574`	`p`
`1573`	`1575`	`\| beam.Create([{`