From 2708ea7280b51bffb71093384dc49664c5440034 Mon Sep 17 00:00:00 2001 From: wym_libra Date: Tue, 8 Dec 2015 15:47:26 +0800 Subject: Add 'nodes' attribute to the scenario definition Defining the 'nodes' attribute which can include more node not only 'host' and 'target' Design etherpad link: https://etherpad.opnfv.org/p/yardstick_framework JIRA:- Change-Id: Ida18ebcda1c73c88d208aa11a10696d1063134ef Signed-off-by: wym_libra --- tests/unit/cmd/commands/test_task.py | 38 ++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 tests/unit/cmd/commands/test_task.py (limited to 'tests/unit/cmd/commands/test_task.py') diff --git a/tests/unit/cmd/commands/test_task.py b/tests/unit/cmd/commands/test_task.py new file mode 100644 index 000000000..89813cb98 --- /dev/null +++ b/tests/unit/cmd/commands/test_task.py @@ -0,0 +1,38 @@ +#!/usr/bin/env python + +############################################################################## +# Copyright (c) 2015 Huawei Technologies Co.,Ltd 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.cmd.commands.task + +import mock +import unittest + +from yardstick.cmd.commands import task + + +class TaskCommandsTestCase(unittest.TestCase): + + @mock.patch('yardstick.cmd.commands.task.Context') + def test_parse_nodes_host_target_same_context(self, mock_context): + nodes = { + "host": "node1.LF", + "target": "node2.LF" + } + scenario_cfg = {"nodes": nodes} + server_info = { + "ip": "10.20.0.3", + "user": "root", + "key_filename": "/root/.ssh/id_rsa" + } + mock_context.get_server.return_value = server_info + context_cfg = task.parse_nodes_with_context(scenario_cfg) + + self.assertEqual(context_cfg["host"], server_info) + self.assertEqual(context_cfg["target"], server_info) -- cgit 1.2.3-korg