def testGetInfo(self, mock_client): # SETUP ds = DomainService() ds.build_cookie = Mock(return_value={"cookie": "value"}) ds.update_cookie = Mock() i = mock_client.return_value i.service.getInfo.return_value = ['foo'] # CALL result = ds.get_info('example.com') # VERIFY ds.build_cookie.assert_called_with(mode=MODE_RO, method='getInfo', parameters=['example.com']) ds.update_cookie.assert_called_with({"cookie": "value"}) i.service.getInfo.assert_called_with('example.com') self.assertEqual(result, ['foo'])
def test_get_info(self, mock_client): # SETUP ds = DomainService(login='******') ds.build_cookie = Mock(return_value={"cookie": "value"}) ds.update_cookie = Mock() i = mock_client.return_value getinfo_result = Mock() getinfo_result.dnsEntries = [ DnsEntry('testentry', 86400, DnsEntry.TYPE_A, '127.0.0.1') ] i.service.getInfo.return_value = getinfo_result # CALL result = ds.get_info('example.com') # VERIFY ds.build_cookie.assert_called_with(mode=MODE_RO, method='getInfo', parameters=['example.com']) ds.update_cookie.assert_called_with({"cookie": "value"}) i.service.getInfo.assert_called_with('example.com') self.assertEqual(result, getinfo_result)