Coverage for src/lib/python/isc/net/tests/parse_test : 97%

# Copyright (C) 2010  Internet Systems Consortium, Inc. ("ISC") 

# 

# 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.parse.""" 

import unittest 

import socket 

from isc.net.parse import port_parse, addr_parse 

class TestCheckPort(unittest.TestCase): 

    """ 

    Testcases for the isc.net.port_parse function 

    """ 

    def test_fail(self): 

        """ 

        Test if it fails on invalid input in the correct way. 

        """ 

        self.assertRaises(ValueError, port_parse, "not a number") 

        self.assertRaises(ValueError, port_parse, -1) 

        self.assertRaises(ValueError, port_parse, 65536) 

    def test_success(self): 

        """ 

        Test if it succeeds on valid inputs and returns the correct output 

        """ 

        self.assertEqual(port_parse(0), 0) 

        self.assertEqual(port_parse("65535"), 65535) 

        self.assertEqual(port_parse(1234), 1234) 

class TestCheckIP(unittest.TestCase): 

    """ 

    Testcases for the isc.net.ip_par function 

    """ 

    def test_fail(self): 

        """ 

        Test if it fails on invalid input the correct way. 

        """ 

        self.assertRaises(ValueError, addr_parse, "not an address") 

        self.assertRaises(ValueError, addr_parse, "123.456.789.012") 

        self.assertRaises(ValueError, addr_parse, "123.0.0.") 

        # Address range not allowed 

        self.assertRaises(ValueError, addr_parse, "192.0.2.0/24") 

        try: 

            # XXX: MacOS X's inet_pton() doesn't reject this form, so we 

            # check the behavior of the underlying library implementation 

            # before the actual test 

            socket.inet_pton(socket.AF_INET, "0000.0.0.0") 

        except socket.error: 

            self.assertRaises(ValueError, addr_parse, "0000.0.0.0") 

        self.assertRaises(ValueError, addr_parse, "bada:ddr0::") 

        self.assertRaises(ValueError, addr_parse, "2001:db8::/32") 

        # This should be one part too long (eg. 9 segments) 

        self.assertRaises(ValueError, addr_parse, "2001:db8:0:0:0:0:0:0:0") 

        # Only one :: allowed 

        self.assertRaises(ValueError, addr_parse, "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_parse("192.0.2.0"))) 

        # The OS could return something else than canonical form, in which 

        # case the test would fail. However, I do not see an easy way to fix 

        # the test, so it is left this way unless someone finds an OS that 

        # does return something else. 

        self.assertEqual("2001:bd8::", str(addr_parse("2001:bd8::"))) 

        # It should strip the unnecesarry parts 

        self.assertEqual("2001:bd8::", str(addr_parse("2001:bd8:0:0:0:0:0:0"))) 

        self.assertEqual("::", str(addr_parse("::"))) 

        self.assertEqual("2001:bd8::", str(addr_parse("2001:bd8::0.0.0.0"))) 

exitif __name__ == "__main__": 

    unittest.main()