def test_instantiate_error(self, *args):
trex_traffic_gen = tg_trex_vpp.TrexTrafficGenVpp('tg0', self.VNFD_0)
trex_traffic_gen.resource_helper = mock.MagicMock()
trex_traffic_gen.setup_helper.setup_vnf_environment = mock.MagicMock()
scenario_cfg = {
"tc": "tc_baremetal_rfc2544_ipv4_1flow_64B",
"nodes": {
"tg0": {},
"vnf0": {}
},
"topology": 'nsb_test_case.yaml',
'options': {
'packetsize': 64,
'traffic_type': 4,
'rfc2544': {
'allowed_drop_rate': '0.8 - 1',
},
'vnf__0': {
'rules': 'acl_1rule.yaml',
'vnf_config': {
'lb_config': 'SW',
'lb_count': 1,
'worker_config': '1C/1T',
'worker_threads': 1,
},
},
},
}
trex_traffic_gen.instantiate(scenario_cfg, {})
def test__start_server(self):
vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
trex_traffic_gen = tg_trex_vpp.TrexTrafficGenVpp('tg0', vnfd)
trex_traffic_gen.ssh_helper = mock.MagicMock()
trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
trex_traffic_gen.scenario_helper.scenario_cfg = {}
self.assertIsNone(trex_traffic_gen._start_server())
def test__check_status(self):
vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
trex_traffic_gen = tg_trex_vpp.TrexTrafficGenVpp('tg0', vnfd)
trex_traffic_gen.ssh_helper = mock.MagicMock()
trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
trex_traffic_gen.resource_helper.ssh_helper.execute.return_value = 0, '', ''
trex_traffic_gen.scenario_helper.scenario_cfg = {}
self.assertEqual(0, trex_traffic_gen._check_status())
def test_collect_kpi(self, *args):
trex_traffic_gen = tg_trex_vpp.TrexTrafficGenVpp('tg0', self.VNFD_0)
trex_traffic_gen.scenario_helper.scenario_cfg = {
'nodes': {
trex_traffic_gen.name: "mock"
}
}
expected = {
'physical_node': 'mock_node',
'collect_stats': {},
}
self.assertEqual(trex_traffic_gen.collect_kpi(), expected)
def test_wait_for_instantiate(self, ssh, *args):
mock_ssh(ssh)
mock_process = mock.Mock(autospec=Process)
mock_process.is_alive.return_value = True
mock_process.exitcode = 432
vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
trex_traffic_gen = tg_trex_vpp.TrexTrafficGenVpp('tg0', vnfd)
trex_traffic_gen.ssh_helper = mock.MagicMock()
trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
trex_traffic_gen.resource_helper.ssh_helper.execute.return_value = 0, '', ''
trex_traffic_gen.scenario_helper.scenario_cfg = {}
trex_traffic_gen._tg_process = mock_process
self.assertEqual(432, trex_traffic_gen.wait_for_instantiate())
def test___init__(self):
trex_traffic_gen = tg_trex_vpp.TrexTrafficGenVpp('tg0', self.VNFD_0)
self.assertIsNotNone(
trex_traffic_gen.resource_helper._terminated.value)