def test_url_pattern_lookup(lookups, url, expected):
router = Router(assert_all_mocked=False)
route = router.get(**lookups) % 418
request = httpx.Request("GET", url)
response = router.handler(request)
assert bool(response.status_code == 418) is expected
assert route.called is expected
def test_side_effect_list():
router = Router()
route = router.get("https://foo.bar/").mock(
return_value=httpx.Response(409),
side_effect=[httpx.Response(404), httpcore.NetworkError, httpx.Response(201)],
)
request = httpx.Request("GET", "https://foo.bar")
response = router.handler(request)
assert response.status_code == 404
assert response.request == request
request = httpx.Request("GET", "https://foo.bar")
with pytest.raises(httpcore.NetworkError):
router.handler(request)
request = httpx.Request("GET", "https://foo.bar")
response = router.handler(request)
assert response.status_code == 201
assert response.request == request
with pytest.raises(StopIteration):
request = httpx.Request("GET", "https://foo.bar")
router.handler(request)
route.side_effect = None
request = httpx.Request("GET", "https://foo.bar")
response = router.handler(request)
assert response.status_code == 409
assert response.request == request
def test_side_effect_decorator():
router = Router()
@router.route(host="ham.spam", path__regex=r"/(?P<slug>\w+)/")
def foobar(request, slug):
return httpx.Response(200, json={"slug": slug})
@router.post("https://example.org/")
def example(request):
return httpx.Response(201, json={"message": "OK"})
request = httpx.Request("GET", "https://ham.spam/egg/")
response = router.handler(request)
assert response.status_code == 200
assert response.json() == {"slug": "egg"}
request = httpx.Request("POST", "https://example.org/")
response = router.handler(request)
assert response.status_code == 201
assert response.json() == {"message": "OK"}
def test_pass_through():
router = Router(assert_all_mocked=False)
route = router.get("https://foo.bar/", path="/baz/").pass_through()
request = httpx.Request("GET", "https://foo.bar/baz/")
matched_route, response = router.match(request)
assert matched_route is route
assert matched_route.is_pass_through
assert response is request
with pytest.raises(PassThrough):
router.handler(request)
route.pass_through(False)
matched_route, response = router.match(request)
assert matched_route is route
assert not matched_route.is_pass_through
assert response is not None
def test_side_effect_with_route_kwarg():
router = Router()
def foobar(request, route, slug):
response = httpx.Response(201, json={"id": route.call_count + 1, "slug": slug})
if route.call_count > 0:
route.mock(return_value=httpx.Response(501))
return response
router.post(path__regex=r"/(?P<slug>\w+)/").mock(side_effect=foobar)
request = httpx.Request("POST", "https://foo.bar/baz/")
response = router.handler(request)
assert response.status_code == 201
assert response.json() == {"id": 1, "slug": "baz"}
response = router.handler(request)
assert response.status_code == 201
assert response.json() == {"id": 2, "slug": "baz"}
response = router.handler(request)
assert response.status_code == 501
def test_side_effect_with_reserved_route_kwarg():
router = Router()
def foobar(request, route):
assert isinstance(route, Route)
return httpx.Response(202)
router.get(path__regex=r"/(?P<route>\w+)/").mock(side_effect=foobar)
with warnings.catch_warnings(record=True) as w:
request = httpx.Request("GET", "https://foo.bar/baz/")
response = router.handler(request)
assert response.status_code == 202
assert len(w) == 1
def test_side_effect_no_match():
router = Router()
def no_match(request):
request.respx_was_here = True
return None
router.get(url__startswith="https://foo.bar/").mock(side_effect=no_match)
router.get(url__eq="https://foo.bar/baz/").mock(return_value=httpx.Response(204))
request = httpx.Request("GET", "https://foo.bar/baz/")
response = router.handler(request)
assert response.status_code == 204
assert response.request.respx_was_here is True
def test_side_effect_exception():
router = Router()
router.get("https://foo.bar/").mock(side_effect=httpx.ConnectError)
router.get("https://ham.spam/").mock(side_effect=httpcore.NetworkError)
router.get("https://egg.plant/").mock(side_effect=httpcore.NetworkError())
request = httpx.Request("GET", "https://foo.bar")
with pytest.raises(httpx.ConnectError) as e:
router.handler(request)
assert e.value.request == request
request = httpx.Request("GET", "https://ham.spam")
with pytest.raises(httpcore.NetworkError):
router.handler(request)
request = httpx.Request("GET", "https://egg.plant")
with pytest.raises(httpcore.NetworkError):
router.handler(request)
def test_rollback():
router = Router()
route = router.get("https://foo.bar/") % 404
pattern = route.pattern
assert route.name is None
router.snapshot() # 1. get 404
route.return_value = httpx.Response(200)
router.post("https://foo.bar/").mock(
side_effect=[httpx.Response(400), httpx.Response(201)]
)
router.snapshot() # 2. get 200, post
_route = router.get("https://foo.bar/", name="foobar")
_route = router.get("https://foo.bar/baz/", name="foobar")
assert _route is route
assert route.name == "foobar"
assert route.pattern != pattern
route.return_value = httpx.Response(418)
request = httpx.Request("GET", "https://foo.bar/baz/")
response = router.handler(request)
assert response.status_code == 418
request = httpx.Request("POST", "https://foo.bar")
response = router.handler(request)
assert response.status_code == 400
assert len(router.routes) == 2
assert router.calls.call_count == 2
assert route.call_count == 1
assert route.return_value.status_code == 418
router.snapshot() # 3. directly rollback, should be identical
router.rollback()
assert len(router.routes) == 2
assert router.calls.call_count == 2
assert route.call_count == 1
assert route.return_value.status_code == 418
router.patch("https://foo.bar/")
assert len(router.routes) == 3
route.rollback() # get 200
assert router.calls.call_count == 2
assert route.call_count == 0
assert route.return_value.status_code == 200
request = httpx.Request("GET", "https://foo.bar")
response = router.handler(request)
assert response.status_code == 200
router.rollback() # 2. get 404, post
request = httpx.Request("POST", "https://foo.bar")
response = router.handler(request)
assert response.status_code == 400
assert len(router.routes) == 2
router.rollback() # 1. get 404
assert len(router.routes) == 1
assert router.calls.call_count == 0
assert route.return_value is None
router.rollback() # Empty inital state
assert len(router.routes) == 0
assert route.return_value is None
# Idempotent
route.rollback()
router.rollback()
assert len(router.routes) == 0
assert route.name is None
assert route.pattern == pattern
assert route.return_value is None
