1 # coding=utf-8

     2 

     3 import unittest

     4 from whois.whois import NICClient

     5 

     6 

     7 class TestNICClient(unittest.TestCase):

     8     def setUp(self):

     9         self.client = NICClient()

    10 

    11     def test_choose_server(self):

    12         domain = 'рнидс.срб'

    13         chosen = self.client.choose_server(domain)

    14         self.assertEqual(

    15             chosen,

    16             'срб'.decode('utf-8').encode('idna') + '.whois-servers.net'

    17         )