def test_iperf_detecting_dualtest_at_client(buffer_connection):
from moler.cmd.unix import iperf2
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_bidirectional_udp_client)
assert iperf_cmd.works_in_dualtest is True
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_basic_client)
assert iperf_cmd.works_in_dualtest is False
def test_iperf_detecting_dualtest_at_server(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_bidirectional_udp_server])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_bidirectional_udp_server)
iperf_cmd()
assert iperf_cmd.works_in_dualtest is True
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_multiple_connections_server])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_server)
iperf_cmd()
assert iperf_cmd.works_in_dualtest is False
def test_iperf_correctly_parses_multiconnection_udp_client_output(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_multiple_connections_udp_client])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_udp_client)
ret = iperf_cmd()
assert ret == iperf2.COMMAND_RESULT_multiple_connections_udp_client
def test_iperf_correctly_parses_singlerun_udp_server_output(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_singlerun_udp_server])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_singlerun_udp_server)
ret = iperf_cmd()
assert ret == iperf2.COMMAND_RESULT_singlerun_udp_server
def test_iperf_ignores_multiple_echo_of_command(buffer_connection):
# running on cygwin may cause getting multiple echo even mixed with prompt
from moler.cmd.unix import iperf2
collected_lines = []
output_with_multiple_echo = "andromeda:/ # /bin/iperf -c 1.2.3.4 -p 5001 -f k -i 1.0 -t 6.0 --dualtest\n" + \
"/bin/iperf -c 1.2.3.4 -p 5001 -f k -i 1.0 -t 6.0 -\n" + \
"andromeda:/ # /bin/iperf -c 1.2.3.4 -p 5001 -f k -i 1.0 -t 6.0 --dualtest\n" + \
"------------------------------------------------------------\n" + \
"[ 3] local 192.168.0.1 port 49597 connected with 1.2.3.4 port 5001\n" + \
"[ ID] Interval Transfer Bandwidth\n" + \
"[ 3] 0.0- 1.0 sec 28.6 MBytes 240 Mbits/sec\n" + \
"[ 3] 0.0-10.0 sec 265 MBytes 222 Mbits/sec\n"+ \
"andromeda:/ # "
buffer_connection.remote_inject_response([output_with_multiple_echo])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
prompt='andromeda:/ # ',
options='-c 1.2.3.4 -p 5001 -f k -i 1.0 -t 6.0 --dualtest')
iperf_cmd.command_path = '/bin/'
oryg_on_new_line = iperf_cmd.__class__.on_new_line
def on_new_line(self, line, is_full_line):
collected_lines.append(line)
oryg_on_new_line(self, line, is_full_line)
with mock.patch.object(iperf_cmd.__class__, "on_new_line", on_new_line):
iperf_cmd()
assert "andromeda:/ # /bin/iperf -c 1.2.3.4 -p 5001 -f k -i 1.0 -t 6.0 --dualtest" not in collected_lines
def test_iperf_singlerun_server_doesnt_use_ctrlc_to_stop_server(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_singlerun_server])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_singlerun_server)
with mock.patch.object(iperf_cmd, "break_cmd") as send_ctrlc:
iperf_cmd()
send_ctrlc.assert_not_called()
def test_iperf_publishes_records_to_subscribed_observers(buffer_connection):
from moler.cmd.unix import iperf2
conn = buffer_connection
conn.remote_inject_response([iperf_server_output_start])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
options='-s -u -i 1')
iperf_stats = []
def iperf_observer(from_client, to_server, data_record=None, report=None):
iperf_record = {}
iperf_record['from_client'] = from_client
iperf_record['to_server'] = to_server
if data_record:
iperf_record['data_record'] = data_record
if report:
iperf_record['report'] = report
iperf_stats.append(iperf_record)
iperf_cmd.subscribe(subscriber=iperf_observer)
iperf_cmd.start()
assert len(iperf_stats) == 0
conn.remote_inject_line("[ ID] Interval Transfer Bandwidth Jitter Lost/Total Datagrams")
assert len(iperf_stats) == 0
conn.remote_inject_line("[904] 0.0- 1.0 sec 1.17 MBytes 9.84 Mbits/sec 1.830 ms 0/ 837 (0%)")
assert len(iperf_stats) == 1
assert iperf_stats[0]['from_client'] == '[email protected]'
assert iperf_stats[0]['to_server'] == '[email protected]'
# iperf progress lines produce data_records
assert iperf_stats[0]['data_record'] == {'Interval': (0.0, 1.0),
'Transfer': 1226833,
'Transfer Raw': u'1.17 MBytes',
'Bandwidth': 1230000,
'Bandwidth Raw': u'9.84 Mbits/sec',
'Jitter': u'1.830 ms',
'Lost_vs_Total_Datagrams': (0, 837),
'Lost_Datagrams_ratio': u'0%'}
conn.remote_inject_line("[904] 1.0- 2.0 sec 1.18 MBytes 9.94 Mbits/sec 1.846 ms 5/ 850 (0.59%)")
assert len(iperf_stats) == 2
assert ('data_record' in iperf_stats[-1]) and ('report' not in iperf_stats[-1])
conn.remote_inject_line("[904] 9.0-10.0 sec 1.19 MBytes 10.0 Mbits/sec 1.801 ms 0/ 851 (0%)")
assert len(iperf_stats) == 3
assert ('data_record' in iperf_stats[-1]) and ('report' not in iperf_stats[-1])
# last line of iperf progress produces report
conn.remote_inject_line("[904] 0.0-10.0 sec 11.8 MBytes 9.86 Mbits/sec 2.618 ms 9/ 8409 (0.11%)")
assert len(iperf_stats) == 4
assert 'data_record' not in iperf_stats[-1]
assert iperf_stats[-1]['from_client'] == '10.6.2.5'
assert iperf_stats[-1]['to_server'] == '[email protected]'
assert iperf_stats[-1]['report'] == {'Interval': (0.0, 10.0),
'Transfer': 12373196,
'Transfer Raw': u'11.8 MBytes',
'Bandwidth': 1232500,
'Bandwidth Raw': u'9.86 Mbits/sec',
'Jitter': u'2.618 ms',
'Lost_vs_Total_Datagrams': (9, 8409),
'Lost_Datagrams_ratio': u'0.11%'}
iperf_cmd.cancel()
def test_iperf_publishes_only_summary_records_when_handling_parallel_clients(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_multiple_connections_udp_server])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_udp_server)
expected_result = iperf2.COMMAND_RESULT_multiple_connections_udp_server
iperf_stats = {}
iperf_report = {}
def iperf_observer(from_client, to_server, data_record=None, report=None):
conn_name = (from_client, to_server)
if data_record:
if conn_name not in iperf_stats:
iperf_stats[conn_name] = []
iperf_stats[conn_name].append(data_record)
if report:
iperf_report[conn_name] = report
iperf_cmd.subscribe(subscriber=iperf_observer)
iperf_cmd()
# published stats should be as
summary_conn_name = ('[email protected]', '[email protected]')
client_conn_name = ('192.168.44.1', '[email protected]')
assert client_conn_name in iperf_report
assert summary_conn_name in iperf_stats
assert len(iperf_stats.keys()) == 1
assert iperf_stats[summary_conn_name] == expected_result['CONNECTIONS'][summary_conn_name][:-1]
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_multiple_connections_udp_client])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_udp_client)
expected_result = iperf2.COMMAND_RESULT_multiple_connections_udp_client
iperf_stats = {}
iperf_report = {}
iperf_cmd.subscribe(subscriber=iperf_observer)
iperf_cmd()
# published stats should be as
summary_conn_name = ('[email protected]', '[email protected]')
client_conn_name = ('192.168.33.5', '[email protected]')
assert client_conn_name in iperf_report
assert summary_conn_name in iperf_stats
assert len(iperf_stats.keys()) == 1
assert iperf_stats[summary_conn_name] == expected_result['CONNECTIONS'][summary_conn_name][:-1]
def test_iperf_stores_ipv6_connections_as_port_at_host_tuple_for_local_and_remote(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_tcp_ipv6_client])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_tcp_ipv6_client)
iperf_cmd()
stored_connections = iperf_cmd.result()['CONNECTIONS'].keys()
# local port@host remote port@host
assert len(stored_connections) == 2
assert ("49597@fd00::2:0", "5901@fd00::1:0") in stored_connections
assert ("fd00::2:0", "5901@fd00::1:0") in stored_connections
def test_iperf_creates_summary_connection_for_parallel_testing(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_multiple_connections])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections)
iperf_cmd()
stored_connections = iperf_cmd.result()['CONNECTIONS'].keys()
# local host:port remote host:port
assert len(stored_connections) == 22
assert ('[email protected]', '[email protected]') in stored_connections # summary
assert ('192.168.0.102', '[email protected]') in stored_connections # result
def test_iperf_stores_connections_as_port_at_host_tuple_for_local_and_remote(buffer_connection):
from moler.cmd.unix import iperf2
buffer_connection.remote_inject_response([iperf2.COMMAND_OUTPUT_bidirectional_udp_server])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_bidirectional_udp_server)
iperf_cmd()
stored_connections = iperf_cmd.result()['CONNECTIONS'].keys()
# local host:port remote host:port
assert len(stored_connections) == 4
assert ('[email protected]', '[email protected]') in stored_connections
assert ('[email protected]', '[email protected]') in stored_connections
assert ('192.168.0.10', '[email protected]') in stored_connections
assert ('192.168.0.12', '[email protected]') in stored_connections
def test_iperf_sends_additional_ctrl_c_after_detecting_to_early_ctrl_c(buffer_connection):
from moler.cmd.unix import iperf2
normal_iperf_output = iperf2.COMMAND_OUTPUT_multiple_connections_server.splitlines(True)
last_line_with_prompt = normal_iperf_output[-1]
normal_iperf_output[-1] = "^CWaiting for server threads to complete. Interrupt again to force quit\n"
normal_iperf_output.append(last_line_with_prompt)
output_with_too_early_ctrl_c = "".join(normal_iperf_output)
buffer_connection.remote_inject_response([output_with_too_early_ctrl_c])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_server)
with mock.patch.object(iperf_cmd, "_stop_server"):
with mock.patch.object(iperf_cmd, "break_cmd") as break_cmd_method:
iperf_cmd()
break_cmd_method.assert_called_once_with()
def test_iperf_server_can_calculate_multiport_summary_record_of_interval(buffer_connection):
from moler.cmd.unix import iperf2
from moler.exceptions import ParsingDone
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_udp_server)
client_connection_lines = [
"[ 1] local 192.168.44.130 port 5016 connected with 192.168.44.1 port 51914",
"[ 2] local 192.168.44.130 port 5016 connected with 192.168.44.1 port 51915",
"[ 3] local 192.168.44.130 port 5016 connected with 192.168.44.1 port 51916",
]
for line in client_connection_lines:
try:
iperf_cmd._parse_connection_name_and_id(line)
except ParsingDone:
pass
parallel_client_1 = ('[email protected]', '[email protected]')
parallel_client_2 = ('[email protected]', '[email protected]')
parallel_client_3 = ('[email protected]', '[email protected]')
first_record = {'Lost_Datagrams_ratio': '0%',
'Jitter': '1.2 ms',
'Transfer': 123904,
'Interval': (0.0, 1.0),
'Transfer Raw': '121 KBytes',
'Bandwidth': 123500,
'Lost_vs_Total_Datagrams': (0, 84),
'Bandwidth Raw': '988 Kbits/sec'}
second_record = dict(first_record)
second_record['Jitter'] = '0.98 ms'
third_record = dict(first_record)
third_record['Jitter'] = '1.48 ms'
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_1] = [first_record]
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_2] = [second_record]
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_3] = [third_record]
iperf_cmd._calculate_multiport_summary_record_of_interval(parallel_client_3)
summary_connection = ('[email protected]', '[email protected]')
assert summary_connection in iperf_cmd.current_ret['CONNECTIONS']
assert iperf_cmd.current_ret['CONNECTIONS'][summary_connection] == [{
'Interval': (0.0, 1.0),
'Transfer': 371712,
'Transfer Raw': '363.0 KBytes',
'Bandwidth': 370500,
'Bandwidth Raw': '2964.0 Kbits/sec',
'Jitter': '1.48 ms',
'Lost_vs_Total_Datagrams': (0, 252),
'Lost_Datagrams_ratio': '0.00%',
}]
def test_iperf_correctly_breaks_server_on_final_inactivity(buffer_connection):
from moler.cmd.unix import iperf2
cmd_output = iperf2.COMMAND_OUTPUT_multiple_connections_udp_server.split("\n")
prompt = cmd_output.pop()
cmd_output_without_prompt = "\n".join(cmd_output) + "\n"
buffer_connection.remote_inject_response([cmd_output_without_prompt])
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_udp_server)
def injecting_break_cmd(self):
self.connection.send("\x03") # ctrl+c
buffer_connection.remote_inject_line(line="^C", add_newline=False)
buffer_connection.remote_inject_line(line=prompt, add_newline=False)
iperf_cmd.break_on_timeout = False # ensuring that break_cmd() is not called via on_timeout()
with mock.patch.object(iperf_cmd.__class__, "break_cmd", injecting_break_cmd):
ret = iperf_cmd()
assert ret == iperf2.COMMAND_RESULT_multiple_connections_udp_server
def test_iperf_server_detects_all_multiport_records_of_interval(buffer_connection):
from moler.cmd.unix import iperf2
from moler.exceptions import ParsingDone
iperf_cmd = iperf2.Iperf2(connection=buffer_connection.moler_connection,
**iperf2.COMMAND_KWARGS_multiple_connections_udp_server)
client_connection_lines = [
"[ 1] local 192.168.44.130 port 5016 connected with 192.168.44.1 port 51914",
"[ 2] local 192.168.44.130 port 5016 connected with 192.168.44.1 port 51915",
"[ 3] local 192.168.44.130 port 5016 connected with 192.168.44.1 port 51916",
]
for line in client_connection_lines:
try:
iperf_cmd._parse_connection_name_and_id(line)
except ParsingDone:
pass
parallel_client_1 = ('[email protected]', '[email protected]')
parallel_client_2 = ('[email protected]', '[email protected]')
parallel_client_3 = ('[email protected]', '[email protected]')
single_record = {'Lost_Datagrams_ratio': '0%',
'Jitter': '1.2 ms',
'Transfer': 123904,
'Interval': (0.0, 1.0),
'Transfer Raw': '121 KBytes',
'Bandwidth': 123500,
'Lost_vs_Total_Datagrams': (0, 84),
'Bandwidth Raw': '988 Kbits/sec'}
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_1] = [single_record]
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_2] = [single_record]
assert iperf_cmd._all_multiport_records_of_interval(connection_name=parallel_client_2) is False
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_3] = [single_record]
assert iperf_cmd._all_multiport_records_of_interval(connection_name=parallel_client_3) is True
second_record = dict(single_record)
second_record['Interval'] = (1.0, 2.0)
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_1] = [second_record]
assert iperf_cmd._all_multiport_records_of_interval(connection_name=parallel_client_1) is False
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_2] = [second_record]
assert iperf_cmd._all_multiport_records_of_interval(connection_name=parallel_client_2) is False
iperf_cmd.current_ret['CONNECTIONS'][parallel_client_3] = [second_record]
assert iperf_cmd._all_multiport_records_of_interval(connection_name=parallel_client_3) is True
