diff options
Diffstat (limited to 'tests/test_tsig.py')
-rw-r--r-- | tests/test_tsig.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/tests/test_tsig.py b/tests/test_tsig.py index b8a38b4..a016cf8 100644 --- a/tests/test_tsig.py +++ b/tests/test_tsig.py @@ -246,7 +246,7 @@ class TSIGTestCase(unittest.TestCase): def test_hmac_sha512_256(self): self._test_truncated_algorithm(dns.tsig.HMAC_SHA512_256, 256) - def text_format(self): + def _test_text_format(self, alg): key = dns.tsig.Key('foo', b'abcdefg', algorithm=alg) q = dns.message.make_query('example', 'a') q.use_tsig(key) @@ -257,10 +257,19 @@ class TSIGTestCase(unittest.TestCase): self.assertEqual(tsig2, q.tsig[0]) q = dns.message.make_query('example', 'a') - q.use_tsig(key, other_data='abc') + q.use_tsig(key, other_data=b'abc') q.use_tsig(key) _ = q.to_wire() text = q.tsig[0].to_text() tsig2 = dns.rdata.from_text('ANY', 'TSIG', text) self.assertEqual(tsig2, q.tsig[0]) + + def test_text_hmac_sha256_128(self): + self._test_text_format(dns.tsig.HMAC_SHA256_128) + + def test_text_hmac_sha384_192(self): + self._test_text_format(dns.tsig.HMAC_SHA384_192) + + def test_text_hmac_sha512_256(self): + self._test_text_format(dns.tsig.HMAC_SHA512_256) |