def test_should_get_datacenter_visual_content(
self, mock_ansible_module, mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.datacenters.get_by.return_value = [{
"name":
"Data Center Name",
"uri":
"/rest/datacenter/id"
}]
mock_ov_instance.datacenters.get_visual_content.return_value = {
"name": "Visual Content"
}
mock_ov_client_from_json_file.return_value = mock_ov_instance
mock_ansible_instance = create_ansible_mock(PARAMS_GET_CONNECTED)
mock_ansible_module.return_value = mock_ansible_instance
DatacenterFactsModule().run()
mock_ansible_instance.exit_json.assert_called_once_with(
changed=False,
ansible_facts={
'datacenter_visual_content': {
'name': 'Visual Content'
},
'datacenters': [{
'name': 'Data Center Name',
'uri': '/rest/datacenter/id'
}]
})
def test_should_get_datacenter_visual_content(self):
self.datacenters.get_by.return_value = [{
"name": "Data Center Name",
"uri": "/rest/datacenter/id"
}]
self.datacenters.get_visual_content.return_value = {
"name": "Visual Content"
}
self.mock_ansible_module.params = PARAMS_GET_CONNECTED
DatacenterFactsModule().run()
self.mock_ansible_module.exit_json.assert_called_once_with(
changed=False,
ansible_facts={
'datacenter_visual_content': {
'name': 'Visual Content'
},
'datacenters': [{
'name': 'Data Center Name',
'uri': '/rest/datacenter/id'
}]
})
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
DatacenterFactsModule()
mock_ov_client_from_json_file.assert_called_once_with('config.json')
mock_ov_client_from_env_vars.not_been_called()
def test_should_get_datacenter_by_name(self):
self.datacenters.get_by.return_value = [{"name": "Data Center Name"}]
self.mock_ansible_module.params = PARAMS_GET_BY_NAME
DatacenterFactsModule().run()
self.mock_ansible_module.exit_json.assert_called_once_with(
changed=False,
ansible_facts=dict(datacenters=([{
"name": "Data Center Name"
}])))
def test_should_get_all_datacenters(self):
self.datacenters.get_all.return_value = {"name": "Data Center Name"}
self.mock_ansible_module.params = PARAMS_GET_ALL
DatacenterFactsModule().run()
self.mock_ansible_module.exit_json.assert_called_once_with(
changed=False,
ansible_facts=dict(datacenters=({
"name": "Data Center Name"
})))
def test_should_fail_when_get_all_raises_exception(
self, mock_ansible_module, mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.datacenters.get_all.side_effect = Exception(ERROR_MSG)
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
DatacenterFactsModule().run()
mock_ansible_instance.fail_json.assert_called_once()
def test_should_get_none_datacenter_visual_content(self):
self.datacenters.get_by.return_value = []
self.mock_ansible_module.params = PARAMS_GET_CONNECTED
DatacenterFactsModule().run()
self.mock_ansible_module.exit_json.assert_called_once_with(
changed=False,
ansible_facts={
'datacenter_visual_content': None,
'datacenters': []
})
def test_should_fail_when_get_visual_content_raises_exception(
self, mock_ansible_module, mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.datacenters.get_visual_content.side_effect = Exception(
ERROR_MSG)
mock_ov_instance.datacenters.get_by.return_value = {
"uri": "/rest/datacenter/id"
}
mock_ov_client_from_json_file.return_value = mock_ov_instance
mock_ansible_instance = create_ansible_mock(PARAMS_GET_CONNECTED)
mock_ansible_module.return_value = mock_ansible_instance
DatacenterFactsModule().run()
mock_ansible_instance.fail_json.assert_called_once()
def test_should_get_none_datacenter_visual_content(
self, mock_ansible_module, mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.datacenters.get_by.return_value = []
mock_ov_client_from_json_file.return_value = mock_ov_instance
mock_ansible_instance = create_ansible_mock(PARAMS_GET_CONNECTED)
mock_ansible_module.return_value = mock_ansible_instance
DatacenterFactsModule().run()
mock_ansible_instance.exit_json.assert_called_once_with(
changed=False,
ansible_facts={
'datacenter_visual_content': None,
'datacenters': []
})
def test_should_get_all_datacenters(self, mock_ansible_module,
mock_ov_client_from_json_file):
mock_ov_instance = mock.Mock()
mock_ov_instance.datacenters.get_all.return_value = {
"name": "Data Center 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
DatacenterFactsModule().run()
mock_ansible_instance.exit_json.assert_called_once_with(
changed=False,
ansible_facts=dict(datacenters=({
"name": "Data Center Name"
})))