def test__query_outlet_state_crashes_when_state_not_found(self):
driver = dli_module.DLIPowerDriver()
call_and_check_mock = self.patch(dli_module, 'call_and_check')
call_and_check_mock.return_value = b'Rubbish'
self.assertRaises(
PowerError, driver._query_outlet_state, sentinel.outlet_id,
sentinel.power_user, sentinel.power_pass, sentinel.power_address)
def test_set_outlet_state_calls_wget(self):
driver = dli_module.DLIPowerDriver()
env = get_env_with_locale()
power_change = factory.make_name("power_change")
outlet_id = choice(["1", "2", "3", "4", "5", "6", "7", "8"])
power_user = factory.make_name("power_user")
power_pass = factory.make_name("power_pass")
power_address = factory.make_name("power_address")
url = "http://%s:%s@%s/outlet?%s=%s" % (
power_user,
power_pass,
power_address,
outlet_id,
power_change,
)
call_and_check_mock = self.patch(dli_module, "call_and_check")
driver._set_outlet_state(power_change, outlet_id, power_user,
power_pass, power_address)
self.assertThat(
call_and_check_mock,
MockCalledOnceWith(
["wget", "--auth-no-challenge", "-O", "/dev/null", url],
env=env,
),
)
def test_power_query(self):
driver = dli_module.DLIPowerDriver()
system_id = factory.make_name('system_id')
context = {'context': factory.make_name('context')}
_query_outlet_state_mock = self.patch(driver, '_query_outlet_state')
driver.power_query(system_id, context)
self.assertThat(
_query_outlet_state_mock, MockCalledOnceWith(**context))
def test__query_outlet_state_crashes_when_wget_exits_nonzero(self):
driver = dli_module.DLIPowerDriver()
call_and_check_mock = self.patch(dli_module, 'call_and_check')
call_and_check_mock.side_effect = (
ExternalProcessError(1, "dli something"))
self.assertRaises(
PowerActionError, driver._query_outlet_state, sentinel.outlet_id,
sentinel.power_user, sentinel.power_pass, sentinel.power_address)
def test_power_off(self):
driver = dli_module.DLIPowerDriver()
system_id = factory.make_name("system_id")
context = {"context": factory.make_name("context")}
_set_outlet_state_mock = self.patch(driver, "_set_outlet_state")
driver.power_off(system_id, context)
self.assertThat(_set_outlet_state_mock,
MockCalledOnceWith("OFF", **context))
def test_power_on_raises_power_error(self):
driver = dli_module.DLIPowerDriver()
system_id = factory.make_name('system_id')
context = {'outlet_id': factory.make_name('outlet_id')}
_query_outlet_state_mock = self.patch(driver, '_query_outlet_state')
_query_outlet_state_mock.side_effect = ('on', 'not-off')
_set_outlet_state_mock = self.patch(driver, '_set_outlet_state')
self.patch(dli_module, 'sleep')
self.assertRaises(PowerError, driver.power_on, system_id, context)
self.expectThat(_query_outlet_state_mock,
MockCallsMatch(call(**context), call(**context)))
self.expectThat(_set_outlet_state_mock,
MockCalledOnceWith('OFF', **context))
def test_power_on(self):
driver = dli_module.DLIPowerDriver()
system_id = factory.make_name('system_id')
context = {'context': factory.make_name('context')}
_query_outlet_state_mock = self.patch(driver, '_query_outlet_state')
_query_outlet_state_mock.side_effect = ('on', 'off')
_set_outlet_state_mock = self.patch(driver, '_set_outlet_state')
self.patch(dli_module, 'sleep')
driver.power_on(system_id, context)
self.expectThat(_query_outlet_state_mock,
MockCallsMatch(call(**context), call(**context)))
self.expectThat(
_set_outlet_state_mock,
MockCallsMatch(call('OFF', **context), call('ON', **context)))
def test_power_on_raises_power_error(self):
driver = dli_module.DLIPowerDriver()
system_id = factory.make_name("system_id")
context = {"outlet_id": factory.make_name("outlet_id")}
_query_outlet_state_mock = self.patch(driver, "_query_outlet_state")
_query_outlet_state_mock.side_effect = ("on", "not-off")
_set_outlet_state_mock = self.patch(driver, "_set_outlet_state")
self.patch(dli_module, "sleep")
self.assertRaises(PowerError, driver.power_on, system_id, context)
self.expectThat(
_query_outlet_state_mock,
MockCallsMatch(call(**context), call(**context)),
)
self.expectThat(_set_outlet_state_mock,
MockCalledOnceWith("OFF", **context))
def test__set_outlet_state_calls_wget(self):
driver = dli_module.DLIPowerDriver()
env = get_env_with_locale()
power_change = factory.make_name('power_change')
outlet_id = choice(['1', '2', '3', '4', '5', '6', '7', '8'])
power_user = factory.make_name('power_user')
power_pass = factory.make_name('power_pass')
power_address = factory.make_name('power_address')
url = 'http://%s:%s@%s/outlet?%s=%s' % (
power_user, power_pass, power_address, outlet_id, power_change)
call_and_check_mock = self.patch(dli_module, 'call_and_check')
driver._set_outlet_state(
power_change, outlet_id, power_user, power_pass, power_address)
self.assertThat(
call_and_check_mock, MockCalledOnceWith(
['wget', '--auth-no-challenge', '-O', '/dev/null', url],
env=env))
def test_power_on(self):
driver = dli_module.DLIPowerDriver()
system_id = factory.make_name("system_id")
context = {"context": factory.make_name("context")}
_query_outlet_state_mock = self.patch(driver, "_query_outlet_state")
_query_outlet_state_mock.side_effect = ("on", "off")
_set_outlet_state_mock = self.patch(driver, "_set_outlet_state")
self.patch(dli_module, "sleep")
driver.power_on(system_id, context)
self.expectThat(
_query_outlet_state_mock,
MockCallsMatch(call(**context), call(**context)),
)
self.expectThat(
_set_outlet_state_mock,
MockCallsMatch(call("OFF", **context), call("ON", **context)),
)
def test__query_outlet_state_queries_on(self):
driver = dli_module.DLIPowerDriver()
env = get_env_with_locale()
outlet_id = choice(['1', '2', '3', '4', '5', '6', '7', '8'])
power_user = factory.make_name('power_user')
power_pass = factory.make_name('power_pass')
power_address = factory.make_name('power_address')
url = 'http://%s:%s@%s/index.htm' % (
power_user, power_pass, power_address)
call_and_check_mock = self.patch(dli_module, 'call_and_check')
call_and_check_mock.return_value = DLI_QUERY_OUTPUT % b'ff'
result = driver._query_outlet_state(
outlet_id, power_user, power_pass, power_address)
self.expectThat(
call_and_check_mock, MockCalledOnceWith(
['wget', '--auth-no-challenge', '-qO-', url],
env=env))
self.expectThat(result, Equals('on'))
def test_query_outlet_state_queries_on(self):
driver = dli_module.DLIPowerDriver()
env = get_env_with_locale()
outlet_id = choice(["1", "2", "3", "4", "5", "6", "7", "8"])
power_user = factory.make_name("power_user")
power_pass = factory.make_name("power_pass")
power_address = factory.make_name("power_address")
url = "http://%s:%s@%s/index.htm" % (
power_user,
power_pass,
power_address,
)
call_and_check_mock = self.patch(dli_module, "call_and_check")
call_and_check_mock.return_value = DLI_QUERY_OUTPUT % b"ff"
result = driver._query_outlet_state(outlet_id, power_user, power_pass,
power_address)
self.expectThat(
call_and_check_mock,
MockCalledOnceWith(["wget", "--auth-no-challenge", "-qO-", url],
env=env),
)
self.expectThat(result, Equals("on"))
def test_no_missing_packages(self):
mock = self.patch(has_command_available)
mock.return_value = True
driver = dli_module.DLIPowerDriver()
missing = driver.detect_missing_packages()
self.assertItemsEqual([], missing)