test(dbapi): add retry to autocommit sample to reduce flakiness (googleapis#214)

larkee · web-flow · commit bc38664891aa · 2021-01-27T10:02:30.000-08:00
Co-authored-by: larkee &lt;larkee@users.noreply.github.com&gt;
diff --git 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()
