diff options
Diffstat (limited to 'tests/unit/benchmark/scenarios')
-rw-r--r-- | tests/unit/benchmark/scenarios/networking/test_sfc.py | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/tests/unit/benchmark/scenarios/networking/test_sfc.py b/tests/unit/benchmark/scenarios/networking/test_sfc.py new file mode 100644 index 000000000..adce0824a --- /dev/null +++ b/tests/unit/benchmark/scenarios/networking/test_sfc.py @@ -0,0 +1,55 @@ +#!/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.sfc + +import mock +import unittest + +from yardstick.benchmark.scenarios.networking import sfc + + +class SfcTestCase(unittest.TestCase): + + def setUp(self): + scenario_cfg = dict() + context_cfg = dict() + + # Used in Sfc.setup() + context_cfg['target'] = dict() + context_cfg['target']['user'] = 'root' + context_cfg['target']['password'] = 'octopus' + context_cfg['target']['ip'] = None + + # Used in Sfc.run() + context_cfg['host'] = dict() + context_cfg['host']['user'] = 'cirros' + context_cfg['host']['password'] = 'cubslose:)' + context_cfg['host']['ip'] = None + context_cfg['target'] = dict() + context_cfg['target']['ip'] = None + + self.sfc = sfc.Sfc(scenario_cfg=scenario_cfg, context_cfg=context_cfg) + + @mock.patch('yardstick.benchmark.scenarios.networking.sfc.ssh') + def test_run_for_success(self, mock_ssh): + # Mock a successfull SSH in Sfc.setup() and Sfc.run() + mock_ssh.SSH().execute.return_value = (0, '100', '') + + result = {} + self.sfc.run(result) + + +def main(): + unittest.main() + +if __name__ == '__main__': + main() |