From 756fff84fa78a8f1386ebe79e2e7e0b71877b4eb Mon Sep 17 00:00:00 2001 From: Clemens Wolff Date: Fri, 10 May 2019 09:28:55 -0400 Subject: [PATCH] Add test case for explicit sslmode enabling --- app/server/tests/test_config.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/app/server/tests/test_config.py b/app/server/tests/test_config.py index 7a175d3d..e80be890 100644 --- a/app/server/tests/test_config.py +++ b/app/server/tests/test_config.py @@ -16,10 +16,14 @@ class TestDatabaseUrl(TestCase): with setenv('DATABASE_URL', 'sqlite:///some/path'): self._assert_sslmode_is(None) - def test_sslmode_can_be_set_via_database_url(self): + def test_sslmode_can_be_disabled_via_database_url(self): with setenv('DATABASE_URL', 'pgsql://u:p@h/d?sslmode=disabled'): self._assert_sslmode_is('disabled') + def test_sslmode_can_be_required_via_database_url(self): + with setenv('DATABASE_URL', 'pgsql://u:p@h/d?sslmode=require'): + self._assert_sslmode_is('require') + def _assert_sslmode_is(self, expected): reload(settings) actual = settings.DATABASES['default'].get('OPTIONS', {}).get('sslmode')