def test_returns_default_route_if_no_path_matches(self): default = Route(path=None, endpoint="404 endpoint") routes = Routes() routes.setdefault(default) route = routes.get("/asdf") self.assertEqual(route, default)
def test_returns_route_that_matches_the_path(self): expected_foo_route = Route("/foo", "foo") expected_bar_route = Route("/bar", "bar") routes = Routes() routes.add(expected_foo_route) routes.add(expected_bar_route) foo_route = routes.get("/foo") bar_route = routes.get("/bar") self.assertEqual(foo_route, expected_foo_route) self.assertEqual(bar_route, expected_bar_route)
def test_raises_NotFound_if_no_matching_path_and_no_set_default(self): routes = Routes() with self.assertRaises(NotFound): routes.get("/asdf")