def test_find_datacenter(self):
mock_account = mock.MagicMock()
self.module.params = {}
mock_account.data = {'primaryDataCenter': 'LOC1'}
self.clc.v2.Account.return_value = mock_account
self.clc.v2.Datacenter.return_value = 'DC1'
ret = ClcModifyServer._find_datacenter(self.clc, self.module)
self.assertEqual(ret, 'DC1')
def test_find_datacenter(self):
mock_account = mock.MagicMock()
self.module.params = {}
mock_account.data = {
'primaryDataCenter' : 'LOC1'
}
self.clc.v2.Account.return_value = mock_account
self.clc.v2.Datacenter.return_value = 'DC1'
ret = ClcModifyServer._find_datacenter(self.clc, self.module)
self.assertEqual(ret, 'DC1')
def test_find_datacenter_error(self):
self.module.params = {'location': 'test'}
mock_account = mock.MagicMock()
self.clc.v2.Account.return_value = mock_account
error = CLCException()
error.message = 'Mock failure message'
self.clc.v2.Datacenter.side_effect = error
ret = ClcModifyServer._find_datacenter(self.clc, self.module)
self.module.fail_json.assert_called_with(
msg='Unable to find location: test. Mock failure message')
self.assertEqual(ret, None)
def test_find_datacenter_error(self):
self.module.params = {
'location': 'test'
}
mock_account = mock.MagicMock()
self.clc.v2.Account.return_value = mock_account
error = CLCException()
error.message = 'Mock failure message'
self.clc.v2.Datacenter.side_effect = error
ret = ClcModifyServer._find_datacenter(self.clc, self.module)
self.module.fail_json.assert_called_with(msg='Unable to find location: test. Mock failure message')
self.assertEqual(ret, None)
