test_starwars_gateway_swapi.py

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

# [...] StarWarsGatewaySWAPITests

    def test_lanca_excecao_quando_personagem_nao_existe(self):
        retorno_json = {
            'count': 0,
            'results': []
        }
        mock_response = mock.Mock(status_code=200)
        mock_response.json.return_value = retorno_json
        self.client.get.return_value = mock_response

        with self.assertRaises(PersonagemNaoEncontradoException) as context:
            self.dados_star_wars_gateway.buscar_personagem(nome='Spock')

        self.assertEqual('Personagem Spock não encontrado', context.exception.message)