def testInitOneToOne(self):
"""Make sure that we are setting things up right."""
self.ips_obj = iperf.IperfSet(self.fake_host_src[0],
self.fake_host_dst[0],
IperfSetTest.dst[0])
self.assertEqual(len(self.ips_obj.client_list), 1)
self.assertEqual(len(self.ips_obj.server_list), 1)
for client in self.ips_obj.client_list:
self.assertTrue(isinstance(client, iperf.IperfClient))
self.assertIn('-c %s' % IperfSetTest.dst[0], client.args)
for server in self.ips_obj.server_list:
self.assertTrue(isinstance(server, iperf.IperfServer))
def testStop(self):
"""Make sure that all server and clients are stopped."""
self.ips_obj = iperf.IperfSet(self.fake_host_src, self.fake_host_dst,
IperfSetTest.dst)
self.ips_obj.Start()
for server in self.ips_obj.server_list:
self.assertGreater(server.child_pid, 0)
for client in self.ips_obj.client_list:
self.assertGreater(client.child_pid, 0)
self.ips_obj.Stop()
for server in self.ips_obj.server_list:
self.assertIsNone(server.child_pid)
for client in self.ips_obj.client_list:
self.assertIsNone(client.child_pid)
def testStart(self):
"""Make sure that all server and clients are started.
Let's also make sure that they have the right hostnames.
"""
self.ips_obj = iperf.IperfSet(self.fake_host_src, self.fake_host_dst,
IperfSetTest.dst)
for server in self.ips_obj.server_list:
self.assertIsNone(server.child_pid)
for client in self.ips_obj.client_list:
self.assertIsNone(client.child_pid)
self.ips_obj.Start()
for server in self.ips_obj.server_list:
self.assertGreater(server.child_pid, 0)
for client in self.ips_obj.client_list:
self.assertGreater(client.child_pid, 0)
def testResults(self):
"""Make sure that all server and clients are saving results.
Let's also make sure that they have the right content.
"""
self.ips_obj = iperf.IperfSet(self.fake_host_src, self.fake_host_dst,
IperfSetTest.dst)
self.ips_obj.Start(length=10)
self.ips_obj.Stop()
results = self.ips_obj.Results()
self.assertEqual(len(results), 2)
self.assertEqual(len(results[0]), len(self.fake_host_dst))
self.assertEqual(len(results[1]), len(self.fake_host_src))
for i in range(0, len(self.fake_host_dst)):
self.assertIn('Server listening', results[0][i])
for i in range(0, len(self.fake_host_src)):
self.assertIn('Client connecting', results[1][i])
def testInitManyToOne(self):
"""Make sure that we are setting things up right."""
self.ips_obj = iperf.IperfSet(self.fake_host_src,
self.fake_host_dst[0],
IperfSetTest.dst[0])
self.assertEqual(len(self.ips_obj.client_list),
len(self.fake_host_src))
self.assertEqual(len(self.ips_obj.server_list), 1)
for i in range(0, len(self.ips_obj.client_list)):
self.assertTrue(
isinstance(self.ips_obj.client_list[i], iperf.IperfClient))
self.assertIn('-c %s' % IperfSetTest.dst[0],
self.ips_obj.client_list[i].args)
self.assertEqual(self.ips_obj.client_list[i].host.host,
IperfSetTest.target_src[i])
for server in self.ips_obj.server_list:
self.assertTrue(isinstance(server, iperf.IperfServer))