Como simular solicitações e respostas

Introdução

O teste de unidade em Python geralmente envolve simulação dependências externas para simular cenários do mundo real. O mock nos permite controlar o comportamento dessas dependências, proporcionando maior estabilidade e previsibilidade aos nossos testes. Este artigo demonstra como simular o módulo de solicitações do Python e suas respostas para facilitar testes de unidade eficazes.

Etapa 1: Módulo de simulação de solicitações

Para simular o módulo de solicitações, utilize o módulo simulado da biblioteca padrão:

import mock

Crie um objeto simulado para o módulo de solicitações na fase de configuração do teste método:

mockedRequests = mock.Mock()

Configure o objeto simulado para retornar respostas específicas para diferentes URLs quando o método get() for chamado:

mockedRequests.get.side_effect = [
    # First URL: Return 'a response'
    mock.Mock(status_code=200, text='a response'),
    # Second URL: Return 'b response'
    mock.Mock(status_code=200, text='b response'),
    # Third URL: Return 'c response'
    mock.Mock(status_code=200, text='c response')
]

Etapa 2: Chamando a View

Com o módulo de solicitações simulado, chame a função na view que faz as solicitações:

res1 = mockedRequests.get('aurl')
res2 = mockedRequests.get('burl')
res3 = mockedRequests.get('curl')

Etapa 3: Verificando as respostas

Como o módulo de solicitações é simulado, as respostas podem ser facilmente verificadas quanto aos valores esperados :

self.assertEqual(res1.text, 'a response')
self.assertEqual(res2.text, 'b response')
self.assertEqual(res3.text, 'c response')