def setUp(self):
super(OmniSwitchTelnetDriverUnitTestClass, self).setUp()
self.telnet_driver = OmniSwitchTelnetDriverTestClass('127.0.0.1', omni_const.OMNISWITCH_6XX,
'admin', 'switch', '->')
self.telnet_driver.clean_commands()
class OmniSwitchTelnetDriverUnitTestClass(unittest.TestCase):
"""Name: OmniSwitchTelnetDriverUnitTestClass
Description: OmniSwitchTelnetDriverUnitTestClass contains all unittest code for
OmniSwitchTelnetDriver(omniswitch_telnet_driver.py module)
Details: unittest and Mock are used for implementing unittest for OmniSwitchTelnetDriverUnitTestClass.
xmlrunner module is used to for generating test repport
To run all test cases in this file, type 'python test_omniswitch_telnet_driver.py'
"""
def setUp(self):
super(OmniSwitchTelnetDriverUnitTestClass, self).setUp()
self.telnet_driver = OmniSwitchTelnetDriverTestClass('127.0.0.1', omni_const.OMNISWITCH_6XX,
'admin', 'switch', '->')
self.telnet_driver.clean_commands()
def __assert_equal(self, expected_commands):
actual_commands = self.telnet_driver.get_commands()
self.assertEqual(actual_commands, expected_commands)
def test_create_vpa(self):
# OMNISWITCH_6XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_6XX)
# Case 1: vlan_id = 1; slotport = 2
vlan_id = '1'
slotport = '2'
self.telnet_driver.clean_commands()
self.telnet_driver.create_vpa(vlan_id, slotport)
self.__assert_equal(['vlan 1 802.1q 2'])
# case 2: vlan_id = 1; slotport = '1/2'
vlan_id = '1'
slotport = '1/2'
self.telnet_driver.clean_commands()
self.telnet_driver.create_vpa(vlan_id, slotport)
self.__assert_equal(['vlan 1 802.1q 1/2'])
# OMNISWITCH_7XX and OMNISWITCH_8XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_7XX)
# Case 1: vlan_id = 1; slotport = 2
vlan_id = '1'
slotport = '2'
self.telnet_driver.clean_commands()
self.telnet_driver.create_vpa(vlan_id, slotport)
self.__assert_equal(['vlan 1 members linkagg 2 tagged'])
# case 2: vlan_id = 1; slotport = '1/2'
self.telnet_driver.clean_commands()
vlan_id = '1'
slotport = '1/2'
self.telnet_driver.create_vpa(vlan_id, slotport)
self.__assert_equal(['vlan 1 members port 1/2 tagged'])
def test_delete_vpa(self):
# OMNISWITCH_6XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_6XX)
# Case 1: vlan_id = 1; slotport = 2
vlan_id = '1'
slotport = '2'
self.telnet_driver.clean_commands()
self.telnet_driver.delete_vpa(vlan_id, slotport)
self.__assert_equal(['no 802.1q 2'])
# case 2: vlan_id = 1; slotport = '1/2'
vlan_id = '1'
slotport = '1/2'
self.telnet_driver.clean_commands()
self.telnet_driver.delete_vpa(vlan_id, slotport)
self.__assert_equal(['no 802.1q 1/2'])
# OMNISWITCH_7XX and OMNISWITCH_8XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_8XX)
# Case 1: vlan_id = 1; slotport = 2
vlan_id = '1'
slotport = '2'
self.telnet_driver.clean_commands()
self.telnet_driver.delete_vpa(vlan_id, slotport)
self.__assert_equal(['no vlan 1 members linkagg 2'])
# case 2: vlan_id = 1; slotport = '1/2'
vlan_id = '1'
slotport = '1/2'
self.telnet_driver.clean_commands()
self.telnet_driver.delete_vpa(vlan_id, slotport)
self.__assert_equal(['no vlan 1 members port 1/2'])
def test_create_vlan(self):
vlan_id = '1'
net_name = 'OpenStack'
self.telnet_driver.clean_commands()
self.telnet_driver.create_vlan(vlan_id, net_name)
self.__assert_equal(['vlan 1 name OpenStack'])
def test_update_vlan(self):
# test_create_vlan
# update_vlan method delegates its work to create_vlan method. no need to test twice
self.skipTest('update_vlan method delegates its work to create_vlan method. no need to test twice')
def test_delete_vlan(self):
vlan_id = '1'
self.telnet_driver.delete_vlan(vlan_id)
self.__assert_equal(['no vlan 1'])
def test_create_network(self):
# omni_const.OMNISWITCH_8XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_8XX)
vlan_id = 1
net_name = 'OpenStack'
self.telnet_driver.clean_commands()
self.telnet_driver.create_network(vlan_id, net_name)
self.__assert_equal(['vlan 1 name OpenStack'])
def test_update_network(self):
# test_create_vlan
# update_network method delegates its work to create_vlan method. no need to test twice
self.skipTest('update_network method delegates its work to create_vlan method. no need to test twice')
def test_delete_network(self):
# omni_const.OMNISWITCH_8XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_8XX)
vlan_id = 1
self.telnet_driver.clean_commands()
self.telnet_driver.delete_network(vlan_id)
self.__assert_equal(['no vlan 1'])
def test_enable_stp_mode_flat(self):
# OMNISWITCH_6XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_6XX)
self.telnet_driver.clean_commands()
self.telnet_driver.enable_stp_mode_flat()
self.__assert_equal(['bridge mode flat'])
# OMNISWITCH_7XX and 8XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_8XX)
self.telnet_driver.clean_commands()
self.telnet_driver.enable_stp_mode_flat()
self.__assert_equal(['spantree mode flat'])
def test_disable_stp_mode_flat(self):
# OMNISWITCH_6XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_6XX)
self.telnet_driver.clean_commands()
self.telnet_driver.disable_stp_mode_flat()
self.__assert_equal(['bridge mode 1x1'])
# OMNISWITCH_7XX and 8XX#
self.telnet_driver.set_switch_type(omni_const.OMNISWITCH_8XX)
self.telnet_driver.clean_commands()
self.telnet_driver.disable_stp_mode_flat()
self.__assert_equal(['spantree mode per-vlan'])
