Skip to content

Commit

Permalink
retry test_big_query_legacy_sql (#31417)
Browse files Browse the repository at this point in the history
* retry test_big_query_legacy_sql

* fix lint

* fixed lint
  • Loading branch information
liferoad authored Jun 6, 2024
1 parent 3cdbd3a commit 03d553e
Showing 1 changed file with 3 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,8 @@

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
Expand Down Expand Up @@ -155,6 +157,7 @@ def _setup_new_types_env(self):
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)
Expand Down

0 comments on commit 03d553e

Please sign in to comment.