|
# Copyright (C) 2010 Internet Systems Consortium.
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SYSTEMS CONSORTIUM
# DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
# INTERNET SYSTEMS CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT,
# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
# FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
# NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
# WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#
# Tests for the opcode part of the pydnspp module
#
import unittest
from pydnspp import *
class OpcodeTest(unittest.TestCase):
def test_init(self):
self.assertRaises(TypeError, Opcode, "wrong")
self.assertEqual(Rcode(0).get_code(), 0)
self.assertEqual(Rcode(Opcode.RESERVED15_CODE).get_code(), 15)
self.assertRaises(OverflowError, Opcode, 16)
def test_constants(self):
self.assertEqual(Opcode.QUERY_CODE, Opcode(0).get_code())
self.assertEqual(Opcode.IQUERY_CODE, Opcode(1).get_code())
self.assertEqual(Opcode.NOTIFY_CODE, Opcode(4).get_code())
self.assertEqual(Opcode.UPDATE_CODE, Opcode(5).get_code())
self.assertEqual(Opcode.RESERVED15_CODE, Opcode(15).get_code())
self.assertEqual(Opcode.QUERY_CODE, Opcode.QUERY().get_code())
self.assertEqual(Opcode.IQUERY_CODE, Opcode.IQUERY().get_code())
self.assertEqual(Opcode.NOTIFY_CODE, Opcode.NOTIFY().get_code())
self.assertEqual(Opcode.UPDATE_CODE, Opcode.UPDATE().get_code())
self.assertEqual(Opcode.RESERVED15_CODE, Opcode.RESERVED15().get_code())
def test_get_code(self):
self.assertEqual(0, Opcode.QUERY().get_code())
self.assertEqual(1, Opcode.IQUERY().get_code())
self.assertEqual(2, Opcode.STATUS().get_code())
self.assertEqual(3, Opcode.RESERVED3().get_code())
self.assertEqual(4, Opcode.NOTIFY().get_code())
self.assertEqual(5, Opcode.UPDATE().get_code())
self.assertEqual(6, Opcode.RESERVED6().get_code())
self.assertEqual(7, Opcode.RESERVED7().get_code())
self.assertEqual(8, Opcode.RESERVED8().get_code())
self.assertEqual(9, Opcode.RESERVED9().get_code())
self.assertEqual(10, Opcode.RESERVED10().get_code())
self.assertEqual(11, Opcode.RESERVED11().get_code())
self.assertEqual(12, Opcode.RESERVED12().get_code())
self.assertEqual(13, Opcode.RESERVED13().get_code())
self.assertEqual(14, Opcode.RESERVED14().get_code())
self.assertEqual(15, Opcode.RESERVED15().get_code())
def test_to_text(self):
self.assertEqual("QUERY", Opcode.QUERY().to_text())
self.assertEqual("QUERY", str(Opcode.QUERY()))
self.assertEqual("IQUERY", Opcode.IQUERY().to_text())
self.assertEqual("STATUS", Opcode.STATUS().to_text())
self.assertEqual("RESERVED3", Opcode.RESERVED3().to_text())
self.assertEqual("NOTIFY", Opcode.NOTIFY().to_text())
self.assertEqual("UPDATE", Opcode.UPDATE().to_text())
self.assertEqual("RESERVED6", Opcode.RESERVED6().to_text())
self.assertEqual("RESERVED7", Opcode.RESERVED7().to_text())
self.assertEqual("RESERVED8", Opcode.RESERVED8().to_text())
self.assertEqual("RESERVED9", Opcode.RESERVED9().to_text())
self.assertEqual("RESERVED10", Opcode.RESERVED10().to_text())
self.assertEqual("RESERVED11", Opcode.RESERVED11().to_text())
self.assertEqual("RESERVED12", Opcode.RESERVED12().to_text())
self.assertEqual("RESERVED13", Opcode.RESERVED13().to_text())
self.assertEqual("RESERVED14", Opcode.RESERVED14().to_text())
self.assertEqual("RESERVED15", Opcode.RESERVED15().to_text())
def test_richcmp(self):
o1 = Opcode.QUERY()
o2 = Opcode.NOTIFY()
o3 = Opcode.NOTIFY()
self.assertTrue(o2 == o3)
self.assertFalse(o2 != o3)
self.assertTrue(o1 != o2)
self.assertFalse(o1 == 1)
self.assertFalse(o1 == o2)
# can't use assertRaises here...
try:
o1 < o2
self.fail("operation that should have raised an error unexpectedly succeeded")
except Exception as err:
self.assertEqual(TypeError, type(err))
try:
o1 <= o2
self.fail("operation that should have raised an error unexpectedly succeeded")
except Exception as err:
self.assertEqual(TypeError, type(err))
try:
o1 > o2
self.fail("operation that should have raised an error unexpectedly succeeded")
except Exception as err:
self.assertEqual(TypeError, type(err))
try:
o1 >= o2
self.fail("operation that should have raised an error unexpectedly succeeded")
except Exception as err:
self.assertEqual(TypeError, type(err))
exitif __name__ == '__main__':
unittest.main()
|
