message_python_test.py 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521
  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 message part of the pydnspp module
  17. #
  18. import unittest
  19. import os
  20. from pydnspp import *
  21. from testutil import *
  22. class MessageFlagTest(unittest.TestCase):
  23. def test_init(self):
  24. self.assertRaises(NotImplementedError, MessageFlag)
  25. def test_get_bit(self):
  26. self.assertEqual(0x8000, MessageFlag.QR().get_bit())
  27. self.assertEqual(0x0400, MessageFlag.AA().get_bit())
  28. self.assertEqual(0x0200, MessageFlag.TC().get_bit())
  29. self.assertEqual(0x0100, MessageFlag.RD().get_bit())
  30. self.assertEqual(0x0080, MessageFlag.RA().get_bit())
  31. self.assertEqual(0x0020, MessageFlag.AD().get_bit())
  32. self.assertEqual(0x0010, MessageFlag.CD().get_bit())
  33. class OpcodeTest(unittest.TestCase):
  34. def test_init(self):
  35. self.assertRaises(NotImplementedError, Opcode)
  36. def test_get_code(self):
  37. self.assertEqual(0, Opcode.QUERY().get_code())
  38. self.assertEqual(1, Opcode.IQUERY().get_code())
  39. self.assertEqual(2, Opcode.STATUS().get_code())
  40. self.assertEqual(3, Opcode.RESERVED3().get_code())
  41. self.assertEqual(4, Opcode.NOTIFY().get_code())
  42. self.assertEqual(5, Opcode.UPDATE().get_code())
  43. self.assertEqual(6, Opcode.RESERVED6().get_code())
  44. self.assertEqual(7, Opcode.RESERVED7().get_code())
  45. self.assertEqual(8, Opcode.RESERVED8().get_code())
  46. self.assertEqual(9, Opcode.RESERVED9().get_code())
  47. self.assertEqual(10, Opcode.RESERVED10().get_code())
  48. self.assertEqual(11, Opcode.RESERVED11().get_code())
  49. self.assertEqual(12, Opcode.RESERVED12().get_code())
  50. self.assertEqual(13, Opcode.RESERVED13().get_code())
  51. self.assertEqual(14, Opcode.RESERVED14().get_code())
  52. self.assertEqual(15, Opcode.RESERVED15().get_code())
  53. def test_to_text(self):
  54. self.assertEqual("QUERY", Opcode.QUERY().to_text())
  55. self.assertEqual("QUERY", str(Opcode.QUERY()))
  56. self.assertEqual("IQUERY", Opcode.IQUERY().to_text())
  57. self.assertEqual("STATUS", Opcode.STATUS().to_text())
  58. self.assertEqual("RESERVED3", Opcode.RESERVED3().to_text())
  59. self.assertEqual("NOTIFY", Opcode.NOTIFY().to_text())
  60. self.assertEqual("UPDATE", Opcode.UPDATE().to_text())
  61. self.assertEqual("RESERVED6", Opcode.RESERVED6().to_text())
  62. self.assertEqual("RESERVED7", Opcode.RESERVED7().to_text())
  63. self.assertEqual("RESERVED8", Opcode.RESERVED8().to_text())
  64. self.assertEqual("RESERVED9", Opcode.RESERVED9().to_text())
  65. self.assertEqual("RESERVED10", Opcode.RESERVED10().to_text())
  66. self.assertEqual("RESERVED11", Opcode.RESERVED11().to_text())
  67. self.assertEqual("RESERVED12", Opcode.RESERVED12().to_text())
  68. self.assertEqual("RESERVED13", Opcode.RESERVED13().to_text())
  69. self.assertEqual("RESERVED14", Opcode.RESERVED14().to_text())
  70. self.assertEqual("RESERVED15", Opcode.RESERVED15().to_text())
  71. def test_richcmp(self):
  72. o1 = Opcode.QUERY()
  73. o2 = Opcode.NOTIFY()
  74. o3 = Opcode.NOTIFY()
  75. self.assertTrue(o2 == o3)
  76. self.assertFalse(o2 != o3)
  77. self.assertTrue(o1 != o2)
  78. self.assertFalse(o1 == 1)
  79. self.assertFalse(o1 == o2)
  80. # can't use assertRaises here...
  81. try:
  82. o1 < o2
  83. self.fail("operation that should have raised an error unexpectedly succeeded")
  84. except Exception as err:
  85. self.assertEqual(TypeError, type(err))
  86. try:
  87. o1 <= o2
  88. self.fail("operation that should have raised an error unexpectedly succeeded")
  89. except Exception as err:
  90. self.assertEqual(TypeError, type(err))
  91. try:
  92. o1 > o2
  93. self.fail("operation that should have raised an error unexpectedly succeeded")
  94. except Exception as err:
  95. self.assertEqual(TypeError, type(err))
  96. try:
  97. o1 >= o2
  98. self.fail("operation that should have raised an error unexpectedly succeeded")
  99. except Exception as err:
  100. self.assertEqual(TypeError, type(err))
  101. class RcodeTest(unittest.TestCase):
  102. def test_init(self):
  103. self.assertRaises(TypeError, Rcode, "wrong")
  104. self.assertRaises(OverflowError, Rcode, 65536)
  105. self.assertEqual(Rcode(0).get_code(), 0)
  106. self.assertEqual(0, Rcode(0).get_code())
  107. self.assertEqual(0xfff, Rcode(0xfff).get_code()) # possible max code
  108. # should fail on attempt of construction with an out of range code
  109. self.assertRaises(OverflowError, Rcode, 0x1000)
  110. self.assertRaises(OverflowError, Rcode, 0xffff)
  111. def test_get_code(self):
  112. self.assertEqual(0, Rcode.NOERROR().get_code())
  113. self.assertEqual(1, Rcode.FORMERR().get_code())
  114. self.assertEqual(2, Rcode.SERVFAIL().get_code())
  115. self.assertEqual(3, Rcode.NXDOMAIN().get_code())
  116. self.assertEqual(4, Rcode.NOTIMP().get_code())
  117. self.assertEqual(5, Rcode.REFUSED().get_code())
  118. self.assertEqual(6, Rcode.YXDOMAIN().get_code())
  119. self.assertEqual(7, Rcode.YXRRSET().get_code())
  120. self.assertEqual(8, Rcode.NXRRSET().get_code())
  121. self.assertEqual(9, Rcode.NOTAUTH().get_code())
  122. self.assertEqual(10, Rcode.NOTZONE().get_code())
  123. self.assertEqual(11, Rcode.RESERVED11().get_code())
  124. self.assertEqual(12, Rcode.RESERVED12().get_code())
  125. self.assertEqual(13, Rcode.RESERVED13().get_code())
  126. self.assertEqual(14, Rcode.RESERVED14().get_code())
  127. self.assertEqual(15, Rcode.RESERVED15().get_code())
  128. self.assertEqual(16, Rcode.BADVERS().get_code())
  129. def test_to_text(self):
  130. self.assertEqual("NOERROR", Rcode(0).to_text())
  131. self.assertEqual("NOERROR", str(Rcode(0)))
  132. self.assertEqual("FORMERR", Rcode(1).to_text())
  133. self.assertEqual("SERVFAIL", Rcode(2).to_text())
  134. self.assertEqual("NXDOMAIN", Rcode(3).to_text())
  135. self.assertEqual("NOTIMP", Rcode(4).to_text())
  136. self.assertEqual("REFUSED", Rcode(5).to_text())
  137. self.assertEqual("YXDOMAIN", Rcode(6).to_text())
  138. self.assertEqual("YXRRSET", Rcode(7).to_text())
  139. self.assertEqual("NXRRSET", Rcode(8).to_text())
  140. self.assertEqual("NOTAUTH", Rcode(9).to_text())
  141. self.assertEqual("NOTZONE", Rcode(10).to_text())
  142. self.assertEqual("RESERVED11", Rcode(11).to_text())
  143. self.assertEqual("RESERVED12", Rcode(12).to_text())
  144. self.assertEqual("RESERVED13", Rcode(13).to_text())
  145. self.assertEqual("RESERVED14", Rcode(14).to_text())
  146. self.assertEqual("RESERVED15", Rcode(15).to_text())
  147. self.assertEqual("BADVERS", Rcode(16).to_text())
  148. self.assertEqual("17", Rcode(Rcode.BADVERS().get_code() + 1).to_text())
  149. self.assertEqual("4095", Rcode(0xfff).to_text())
  150. def test_richcmp(self):
  151. r1 = Rcode.NOERROR()
  152. r2 = Rcode.FORMERR()
  153. r3 = Rcode.FORMERR()
  154. self.assertTrue(r2 == r3)
  155. self.assertTrue(r1 != r2)
  156. self.assertFalse(r1 == r2)
  157. self.assertFalse(r1 != 1)
  158. # can't use assertRaises here...
  159. try:
  160. r1 < r2
  161. self.fail("operation that should have raised an error unexpectedly succeeded")
  162. except Exception as err:
  163. self.assertEqual(TypeError, type(err))
  164. try:
  165. r1 <= r2
  166. self.fail("operation that should have raised an error unexpectedly succeeded")
  167. except Exception as err:
  168. self.assertEqual(TypeError, type(err))
  169. try:
  170. r1 > r2
  171. self.fail("operation that should have raised an error unexpectedly succeeded")
  172. except Exception as err:
  173. self.assertEqual(TypeError, type(err))
  174. try:
  175. r1 >= r2
  176. self.fail("operation that should have raised an error unexpectedly succeeded")
  177. except Exception as err:
  178. self.assertEqual(TypeError, type(err))
  179. class SectionTest(unittest.TestCase):
  180. def test_init(self):
  181. self.assertRaises(NotImplementedError, Section)
  182. def test_get_code(self):
  183. self.assertEqual(0, Section.QUESTION().get_code())
  184. self.assertEqual(1, Section.ANSWER().get_code())
  185. self.assertEqual(2, Section.AUTHORITY().get_code())
  186. self.assertEqual(3, Section.ADDITIONAL().get_code())
  187. def test_richcmp(self):
  188. s1 = Section.QUESTION()
  189. s2 = Section.ANSWER()
  190. s3 = Section.ANSWER()
  191. self.assertTrue(s2 == s3)
  192. self.assertTrue(s1 != s2)
  193. self.assertFalse(s1 == s2)
  194. self.assertFalse(s1 == 1)
  195. # can't use assertRaises here...
  196. try:
  197. s1 < s2
  198. self.fail("operation that should have raised an error unexpectedly succeeded")
  199. except Exception as err:
  200. self.assertEqual(TypeError, type(err))
  201. try:
  202. s1 <= s2
  203. self.fail("operation that should have raised an error unexpectedly succeeded")
  204. except Exception as err:
  205. self.assertEqual(TypeError, type(err))
  206. try:
  207. s1 > s2
  208. self.fail("operation that should have raised an error unexpectedly succeeded")
  209. except Exception as err:
  210. self.assertEqual(TypeError, type(err))
  211. try:
  212. s1 >= s2
  213. self.fail("operation that should have raised an error unexpectedly succeeded")
  214. except Exception as err:
  215. self.assertEqual(TypeError, type(err))
  216. # helper functions for tests taken from c++ unittests
  217. if "TESTDATA_PATH" in os.environ:
  218. testdata_path = os.environ["TESTDATA_PATH"]
  219. else:
  220. testdata_path = "../tests/testdata"
  221. def factoryFromFile(message, file):
  222. data = read_wire_data(file)
  223. message.from_wire(data)
  224. pass
  225. # we don't have direct comparison for rrsets right now (should we?
  226. # should go in the cpp version first then), so also no direct list
  227. # comparison. Created a helper function
  228. def compare_rrset_list(list1, list2):
  229. if len(list1) != len(list2):
  230. return False
  231. for i in range(0, len(list1)):
  232. if str(list1[i]) != str(list2[i]):
  233. return False
  234. return True
  235. # a complete message taken from cpp tests, for testing towire and totext
  236. def create_message():
  237. message_render = Message(Message.RENDER)
  238. message_render.set_qid(0x1035)
  239. message_render.set_opcode(Opcode.QUERY())
  240. message_render.set_rcode(Rcode.NOERROR())
  241. message_render.set_header_flag(MessageFlag.QR())
  242. message_render.set_header_flag(MessageFlag.RD())
  243. message_render.set_header_flag(MessageFlag.AA())
  244. message_render.add_question(Question(Name("test.example.com"), RRClass("IN"), RRType("A")))
  245. rrset = RRset(Name("test.example.com"), RRClass("IN"),
  246. RRType("A"), RRTTL(3600))
  247. rrset.add_rdata(Rdata(RRType("A"), RRClass("IN"), "192.0.2.1"))
  248. rrset.add_rdata(Rdata(RRType("A"), RRClass("IN"), "192.0.2.2"))
  249. message_render.add_rrset(Section.ANSWER(), rrset)
  250. return message_render
  251. class MessageTest(unittest.TestCase):
  252. def setUp(self):
  253. self.p = Message(Message.PARSE)
  254. self.r = Message(Message.RENDER)
  255. def test_init(self):
  256. self.assertRaises(TypeError, Message, 3)
  257. self.assertRaises(TypeError, Message, "wrong")
  258. def test_get_header_flag(self):
  259. self.assertRaises(TypeError, self.p.get_header_flag, "wrong")
  260. self.assertFalse(self.p.get_header_flag(MessageFlag.AA()))
  261. def test_set_clear_header_flag(self):
  262. self.assertRaises(TypeError, self.r.set_header_flag, "wrong")
  263. self.assertRaises(TypeError, self.r.clear_header_flag, "wrong")
  264. self.assertFalse(self.r.get_header_flag(MessageFlag.AA()))
  265. self.r.set_header_flag(MessageFlag.AA())
  266. self.assertTrue(self.r.get_header_flag(MessageFlag.AA()))
  267. self.r.clear_header_flag(MessageFlag.AA())
  268. self.assertFalse(self.r.get_header_flag(MessageFlag.AA()))
  269. self.assertRaises(InvalidMessageOperation,
  270. self.p.set_header_flag, MessageFlag.AA())
  271. self.assertRaises(InvalidMessageOperation,
  272. self.p.clear_header_flag, MessageFlag.AA())
  273. def test_set_qid(self):
  274. self.assertRaises(TypeError, self.r.set_qid, "wrong")
  275. self.assertRaises(InvalidMessageOperation,
  276. self.p.set_qid, 123)
  277. self.r.set_qid(1234)
  278. self.assertEqual(1234, self.r.get_qid())
  279. def test_set_rcode(self):
  280. self.assertRaises(TypeError, self.r.set_rcode, "wrong")
  281. rcode = Rcode.BADVERS()
  282. self.r.set_rcode(rcode)
  283. self.assertEqual(rcode, self.r.get_rcode())
  284. self.assertRaises(InvalidMessageOperation,
  285. self.p.set_rcode, rcode)
  286. def test_set_opcode(self):
  287. self.assertRaises(TypeError, self.r.set_opcode, "wrong")
  288. opcode = Opcode.IQUERY()
  289. self.r.set_opcode(opcode)
  290. self.assertEqual(opcode, self.r.get_opcode())
  291. self.assertRaises(InvalidMessageOperation,
  292. self.p.set_opcode, opcode)
  293. def test_get_edns(self):
  294. self.assertEqual(None, self.p.get_edns())
  295. message_parse = Message(Message.PARSE)
  296. factoryFromFile(message_parse, "message_fromWire10.wire")
  297. edns = message_parse.get_edns()
  298. self.assertEqual(0, edns.get_version())
  299. self.assertEqual(4096, edns.get_udp_size())
  300. self.assertTrue(edns.get_dnssec_awareness())
  301. def test_set_edns(self):
  302. self.assertRaises(InvalidMessageOperation, self.p.set_edns, EDNS())
  303. edns = EDNS()
  304. edns.set_udp_size(1024)
  305. self.r.set_edns(edns)
  306. self.assertEqual(1024, self.r.get_edns().get_udp_size())
  307. def test_get_section(self):
  308. self.assertRaises(TypeError, self.r.get_section, "wrong")
  309. rrset = RRset(Name("example.com"), RRClass("IN"), RRType("A"), RRTTL(3600))
  310. rrset.add_rdata(Rdata(RRType("A"), RRClass("IN"), "192.0.2.1"))
  311. rrset.add_rdata(Rdata(RRType("A"), RRClass("IN"), "192.0.2.2"))
  312. section_rrset = [rrset]
  313. self.assertRaises(InvalidMessageOperation, self.p.add_rrset,
  314. Section.ANSWER(), rrset)
  315. self.assertFalse(compare_rrset_list(section_rrset, self.r.get_section(Section.ANSWER())))
  316. self.assertEqual(0, self.r.get_rr_count(Section.ANSWER()))
  317. self.r.add_rrset(Section.ANSWER(), rrset)
  318. self.assertTrue(compare_rrset_list(section_rrset, self.r.get_section(Section.ANSWER())))
  319. self.assertEqual(2, self.r.get_rr_count(Section.ANSWER()))
  320. self.assertFalse(compare_rrset_list(section_rrset, self.r.get_section(Section.AUTHORITY())))
  321. self.assertEqual(0, self.r.get_rr_count(Section.AUTHORITY()))
  322. self.r.add_rrset(Section.AUTHORITY(), rrset)
  323. self.assertTrue(compare_rrset_list(section_rrset, self.r.get_section(Section.AUTHORITY())))
  324. self.assertEqual(2, self.r.get_rr_count(Section.AUTHORITY()))
  325. self.assertFalse(compare_rrset_list(section_rrset, self.r.get_section(Section.ADDITIONAL())))
  326. self.assertEqual(0, self.r.get_rr_count(Section.ADDITIONAL()))
  327. self.r.add_rrset(Section.ADDITIONAL(), rrset)
  328. self.assertTrue(compare_rrset_list(section_rrset, self.r.get_section(Section.ADDITIONAL())))
  329. self.assertEqual(2, self.r.get_rr_count(Section.ADDITIONAL()))
  330. def test_get_rr_count(self):
  331. self.assertRaises(TypeError, self.r.get_rr_count, "wrong")
  332. # counts also tested in add_section
  333. def test_add_question(self):
  334. self.assertRaises(TypeError, self.r.add_question, "wrong", "wrong")
  335. q = Question(Name("example.com"), RRClass("IN"), RRType("A"))
  336. qs = [q]
  337. self.assertFalse(compare_rrset_list(qs, self.r.get_question()))
  338. self.assertEqual(0, self.r.get_rr_count(Section.QUESTION()))
  339. self.r.add_question(q)
  340. self.assertTrue(compare_rrset_list(qs, self.r.get_question()))
  341. self.assertEqual(1, self.r.get_rr_count(Section.QUESTION()))
  342. def test_add_rrset(self):
  343. self.assertRaises(TypeError, self.r.add_rrset, "wrong")
  344. # actual addition already tested in get_section
  345. def test_clear(self):
  346. self.assertEqual(None, self.r.clear(Message.PARSE))
  347. self.assertEqual(None, self.r.clear(Message.RENDER))
  348. self.assertRaises(TypeError, self.r.clear, "wrong")
  349. self.assertRaises(TypeError, self.r.clear, 3)
  350. def test_to_wire(self):
  351. self.assertRaises(TypeError, self.r.to_wire, 1)
  352. self.assertRaises(InvalidMessageOperation,
  353. self.p.to_wire, MessageRenderer())
  354. message_render = create_message()
  355. renderer = MessageRenderer()
  356. message_render.to_wire(renderer)
  357. self.assertEqual(b'\x105\x85\x00\x00\x01\x00\x02\x00\x00\x00\x00\x04test\x07example\x03com\x00\x00\x01\x00\x01\xc0\x0c\x00\x01\x00\x01\x00\x00\x0e\x10\x00\x04\xc0\x00\x02\x01\xc0\x0c\x00\x01\x00\x01\x00\x00\x0e\x10\x00\x04\xc0\x00\x02\x02',
  358. renderer.get_data())
  359. def test_to_text(self):
  360. message_render = create_message()
  361. msg_str =\
  362. """;; ->>HEADER<<- opcode: QUERY, status: NOERROR, id: 4149
  363. ;; flags: qr aa rd ; QUESTION: 1, ANSWER: 2, AUTHORITY: 0, ADDITIONAL: 0
  364. ;; QUESTION SECTION:
  365. ;test.example.com. IN A
  366. ;; ANSWER SECTION:
  367. test.example.com. 3600 IN A 192.0.2.1
  368. test.example.com. 3600 IN A 192.0.2.2
  369. """
  370. self.assertEqual(msg_str, message_render.to_text())
  371. self.assertEqual(msg_str, str(message_render))
  372. def test_from_wire(self):
  373. self.assertRaises(TypeError, self.r.from_wire, 1)
  374. self.assertRaises(InvalidMessageOperation,
  375. Message.from_wire, self.r, bytes())
  376. self.assertRaises(MessageTooShort,
  377. Message.from_wire, self.p, bytes())
  378. test_name = Name("test.example.com");
  379. message_parse = Message(0)
  380. factoryFromFile(message_parse, "message_fromWire1")
  381. self.assertEqual(0x1035, message_parse.get_qid())
  382. self.assertEqual(Opcode.QUERY(), message_parse.get_opcode())
  383. self.assertEqual(Rcode.NOERROR(), message_parse.get_rcode())
  384. self.assertTrue(message_parse.get_header_flag(MessageFlag.QR()))
  385. self.assertTrue(message_parse.get_header_flag(MessageFlag.RD()))
  386. self.assertTrue(message_parse.get_header_flag(MessageFlag.AA()))
  387. #QuestionPtr q = *message_parse.beginQuestion()
  388. q = message_parse.get_question()[0]
  389. self.assertEqual(test_name, q.get_name())
  390. self.assertEqual(RRType("A"), q.get_type())
  391. self.assertEqual(RRClass("IN"), q.get_class())
  392. self.assertEqual(1, message_parse.get_rr_count(Section.QUESTION()))
  393. self.assertEqual(2, message_parse.get_rr_count(Section.ANSWER()))
  394. self.assertEqual(0, message_parse.get_rr_count(Section.AUTHORITY()))
  395. self.assertEqual(0, message_parse.get_rr_count(Section.ADDITIONAL()))
  396. #RRsetPtr rrset = *message_parse.beginSection(Section.ANSWER())
  397. rrset = message_parse.get_section(Section.ANSWER())[0]
  398. self.assertEqual(test_name, rrset.get_name())
  399. self.assertEqual(RRType("A"), rrset.get_type())
  400. self.assertEqual(RRClass("IN"), rrset.get_class())
  401. ## TTL should be 3600, even though that of the 2nd RR is 7200
  402. self.assertEqual(RRTTL(3600), rrset.get_ttl())
  403. rdata = rrset.get_rdata();
  404. self.assertEqual("192.0.2.1", rdata[0].to_text())
  405. self.assertEqual("192.0.2.2", rdata[1].to_text())
  406. self.assertEqual(2, len(rdata))
  407. def test_EDNS0ExtCode(self):
  408. # Extended Rcode = BADVERS
  409. message_parse = Message(Message.PARSE)
  410. factoryFromFile(message_parse, "message_fromWire10.wire")
  411. self.assertEqual(Rcode.BADVERS(), message_parse.get_rcode())
  412. # Maximum extended Rcode
  413. message_parse.clear(Message.PARSE)
  414. factoryFromFile(message_parse, "message_fromWire11.wire")
  415. self.assertEqual(0xfff, message_parse.get_rcode().get_code())
  416. def test_BadEDNS0(self):
  417. message_parse = Message(Message.PARSE)
  418. # OPT RR in the answer section
  419. self.assertRaises(DNSMessageFORMERR,
  420. factoryFromFile,
  421. message_parse,
  422. "message_fromWire4")
  423. # multiple OPT RRs (in the additional section)
  424. message_parse.clear(Message.PARSE)
  425. self.assertRaises(DNSMessageFORMERR,
  426. factoryFromFile,
  427. message_parse,
  428. "message_fromWire5")
  429. ## OPT RR of a non root name
  430. message_parse.clear(Message.PARSE)
  431. self.assertRaises(DNSMessageFORMERR,
  432. factoryFromFile,
  433. message_parse,
  434. "message_fromWire6")
  435. # Compressed owner name of OPT RR points to a root name.
  436. # Not necessarily bogus, but very unusual and mostly pathological.
  437. # We accept it, but is it okay?
  438. message_parse.clear(Message.PARSE)
  439. factoryFromFile(message_parse, "message_fromWire7")
  440. # Unsupported Version
  441. message_parse.clear(Message.PARSE)
  442. self.assertRaises(DNSMessageBADVERS,
  443. factoryFromFile,
  444. message_parse,
  445. "message_fromWire9")
  446. if __name__ == '__main__':
  447. unittest.main()