Search code examples
pythondjangodjango-unittest

How to test a view using the request object


I need to make tests to see if my view is working properly. However, I did not find how to use the request object in the test file.

My view function is:

def track_view(request):
    user_email, user_order = request.session['email'], request.session['order']
    if request.method == 'POST':
        client_id = request.META.get('HTTP_HOST').split(':8000')[0]
        payload = "{ \n\"description\": \"Rastreio de pedido\", \n\"subject\": \"Rastreamento " + user_order + ' ' + user_email + "\", \n\"email\": \"" + user_email + "\", \n\"priority\": " + str(
            priority['low']) + ", \n\"status\": " + str(status['closed']) + ", \n\"group_id\": " + str(group[
                                                                                                          'Logística']) + ", \n\"type\": \"Rastrear Pedido\", \n\"product_id\": " + str(client[client_id]) + "}"
        headers = {
            'Content-Type': "application/json",
            'Cache-Control': "no-cache"
        }

        response = requests.request("POST", url, data=payload, headers=headers, auth=('XXX', 'X'))

        print(response.text)
        context = {'email_variable': user_email,
                   'order_variable': user_order}
        return redirect('atendimento:final_view')
    logo_image = 'client_logo/' + request.META.get('HTTP_HOST').split(':8000')[0] + '.png'
    context = {'logo_image': logo_image,
               'user_email': request.POST.get('email'),
               'user_order': request.session['order']}
    return render(request, 'atendimento/track_page.html', context)

How can I make that work to test different inputs to that view?


Solution

  • You should use the test client to test requests.

    def test_track_view(self):
        session = self.client.session
        session['email'] = 'foo@bar.com'
        session['user_order'] = 'whatever'
        url = reverse('my-url')
        response = self.client.get(url)
        self.assertEqual(response.status_code, 200)
    

    Note, you'll probably also want to mock out the requests object so that your tests don't actually make a request to the external API.

    Also note, it's really not good practice to build up JSON strings manually. Instead, create a dict, and dump it to JSON:

    payload = { "description": "Rastreio de pedido", ... "priority": priority['low'], "status": status['closed'], "group_id": group['Logística']...}
    response = requests.request("POST", url, data=json.dumps(payload), headers=headers, auth=('XXX', 'X'))