tobiko/tobiko/tests/functional/shell/test_reboot.py

93 lines
3.1 KiB
Python

# Copyright (c) 2019 Red Hat, 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.
from __future__ import absolute_import
import time
from oslo_log import log
import testtools
import tobiko
from tobiko.shell import sh
from tobiko.openstack import nova
from tobiko.openstack import stacks
LOG = log.getLogger(__name__)
class RebootableServer(stacks.CirrosServerStackFixture):
"Server to be rebooted"
class RebootHostTest(testtools.TestCase):
stack = tobiko.required_setup_fixture(RebootableServer)
def test_reboot_host(self, **params):
server = nova.activate_server(self.stack.server_id)
self.assertEqual('ACTIVE', server.status)
ssh_client = self.stack.ssh_client
uptime_0 = sh.get_uptime(ssh_client=ssh_client)
LOG.debug("Testing reboot command on remote host: "
"uptime=%r", uptime_0)
boottime_0 = time.time() - uptime_0
sh.reboot_host(ssh_client=ssh_client, **params)
server = nova.wait_for_server_status(server, 'ACTIVE')
self.assertEqual('ACTIVE', server.status)
wait = params.get('wait', True)
if wait:
self.assert_is_connected(ssh_client)
uptime_1 = sh.get_uptime(ssh_client=ssh_client)
boottime_1 = time.time() - uptime_1
LOG.debug("Reboot operation executed on remote host: "
"uptime=%r", uptime_1)
self.assertGreater(boottime_1, boottime_0)
else:
self.assert_is_not_connected(ssh_client)
def test_reboot_host_with_wait(self):
self.test_reboot_host(wait=True)
def test_reboot_host_with_no_wait(self):
self.test_reboot_host(wait=False)
def test_reboot_server_after_shutoff(self):
server = nova.activate_server(self.stack.server_id)
self.assertEqual('ACTIVE', server.status)
ssh_client = self.stack.ssh_client
ssh_client.connect()
self.assert_is_connected(ssh_client)
server = nova.shutoff_server(self.stack.server_id)
self.assertEqual('SHUTOFF', server.status)
self.assertRaises(sh.HostNameError, sh.reboot_host,
ssh_client=ssh_client, timeout=5.0)
self.assert_is_not_connected(ssh_client)
server = nova.wait_for_server_status(self.stack.server_id, 'SHUTOFF')
self.assertEqual('SHUTOFF', server.status)
def assert_is_connected(self, ssh_client):
self.assertIsNotNone(ssh_client.client)
def assert_is_not_connected(self, ssh_client):
self.assertIsNone(ssh_client.client)