def test_parameter_limit(): httpretty.register_uri(httpretty.GET, BVG_URL, body=DEPARTURE_HTML) _, ok = request_departures('any id', limit=2) request = httpretty.last_request() assert hasattr(request, 'querystring') assert 'maxJourneys' in request.querystring assert '2' in request.querystring['maxJourneys']
def test_parameter_products_filter(): httpretty.register_uri(httpretty.GET, BVG_URL, body=DEPARTURE_HTML) products_filter = create_products_filter(select='US') _, ok = request_departures('any id', 2, products_filter) request = httpretty.last_request() assert hasattr(request, 'querystring') assert 'productsFilter' in request.querystring assert '11000000' in request.querystring['productsFilter']
def test_limit(): httpretty.register_uri(httpretty.GET, BVG_URL, body=DEPARTURE_HTML) departures, ok = request_departures('9120025', limit=2) assert ok is True assert len(departures) == 2
def test_request_departures_timeout(): httpretty.register_uri(httpretty.GET, BVG_URL, body=mock_timeout_callback) _, ok = request_departures('any id', limit=10) assert ok is False
def test_request_departures_server_error(): httpretty.register_uri(httpretty.GET, BVG_URL, status=500) _, ok = request_departures('any id', limit=10) assert ok is False