chore: import upstream snapshot with attribution

python/ray/dashboard/modules/tests/test_agent.py

@@ -0,0 +1,49 @@
+import logging
+
+import aiohttp.web
+
+import ray.dashboard.modules.tests.test_consts as test_consts
+import ray.dashboard.modules.tests.test_utils as test_utils
+import ray.dashboard.optional_utils as dashboard_optional_utils
+import ray.dashboard.utils as dashboard_utils
+from ray._private.ray_constants import env_bool
+
+logger = logging.getLogger(__name__)
+routes = dashboard_optional_utils.DashboardAgentRouteTable
+
+
+@dashboard_utils.dashboard_module(
+    enable=env_bool(test_consts.TEST_MODULE_ENVIRONMENT_KEY, False)
+)
+class TestAgent(dashboard_utils.DashboardAgentModule):
+    def __init__(self, dashboard_agent):
+        super().__init__(dashboard_agent)
+
+    @staticmethod
+    def is_minimal_module():
+        return False
+
+    @routes.get("/test/http_get_from_agent")
+    async def get_url(self, req) -> aiohttp.web.Response:
+        url = req.query.get("url")
+        result = await test_utils.http_get(self._dashboard_agent.http_session, url)
+        return aiohttp.web.json_response(result)
+
+    @routes.head("/test/route_head")
+    async def route_head(self, req) -> aiohttp.web.Response:
+        pass
+
+    @routes.post("/test/route_post")
+    async def route_post(self, req) -> aiohttp.web.Response:
+        pass
+
+    @routes.patch("/test/route_patch")
+    async def route_patch(self, req) -> aiohttp.web.Response:
+        pass
+
+    async def run(self, server):
+        pass
+
+
+if __name__ == "__main__":
+    pass