async def test_http_cache_middleware_2(): class RequestMock: url = None m = HTTPCacheMiddleware(maxsize=1_000_000, ttl=2) request = RequestMock() request.url = "http:127.0.0.1:9090/v1/test1" response = HTTPResponse() response.body = {"data": "datatatata"} response.headers = {"custom_header_name": "custom_header_value"} assert await m.middleware_request(request) is None await m.middleware_response(request, response) assert response.headers.get("x-cache") == "MISS" response_check = await m.middleware_request(request) assert response assert response_check.body == response.body assert response_check.headers["x-cache"] == "HIT" del response.headers["x-cache"] del response_check.headers["x-cache"] assert response_check.headers == response.headers assert response_check.status == response.status await asyncio.sleep(2) assert await m.middleware_request(request) is None
def test_multiple_set_cookie_headers(request): resp = HTTPResponse(body="Foo bar baz") resp.headers = CIDict() resp.headers['set-cookie'] = 'foo' resp.headers['set-cookie'] = 'bar' return resp