Add test defs

This commit is contained in:
Rachid Mrad 2024-02-02 15:16:38 -05:00
parent ad6d080ad9
commit fa8122b217
No known key found for this signature in database

View file

@ -20,10 +20,12 @@ class MyTestCase(TestCase):
@override_settings(IS_PRODUCTION=True)
def test_production_environment(self):
"""No banner on prod."""
home_page = self.client.get("/")
self.assertNotContains(home_page, "You are not on production.")
@override_settings(IS_PRODUCTION=False)
def test_non_production_environment(self):
"""Banner on non-prod."""
home_page = self.client.get("/")
self.assertContains(home_page, "You are not on production.")