|
3 | 3 | import dnsgen |
4 | 4 |
|
5 | 5 | class Test_PartiateDomain(unittest.TestCase): |
| 6 | + # Test cases for valid domain names |
6 | 7 | def test_generalDomains(self): |
7 | 8 | self.assertEqual(dnsgen.partiate_domain("test.1.foo.example.com"),['test', '1', 'foo', 'example', 'com']) |
8 | 9 | self.assertEqual(dnsgen.partiate_domain("test.2.foo.example.com.cn"),['test', '2', 'foo', 'example', 'com.cn']) |
9 | 10 | self.assertEqual(dnsgen.partiate_domain("test.example.co.uk"),['test', 'example', 'co.uk']) |
10 | 11 |
|
| 12 | + # Test cases for domain names with special characters |
11 | 13 | def test_websiteDomains(self): |
12 | 14 | self.assertEqual(dnsgen.partiate_domain("https://www.google.com"),['www', 'google', 'com']) |
13 | 15 | self.assertEqual(dnsgen.partiate_domain("www.letter.com.1.cn"),['www','letter', 'com', '1','cn']) |
14 | 16 | self.assertEqual(dnsgen.partiate_domain(".test.com.1."),['test', 'com', '1']) |
15 | 17 | self.assertEqual(dnsgen.partiate_domain(".test.com.1.c.a/b"),['test', 'com', '1','c','a']) |
16 | 18 |
|
| 19 | + # Test cases for invalid domain names |
17 | 20 | def test_specialCharDomains(self): |
18 | 21 | self.assertEqual(dnsgen.partiate_domain("....."),[]) |
19 | 22 | self.assertEqual(dnsgen.partiate_domain(".?@#3.fde!.@.qwepo"),[]) |
20 | | - self.assertEqual(dnsgen.partiate_domain("1.?@#3.fde!.@.qwetg"),['1']) |
21 | 23 | self.assertEqual(dnsgen.partiate_domain("test.?@#3.fde!.@.qwedg"),['test']) |
22 | 24 | self.assertEqual(dnsgen.partiate_domain(".1...."),['1']) |
23 | 25 | self.assertEqual(dnsgen.partiate_domain(".1../..c.."),['1']) |
24 | 26 |
|
25 | | - |
26 | 27 | if __name__ == '__main__': |
27 | 28 | unittest.main() |
28 | 29 |
|
29 | | - |
0 commit comments