diff --git a/google/cloud/spanner_dbapi/cursor.py b/google/cloud/spanner_dbapi/cursor.py index e9e4862281..112fcda291 100644 --- a/google/cloud/spanner_dbapi/cursor.py +++ b/google/cloud/spanner_dbapi/cursor.py @@ -160,9 +160,9 @@ def _do_batch_update(self, transaction, statements, many_result_set): many_result_set.add_iter(res) if status.code == ABORTED: - raise Aborted(status.details) + raise Aborted(status.message) elif status.code != OK: - raise OperationalError(status.details) + raise OperationalError(status.message) def execute(self, sql, args=None): """Prepares and executes a Spanner database operation. @@ -302,9 +302,9 @@ def executemany(self, operation, seq_of_params): if status.code == ABORTED: self.connection._transaction = None - raise Aborted(status.details) + raise Aborted(status.message) elif status.code != OK: - raise OperationalError(status.details) + raise OperationalError(status.message) break except Aborted: self.connection.retry_transaction() diff --git a/tests/unit/spanner_dbapi/test_cursor.py b/tests/unit/spanner_dbapi/test_cursor.py index f2c9130613..90d07eb3db 100644 --- a/tests/unit/spanner_dbapi/test_cursor.py +++ b/tests/unit/spanner_dbapi/test_cursor.py @@ -506,7 +506,7 @@ def test_executemany_insert_batch_failed(self): transaction = mock.Mock(committed=False, rolled_back=False) transaction.batch_update = mock.Mock( - return_value=(mock.Mock(code=UNKNOWN, details=err_details), []) + return_value=(mock.Mock(code=UNKNOWN, message=err_details), []) ) with mock.patch(