Add unit test file for ArithmeticRunner 15/58715/5
authorMiikka Koistinen <miikka.koistinen@nokia.com>
Fri, 15 Jun 2018 12:29:21 +0000 (15:29 +0300)
committerMiikka Koistinen <miikka.koistinen@nokia.com>
Fri, 27 Jul 2018 09:34:01 +0000 (09:34 +0000)
* Initial test: test ArithmeticRunner._run_benchmark called arguments

JIRA: YARDSTICK-1245

Change-Id: Id18c36bd1345052863ab777bf899c00fe66fd713
Signed-off-by: Miikka Koistinen <miikka.koistinen@nokia.com>
yardstick/tests/unit/benchmark/runner/test_arithmetic.py [new file with mode: 0644]

diff --git a/yardstick/tests/unit/benchmark/runner/test_arithmetic.py b/yardstick/tests/unit/benchmark/runner/test_arithmetic.py
new file mode 100644 (file)
index 0000000..7fda011
--- /dev/null
@@ -0,0 +1,38 @@
+##############################################################################
+# Copyright (c) 2018 Nokia 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
+##############################################################################
+
+import mock
+import unittest
+import multiprocessing
+import os
+
+from yardstick.benchmark.runners import arithmetic
+
+
+class ArithmeticRunnerTest(unittest.TestCase):
+    def setUp(self):
+        self.scenario_cfg = {
+            'type': 'some_type'
+        }
+
+    @mock.patch.object(os, 'getpid')
+    @mock.patch.object(multiprocessing, 'Process')
+    def test__run_benchmark_called_with(self, mock_multiprocessing_process,
+                                        mock_os_getpid):
+        mock_os_getpid.return_value = 101
+
+        runner = arithmetic.ArithmeticRunner({})
+        benchmark_cls = mock.Mock()
+        runner._run_benchmark(benchmark_cls, 'my_method', self.scenario_cfg,
+                              {})
+        mock_multiprocessing_process.assert_called_once_with(
+            name='Arithmetic-some_type-101',
+            target=arithmetic._worker_process,
+            args=(runner.result_queue, benchmark_cls, 'my_method',
+                  self.scenario_cfg, {}, runner.aborted, runner.output_queue))