def test_dispatch_with_suffix(self):
class TestResource:
def on_get_first(self, req):
return "GET/OK"
def on_post(self, req):
return "POST/NOK"
def on_post_second(self, req):
return "POST/OK"
router = ResourceRouter()
resource = TestResource()
router.add_route("/health/first", resource, suffix="first")
router.add_route("/health/second", resource, suffix="second")
assert router.dispatch(Request("GET", "/health/first", b"",
{})) == "GET/OK"
assert router.dispatch(Request("POST", "/health/first", b"",
{})) == ResourceRouter.NO_ROUTE
assert router.dispatch(Request("GET", "/health/second", b"",
{})) == ResourceRouter.NO_ROUTE
assert router.dispatch(Request("POST", "/health/second", b"",
{})) == "POST/OK"
def test_put_overwrite_and_get(self):
service_manager = ServiceManager()
service_manager.get_states = mock.MagicMock(
return_value={"foo": ServiceState.AVAILABLE})
resource = HealthResource(service_manager)
resource.on_put(
Request(
"PUT",
"/",
b'{"features:initScripts": "initializing","features:persistence": "disabled"}',
{},
))
resource.on_put(
Request("PUT", "/", b'{"features:initScripts": "initialized"}',
{}))
state = resource.on_get(Request("GET", "/", b"None", {}))
assert state == {
"features": {
"initScripts": "initialized",
"persistence": "disabled",
},
"services": {
"foo": "available",
},
"version": VERSION,
}
def test_matches_with_host(self):
rule = RoutingRule("http://localhost:4566/foo/bar", match_host=True)
assert rule.matches(
Request("GET", "/foo/bar", b"", {"Host": "localhost:4566"}))
assert not rule.matches(
Request("GET", "/foo/bar", b"", {"Host": "localhost"}))
assert not rule.matches(
Request("GET", "/foo/bar", b"", {"Host": "example.com"}))
assert not rule.matches(Request("GET", "/foo/bar", b"", {}))
def test_dispatch_to_non_existing_function_returns_no_route(self):
class TestResource:
def on_post(self, request):
return "POST/OK"
router = ResourceRouter()
router.add_route("/health", TestResource())
assert router.dispatch(Request("GET", "/health", b"",
{})) == ResourceRouter.NO_ROUTE
assert router.dispatch(Request("POST", "/health", b"",
{})) == "POST/OK"
def test_matches_strict(self):
rule = RoutingRule("/foo/bar")
assert rule.matches(Request("GET", "/foo/bar", b"", {}))
assert rule.matches(Request("GET", "/foo/bar/", b"", {}))
assert rule.matches(Request("POST", "/foo/bar", b"", {}))
assert not rule.matches(Request("GET", "/foo/ba", b"", {}))
assert not rule.matches(Request("GET", "/foo/bare", b"", {}))
assert not rule.matches(Request("GET", "/foo/bar/ed", b"", {}))
assert not rule.matches(Request("GET", "/foo", b"", {}))
def test_dispatch_to_correct_function(self):
requests = list()
class TestResource:
def on_get(self, req):
requests.append(req)
return "GET/OK"
def on_post(self, req):
requests.append(req)
return "POST/OK"
router = ResourceRouter()
router.add_route("/health", TestResource())
request = Request("GET", "/health", b"", {})
assert router.dispatch(request) == "GET/OK"
assert router.dispatch(request) == "GET/OK"
assert len(requests) == 2
assert requests[0] is request
assert requests[1] is request
def test_dispatch_returns_no_route(self):
router = ResourceRouter()
assert router.dispatch(Request("GET", "/health", b"",
{})) == ResourceRouter.NO_ROUTE
def test_matches_with_host_no_pattern_errors(self):
rule = RoutingRule("/foo/bar", match_host=True)
with pytest.raises(ValueError):
rule.matches(Request("GET", "/foo/bar", b"", {}))
def test_matches_with_query(self):
rule = RoutingRule("/foo/bar")
assert rule.matches(Request("GET", "/foo/bar?", b"", {}))
assert rule.matches(Request("GET", "/foo/bar?help", b"", {}))
assert rule.matches(Request("GET", "/foo/bar?help=me&ok", b"", {}))