def test_connect(self, mock_sys): ixnet_gen = IxNextgen() ixnet_gen.get_config = mock.MagicMock() ixnet_gen.get_ixnet = mock.MagicMock() self.assertRaises(ImportError, ixnet_gen._connect, {"py_lib_path": "/tmp"})
def test_connect(self, mock_sys, mock_ix_network): mock_ix_network.IxNet.return_value = mock_ixnet = mock.Mock() ixnet_gen = IxNextgen() ixnet_gen.get_config = mock.MagicMock() ixnet_gen.get_ixnet = mock.MagicMock() result = ixnet_gen._connect({"py_lib_path": "/tmp"}) self.assertIsNotNone(result) self.assertEqual(mock_ix_network.IxNet.call_count, 1) self.assertEqual(mock_ixnet.connect.call_count, 1)
def test_get_config(self): tg_cfg = { "vdu": [ { "external-interface": [ { "virtual-interface": { "vpci": "0000:07:00.1", }, }, { "virtual-interface": { "vpci": "0001:08:01.2", }, }, ], }, ], "mgmt-interface": { "ip": "test1", "tg-config": { "dut_result_dir": "test2", "version": "test3", "ixchassis": "test4", "tcl_port": "test5", "py_lib_path": "test6", }, } } expected = { 'py_lib_path': 'test6', 'machine': 'test1', 'port': 'test5', 'chassis': 'test4', 'card1': '0000', 'port1': '07', 'card2': '0001', 'port2': '08', 'output_dir': 'test2', 'version': 'test3', 'bidir': True, } result = IxNextgen.get_config(tg_cfg) self.assertDictEqual(result, expected)