use connection.Cursor in patch

larkee · Dec 30, 2020 · Nov 19, 2020 · Nov 19, 2020 · Nov 23, 2020 · Nov 24, 2020
commit 09c52464f557e0a97078d6d80f31f519f1925b17
diff --git a/samples/samples/autocommit_test.py b/samples/samples/autocommit_test.py
@@ -50,10 +50,15 @@ def database(spanner_instance):
 def test_enable_autocommit_mode(capsys, database):
     import mock
     from google.cloud.spanner_dbapi.checksum import ResultsChecksum
+    from google.cloud.spanner_dbapi import connect
+
+    connection = connect(INSTANCE_ID, DATABASE_ID)
+    cursor = connection.cursor()
+    cursor._checksum = ResultsChecksum()
 
     with mock.patch(
-        "google.cloud.spanner_dbapi.cursor.Cursor._checksum",
-        new_callable=ResultsChecksum(),
+        "google.cloud.spanner_dbapi.connection.Cursor",
+        return_value=cursor,
     ):
         autocommit.enable_autocommit_mode(INSTANCE_ID, DATABASE_ID)
         out, _ = capsys.readouterr()