9534b5f682
Realistically we just want to execute a command, potentially inside a container. No need for all the context managers cleverness. Change-Id: I1a85dcd310f9ec0b24435dcd2d8446de7f48d65b
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
# Copyright 2016
|
|
# 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.
|
|
|
|
from oslo_log import log as logging
|
|
from tempest import config
|
|
from tempest.lib.common import ssh
|
|
|
|
CONF = config.CONF
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def sudo(command):
|
|
return 'sudo %s' % command
|
|
|
|
|
|
def in_container(container_name, command):
|
|
if CONF.whitebox.containers:
|
|
return 'docker exec -u root %s %s' % (container_name, command)
|
|
return command
|
|
|
|
|
|
class SSHClient(object):
|
|
"""A client to execute remote commands, based on tempest.lib.common.ssh."""
|
|
|
|
def __init__(self, hostname):
|
|
self.ssh_key = CONF.whitebox.ctlplane_ssh_private_key_path
|
|
self.ssh_user = CONF.whitebox.ctlplane_ssh_username
|
|
self.host = hostname
|
|
|
|
def execute(self, command):
|
|
ssh_client = ssh.Client(self.host, self.ssh_user,
|
|
key_filename=self.ssh_key)
|
|
LOG.debug("Executing %s", command)
|
|
return ssh_client.exec_command(command)
|
|
|
|
|
|
class VirshXMLClient(SSHClient):
|
|
"""A client to obtain libvirt XML from a remote host."""
|
|
|
|
def dumpxml(self, domain):
|
|
command = sudo(in_container('nova_libvirt',
|
|
'virsh dumpxml %s' % domain))
|
|
return self.execute(command)
|
|
|
|
def capabilities(self):
|
|
command = sudo(in_container('nova_libvirt', 'virsh capabilities'))
|
|
return self.execute(command)
|