rdata_python_test.py 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. # Copyright (C) 2010 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. #
  16. # Tests for the rdata part of the libdns_python module
  17. #
  18. import unittest
  19. import os
  20. from libdns_python import *
  21. class RdataTest(unittest.TestCase):
  22. def setUp(self):
  23. c = RRClass("IN")
  24. t = RRType("A")
  25. self.rdata1 = Rdata(t, c, "192.0.2.98")
  26. self.rdata2 = Rdata(t, c, "192.0.2.99")
  27. t = RRType("TXT")
  28. self.rdata3 = Rdata(t, c, "asdfasdfasdf")
  29. self.rdata4 = Rdata(t, c, "foo")
  30. def test_init(self):
  31. self.assertRaises(TypeError, Rdata, "wrong", RRClass("IN"), "192.0.2.99")
  32. self.assertRaises(TypeError, Rdata, RRType("A"), "wrong", "192.0.2.99")
  33. self.assertRaises(TypeError, Rdata, RRType("A"), RRClass("IN"), 1)
  34. def test_rdata_to_wire(self):
  35. b = bytearray()
  36. self.rdata1.to_wire(b)
  37. self.assertEqual(b'\xc0\x00\x02b', b)
  38. b = bytearray()
  39. self.rdata2.to_wire(b)
  40. self.assertEqual(b'\xc0\x00\x02c', b)
  41. b = bytearray()
  42. self.rdata3.to_wire(b)
  43. self.assertEqual(b'\x0casdfasdfasdf', b)
  44. b = bytearray()
  45. self.rdata4.to_wire(b)
  46. self.assertEqual(b'\x03foo', b)
  47. self.assertRaises(TypeError, self.rdata1.to_wire, 1)
  48. def test_rdata_to_text(self):
  49. self.assertEqual("192.0.2.98", self.rdata1.to_text())
  50. self.assertEqual("192.0.2.99", self.rdata2.to_text())
  51. self.assertEqual("\"asdfasdfasdf\"", self.rdata3.to_text())
  52. self.assertEqual("\"foo\"", self.rdata4.to_text())
  53. if __name__ == '__main__':
  54. unittest.main()