Skip to content

Commit

Permalink
style: fix lint errors
Browse files Browse the repository at this point in the history
  • Loading branch information
larkee committed Feb 5, 2021
1 parent ede3345 commit 613315d
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions tests/unit/test_database.py
Expand Up @@ -1368,16 +1368,13 @@ def test_context_mgr_w_commit_stats(self):
session=self.SESSION_NAME,
mutations=[],
single_use_transaction=expected_txn_options,
return_commit_stats=True

return_commit_stats=True,
)
api.commit.assert_called_once_with(
request=request, metadata=[("google-cloud-resource-prefix", database.name)],
)

database.logger.info.assert_called_once_with(
"Transaction mutation count: 4"
)
database.logger.info.assert_called_once_with("Transaction mutation count: 4")

def test_context_mgr_failure(self):
from google.cloud.spanner_v1.batch import Batch
Expand Down Expand Up @@ -1972,6 +1969,7 @@ def __init__(self, name, instance=None):
self.database_id = name.rsplit("/", 1)[1]
self._instance = instance
from logging import Logger

self.logger = mock.create_autospec(Logger, instance=True)


Expand Down

0 comments on commit 613315d

Please sign in to comment.