Files
OpenMensa-Parser-STW-Potsdam/tests/test_retrieval.py
T
2019-12-28 17:36:44 +01:00

54 lines
1.5 KiB
Python

# -*- encoding: utf-8 -*-
import logging
import os
import pytest
from stw_potsdam import feed
from stw_potsdam.config import read_canteen_config
from stw_potsdam.canteen_api import download_menu
from stw_potsdam.canteen_api import MenuParams
# pragma pylint: disable=invalid-name,redefined-outer-name
ENV_ENABLED = 'ENABLE_API_QUERY'
# Because log messages are automatically part of the Pytest report, below
# explicitly avoids adding log handlers via logging#basicConfig, for example.
logging.getLogger().setLevel(logging.DEBUG)
CANTEENS = read_canteen_config()
@pytest.fixture(params=CANTEENS.values(), ids=lambda canteen: canteen.name)
def canteen(request):
return request.param
def is_enabled():
user_enabled = bool(os.getenv(ENV_ENABLED))
travis_enabled = os.getenv('TRAVIS_EVENT_TYPE') == 'cron'
return user_enabled or travis_enabled
requires_online_api = pytest.mark.skipif(
not is_enabled(),
reason='Querying the online API is disabled. '
'Turn on by setting env variable %s.' % ENV_ENABLED
)
@requires_online_api
def test_retrieval(canteen):
feed.render_meta(canteen, "/canteens/{}/menu".format(canteen.key))
params = MenuParams(canteen_id=canteen.id, chash=canteen.chash)
try:
menu = download_menu(params)
except ValueError as e:
if e.message == 'No JSON object could be decoded':
pytest.xfail('JSON endpoint returned garbage (issue #6)')
raise e
feed.render_menu(menu)