# 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'
        )