test/test_nicclient.py
author Evgeni Kunev <evgeni.kunev@gmail.com>
Mon, 10 Nov 2014 13:42:37 +0200
changeset 45 52ce01013731
parent 36 af839b9c0ed1
child 70 1fe2c20adeba
permissions -rw-r--r--
Handle experimental lookup method for bad domains Attempting to use NICClient with bad domains could hit a bad logical path that leads to an attempt to reference a variable that hasn't been defined. So in case we don't manage to find a nichost for a domain, just set `result` to `None`

# coding=utf-8

import unittest
from whois.whois import NICClient


class TestNICClient(unittest.TestCase):
    def setUp(self):
        self.client = NICClient()

    def test_choose_server(self):
        domain = 'рнидс.срб'
        chosen = self.client.choose_server(domain)
        self.assertEqual(
            chosen,
            'срб'.decode('utf-8').encode('idna') + '.whois-servers.net'
        )