From 63f71794a3726a00a08040012878c2286107f470 Mon Sep 17 00:00:00 2001 From: matthewt Date: Mon, 18 Nov 2024 08:44:55 +0200 Subject: [PATCH] waffle: adding template tag tests --- waffle/tests/test_templates.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/waffle/tests/test_templates.py b/waffle/tests/test_templates.py index a368a9c3..fad9a4d9 100644 --- a/waffle/tests/test_templates.py +++ b/waffle/tests/test_templates.py @@ -9,6 +9,7 @@ from test_app import views from waffle.middleware import WaffleMiddleware from waffle.tests.base import TestCase +from waffle.testutils import override_switch, override_flag, override_sample def get(): @@ -37,6 +38,26 @@ def test_django_tags(self): self.assertContains(response, 'switch_var off') self.assertContains(response, 'sample_var') self.assertContains(response, 'window.waffle =') + self.assertContains(response, 'flag_is_active off') + self.assertContains(response, 'switch_is_active off') + self.assertContains(response, 'sample_is_active off') + + @override_flag("flag", active=True) + @override_sample("sample", active=True) + @override_switch("switch", active=True) + def test_django_tags_enabled(self): + request = get() + response = process_request(request, views.flag_in_django) + self.assertContains(response, 'flag on') + self.assertContains(response, 'switch on') + self.assertContains(response, 'sample on') + self.assertNotContains(response, 'flag off') + self.assertNotContains(response, 'switch off') + self.assertNotContains(response, 'sample off') + self.assertContains(response, 'window.waffle =') + self.assertContains(response, 'flag_is_active on') + self.assertContains(response, 'switch_is_active on') + self.assertContains(response, 'sample_is_active on') def test_get_nodes_by_type(self): """WaffleNode.get_nodes_by_type() should find all child nodes."""