From e50a7606564d9ee1e53c051421559e31899d7513 Mon Sep 17 00:00:00 2001 From: user Date: Thu, 12 Dec 2013 17:37:48 +0400 Subject: [PATCH] Tests for Router class. --- tests/pdu_tests.py | 1 - tests/router_tests.py | 39 +++++++++++++++++++++++++++++++++++++++ 2 files changed, 39 insertions(+), 1 deletion(-) create mode 100644 tests/router_tests.py diff --git a/tests/pdu_tests.py b/tests/pdu_tests.py index 17acb83..8346372 100644 --- a/tests/pdu_tests.py +++ b/tests/pdu_tests.py @@ -1,7 +1,6 @@ import sys sys.path.append("/home/user/_cluster/cntrl_hac") sys.path.append("/home/user/_cluster/testutil_hac") -from cluster import Cluster from exceptionz import DeviceError from pdu import PDU diff --git a/tests/router_tests.py b/tests/router_tests.py new file mode 100644 index 0000000..130b307 --- /dev/null +++ b/tests/router_tests.py @@ -0,0 +1,39 @@ +import sys +sys.path.append("/home/user/_cluster/cntrl_hac") +sys.path.append("/home/user/_cluster/testutil_hac") +from exceptionz import DeviceError +from router import Router + +import unittest + + +# Working router. +ROUTER_IP = "192.168.50.174" +# Some device connected to router. +DUMMY_DEVICE = Router("dummy", {"ip": None}) +DUMMY_DEVICE_PORT = 13 + + +class TestsForRouter(unittest.TestCase): + def test_correct_settings(self): + router = Router("my_router", {"ip": ROUTER_IP, + "port%i" % (DUMMY_DEVICE_PORT): DUMMY_DEVICE.id}) + router.disable_port(DUMMY_DEVICE) + router.enable_port(DUMMY_DEVICE) + + + def test_wrong_ip(self): + router = Router("my_router", {"ip": "127.0.0.1", + "port%i" % (DUMMY_DEVICE_PORT): DUMMY_DEVICE.id}) + with self.assertRaises(DeviceError): + router.disable_port(DUMMY_DEVICE) + + + def test_wrong_ports_config(self): + router = Router("my_router", {"ip": ROUTER_IP}) + with self.assertRaises(DeviceError): + router.disable_port(DUMMY_DEVICE) + + +if (__name__ == '__main__'): + unittest.main()