Python: Aiohttp add response tests

Author: RasmusWL

python/ql/test/library-tests/frameworks/aiohttp/response_test.py

@@ -0,0 +1,72 @@
+from aiohttp import web
+
+
+routes = web.RouteTableDef()
+
+
+@routes.get("/raw_text") # $ routeSetup="/raw_text"
+async def raw_text(request): # $ requestHandler
+    return web.Response(text="foo") # $ MISSING: HttpResponse
+
+
+@routes.get("/raw_body") # $ routeSetup="/raw_body"
+async def raw_body(request): # $ requestHandler
+    return web.Response(body=b"foo") # $ MISSING: HttpResponse
+
+
+@routes.get("/html_text") # $ routeSetup="/html_text"
+async def html_text(request): # $ requestHandler
+    return web.Response(text="foo", content_type="text/html") # $ MISSING: HttpResponse
+
+
+@routes.get("/html_body") # $ routeSetup="/html_body"
+async def html_body(request): # $ requestHandler
+    return web.Response(body=b"foo", content_type="text/html") # $ MISSING: HttpResponse
+
+
+@routes.get("/html_body_set_later") # $ routeSetup="/html_body_set_later"
+async def html_body_set_later(request): # $ requestHandler
+    resp = web.Response(body=b"foo") # $ MISSING: HttpResponse
+    resp.content_type = "text/html"
+    return resp
+
+# Each HTTP status code has an exception
+# see https://docs.aiohttp.org/en/stable/web_quickstart.html#exceptions
+
+@routes.get("/through_200_exception") # $ routeSetup="/through_200_exception"
+async def through_200_exception(request): # $ requestHandler
+    raise web.HTTPOk(text="foo") # $ MISSING: HttpResponse
+
+
+@routes.get("/through_200_exception_html") # $ routeSetup="/through_200_exception_html"
+async def through_200_exception(request): # $ requestHandler
+    exception = web.HTTPOk(text="foo") # $ MISSING: HttpResponse
+    exception.content_type = "text/html"
+    raise exception
+
+
+@routes.get("/through_404_exception") # $ routeSetup="/through_404_exception"
+async def through_404_exception(request): # $ requestHandler
+    raise web.HTTPNotFound(text="foo") # $ MISSING: HttpResponse
+
+
+@routes.get("/redirect_301") # $ routeSetup="/redirect_301"
+async def redirect_301(request): # $ requestHandler
+    if not "kwarg" in request.url.query:
+        raise web.HTTPMovedPermanently("/login") # $ MISSING: HttpResponse HttpRedirectResponse
+    else:
+        raise web.HTTPMovedPermanently(location="/logout") # $ MISSING: HttpResponse HttpRedirectResponse
+
+
+@routes.get("/redirect_302") # $ routeSetup="/redirect_302"
+async def redirect_302(request): # $ requestHandler
+    if not "kwarg" in request.url.query:
+        raise web.HTTPFound("/login") # $ MISSING: HttpResponse HttpRedirectResponse
+    else:
+        raise web.HTTPFound(location="/logout") # $ MISSING: HttpResponse HttpRedirectResponse
+
+
+if __name__ == "__main__":
+    app = web.Application()
+    app.add_routes(routes)
+    web.run_app(app)