/usr/lib/python3/dist-packages/provisioningserver/drivers/power/tests/test_virsh.py is in python3-maas-provisioningserver 2.0.0~beta3+bzr4941-0ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | # Copyright 2015 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""Tests for `provisioningserver.drivers.power.virsh`."""
__all__ = []
from maastesting.factory import factory
from maastesting.matchers import MockCalledOnceWith
from maastesting.testcase import MAASTestCase
from provisioningserver.drivers.power import virsh as virsh_module
from provisioningserver.drivers.power.virsh import (
extract_virsh_parameters,
VirshPowerDriver,
)
from provisioningserver.utils.shell import has_command_available
from testtools.matchers import Equals
class TestVirshPowerDriver(MAASTestCase):
def test_missing_packages(self):
mock = self.patch(has_command_available)
mock.return_value = False
driver = virsh_module.VirshPowerDriver()
missing = driver.detect_missing_packages()
self.assertItemsEqual(['libvirt-bin'], missing)
def test_no_missing_packages(self):
mock = self.patch(has_command_available)
mock.return_value = True
driver = virsh_module.VirshPowerDriver()
missing = driver.detect_missing_packages()
self.assertItemsEqual([], missing)
def make_parameters(self):
system_id = factory.make_name('system_id')
poweraddr = factory.make_name('power_address')
machine = factory.make_name('power_id')
password = factory.make_name('power_pass')
context = {
'system_id': system_id,
'power_address': poweraddr,
'power_id': machine,
'power_pass': password,
}
return system_id, poweraddr, machine, password, context
def test_extract_virsh_parameters_extracts_parameters(self):
system_id, poweraddr, machine, password, context = (
self.make_parameters())
self.assertItemsEqual(
(poweraddr, machine, password),
extract_virsh_parameters(context))
def test_power_on_calls_power_control_virsh(self):
power_change = 'on'
system_id, poweraddr, machine, password, context = (
self.make_parameters())
virsh_power_driver = VirshPowerDriver()
power_control_virsh = self.patch(
virsh_module, 'power_control_virsh')
virsh_power_driver.power_on(system_id, context)
self.assertThat(
power_control_virsh, MockCalledOnceWith(
poweraddr, machine, power_change, password))
def test_power_off_calls_power_control_virsh(self):
power_change = 'off'
system_id, poweraddr, machine, password, context = (
self.make_parameters())
virsh_power_driver = VirshPowerDriver()
power_control_virsh = self.patch(
virsh_module, 'power_control_virsh')
virsh_power_driver.power_off(system_id, context)
self.assertThat(
power_control_virsh, MockCalledOnceWith(
poweraddr, machine, power_change, password))
def test_power_query_calls_power_state_virsh(self):
system_id, poweraddr, machine, password, context = (
self.make_parameters())
virsh_power_driver = VirshPowerDriver()
power_state_virsh = self.patch(
virsh_module, 'power_state_virsh')
power_state_virsh.return_value = 'off'
expected_result = virsh_power_driver.power_query(system_id, context)
self.expectThat(
power_state_virsh, MockCalledOnceWith(
poweraddr, machine, password))
self.expectThat(expected_result, Equals('off'))
|