def test_rollback(self, mock_warn):
from google.cloud.spanner_dbapi import Connection
from google.cloud.spanner_dbapi.connection import AUTOCOMMIT_MODE_WARNING
connection = Connection(INSTANCE, DATABASE)
with mock.patch(
"google.cloud.spanner_dbapi.connection.Connection._release_session"
) as mock_release:
connection.rollback()
mock_release.assert_not_called()
mock_transaction = mock.MagicMock()
connection._transaction = mock_transaction
mock_rollback = mock.MagicMock()
mock_transaction.rollback = mock_rollback
with mock.patch(
"google.cloud.spanner_dbapi.connection.Connection._release_session"
) as mock_release:
connection.rollback()
mock_rollback.assert_called_once_with()
mock_release.assert_called_once_with()
connection._autocommit = True
connection.rollback()
mock_warn.assert_called_once_with(
AUTOCOMMIT_MODE_WARNING, UserWarning, stacklevel=2
)
def test_transaction_checkout(self):
from google.cloud.spanner_dbapi import Connection
connection = Connection(INSTANCE, DATABASE)
mock_checkout = mock.MagicMock(autospec=True)
connection._session_checkout = mock_checkout
connection.transaction_checkout()
mock_checkout.assert_called_once_with()
mock_transaction = mock.MagicMock()
mock_transaction.committed = mock_transaction.rolled_back = False
connection._transaction = mock_transaction
self.assertEqual(connection.transaction_checkout(), mock_transaction)
connection._autocommit = True
self.assertIsNone(connection.transaction_checkout())