3d8b423db3
This patch allows VM runcommand scenario attach a floating ip to the instance to perform ssh connection. Blueprint: benchmark-scenarios-vm-floating-ip Change-Id: I2e77d6be34a64d5feb9d3ecc5e35c8a712b10bed
107 lines
4.0 KiB
Python
107 lines
4.0 KiB
Python
# Copyright 2013: Mirantis Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import mock
|
|
import subprocess
|
|
|
|
from oslotest import mockpatch
|
|
from rally.benchmark.scenarios.vm import utils
|
|
from tests import fakes
|
|
from tests import test
|
|
|
|
|
|
VMTASKS_UTILS = "rally.benchmark.scenarios.vm.utils"
|
|
|
|
|
|
class VMScenarioTestCase(test.TestCase):
|
|
|
|
def setUp(self):
|
|
super(VMScenarioTestCase, self).setUp()
|
|
self.wait_for = mockpatch.Patch(VMTASKS_UTILS +
|
|
".bench_utils.wait_for")
|
|
self.useFixture(self.wait_for)
|
|
|
|
@mock.patch('__builtin__.open')
|
|
def test_run_action(self, mock_open):
|
|
mock_ssh = mock.MagicMock()
|
|
mock_file_handle = mock.MagicMock()
|
|
mock_open.return_value = mock_file_handle
|
|
vm_scenario = utils.VMScenario()
|
|
vm_scenario.run_action(mock_ssh, 'interpreter', 'script')
|
|
mock_ssh.execute.assert_called_once_with('interpreter',
|
|
stdin=mock_file_handle)
|
|
|
|
def test_wait_for_ssh(self):
|
|
ssh = mock.MagicMock()
|
|
vm_scenario = utils.VMScenario()
|
|
vm_scenario.wait_for_ssh(ssh)
|
|
ssh.wait.assert_called_once_with()
|
|
|
|
@mock.patch(VMTASKS_UTILS + ".VMScenario.ping_ip_address")
|
|
def test_wait_for_ping(self, mock_ping):
|
|
mock_ping.return_value = True
|
|
vm_scenario = utils.VMScenario()
|
|
vm_scenario.wait_for_ping("1.2.3.4")
|
|
self.wait_for.mock.assert_called_once_with("1.2.3.4",
|
|
is_ready=mock_ping,
|
|
timeout=120)
|
|
|
|
@mock.patch(VMTASKS_UTILS + ".VMScenario.run_action")
|
|
@mock.patch(VMTASKS_UTILS + ".VMScenario.wait_for_ping")
|
|
@mock.patch("rally.sshutils.SSH")
|
|
def test_run_command(self, mock_ssh_class, mock_wait_ping,
|
|
mock_run_action):
|
|
mock_ssh_instance = mock.MagicMock()
|
|
mock_ssh_class.return_value = mock_ssh_instance
|
|
|
|
vm_scenario = utils.VMScenario()
|
|
vm_scenario._context = {"user": {"keypair": {"private": "ssh"}}}
|
|
vm_scenario.run_command("1.2.3.4", 22, "username", "int", "script")
|
|
|
|
mock_wait_ping.assert_called_once_with("1.2.3.4")
|
|
mock_ssh_class.assert_called_once_with("username", "1.2.3.4", port=22,
|
|
pkey="ssh")
|
|
mock_ssh_instance.wait.assert_called_once_with()
|
|
mock_run_action.assert_called_once_with(mock_ssh_instance,
|
|
"int", "script")
|
|
|
|
def test_check_network(self):
|
|
vm_scenario = utils.VMScenario()
|
|
fake_server = fakes.FakeServer()
|
|
fake_server.addresses = {}
|
|
self.assertRaises(ValueError,
|
|
vm_scenario.check_network, fake_server, "private")
|
|
fake_server.addresses["private_1"] = {
|
|
"version": 4,
|
|
"addr": "1.2.3.4"
|
|
}
|
|
vm_scenario.check_network(fake_server, "private_1")
|
|
|
|
@mock.patch("subprocess.Popen")
|
|
def test_ping_ip_address(self, mock_subprocess):
|
|
|
|
ping_process = mock.MagicMock()
|
|
ping_process.returncode = 0
|
|
mock_subprocess.return_value = ping_process
|
|
|
|
vm_scenario = utils.VMScenario()
|
|
host_ip = "1.2.3.4"
|
|
self.assertTrue(vm_scenario.ping_ip_address(host_ip))
|
|
|
|
mock_subprocess.assert_called_once_with(
|
|
['ping', '-c1', '-w1', host_ip],
|
|
stderr=subprocess.PIPE, stdout=subprocess.PIPE)
|
|
ping_process.wait.assert_called_once_with()
|