|
# 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 rrttl part of the pydnspp module
#
import unittest
import os
from pydnspp import *
class RRTTLTest(unittest.TestCase):
def setUp(self):
self.t1 = RRTTL(1)
self.t2 = RRTTL(3600)
def test_init(self):
self.assertRaises(InvalidRRTTL, RRTTL, "wrong")
self.assertRaises(TypeError, RRTTL, Exception())
b = bytearray(1)
b[0] = 123
self.assertRaises(IncompleteRRTTL, RRTTL, b)
self.assertRaises(InvalidRRTTL, RRTTL, "4294967296")
b = bytearray(4)
b[0] = 0
b[1] = 0
b[2] = 0
b[3] = 15
self.assertEqual(15, RRTTL(b).get_value())
# Range check. We need to do this at the binding level, so we need
# explicit tests for it.
self.assertRaises(TypeError, RRTTL, -1)
self.assertRaises(ValueError, RRTTL, 4294967296)
self.assertEqual(0, RRTTL(0).get_value())
self.assertEqual(4294967295, RRTTL(4294967295).get_value())
def test_rrttl_to_text(self):
self.assertEqual("1", self.t1.to_text())
self.assertEqual("1", str(self.t1))
self.assertEqual("3600", self.t2.to_text())
def test_rrttl_to_wire(self):
b = bytearray()
self.t1.to_wire(b)
self.assertEqual(b'\x00\x00\x00\x01', b)
b = bytearray()
self.t2.to_wire(b)
self.assertEqual(b'\x00\x00\x0e\x10', b)
mr = MessageRenderer()
self.t2.to_wire(mr)
self.assertEqual(b'\x00\x00\x0e\x10', mr.get_data())
self.assertRaises(TypeError, self.t1.to_wire, 1)
def test_rrttl_richcmp(self):
self.assertTrue(self.t1 == RRTTL(1))
self.assertFalse(self.t1 != RRTTL(1))
self.assertFalse(self.t1 == 1)
self.assertTrue(self.t1 < self.t2)
self.assertTrue(self.t1 <= self.t2)
self.assertFalse(self.t1 > self.t2)
self.assertFalse(self.t1 >= self.t2)
self.assertFalse(self.t1 != RRTTL(1))
self.assertTrue(self.t1 == RRTTL(1))
self.assertFalse(self.t1 > self.t2)
self.assertFalse(self.t1 >= self.t2)
self.assertTrue(self.t1 < self.t2)
self.assertTrue(self.t1 <= self.t2)
exitif __name__ == '__main__':
unittest.main()
|
