tsigerror_python_test.py 4.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697
  1. # Copyright (C) 2011 Internet Systems Consortium.
  2. #
  3. # Permission to use, copy, modify, and distribute this software for any
  4. # purpose with or without fee is hereby granted, provided that the above
  5. # copyright notice and this permission notice appear in all copies.
  6. #
  7. # THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SYSTEMS CONSORTIUM
  8. # DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
  9. # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
  10. # INTERNET SYSTEMS CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT,
  11. # INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
  12. # FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
  13. # NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
  14. # WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15. import unittest
  16. import sys
  17. from pydnspp import *
  18. class TSIGErrorTest(unittest.TestCase):
  19. def test_from_code(self):
  20. self.assertEqual(0, TSIGError(0).get_code())
  21. self.assertEqual(18, TSIGError(18).get_code())
  22. self.assertEqual(65535, TSIGError(65535).get_code())
  23. self.assertRaises(ValueError, TSIGError, 65536)
  24. self.assertRaises(ValueError, TSIGError, -1)
  25. self.assertRaises(TypeError, TSIGError, "not yet supported")
  26. def test_from_rcode(self):
  27. # We use RCODE for code values from 0-15.
  28. self.assertEqual(0, TSIGError(Rcode.NOERROR).get_code())
  29. self.assertEqual(15, TSIGError(Rcode(15)).get_code())
  30. # From error code 16 TSIG errors define a separate space, so passing
  31. # corresponding RCODE for such code values should be prohibited.
  32. self.assertRaises(ValueError, TSIGError, Rcode(16))
  33. def test_constants(self):
  34. # We'll only test arbitrarily chosen subsets of the codes.
  35. # This class is quite simple, so it should be suffice.
  36. self.assertEqual(TSIGError.BAD_SIG_CODE, TSIGError(16).get_code())
  37. self.assertEqual(TSIGError.BAD_KEY_CODE, TSIGError(17).get_code())
  38. self.assertEqual(TSIGError.BAD_TIME_CODE, TSIGError(18).get_code())
  39. self.assertEqual(0, TSIGError.NOERROR.get_code())
  40. self.assertEqual(9, TSIGError.NOTAUTH.get_code())
  41. self.assertEqual(14, TSIGError.RESERVED14.get_code())
  42. self.assertEqual(TSIGError.BAD_SIG_CODE, TSIGError.BAD_SIG.get_code())
  43. self.assertEqual(TSIGError.BAD_KEY_CODE, TSIGError.BAD_KEY.get_code())
  44. self.assertEqual(TSIGError.BAD_TIME_CODE, TSIGError.BAD_TIME.get_code())
  45. def test_equal(self):
  46. self.assertTrue(TSIGError.NOERROR == TSIGError(Rcode.NOERROR))
  47. self.assertTrue(TSIGError(Rcode.NOERROR) == TSIGError.NOERROR)
  48. self.assertTrue(TSIGError.BAD_SIG == TSIGError(16))
  49. self.assertTrue(TSIGError(16) == TSIGError.BAD_SIG)
  50. def test_nequal(self):
  51. self.assertTrue(TSIGError.BAD_KEY != TSIGError(Rcode.NOERROR))
  52. self.assertTrue(TSIGError(Rcode.NOERROR) != TSIGError.BAD_KEY)
  53. def test_to_text(self):
  54. # TSIGError derived from the standard Rcode
  55. self.assertEqual("NOERROR", TSIGError(Rcode.NOERROR).to_text())
  56. # Well known TSIG errors
  57. self.assertEqual("BADSIG", TSIGError.BAD_SIG.to_text())
  58. self.assertEqual("BADKEY", TSIGError.BAD_KEY.to_text())
  59. self.assertEqual("BADTIME", TSIGError.BAD_TIME.to_text())
  60. # Unknown (or not yet supported) codes. Simply converted as numeric.
  61. self.assertEqual("19", TSIGError(19).to_text());
  62. self.assertEqual("65535", TSIGError(65535).to_text());
  63. # also check str() works same way
  64. self.assertEqual("NOERROR", str(TSIGError(Rcode.NOERROR)))
  65. self.assertEqual("BADSIG", str(TSIGError.BAD_SIG))
  66. def test_to_rcode(self):
  67. # TSIGError derived from the standard Rcode
  68. self.assertEqual(Rcode.NOERROR, TSIGError(Rcode.NOERROR).to_rcode())
  69. # Well known TSIG errors
  70. self.assertEqual(Rcode.NOTAUTH, TSIGError.BAD_SIG.to_rcode())
  71. self.assertEqual(Rcode.NOTAUTH, TSIGError.BAD_KEY.to_rcode())
  72. self.assertEqual(Rcode.NOTAUTH, TSIGError.BAD_TIME.to_rcode())
  73. # Unknown (or not yet supported) codes are treated as SERVFAIL.
  74. self.assertEqual(Rcode.SERVFAIL, TSIGError(19).to_rcode())
  75. self.assertEqual(Rcode.SERVFAIL, TSIGError(65535).to_rcode())
  76. # Check there's no redundant refcount (which would cause leak)
  77. self.assertEqual(1, sys.getrefcount(TSIGError.BAD_SIG.to_rcode()))
  78. if __name__ == '__main__':
  79. unittest.main()