From 66f29649287d30d80a06664b68dcbd263920cc0f Mon Sep 17 00:00:00 2001 From: larkee Date: Fri, 22 Jan 2021 17:33:11 +1100 Subject: [PATCH] test(dbapi): add retry to autocommit sample to reduce flakiness --- samples/samples/autocommit_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/samples/samples/autocommit_test.py b/samples/samples/autocommit_test.py index c906f060e0..a98744968a 100644 --- a/samples/samples/autocommit_test.py +++ b/samples/samples/autocommit_test.py @@ -6,10 +6,12 @@ import uuid +from google.api_core.exceptions import Aborted from google.cloud import spanner from google.cloud.spanner_dbapi import connect import mock import pytest +from test_utils.retry import RetryErrors import autocommit @@ -49,6 +51,7 @@ def database(spanner_instance): db.drop() +@RetryErrors(exception=Aborted, max_tries=2) def test_enable_autocommit_mode(capsys, database): connection = connect(INSTANCE_ID, DATABASE_ID) cursor = connection.cursor()