feat:added toggle methods to basemodel

edx_toggles/tests/test_waffle.py

@@ -2,6 +2,7 @@
 Unit tests for waffle classes.
 """
 
+from unittest.mock import patch
 from django.test import TestCase
 
 from edx_toggles.toggles import NonNamespacedWaffleFlag, NonNamespacedWaffleSwitch, WaffleFlag, WaffleSwitch
@@ -29,6 +30,29 @@ def test_constructor(self):
         self.assertEqual("module1", waffle.module_name)
         self.assertEqual(1, len(NaiveWaffle.get_instances()))
 
+    def test_toggle_methods_with_enabled_waffle(self):
+        waffle = NaiveWaffle("namespaced.name", "module1")
+        # test is_enabled method
+        self.assertEqual(True, waffle.is_enabled())
+        # test is_disabled method
+        self.assertEqual(False, waffle.is_disabled())
+        # test is_toggle_on method
+        self.assertEqual(True, waffle.is_toggle_on())
+        # test is_toggle_off method
+        self.assertEqual(False, waffle.is_toggle_off())
+
+    def test_toggle_methods_with_disabled_waffle(self):
+        with patch.object(NaiveWaffle, "is_enabled", return_value=False):
+            waffle = NaiveWaffle("namespaced.name", "module1")
+            # test is_enabled method
+            self.assertEqual(False, waffle.is_enabled())
+            # test is_disabled method
+            self.assertEqual(True, waffle.is_disabled())
+            # test is_toggle_on method
+            self.assertEqual(False, waffle.is_toggle_on())
+            # test is_toggle_off method
+            self.assertEqual(True, waffle.is_toggle_off())
+
 
 class WaffleFlagTests(TestCase):
     """

edx_toggles/toggles/internal/base.py

@@ -37,3 +37,12 @@ def get_instances(cls):
         Return the list of class instances sorted by name.
         """
         return sorted(cls._class_instances, key=lambda instance: instance.name)
+
+    def is_disabled(self):
+        return not self.is_enabled()
+
+    def is_toggle_on(self):
+        return self.is_enabled()
+
+    def is_toggle_off(self):
+        return self.is_disabed()