2013-04-17 9 views
9

Итак, я только начал использовать mock с проектом Django. Я пытаюсь высмеять часть представления, которое делает запрос к удаленному API для подтверждения запроса на подписку подлинным (форма проверки в соответствии с спецификацией, над которой я работаю).Python mock, django и запросы

То, что я похож на:

class SubscriptionView(View): 
    def post(self, request, **kwargs): 
     remote_url = request.POST.get('remote_url') 
     if remote_url: 
      response = requests.get(remote_url, params={'verify': 'hello'}) 

     if response.status_code != 200: 
      return HttpResponse('Verification of request failed') 

Что теперь я хочу сделать, это использовать издеваться издеваться вне requests.get вызов, чтобы изменить ответ, но я не могу работать, как сделать это для патч-декоратор. Я думал, вы что-то вроде:

@patch(requests.get) 
def test_response_verify(self): 
    # make a call to the view using self.app.post (WebTest), 
    # requests.get makes a suitable fake response from the mock object 

Как это достичь?

+0

Замедление при использовании mocks? Существует также django.test.client.RequestFactory - https://docs.djangoproject.com/en/1.5/topics/testing/advanced/#module-django.test.client – David

+3

Просто для будущих зрителей, вопросник хотел насмехаться внешний вызов API. Не вызов самого представления. В этой ситуации маки кажутся очень разумными. – aychedee

+0

В соответствии с @aychedee это действительно то, к чему я стремился с этим вопросом. – jvc26

ответ

11

Вы почти там. Вы просто неправильно назовете это.

from mock import call, patch 


@patch('my_app.views.requests') 
def test_response_verify(self, mock_requests): 
    # We setup the mock, this may look like magic but it works, return_value is 
    # a special attribute on a mock, it is what is returned when it is called 
    # So this is saying we want the return value of requests.get to have an 
    # status code attribute of 200 
    mock_requests.get.return_value.status_code = 200 

    # Here we make the call to the view 
    response = SubscriptionView().post(request, {'remote_url': 'some_url'}) 

    self.assertEqual(
     mock_requests.get.call_args_list, 
     [call('some_url', params={'verify': 'hello'})] 
    ) 

Вы также можете проверить, что ответ правильный, и имеет правильное содержание.

3

Это все в the documentation:

патч (цель, новое = DEFAULT, спецификация = None, не создают = False, spec_set = None, Autospec = None, new_callable = None, ** kwargs)

Цель должна быть строкой в ​​форме 'package.module.ClassName'.

from mock import patch 

# or @patch('requests.get') 
@patch.object(requests, 'get') 
def test_response_verify(self): 
    # make a call to the view using self.app.post (WebTest), 
    # requests.get makes a suitable fake response from the mock object 
Смежные вопросы