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)
