Skip to content

Commit

Permalink
Modified the test cases for partiate domain method
Browse files Browse the repository at this point in the history
  • Loading branch information
Shree7 committed Oct 14, 2019
1 parent b0ee88f commit c8cdd30
Show file tree
Hide file tree
Showing 2 changed files with 32 additions and 3 deletions.
6 changes: 3 additions & 3 deletions dnsgen/test_dnsgen.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,27 +3,27 @@
import dnsgen

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

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

# Test cases for invalid domain names
def test_specialCharDomains(self):
self.assertEqual(dnsgen.partiate_domain("....."),[])
self.assertEqual(dnsgen.partiate_domain(".?@#[email protected]"),[])
self.assertEqual(dnsgen.partiate_domain("1.?@#[email protected]"),['1'])
self.assertEqual(dnsgen.partiate_domain("test.?@#[email protected]"),['test'])
self.assertEqual(dnsgen.partiate_domain(".1...."),['1'])
self.assertEqual(dnsgen.partiate_domain(".1../..c.."),['1'])


if __name__ == '__main__':
unittest.main()


29 changes: 29 additions & 0 deletions tests/test_dnsgen.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@

import unittest
import dnsgen

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

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

# Test cases for invalid domain names
def test_specialCharDomains(self):
self.assertEqual(dnsgen.partiate_domain("....."),[])
self.assertEqual(dnsgen.partiate_domain(".?@#[email protected]"),[])
self.assertEqual(dnsgen.partiate_domain("test.?@#[email protected]"),['test'])
self.assertEqual(dnsgen.partiate_domain(".1...."),['1'])
self.assertEqual(dnsgen.partiate_domain(".1../..c.."),['1'])

if __name__ == '__main__':
unittest.main()

0 comments on commit c8cdd30

Please sign in to comment.