def test_list_datacenters(self):
mock_list_datacenters = MagicMock()
with patch.dict(
esxdatacenter.__salt__,
{'vsphere.list_datacenters_via_proxy': mock_list_datacenters}):
esxdatacenter.datacenter_configured('fake_dc')
mock_list_datacenters.assert_called_once_with(
datacenter_names=['fake_dc'], service_instance=self.mock_si)
def test_get_service_instance(self):
mock_get_service_instance = MagicMock()
with patch.dict(esxdatacenter.__salt__, {
'vsphere.get_service_instance_via_proxy':
mock_get_service_instance
}):
esxdatacenter.datacenter_configured('fake_dc')
mock_get_service_instance.assert_called_once_with()
def test_create_datacenter(self):
mock_create_datacenter = MagicMock()
with patch.dict(
esxdatacenter.__salt__,
{
"vsphere.list_datacenters_via_proxy":
MagicMock(return_value=[]),
"vsphere.create_datacenter": mock_create_datacenter,
},
):
res = esxdatacenter.datacenter_configured("fake_dc")
mock_create_datacenter.assert_called_once_with("fake_dc", self.mock_si)
self.assertDictEqual(
res,
{
"name": "fake_dc",
"changes": {
"new": {
"name": "fake_dc"
}
},
"result": True,
"comment": "Created datacenter 'fake_dc'.",
},
)
def test_state_raise_command_execution_error_test_mode(self):
with patch.dict(esxdatacenter.__opts__, {'test': True}):
with patch.dict(
esxdatacenter.__salt__, {
'vsphere.list_datacenters_via_proxy':
MagicMock(side_effect=CommandExecutionError('Error'))
}):
res = esxdatacenter.datacenter_configured('fake_dc')
self.assertDictEqual(res, {
'name': 'fake_dc',
'changes': {},
'result': None,
'comment': 'Error'
})
def test_state_raise_command_execution_error_test_mode(self):
with patch.dict(esxdatacenter.__opts__, {"test": True}):
with patch.dict(
esxdatacenter.__salt__,
{
"vsphere.list_datacenters_via_proxy": MagicMock(
side_effect=CommandExecutionError("Error")
)
},
):
res = esxdatacenter.datacenter_configured("fake_dc")
self.assertDictEqual(
res, {"name": "fake_dc", "changes": {}, "result": None, "comment": "Error"}
)
def test_dc_name_different_proxy(self):
with patch.dict(
esxdatacenter.__salt__,
{"vsphere.get_proxy_type": MagicMock(return_value="different_proxy")},
):
res = esxdatacenter.datacenter_configured("fake_dc")
self.assertDictEqual(
res,
{
"name": "fake_dc",
"changes": {},
"result": True,
"comment": "Datacenter 'fake_dc' already "
"exists. Nothing to be done.",
},
)
def test_state_raise_command_execution_error_after_si(self):
mock_disconnect = MagicMock()
with patch.dict(
esxdatacenter.__salt__,
{
"vsphere.disconnect": mock_disconnect,
"vsphere.list_datacenters_via_proxy": MagicMock(
side_effect=CommandExecutionError("Error")
),
},
):
res = esxdatacenter.datacenter_configured("fake_dc")
mock_disconnect.assert_called_once_with(self.mock_si)
self.assertDictEqual(
res, {"name": "fake_dc", "changes": {}, "result": False, "comment": "Error"}
)
def test_state_get_service_instance_raise_command_execution_error(self):
mock_disconnect = MagicMock()
with patch.dict(
esxdatacenter.__salt__,
{
"vsphere.disconnect": mock_disconnect,
"vsphere.get_service_instance_via_proxy": MagicMock(
side_effect=CommandExecutionError("Error")
),
},
):
res = esxdatacenter.datacenter_configured("fake_dc")
self.assertEqual(mock_disconnect.call_count, 0)
self.assertDictEqual(
res, {"name": "fake_dc", "changes": {}, "result": False, "comment": "Error"}
)
def test_create_datacenter_test_mode(self):
with patch.dict(esxdatacenter.__opts__, {"test": True}):
with patch.dict(
esxdatacenter.__salt__,
{"vsphere.list_datacenters_via_proxy": MagicMock(return_value=[])},
):
res = esxdatacenter.datacenter_configured("fake_dc")
self.assertDictEqual(
res,
{
"name": "fake_dc",
"changes": {"new": {"name": "fake_dc"}},
"result": None,
"comment": "State will create " "datacenter 'fake_dc'.",
},
)
def test_dc_name_different_proxy(self):
with patch.dict(esxdatacenter.__salt__, {
'vsphere.get_proxy_type':
MagicMock(return_value='different_proxy')
}):
res = esxdatacenter.datacenter_configured('fake_dc')
self.assertDictEqual(
res, {
'name':
'fake_dc',
'changes': {},
'result':
True,
'comment':
'Datacenter \'fake_dc\' already '
'exists. Nothing to be done.'
})
def test_state_raise_command_execution_error_after_si(self):
mock_disconnect = MagicMock()
with patch.dict(
esxdatacenter.__salt__, {
'vsphere.disconnect':
mock_disconnect,
'vsphere.list_datacenters_via_proxy':
MagicMock(side_effect=CommandExecutionError('Error'))
}):
res = esxdatacenter.datacenter_configured('fake_dc')
mock_disconnect.assert_called_once_with(self.mock_si)
self.assertDictEqual(res, {
'name': 'fake_dc',
'changes': {},
'result': False,
'comment': 'Error'
})
def test_state_get_service_instance_raise_command_execution_error(self):
mock_disconnect = MagicMock()
with patch.dict(
esxdatacenter.__salt__, {
'vsphere.disconnect':
mock_disconnect,
'vsphere.get_service_instance_via_proxy':
MagicMock(side_effect=CommandExecutionError('Error'))
}):
res = esxdatacenter.datacenter_configured('fake_dc')
self.assertEqual(mock_disconnect.call_count, 0)
self.assertDictEqual(res, {
'name': 'fake_dc',
'changes': {},
'result': False,
'comment': 'Error'
})
def test_create_datacenter_test_mode(self):
with patch.dict(esxdatacenter.__opts__, {'test': True}):
with patch.dict(esxdatacenter.__salt__, {
'vsphere.list_datacenters_via_proxy':
MagicMock(return_value=[])
}):
res = esxdatacenter.datacenter_configured('fake_dc')
self.assertDictEqual(
res, {
'name': 'fake_dc',
'changes': {
'new': {
'name': 'fake_dc'
}
},
'result': None,
'comment': 'State will create '
'datacenter \'fake_dc\'.'
})
def test_create_datacenter(self):
mock_create_datacenter = MagicMock()
with patch.dict(
esxdatacenter.__salt__, {
'vsphere.list_datacenters_via_proxy':
MagicMock(return_value=[]),
'vsphere.create_datacenter': mock_create_datacenter
}):
res = esxdatacenter.datacenter_configured('fake_dc')
mock_create_datacenter.assert_called_once_with('fake_dc', self.mock_si)
self.assertDictEqual(
res, {
'name': 'fake_dc',
'changes': {
'new': {
'name': 'fake_dc'
}
},
'result': True,
'comment': 'Created datacenter \'fake_dc\'.'
})
def test_dc_name_esxdatacenter_proxy(self):
with patch.dict(
esxdatacenter.__salt__,
{
"vsphere.get_proxy_type": MagicMock(return_value="esxdatacenter"),
"esxdatacenter.get_details": MagicMock(
return_value={"datacenter": "proxy_dc"}
),
},
):
res = esxdatacenter.datacenter_configured("fake_dc")
self.assertDictEqual(
res,
{
"name": "fake_dc",
"changes": {},
"result": True,
"comment": "Datacenter 'proxy_dc' "
"already exists. Nothing to be done.",
},
)