| #!/usr/bin/python |
| # Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Test iperf parser.""" |
| |
| import unittest, iperf |
| |
| |
| class test_iperf(unittest.TestCase): |
| def test_parse_warning(self): |
| text = """ |
| WARNING: option -b implies udp testing |
| WARNING: option -b is not valid for server mode |
| ------------------------------------------------------------ |
| Client connecting to 172.31.206.152, UDP port 5001 |
| Sending 1470 byte datagrams |
| UDP buffer size: 0.11 MByte (default) |
| ------------------------------------------------------------ |
| [ 3] local 172.31.206.145 port 39777 connected with 172.31.206.152 port 5001 |
| [ ID] Interval Transfer Bandwidth |
| [ 3] 0.0-15.0 sec 1.88 MBytes 1.05 Mbits/sec |
| [ 3] Sent 1339 datagrams |
| [ 3] Server Report: |
| [ ID] Interval Transfer Bandwidth Jitter Lost/Total Datagrams |
| [ 3] 0.0-15.0 sec 1.88 MBytes 1.05 Mbits/sec 0.011 ms 1/ 1340 (0.075%) |
| """.lstrip().rstrip() |
| self.assertRaises(iperf.Error, iperf.ParseIperfOutput, text) |
| |
| def test_connect_failed(self): |
| text = """connect failed: Connection refused |
| ------------------------------------------------------------ |
| Client connecting to 172.31.206.152, TCP port 5001 |
| TCP window size: 0.02 MByte (default) |
| ------------------------------------------------------------ |
| write1 failed: Broken pipe |
| [ 3] local 0.0.0.0 port 57288 connected with 172.31.206.152 port 5001 |
| write2 failed: Broken pipe |
| [ ID] Interval Transfer Bandwidth |
| [ 3] 0.0- 0.0 sec 0.00 MBytes 0.00 Mbits/sec |
| """.lstrip().rstrip() |
| self.assertRaises(iperf.Error, iperf.ParseIperfOutput, text) |
| |
| def test_bad_udp(self): |
| text = """------------------------------------------------------------ |
| Client connecting to 172.31.206.152, UDP port 5001 |
| Sending 1470 byte datagrams |
| UDP buffer size: 0.11 MByte (default) |
| ------------------------------------------------------------ |
| [ 3] local 172.31.206.145 port 51532 connected with 172.31.206.152 port 5001 |
| [ ID] Interval Transfer Bandwidth |
| [ 3] 0.0- 5.2 sec 0.06 MBytes 0.10 Mbits/sec |
| [ 3] Sent 44 datagrams |
| [ 3] Server Report: |
| |
| [ ID] Interval Transfer Bandwidth Jitter Lost/Total Datagrams |
| [ 3] 0.0- 5.2 sec 0.06 bogus 0.10 Mbits/sec 0.014 ms 1/ 45 (2.2%) |
| """.lstrip().rstrip() |
| self.assertRaises(iperf.Error, iperf.ParseIperfOutput, text) |
| |
| def test_udp(self): |
| text = """------------------------------------------------------------ |
| Client connecting to 172.31.206.152, UDP port 5001 |
| Sending 1470 byte datagrams |
| UDP buffer size: 0.11 MByte (default) |
| ------------------------------------------------------------ |
| [ 3] local 172.31.206.145 port 51532 connected with 172.31.206.152 port 5001 |
| [ ID] Interval Transfer Bandwidth |
| [ 3] 0.0- 5.2 sec 0.06 MBytes 0.10 Mbits/sec |
| [ 3] Sent 44 datagrams |
| [ 3] Server Report: |
| [ ID] Interval Transfer Bandwidth Jitter Lost/Total Datagrams |
| [ 3] 0.0- 5.2 sec 0.06 MBytes 0.10 Mbits/sec 0.014 ms 1/ 45 (2.2%) |
| """.lstrip().rstrip() |
| expected = {'jitter': 0.014, |
| 'throughput': 100000.0, |
| 'lost': 2.2} |
| expected.update(dict([('uplink_' + a, b) |
| for (a,b) in expected.items()])) |
| self.assertEqual(expected, |
| iperf.ParseIperfOutput(text)) |
| |
| |
| def test_udp_tradeoff(self): |
| text = """------------------------------------------------------------ |
| Client connecting to 172.31.206.152, UDP port 5001 |
| Sending 1470 byte datagrams |
| UDP buffer size: 0.11 MByte (default) |
| ------------------------------------------------------------ |
| [ 3] local 172.31.206.145 port 51532 connected with 172.31.206.152 port 5001 |
| [ ID] Interval Transfer Bandwidth |
| [ 3] 0.0- 5.2 sec 0.06 MBytes 0.10 Mbits/sec |
| [ 3] Sent 44 datagrams |
| [ 3] Server Report: |
| [ ID] Interval Transfer Bandwidth Jitter Lost/Total Datagrams |
| [ 3] 0.0- 5.2 sec 0.06 MBytes 0.10 Mbits/sec 0.014 ms 1/ 45 (2.2%) |
| ------------------------------------------------------------ |
| Server listening on UDP port 5001 |
| Receiving 1470 byte datagrams |
| UDP buffer size: 110 KByte (default) |
| ------------------------------------------------------------ |
| [ 3] local 172.31.206.145 port 5001 connected with 172.31.206.152 port 57416 |
| [ ID] Interval Transfer Bandwidth Jitter Lost/Total Datagrams |
| [ 3] 0.0- 3.0 sec 386 KBytes 1.05 Mbits/sec 0.085 ms 1/ 270 (0.37%) |
| """.lstrip().rstrip() |
| expected = {'jitter': 0.014, |
| 'throughput': 100000.0, |
| 'lost': 2.2} |
| expected.update(dict([('uplink_' + a, b) |
| for (a,b) in expected.items()])) |
| expected.update({'downlink_throughput': 1.05e6, |
| 'downlink_jitter': 0.085, |
| 'downlink_lost': 0.37}) |
| |
| self.assertEqual(expected, |
| iperf.ParseIperfOutput(text)) |
| |
| |
| |
| def test_tcp(self): |
| text = """------------------------------------------------------------ |
| Client connecting to 172.31.206.152, TCP port 5001 |
| TCP window size: 0.02 MByte (default) |
| ------------------------------------------------------------ |
| [ 3] local 172.31.206.145 port 38376 connected with 172.31.206.152 port 5001 |
| [ ID] Interval Transfer Bandwidth |
| [ 3] 0.0- 3.0 sec 34.1 MBytes 95.4 Mbits/sec""".lstrip().rstrip() |
| self.assertEqual({'throughput': 95.4e6, |
| 'uplink_throughput': 95.4e6}, |
| iperf.ParseIperfOutput(text)) |
| def test_tcp_tradeoff(self): |
| text = """------------------------------------------------------------ |
| Client connecting to 172.31.206.152, TCP port 5001 |
| TCP window size: 0.02 MByte (default) |
| ------------------------------------------------------------ |
| [ 3] local 172.31.206.145 port 38376 connected with 172.31.206.152 port 5001 |
| [ ID] Interval Transfer Bandwidth |
| [ 3] 0.0- 3.0 sec 34.1 MBytes 95.4 Mbits/sec |
| ------------------------------------------------------------ |
| Server listening on TCP port 5001 |
| TCP window size: 85.3 KByte (default) |
| ------------------------------------------------------------ |
| [ 4] local 172.31.206.145 port 5001 connected with 172.31.206.152 port 56542 |
| [ ID] Interval Transfer Bandwidth |
| [ 4] 0.0- 3.0 sec 34.1 MBytes 94.7 Mbits/sec""".lstrip().rstrip() |
| self.assertEqual({'throughput': 95.4e6, |
| 'uplink_throughput': 95.4e6, |
| 'downlink_throughput': 94.7e6}, |
| iperf.ParseIperfOutput(text)) |
| |
| |
| def test_Multipliers(self): |
| self.assertEqual(1, iperf.ParseMultiplier(1, 'bits/sec')) |
| self.assertEqual(1, iperf.ParseMultiplier('1', 'bits/sec')) |
| self.assertEqual(1000, iperf.ParseMultiplier('1', 'Kbits/sec')) |
| self.assertEqual(1000000, iperf.ParseMultiplier('1', 'Mbits/sec')) |
| self.assertEqual(1000000000, iperf.ParseMultiplier('1', 'Gbits/sec')) |
| self.assertEqual(1000000000, iperf.ParseMultiplier('1000', 'Mbits/sec')) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
