This is an automated email from the ASF dual-hosted git repository.
damondouglas pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new 03d553e1f9c retry test_big_query_legacy_sql (#31417)
03d553e1f9c is described below
commit 03d553e1f9c58eb8a1cd29352e9281427cd8669e
Author: liferoad <[email protected]>
AuthorDate: Thu Jun 6 16:43:01 2024 -0400
retry test_big_query_legacy_sql (#31417)
* retry test_big_query_legacy_sql
* fix lint
* fixed lint
---
sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py
b/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py
index e8cd8884219..052790c4a20 100644
--- a/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py
+++ b/sdks/python/apache_beam/io/gcp/big_query_query_to_table_it_test.py
@@ -30,6 +30,8 @@ import unittest
import pytest
from hamcrest.core.core.allof import all_of
+from tenacity import retry
+from tenacity import stop_after_attempt
from apache_beam.io.gcp import big_query_query_to_table_pipeline
from apache_beam.io.gcp.bigquery_tools import BigQueryWrapper
@@ -155,6 +157,7 @@ class BigQueryQueryToTableIT(unittest.TestCase):
self.assertTrue(passed, 'Error in BQ setup: %s' % errors)
@pytest.mark.it_postcommit
+ @retry(reraise=True, stop=stop_after_attempt(3))
def test_big_query_legacy_sql(self):
verify_query = DIALECT_OUTPUT_VERIFY_QUERY % self.output_table
expected_checksum = test_utils.compute_hash(DIALECT_OUTPUT_EXPECTED)
