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)
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"