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)