test_starwars_gateway_swapi.py

# coding=utf-8
import unittest
import requests
import mock
from star_wars.starwars_gateway_swapi import StarWarsGatewaySWAPI


class StarWarsGatewaySWAPITests(unittest.TestCase):
    def setUp(self):
        self.client = mock.Mock(spec=requests)  # 1
        self.dados_star_wars_gateway = StarWarsGatewaySWAPI(self.client)

    def test_obtem_resposta_do_servidor(self):
        self.client.get.return_value = mock.Mock(status_code=200)  # 2

        resposta = self.dados_star_wars_gateway.buscar_dados()

        self.assertEqual(200, resposta.status_code)

    def test_obtem_personagem(self):
        retorno_json = {  # 3
            'count': 1,
            'results': [
                {'name': 'Anakin Skywalker'}
            ]
        }
        mock_response = mock.Mock(status_code=200)  # 4
        mock_response.json.return_value = retorno_json
        self.client.get.return_value = mock_response

        resposta = self.dados_star_wars_gateway.buscar_personagem(nome='Anakin')

        conteudo = resposta.json()
        self.assertEqual('Anakin Skywalker', conteudo['results'][0]['name'])
        # 5
        self.client.get.assert_called_with('http://swapi.co/api/people/?search=Anakin')