Skip to content
This repository was archived by the owner on Jun 8, 2026. It is now read-only.
Closed
Prev Previous commit
Next Next commit
move ResultsChecksum() to test
  • Loading branch information
AlisskaPie committed Nov 30, 2020
commit 1fb70190968286ed99caa5e346caadb4583d81a3
2 changes: 0 additions & 2 deletions samples/samples/autocommit.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,14 +12,12 @@
def enable_autocommit_mode(instance_id, database_id):
"""Enables autocommit mode."""
# [START spanner_enable_autocommit_mode]
from google.cloud.spanner_dbapi.checksum import ResultsChecksum

connection = connect(instance_id, database_id)
connection.autocommit = True
print("Autocommit mode is enabled.")

cursor = connection.cursor()
cursor._checksum = ResultsChecksum()

cursor.execute(
"""CREATE TABLE Singers (
Expand Down
15 changes: 11 additions & 4 deletions samples/samples/autocommit_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,14 @@ def database(spanner_instance):


def test_enable_autocommit_mode(capsys, database):
autocommit.enable_autocommit_mode(INSTANCE_ID, DATABASE_ID)
out, _ = capsys.readouterr()
assert "Autocommit mode is enabled." in out
assert "SingerId: 13, AlbumId: Russell, AlbumTitle: Morales" in out
import mock
from google.cloud.spanner_dbapi.checksum import ResultsChecksum

with mock.patch(
"google.cloud.spanner_dbapi.cursor.Cursor._checksum",
new_callable=ResultsChecksum(),
):
autocommit.enable_autocommit_mode(INSTANCE_ID, DATABASE_ID)
out, _ = capsys.readouterr()
assert "Autocommit mode is enabled." in out
assert "SingerId: 13, AlbumId: Russell, AlbumTitle: Morales" in out