test/test_nicclient.py
changeset 97 44522cd37b07
parent 70 1fe2c20adeba
--- a/test/test_nicclient.py	Fri Apr 15 09:21:42 2016 -0600
+++ b/test/test_nicclient.py	Fri Apr 15 18:29:24 2016 -0600
@@ -18,7 +18,6 @@
     def test_choose_server(self):
         domain = 'рнидс.срб'
         chosen = self.client.choose_server(domain)
-        self.assertEqual(
-            chosen,
-            'срб'.decode('utf-8').encode('idna') + '.whois-servers.net'
-        )
+        suffix = domain.split('.')[-1].encode('idna').decode('utf-8')
+        correct = '{}.whois-servers.net'.format(suffix)
+        self.assertEqual(chosen, correct)