def test_should_get_power_device_by_name_with_options(self):
self.power_devices.get_by.return_value = [{
"name": "Power Device Name",
"uri": "resuri"
}]
self.power_devices.get_power_state.return_value = {'subresource': 'ps'}
self.power_devices.get_uid_state.return_value = {'subresource': 'uid'}
self.power_devices.get_utilization.return_value = {
'subresource': 'util'
}
self.mock_ansible_module.params = self.PARAMS_WITH_OPTIONS
PowerDeviceFactsModule().run()
self.mock_ansible_module.exit_json.assert_called_once_with(
changed=False,
ansible_facts={
'power_devices': [{
'name': 'Power Device Name',
'uri': 'resuri'
}],
'power_device_power_state': {
'subresource': 'ps'
},
'power_device_uid_state': {
'subresource': 'uid'
},
'power_device_utilization': {
'subresource': 'util'
},
})
def test_should_get_power_device_by_name(self):
self.power_devices.get_by.return_value = {"name": "Power Device Name"}
self.mock_ansible_module.params = self.PARAMS_GET_BY_NAME
PowerDeviceFactsModule().run()
self.mock_ansible_module.exit_json.assert_called_once_with(
changed=False,
ansible_facts=dict(power_devices=({
"name": "Power Device Name"
})))
def test_should_load_config_from_file(self, mock_ansible_module,
mock_ov_client_from_env_vars,
mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_client_from_json_file.return_value = mock_ov_instance
mock_ansible_instance = create_ansible_mock({'config': 'config.json'})
mock_ansible_module.return_value = mock_ansible_instance
PowerDeviceFactsModule()
mock_ov_client_from_json_file.assert_called_once_with('config.json')
mock_ov_client_from_env_vars.not_been_called()
def test_should_fail_when_get_by_name_raises_exception(
self, mock_ansible_module, mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.power_devices.get_by.side_effect = Exception(
ERROR_MSG)
mock_ov_client_from_json_file.return_value = mock_ov_instance
mock_ansible_instance = create_ansible_mock(PARAMS_GET_BY_NAME)
mock_ansible_module.return_value = mock_ansible_instance
PowerDeviceFactsModule().run()
mock_ansible_instance.fail_json.assert_called_once()
def test_should_get_all_power_devices(self, mock_ansible_module,
mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.power_devices.get_all.return_value = {
"name": "Power Device Name"
}
mock_ov_client_from_json_file.return_value = mock_ov_instance
mock_ansible_instance = create_ansible_mock(PARAMS_GET_ALL)
mock_ansible_module.return_value = mock_ansible_instance
PowerDeviceFactsModule().run()
mock_ansible_instance.exit_json.assert_called_once_with(
changed=False,
ansible_facts=dict(power_devices=({
"name": "Power Device Name"
})))
def test_should_get_power_device_by_name_with_options(
self, mock_ansible_module, mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.power_devices.get_by.return_value = [{
"name": "Power Device Name",
"uri": "resuri"
}]
mock_ov_instance.power_devices.get_power_state.return_value = {
'subresource': 'ps'
}
mock_ov_instance.power_devices.get_uid_state.return_value = {
'subresource': 'uid'
}
mock_ov_instance.power_devices.get_utilization.return_value = {
'subresource': 'util'
}
mock_ov_client_from_json_file.return_value = mock_ov_instance
mock_ansible_instance = create_ansible_mock(PARAMS_WITH_OPTIONS)
mock_ansible_module.return_value = mock_ansible_instance
PowerDeviceFactsModule().run()
mock_ansible_instance.exit_json.assert_called_once_with(
changed=False,
ansible_facts={
'power_devices': [{
'name': 'Power Device Name',
'uri': 'resuri'
}],
'power_device_power_state': {
'subresource': 'ps'
},
'power_device_uid_state': {
'subresource': 'uid'
},
'power_device_utilization': {
'subresource': 'util'
},
})