12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- import unittest
- import os
- from libdns_python import *
- class RdataTest(unittest.TestCase):
- def setUp(self):
- c = RRClass("IN")
- t = RRType("A")
- self.rdata1 = Rdata(t, c, "192.0.2.98")
- self.rdata2 = Rdata(t, c, "192.0.2.99")
- t = RRType("TXT")
- self.rdata3 = Rdata(t, c, "asdfasdfasdf")
- self.rdata4 = Rdata(t, c, "foo")
-
- def test_init(self):
- self.assertRaises(TypeError, Rdata, "wrong", RRClass("IN"), "192.0.2.99")
- self.assertRaises(TypeError, Rdata, RRType("A"), "wrong", "192.0.2.99")
- self.assertRaises(TypeError, Rdata, RRType("A"), RRClass("IN"), 1)
- def test_rdata_to_wire(self):
- b = bytearray()
- self.rdata1.to_wire(b)
- self.assertEqual(b'\xc0\x00\x02b', b)
- b = bytearray()
- self.rdata2.to_wire(b)
- self.assertEqual(b'\xc0\x00\x02c', b)
- b = bytearray()
- self.rdata3.to_wire(b)
- self.assertEqual(b'\x0casdfasdfasdf', b)
- b = bytearray()
- self.rdata4.to_wire(b)
- self.assertEqual(b'\x03foo', b)
- self.assertRaises(TypeError, self.rdata1.to_wire, 1)
- def test_rdata_to_text(self):
- self.assertEqual("192.0.2.98", self.rdata1.to_text())
- self.assertEqual("192.0.2.99", self.rdata2.to_text())
- self.assertEqual("\"asdfasdfasdf\"", self.rdata3.to_text())
- self.assertEqual("\"foo\"", self.rdata4.to_text())
- if __name__ == '__main__':
- unittest.main()
|