class TestTitulosApi(TestCase):
def setUp(self):
self.api = TitulosApi(URL_TITULOS_EXTRANJEROS)
@vcr.use_cassette('test/cassettes/extranjeros/titulos_encontrados.yaml')
def test_titulos_encontrados(self):
titulos = self.api.obtener_titulos('1111111116')
assert_equal(2, len(titulos))
assert_equal(4, titulos[0]['numeroRegistro'])
assert_equal(5, titulos[1]['numeroRegistro'])
@vcr.use_cassette('test/cassettes/extranjeros/titulos_no_encontrados.yaml')
def test_titulos_no_encontrados(self):
titulos = self.api.obtener_titulos('1111111112')
assert_equal([], titulos)
@vcr.use_cassette('test/cassettes/extranjeros/titulos_error_interno.yaml')
def test_titulos_error_interno(self):
with self.assertRaises(ConexionError):
titulos = self.api.obtener_titulos('1111111116')
def test_titulos_caido(self):
self.api = TitulosApi('http://127.0.0.1:9999/url-invalida')
with self.assertRaises(ConexionError):
self.api.obtener_titulos('1111111116')
def obtener_titulos_por_portador(busqueda):
titulos_nacionales_api = TitulosApi(URL_TITULOS_NACIONALES)
titulos_extranjeros_api = TitulosApi(URL_TITULOS_EXTRANJEROS)
try:
titulos = titulos_nacionales_api.obtener_titulos(busqueda)
titulos.extend(titulos_extranjeros_api.obtener_titulos(busqueda))
except ConexionError:
return '', codes.BAD_GATEWAY
return jsonify(elementos=titulos), codes.OK