github/sprockets-postgres
1
0
Fork 0
mirror of https://github.com/sprockets/sprockets-postgres.git synced 2024-12-28 19:29:21 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix indentation, duplicate test name

Browse source
This commit is contained in:
Gavin M. Roy 2020-07-07 17:31:16 -04:00
parent a1736bccae
commit aaf0548bdf
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
sprockets_postgres.py
View file

@ -351,8 +351,8 @@ class ApplicationMixin:
if retry: if retry:
await self._postgres_connected.wait() await self._postgres_connected.wait()
async with self.postgres_connector( async with self.postgres_connector(
on_error, on_duration, timeout, on_error, on_duration, timeout,
_attempt + 1) as connector: _attempt + 1) as connector:
yield connector yield connector
return return
message = 'disconnected' message = 'disconnected'

2
tests.py
View file

@ -426,7 +426,7 @@ class RequestHandlerMixinTestCase(TestCase):
aiopg.connection.Connection.cursor = original aiopg.connection.Connection.cursor = original
@mock.patch('aiopg.connection.Connection.cursor') @mock.patch('aiopg.connection.Connection.cursor')
def test_postgres_cursor_raises(self, cursor): def test_postgres_cursor_raises_on_failed_reconnect(self, cursor):
cursor.side_effect = psycopg2.OperationalError() cursor.side_effect = psycopg2.OperationalError()
with mock.patch.object(self.app, '_postgres_connect') as connect: with mock.patch.object(self.app, '_postgres_connect') as connect:
connect.return_value = False connect.return_value = False