def setUp(self):
lm = LogManager('./test-logs')
ptm = PhysicalTopologyManager(root_dir=os.path.dirname(os.path.abspath(__file__)) + '/../..', log_manager=lm)
self.hypervisor = None
self.root_host = None
root_hostcfg = HostDef('root',
interfaces={'hveth0': InterfaceDef('hveth0')})
root_host_implcfg = ImplementationDef('test', 'RootHost')
hypervisorcfg = HostDef('hv',
interfaces={'eth0': InterfaceDef('eth0', [IP('192.168.1.3')])})
hypervisor_implcfg = ImplementationDef('test', 'ComputeHost')
self.root_host = RootHost(root_hostcfg.name, ptm)
self.hypervisor = ComputeHost(hypervisorcfg.name, ptm)
log = lm.add_file_logger('test.log', 'test')
self.root_host.set_logger(log)
self.hypervisor.set_logger(log)
# Now configure the host with the definition and impl configs
self.root_host.config_from_ptc_def(root_hostcfg, root_host_implcfg)
self.hypervisor.config_from_ptc_def(hypervisorcfg, hypervisor_implcfg)
self.root_host.link_interface(self.root_host.interfaces['hveth0'],
self.hypervisor, self.hypervisor.interfaces['eth0'])
self.root_host.create()
self.hypervisor.create()
self.root_host.boot()
self.hypervisor.boot()
self.root_host.net_up()
self.hypervisor.net_up()
self.root_host.net_finalize()
self.hypervisor.net_finalize()
class VMHostTest(unittest.TestCase):
def setUp(self):
lm = LogManager('./test-logs')
ptm = PhysicalTopologyManager(root_dir=os.path.dirname(os.path.abspath(__file__)) + '/../..', log_manager=lm)
self.hypervisor = None
self.root_host = None
root_hostcfg = HostDef('root',
interfaces={'hveth0': InterfaceDef('hveth0')})
root_host_implcfg = ImplementationDef('test', 'RootHost')
hypervisorcfg = HostDef('hv',
interfaces={'eth0': InterfaceDef('eth0', [IP('192.168.1.3')])})
hypervisor_implcfg = ImplementationDef('test', 'ComputeHost')
self.root_host = RootHost(root_hostcfg.name, ptm)
self.hypervisor = ComputeHost(hypervisorcfg.name, ptm)
log = lm.add_file_logger('test.log', 'test')
self.root_host.set_logger(log)
self.hypervisor.set_logger(log)
# Now configure the host with the definition and impl configs
self.root_host.config_from_ptc_def(root_hostcfg, root_host_implcfg)
self.hypervisor.config_from_ptc_def(hypervisorcfg, hypervisor_implcfg)
self.root_host.link_interface(self.root_host.interfaces['hveth0'],
self.hypervisor, self.hypervisor.interfaces['eth0'])
self.root_host.create()
self.hypervisor.create()
self.root_host.boot()
self.hypervisor.boot()
self.root_host.net_up()
self.hypervisor.net_up()
self.root_host.net_finalize()
self.hypervisor.net_finalize()
def dtest_create_vm(self):
vm_host = self.hypervisor.create_vm('test_vm')
self.assertIs(vm_host, self.hypervisor.get_vm('test_vm'))
vm_host.create()
vm_host.boot()
vm_host.net_up()
vm_host.net_finalize()
vm_host.shutdown()
vm_host.remove()
def dtest_create_vm_interface(self):
vm_host = self.hypervisor.create_vm('test_vm')
self.assertIs(vm_host, self.hypervisor.get_vm('test_vm'))
new_if = vm_host.create_interface('eth0', ip_list=[IP('10.50.50.3')])
self.assertTrue(vm_host.cli.grep_cmd('ip l', 'eth0'))
self.assertTrue(self.hypervisor.cli.grep_cmd('ip l', 'test_vmeth0'))
vm_host.shutdown()
vm_host.remove()
def test_packet_communication(self):
vm_host1 = self.hypervisor.create_vm('test_vm1')
try:
vm_host1.create_interface('eth0', ip_list=[IP('10.50.50.3')])
vm_host1.start_capture('lo', save_dump_file=True, save_dump_filename='tcp.vmhost.out')
ping_ret = vm_host1.ping('10.50.50.3')
vm_host1.send_tcp_packet(iface='lo', dest_ip='10.50.50.3', source_port=6015, dest_port=6055)
ret1 = vm_host1.capture_packets('lo', count=1, timeout=5)
ret2 = vm_host1.capture_packets('lo', count=1, timeout=5)
vm_host1.stop_capture('lo')
self.assertTrue(ping_ret)
self.assertEquals(1, len(ret1))
finally:
vm_host1.shutdown()
vm_host1.remove()
def tearDown(self):
self.root_host.net_down()
self.hypervisor.net_down()
self.hypervisor.shutdown()
self.root_host.shutdown()
self.hypervisor.remove()
self.root_host.remove()
LinuxCLI().cmd('ip netns del test_vm')
LinuxCLI().rm('tcp.vmhost.out')