fix test and config default params

jazzband · Dec 12, 2022 · a36ecda · a36ecda
1 parent 9740c13
commit a36ecda
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/dj_database_url.py b/dj_database_url.py
@@ -50,14 +50,15 @@ def config(
     default=None,
     engine=None,
     conn_max_age=0,
+    conn_health_checks=False,
     ssl_require=False,
     test_options={},
 ):
     """Returns configured DATABASE dictionary from DATABASE_URL."""
     s = os.environ.get(env, default)
 
     if s:
-        return parse(s, engine, conn_max_age, ssl_require, test_options)
+        return parse(s, engine, conn_max_age, conn_health_checks, ssl_require, test_options)
 
     return {}
 
@@ -131,7 +132,7 @@ def parse(
         }
     )
     if test_options:
-        config.update(
+        parsed_config.update(
             {
                 'TEST': test_options,
             }

diff --git a/test_dj_database_url.py b/test_dj_database_url.py
@@ -118,6 +118,9 @@ def test_mysql_connector_parsing(self):
         assert url["PORT"] == 5431
 
     def test_config_test_options(self):
+        os.environ[
+            "DATABASE_URL"
+        ] = "postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?"
         test_db_config = {
             'NAME': 'mytestdatabase',
         }