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)