response = self.sendUDPQuery(query)
self.assertEqual(len(response.options), 1)
- self.assertEqual(response.options[0].data, self._servername.encode('ascii'))
+ if dns.version.MAJOR < 2 or (dns.version.MAJOR == 2 and dns.version.MINOR < 6):
+ self.assertEqual(response.options[0].data, self._servername.encode('ascii'))
+ else:
+ self.assertEqual(response.options[0].to_text(), 'NSID ' + self._servername)
def testNSIDTCP(self):
"""
response = self.sendTCPQuery(query)
self.assertEqual(len(response.options), 1)
- self.assertEqual(response.options[0].data, self._servername.encode('ascii'))
+ if dns.version.MAJOR < 2 or (dns.version.MAJOR == 2 and dns.version.MINOR < 6):
+ self.assertEqual(response.options[0].data, self._servername.encode('ascii'))
+ else:
+ self.assertEqual(response.options[0].to_text(), 'NSID ' + self._servername)