# Copyright (C) 2010 CZ NIC # # 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 isc.net.check.""" import unittest from isc.net.check import port_check, addr_check class TestCheckPort(unittest.TestCase): """ Testcases for the isc.net.port_check function """ def test_fail(self): """ Test if it fails on invalid input in the correct way. """ self.assertRaises(ValueError, port_check, "not a number") self.assertRaises(ValueError, port_check, -1) self.assertRaises(ValueError, port_check, 65536) def test_success(self): """ Test if it succeeds on valid inputs and returns the correct output """ self.assertEqual(port_check(0), 0) self.assertEqual(port_check("65535"), 65535) self.assertEqual(port_check(1234), 1234) class TestCheckIP(unittest.TestCase): """ Testcases for the isc.net.ip_check function """ def test_fail(self): """ Test if it fails on invalid input the correct way. """ self.assertRaises(ValueError, addr_check, "not an address") self.assertRaises(ValueError, addr_check, "123.456.789.012") self.assertRaises(ValueError, addr_check, "123.0.0.") # Address range not allowed self.assertRaises(ValueError, addr_check, "192.0.2.0/24") self.assertRaises(ValueError, addr_check, "0000.0.0.0") self.assertRaises(ValueError, addr_check, "bada:ddr0::") self.assertRaises(ValueError, addr_check, "2001:db8::/32") # This should be one part too long (eg. 9 segments) self.assertRaises(ValueError, addr_check, "2001:db8:0:0:0:0:0:0:0") # Only one :: allowed self.assertRaises(ValueError, addr_check, "2001::db8::c") def test_success(self): """ Test if it succeeds on valid inputs and returns addresses that look the same. """ self.assertEqual("192.0.2.0", str(addr_check("192.0.2.0"))) self.assertEqual("2001:bd8::", str(addr_check("2001:bd8::"))) # It should strip the unnecesarry parts self.assertEqual("2001:bd8::", str(addr_check("2001:bd8:0:0:0:0:0:0"))) self.assertEqual("::", str(addr_check("::"))) if __name__ == "__main__": unittest.main()