Преглед на файлове

fix(api): fix domain registrable tests

Peter Thomassen преди 5 години
родител
ревизия
ef4eb3641c
променени са 1 файла, в които са добавени 5 реда и са изтрити 5 реда
  1. 5 5
      api/desecapi/tests/test_domains.py

+ 5 - 5
api/desecapi/tests/test_domains.py

@@ -54,19 +54,19 @@ class IsRegistrableTestCase(DesecTestCase, PublicSuffixMockMixin):
             global_public_suffixes=['com', 'de', 'xxx', 'com.uk'],
             global_public_suffixes=['com', 'de', 'xxx', 'com.uk'],
             local_public_suffixes=['something.else'],
             local_public_suffixes=['something.else'],
         ):
         ):
+            self.assertNotRegistrable('tld')
             self.assertNotRegistrable('xxx')
             self.assertNotRegistrable('xxx')
             self.assertNotRegistrable('com.uk')
             self.assertNotRegistrable('com.uk')
             self.assertRegistrable('something.else')
             self.assertRegistrable('something.else')
 
 
     def test_can_register_local_public_suffix(self):
     def test_can_register_local_public_suffix(self):
+        local_public_suffixes = ['something.else', 'our.public.suffix', 'com', 'com.uk']
         with self.mock(
         with self.mock(
             global_public_suffixes=['com', 'de', 'xxx', 'com.uk'],
             global_public_suffixes=['com', 'de', 'xxx', 'com.uk'],
-            local_public_suffixes=['something.else', 'our.public.suffix', 'com', 'com.uk'],
+            local_public_suffixes=local_public_suffixes,
         ):
         ):
-            self.assertRegistrable('something.else')
-            self.assertRegistrable('out.public.suffix')
-            self.assertRegistrable('com')
-            self.assertRegistrable('com.uk')
+            for local_public_suffix in local_public_suffixes:
+                self.assertRegistrable(local_public_suffix)
             self.assertRegistrable('foo.bar.com')
             self.assertRegistrable('foo.bar.com')
 
 
     def test_cant_register_descendants_of_children_of_public_suffixes(self):
     def test_cant_register_descendants_of_children_of_public_suffixes(self):