Skip to content

Commit 618a39a

Browse files
committed
Modified the test cases for partiate domain method
1 parent b0ee88f commit 618a39a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dnsgen/test_dnsgen.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,27 +3,27 @@
33
import dnsgen
44

55
class Test_PartiateDomain(unittest.TestCase):
6+
# Test cases for valid domain names
67
def test_generalDomains(self):
78
self.assertEqual(dnsgen.partiate_domain("test.1.foo.example.com"),['test', '1', 'foo', 'example', 'com'])
89
self.assertEqual(dnsgen.partiate_domain("test.2.foo.example.com.cn"),['test', '2', 'foo', 'example', 'com.cn'])
910
self.assertEqual(dnsgen.partiate_domain("test.example.co.uk"),['test', 'example', 'co.uk'])
1011

12+
# Test cases for domain names with special characters
1113
def test_websiteDomains(self):
1214
self.assertEqual(dnsgen.partiate_domain("https://www.google.com"),['www', 'google', 'com'])
1315
self.assertEqual(dnsgen.partiate_domain("www.letter.com.1.cn"),['www','letter', 'com', '1','cn'])
1416
self.assertEqual(dnsgen.partiate_domain(".test.com.1."),['test', 'com', '1'])
1517
self.assertEqual(dnsgen.partiate_domain(".test.com.1.c.a/b"),['test', 'com', '1','c','a'])
1618

19+
# Test cases for invalid domain names
1720
def test_specialCharDomains(self):
1821
self.assertEqual(dnsgen.partiate_domain("....."),[])
1922
self.assertEqual(dnsgen.partiate_domain(".?@#3.fde!.@.qwepo"),[])
20-
self.assertEqual(dnsgen.partiate_domain("1.?@#3.fde!.@.qwetg"),['1'])
2123
self.assertEqual(dnsgen.partiate_domain("test.?@#3.fde!.@.qwedg"),['test'])
2224
self.assertEqual(dnsgen.partiate_domain(".1...."),['1'])
2325
self.assertEqual(dnsgen.partiate_domain(".1../..c.."),['1'])
2426

25-
2627
if __name__ == '__main__':
2728
unittest.main()
2829

29-

0 commit comments

Comments
 (0)