def test_clean_restart(self): err = command.clean_all(test_topo[0][2], port) self.assertIsNone(err) err = command.clean_all(test_topo[0][3], port) self.assertIsNone(err) err = command.receive(test_topo[0][3], port, 0, 1000) self.assertIsNone(err) err = command.send(test_topo[0][2], port, 0, test_topo[0][3], 1000, 1000000) self.assertIsNone(err) time.sleep(5) err, res = command.get_progress(test_topo[0][3], port, 0) self.assertIsNone(err) self.assertGreater(res['progress'], 0) err, res = command.get_progress(test_topo[0][2], port, 0) self.assertIsNone(err) self.assertGreater(res['progress'], 0)
def test_active_transfer(self): err, res_json = command.active_transfer(test_topo[0][2], port) self.assertIsNone(err) self.assertEqual(len(res_json['senders']), 3) err, res_json = command.active_transfer(test_topo[0][3], port) self.assertIsNone(err) self.assertEqual(len(res_json['receivers']), 3) err = command.clean_all(test_topo[0][2], port) self.assertIsNone(err) err = command.clean_all(test_topo[0][3], port) self.assertIsNone(err) time.sleep(2) err, res_json = command.active_transfer(test_topo[0][2], port) self.assertIsNone(err) self.assertEqual(len(res_json['senders']), 0) err, res_json = command.active_transfer(test_topo[0][3], port) self.assertIsNone(err) self.assertEqual(len(res_json['receivers']), 0)
def test_issue(): commands = parse("issue2.log") id_ip_map = {} cur_time = 0 for command_time in commands: if command_time > cur_time: time.sleep(command_time - cur_time) cur_time = command_time for command_item in commands[command_time]: print(command_item) if command_item["type"] == "cleanAll": err = command.clean_all(command_item["host"], port) if err: print(cur_time) print(command_item) print(err) elif command_item["type"] == "progress": err = command.get_progress(command_item["host"], port, command_item["id"]) if err: print(cur_time) print(command_item) print(err) elif command_item["type"] == "updateRate": err = command.update_rate(command_item["host"], port, command_item["id"], command_item["rate"]) if err: print(cur_time) print(command_item) print(err) elif command_item["type"] == "receive": err = command.receive(command_item["host"], port, command_item["id"], command_item["port"]) if err: print(cur_time) print(command_item) print(err) id_ip_map[command_item["id"]] = command_item["host"] elif command_item["type"] == "send": if id_ip_map.get(command_item["id"], None) is None: print("id_ip_map is none") err = command.send(command_item["host"], port, command_item["id"], id_ip_map[command_item["id"]], command_item["port"], command_item["rate"]) if err: print(cur_time) print(command_item) print(err)
def setUp(self): for i in range(2, 14): err = command.clean_all("192.168.1.{}".format(i), port) self.assertIsNone(err) time.sleep(2)
def tearDown(self): err = command.clean_all(test_topo[0][2], port) self.assertIsNone(err) err = command.clean_all(test_topo[0][3], port) self.assertIsNone(err)