12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697 |
- import unittest
- import sys
- from pydnspp import *
- class TSIGErrorTest(unittest.TestCase):
- def test_from_code(self):
- self.assertEqual(0, TSIGError(0).get_code())
- self.assertEqual(18, TSIGError(18).get_code())
- self.assertEqual(65535, TSIGError(65535).get_code())
- self.assertRaises(ValueError, TSIGError, 65536)
- self.assertRaises(ValueError, TSIGError, -1)
- self.assertRaises(TypeError, TSIGError, "not yet supported")
- def test_from_rcode(self):
-
- self.assertEqual(0, TSIGError(Rcode.NOERROR).get_code())
- self.assertEqual(15, TSIGError(Rcode(15)).get_code())
-
-
- self.assertRaises(ValueError, TSIGError, Rcode(16))
- def test_constants(self):
-
-
- self.assertEqual(TSIGError.BAD_SIG_CODE, TSIGError(16).get_code())
- self.assertEqual(TSIGError.BAD_KEY_CODE, TSIGError(17).get_code())
- self.assertEqual(TSIGError.BAD_TIME_CODE, TSIGError(18).get_code())
- self.assertEqual(0, TSIGError.NOERROR.get_code())
- self.assertEqual(9, TSIGError.NOTAUTH.get_code())
- self.assertEqual(14, TSIGError.RESERVED14.get_code())
- self.assertEqual(TSIGError.BAD_SIG_CODE, TSIGError.BAD_SIG.get_code())
- self.assertEqual(TSIGError.BAD_KEY_CODE, TSIGError.BAD_KEY.get_code())
- self.assertEqual(TSIGError.BAD_TIME_CODE, TSIGError.BAD_TIME.get_code())
- def test_equal(self):
- self.assertTrue(TSIGError.NOERROR == TSIGError(Rcode.NOERROR))
- self.assertTrue(TSIGError(Rcode.NOERROR) == TSIGError.NOERROR)
- self.assertTrue(TSIGError.BAD_SIG == TSIGError(16))
- self.assertTrue(TSIGError(16) == TSIGError.BAD_SIG)
- def test_nequal(self):
- self.assertTrue(TSIGError.BAD_KEY != TSIGError(Rcode.NOERROR))
- self.assertTrue(TSIGError(Rcode.NOERROR) != TSIGError.BAD_KEY)
- def test_to_text(self):
-
- self.assertEqual("NOERROR", TSIGError(Rcode.NOERROR).to_text())
-
- self.assertEqual("BADSIG", TSIGError.BAD_SIG.to_text())
- self.assertEqual("BADKEY", TSIGError.BAD_KEY.to_text())
- self.assertEqual("BADTIME", TSIGError.BAD_TIME.to_text())
-
- self.assertEqual("19", TSIGError(19).to_text());
- self.assertEqual("65535", TSIGError(65535).to_text());
-
- self.assertEqual("NOERROR", str(TSIGError(Rcode.NOERROR)))
- self.assertEqual("BADSIG", str(TSIGError.BAD_SIG))
- def test_to_rcode(self):
-
- self.assertEqual(Rcode.NOERROR, TSIGError(Rcode.NOERROR).to_rcode())
-
- self.assertEqual(Rcode.NOTAUTH, TSIGError.BAD_SIG.to_rcode())
- self.assertEqual(Rcode.NOTAUTH, TSIGError.BAD_KEY.to_rcode())
- self.assertEqual(Rcode.NOTAUTH, TSIGError.BAD_TIME.to_rcode())
-
- self.assertEqual(Rcode.SERVFAIL, TSIGError(19).to_rcode())
- self.assertEqual(Rcode.SERVFAIL, TSIGError(65535).to_rcode())
-
- self.assertEqual(1, sys.getrefcount(TSIGError.BAD_SIG.to_rcode()))
- if __name__ == '__main__':
- unittest.main()
|