def test_get_leads_by_filter_type(self): responses.add(responses.GET, self.AUTH_URL, body=json.dumps(self.AUTH_RESPONSE), status=200) response = { u'requestId': u'20a6#1511d9f4935', u'success': True, u'result': [{ u'firstName': u'Brian', u'lastName': u'Zindler', u'id': 5517997, u'updatedAt': u'2015-11-05T04:13:30Z', u'email': u'*****@*****.**', u'createdAt': u'2015-05-18T20:32:02Z' }] } responses.add(responses.GET, '{0}{1}'.format(self.URL_ENDPOINT, '/rest/v1/leads.json'), body=json.dumps(response), status=200) m = MarketoRest('asdfsaf', 'asdfdsafs') self.assertEqual( m.get_leads_by_filter_type('email', ['*****@*****.**']), [{ u'firstName': u'Brian', u'lastName': u'Zindler', u'id': 5517997, u'updatedAt': u'2015-11-05T04:13:30Z', u'email': u'*****@*****.**', u'createdAt': u'2015-05-18T20:32:02Z' }] )
def test_retry_access_token(self): self.NEW_TOKEN = 'adfdfdsadfss' def request_callback(request): if request_callback.times_called == 0: resp_body = self.AUTH_RESPONSE else: resp_body = {u'access_token': self.NEW_TOKEN} request_callback.times_called += 1 return (200, {}, json.dumps(resp_body)) request_callback.times_called = 0 responses.add_callback( responses.GET, self.AUTH_URL, callback=request_callback, content_type='application/json', ) def describe_callback(request): if describe_callback.times_called == 0: resp_body = { u'errors': [{u'message': u'Access token invalid', u'code': u'602'}], u'requestId': u'11082#1511d9aa5d6', u'success': False } else: resp_body = { u'result': [{'thing': 1}, {'thing': 2}, {'thing': 3}], u'requestId': u'11082#1511d9aa5d6', u'success': True } describe_callback.times_called += 1 return (200, {}, json.dumps(resp_body)) describe_callback.times_called = 0 responses.add_callback(responses.GET, self.DESCRIBE_URL, callback=describe_callback, content_type='application/json') m = MarketoRest('asdfsaf', 'asdfdsafs') m.describeLead() self.assertEqual(m.tokenParams['access_token'], self.NEW_TOKEN)