[change] Add more tests to improve coverage

openwisp · Jan 2, 2021 · c5890b7 · c5890b7
1 parent e685000
commit c5890b7
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/openwisp_ipam/tests/test_multitenant.py b/openwisp_ipam/tests/test_multitenant.py
@@ -7,6 +7,14 @@
 
 from . import CreateModelsMixin, PostDataMixin
 
+from openwisp_ipam.api.utils import (
+    FilterByOrganization,
+    FilterByParent,
+    AuthorizeCSVImport,
+    FilterSerializerByOrganization,
+)
+
+
 User = get_user_model()
 IpAddress = load_model('openwisp_ipam', 'IPAddress')
 Subnet = load_model('openwisp_ipam', 'Subnet')
@@ -403,3 +411,22 @@ def test_browsable_api_ipaddress_list(self):
             )
             self.assertContains(response, '10.0.0.0/24</option>')
             self.assertContains(response, '10.10.0.0/24</option>')
+
+    def test_not_implemented_error(self):
+        with self.assertRaises(NotImplementedError):
+            FilterByOrganization.get_organization_queryset(self)
+
+        with self.assertRaises(NotImplementedError):
+            FilterByParent.get_parent_queryset(self)
+
+        with self.assertRaises(NotImplementedError):
+            FilterByParent.get_organization_queryset(self)
+
+        with self.assertRaises(NotImplementedError):
+            AuthorizeCSVImport.get_csv_organization(self)
+
+        with self.assertRaises(NotImplementedError):
+            AuthorizeCSVImport.get_user_organizations(self)
+
+        with self.assertRaises(NotImplementedError):
+            FilterSerializerByOrganization.filter_fields(self)