test/test_main.py
author Evgeni Kunev <evgeni.kunev@gmail.com>
Mon, 25 Aug 2014 15:00:56 +0300
changeset 43 f7bf8d6f0547
parent 35 0de2468a27e8
child 70 1fe2c20adeba
permissions -rw-r--r--
[#59] Add support for dateutil.parser dayfirst and yearfirst arguments The WhoisEntry object should expose a dayfirst and yearfist argument to signify how the datetime format for the TLD should be parsed by dateutil. The base WhoisEntry class gives both those fields a value of False. Those are also the default values dateutil assumes.

# coding=utf-8

import unittest
from whois import extract_domain


class TestExtractDomain(unittest.TestCase):
    def test_simple_ascii_domain(self):
        url = 'google.com'
        domain = url
        self.assertEqual(domain, extract_domain(url))

    def test_ascii_with_schema_path_and_query(self):
        url = 'https://www.google.com/search?q=why+is+domain+whois+such+a+mess'
        domain = 'google.com'
        self.assertEqual(domain, extract_domain(url))

    def test_simple_unicode_domain(self):
        url = 'http://нарояци.com/'
        domain = 'нарояци.com'
        self.assertEqual(domain, extract_domain(url))

    def test_unicode_domain_and_tld(self):
        url = 'http://россия.рф/'
        domain = 'россия.рф'
        self.assertEqual(domain, extract_domain(url))