Update config test_options adding unitest

This commit is contained in:
Matt Seymour 2022-12-13 09:05:04 +00:00
parent 3c1a39b437
commit 7aa7afd764
No known key found for this signature in database
2 changed files with 12 additions and 1 deletions

View file

@ -52,7 +52,7 @@ def config(
conn_max_age=0,
conn_health_checks=False,
ssl_require=False,
test_options={},
test_options=None,
):
"""Returns configured DATABASE dictionary from DATABASE_URL."""
s = os.environ.get(env, default)

View file

@ -1,5 +1,6 @@
import os
import unittest
from unittest import mock
import dj_database_url
@ -530,6 +531,16 @@ class DatabaseTestSuite(unittest.TestCase):
assert "CONN_MAX_AGE" not in url
assert "CONN_HEALTH_CHECKS" not in url
@mock.patch.dict(
os.environ,
{"DATABASE_URL": "postgres://user:password@instance.amazonaws.com:5431/d8r8?"},
)
def test_persistent_connection_variables_config(self):
url = dj_database_url.config(conn_max_age=600, conn_health_checks=True)
assert url["CONN_MAX_AGE"] == 600
assert url["CONN_HEALTH_CHECKS"] is True
if __name__ == "__main__":
unittest.main()