class VagrantLatencyTests(unittest.TestCase): def setUp(self): #defaults to vagrantfile in current directory self.vagrant_chaos = VagrantChaos() self.vagrant_latency = VagrantLatency() self.node_ip = "172.28.128.199" self.node_name = "node1" def test_add_latency(self): self.vagrant_chaos.turn_on_node(self.node_name) command = "ping -c 1 " + self.node_ip output = subprocess.check_output(command, shell=True) regex = "round-trip min/avg/max/stddev = (?P<min>\d+\.\d+)/(?P<avg>\d+\.\d+)/(?P<max>\d+\.\d+)" (_, _, old_ping_avg) = re.search(regex, output).groups() self.vagrant_latency.add_latency(self.vagrant_latency, self.node_name, 'eth1', 100, 0) output = subprocess.check_output(command, shell=True) regex = "round-trip min/avg/max/stddev = (?P<min>\d+\.\d+)/(?P<avg>\d+\.\d+)/(?P<max>\d+\.\d+)" (_, _, new_ping_avg) = re.search(regex, output).groups() self.vagrant_latency.clear_config(self.vagrant_latency, self.node_name, "eth1") self.assertTrue(new_ping_avg > old_ping_avg) self.assertTrue(new_ping_avg > 100)
class VagrantChaosTests(unittest.TestCase): def setUp(self): #defaults to vagrantfile in current directory self.vagrant_chaos = VagrantChaos() self.node_ip = "172.28.128.199" self.node_name = "node1" def tearDown(self): #cleanup everything #halting the machines makes test iteration a bit faster #for a more serious deployment change this to #self.vagrant_instance.destroy() #self.vagrant_instance.halt() pass def test_turn_on(self): self.vagrant_chaos.turn_on_node(self.node_name) status_list = self.vagrant_chaos.status() name_list = [] for status in status_list: if status.state == 'running': name_list.append(status.name) self.assertEqual(name_list, ["node1"]) def test_active_nodes_list(self): self.vagrant_chaos.turn_on_node(self.node_name) status_list = self.vagrant_chaos.status() name_list = self.vagrant_chaos.active_nodes_list() self.assertEqual(name_list, ["node1"]) def test_turn_off(self): self.vagrant_chaos.turn_on_node(self.node_name) node_state = self.vagrant_chaos.vagrant_instance.status(vm_name = self.node_name)[0].state self.assertEqual(node_state, "running") self.vagrant_chaos.turn_off_node(self.node_name) node_state = self.vagrant_chaos.vagrant_instance.status(vm_name = self.node_name)[0].state self.assertEqual(node_state, "poweroff")