def test_build_versioned_handlers_with_health(self):
endpoint = MagicMock()
endpoint.name = "foobar"
endpoint.handlers = [
("/test0", "TEST0"),
("/test1", "TEST1"),
]
endpoint.health_checks = True # must evaluate to True
active_version = '2'
deprecated_versions = ['1']
deprecated_handler = "DEPRECATED"
expected = [
("/v2/foobar/_health", HealthHandler, {"endpoint": endpoint}),
("/v2/foobar/test0", "TEST0", {"endpoint": endpoint}),
("/v2/foobar/test1", "TEST1", {"endpoint": endpoint}),
("/v1/foobar/test0", "DEPRECATED"),
("/v1/foobar/test1", "DEPRECATED"),
]
versioned = build_versioned_handlers(
endpoint,
active_version,
deprecated_versions,
deprecated_handler)
self.assertEqual(versioned, expected)
def test_build_versioned_handlers_long_handler(self):
endpoint = MagicMock()
endpoint.name = "foobar"
endpoint.handlers = [
("/test0", "TEST0", {"ADDITIONAL": True}),
("/test1", "TEST1"),
]
endpoint.health_checks = []
active_version = '2'
deprecated_versions = ['1']
deprecated_handler = "DEPRECATED"
expected = [
("/v2/foobar/test0", "TEST0", {"ADDITIONAL": True,
"endpoint": endpoint}),
("/v2/foobar/test1", "TEST1", {"endpoint": endpoint}),
("/v1/foobar/test0", "DEPRECATED"),
("/v1/foobar/test1", "DEPRECATED"),
]
versioned = build_versioned_handlers(
endpoint,
active_version,
deprecated_versions,
deprecated_handler)
self.assertEqual(versioned, expected)
