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
