Merge pull request #8 from nvllsvm/fix_before_connect

Fix postgres_status before first connection attempt
This commit is contained in:
Gavin M. Roy 2021-01-20 16:46:41 -05:00 committed by GitHub
commit 3d7940ee9c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 1 deletions

View file

@ -408,7 +408,8 @@ class ApplicationMixin:
} }
""" """
if not self._postgres_connected.is_set(): if not self._postgres_connected or \
not self._postgres_connected.is_set():
return { return {
'available': False, 'available': False,
'pool_size': 0, 'pool_size': 0,

View file

@ -349,6 +349,18 @@ class TestCase(testing.SprocketsHttpTestCase):
return self.app return self.app
class PostgresStatusTestCase(asynctest.TestCase):
async def test_postgres_status_before_first_connection(self):
app = Application()
status = await app.postgres_status()
self.assertEqual(
status,
{'available': False,
'pool_size': 0,
'pool_free': 0})
class RequestHandlerMixinTestCase(TestCase): class RequestHandlerMixinTestCase(TestCase):
def test_postgres_status(self): def test_postgres_status(self):