diff options
author | 2015-06-16 08:57:07 +0000 | |
---|---|---|
committer | 2015-06-16 08:57:07 +0000 | |
commit | f68ce570fd0eefb17bfc9e1e75c57b24050db89e (patch) | |
tree | c9d92089e7bb0a839fa8c6119287a765559fd084 /tests/unit/benchmark/scenarios/networking/test_ping.py | |
parent | 585a2483e9fb021d853b0946f185d066335adea4 (diff) | |
parent | 17e94ff1e86c2a98895910d63828081ad5867052 (diff) |
Merge "add unit test for ping"
Diffstat (limited to 'tests/unit/benchmark/scenarios/networking/test_ping.py')
-rw-r--r-- | tests/unit/benchmark/scenarios/networking/test_ping.py | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/tests/unit/benchmark/scenarios/networking/test_ping.py b/tests/unit/benchmark/scenarios/networking/test_ping.py new file mode 100644 index 000000000..d930adcee --- /dev/null +++ b/tests/unit/benchmark/scenarios/networking/test_ping.py @@ -0,0 +1,91 @@ +#!/usr/bin/env python + +############################################################################## +# Copyright (c) 2015 Ericsson AB and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## + +# Unittest for yardstick.benchmark.scenarios.networking.ping.Ping + +import mock +import unittest + +from yardstick.benchmark.scenarios.networking import ping + + +class PingTestCase(unittest.TestCase): + + def setUp(self): + self.ctx = { + 'host': '172.16.0.137', + 'user': 'cirros', + 'key_filename': "mykey.key" + } + + @mock.patch('yardstick.benchmark.scenarios.networking.ping.ssh') + def test_ping_successful_no_sla(self, mock_ssh): + + p = ping.Ping(self.ctx) + + args = { + 'options': {'packetsize': 200}, + 'ipaddr': '172.16.0.138' + } + + mock_ssh.SSH().execute.return_value = (0, '100', '') + result = p.run(args) + self.assertEqual(result, float(mock_ssh.SSH().execute.return_value[1])) + + @mock.patch('yardstick.benchmark.scenarios.networking.ping.ssh') + def test_ping_successful_sla(self, mock_ssh): + + p = ping.Ping(self.ctx) + + args = { + 'options': {'packetsize': 200}, + 'ipaddr': '172.16.0.138', + 'sla': {'max_rtt': 150} + } + + mock_ssh.SSH().execute.return_value = (0, '100', '') + result = p.run(args) + self.assertEqual(result, float(mock_ssh.SSH().execute.return_value[1])) + + @mock.patch('yardstick.benchmark.scenarios.networking.ping.ssh') + def test_ping_unsuccessful_sla(self, mock_ssh): + + p = ping.Ping(self.ctx) + + args = { + 'options': {'packetsize': 200}, + 'ipaddr': '172.16.0.138', + 'sla': {'max_rtt': 50} + } + + mock_ssh.SSH().execute.return_value = (0, '100', '') + self.assertRaises(AssertionError, p.run, args) + + @mock.patch('yardstick.benchmark.scenarios.networking.ping.ssh') + def test_ping_unsuccessful_script_error(self, mock_ssh): + + p = ping.Ping(self.ctx) + + args = { + 'options': {'packetsize': 200}, + 'ipaddr': '172.16.0.138', + 'sla': {'max_rtt': 50} + } + + mock_ssh.SSH().execute.return_value = (1, '', 'FOOBAR') + self.assertRaises(RuntimeError, p.run, args) + + +def main(): + unittest.main() + +if __name__ == '__main__': + main() |