1 # coding=utf-8

     2 

     3 import unittest

     4 from whois import extract_domain

     5 

     6 

     7 class TestExtractDomain(unittest.TestCase):

     8     def test_simple_ascii_domain(self):

     9         url = 'google.com'

    10         domain = url

    11         self.assertEqual(domain, extract_domain(url))

    12 

    13     def test_ascii_with_schema_path_and_query(self):

    14         url = 'https://www.google.com/search?q=why+is+domain+whois+such+a+mess'

    15         domain = 'google.com'

    16         self.assertEqual(domain, extract_domain(url))

    17 

    18     def test_simple_unicode_domain(self):

    19         url = 'http://нарояци.com/'

    20         domain = 'нарояци.com'

    21         self.assertEqual(domain, extract_domain(url))

    22 

    23     def test_unicode_domain_and_tld(self):

    24         url = 'http://россия.рф/'

    25         domain = 'россия.рф'

    26         self.assertEqual(domain, extract_domain(url))