import pytest from sanic.blueprints import Blueprint from sanic.constants import HTTP_METHODS from sanic.exceptions import InvalidUsage from sanic.request import Request from sanic.response import HTTPResponse, text from sanic.views import HTTPMethodView @pytest.mark.parametrize("method", HTTP_METHODS) def test_methods(app, method): class DummyView(HTTPMethodView): async def get(self, request): return text("", headers={"method": "GET"}) def post(self, request): return text("", headers={"method": "POST"}) async def put(self, request): return text("", headers={"method": "PUT"}) def head(self, request): return text("", headers={"method": "HEAD"}) def options(self, request): return text("", headers={"method": "OPTIONS"}) async def patch(self, request): return text("", headers={"method": "PATCH"}) def delete(self, request): return text("", headers={"method": "DELETE"}) app.add_route(DummyView.as_view(), "/") request, response = getattr(app.test_client, method.lower())("/") assert response.headers["method"] == method def test_unexisting_methods(app): class DummyView(HTTPMethodView): def get(self, request): return text("I am get method") app.add_route(DummyView.as_view(), "/") request, response = app.test_client.get("/") assert response.body == b"I am get method" request, response = app.test_client.post("/") assert b"Method POST not allowed for URL /" in response.body def test_argument_methods(app): class DummyView(HTTPMethodView): def get(self, request, my_param_here): return text("I am get method with %s" % my_param_here) app.add_route(DummyView.as_view(), "/") request, response = app.test_client.get("/test123") assert response.text == "I am get method with test123" def test_with_bp(app): bp = Blueprint("test_text") class DummyView(HTTPMethodView): def get(self, request): return text("I am get method") bp.add_route(DummyView.as_view(), "/") app.blueprint(bp) request, response = app.test_client.get("/") assert response.text == "I am get method" def test_with_attach(app): class DummyView(HTTPMethodView): def get(self, request): return text("I am get method") DummyView.attach(app, "/") request, response = app.test_client.get("/") assert response.text == "I am get method" def test_with_sub_init(app): class DummyView(HTTPMethodView, attach=app, uri="/"): def get(self, request): return text("I am get method") request, response = app.test_client.get("/") assert response.text == "I am get method" def test_with_attach_and_bp(app): bp = Blueprint("test_text") class DummyView(HTTPMethodView): def get(self, request): return text("I am get method") DummyView.attach(bp, "/") app.blueprint(bp) request, response = app.test_client.get("/") assert response.text == "I am get method" def test_with_sub_init_and_bp(app): bp = Blueprint("test_text") class DummyView(HTTPMethodView, attach=bp, uri="/"): def get(self, request): return text("I am get method") app.blueprint(bp) request, response = app.test_client.get("/") assert response.text == "I am get method" def test_with_bp_with_url_prefix(app): bp = Blueprint("test_text", url_prefix="/test1") class DummyView(HTTPMethodView): def get(self, request): return text("I am get method") bp.add_route(DummyView.as_view(), "/") app.blueprint(bp) request, response = app.test_client.get("/test1/") assert response.text == "I am get method" def test_with_middleware(app): class DummyView(HTTPMethodView): def get(self, request): return text("I am get method") app.add_route(DummyView.as_view(), "/") results = [] @app.middleware async def handler(request): results.append(request) request, response = app.test_client.get("/") assert response.text == "I am get method" assert type(results[0]) is Request def test_with_middleware_response(app): results = [] @app.middleware("request") async def process_request(request): results.append(request) @app.middleware("response") async def process_response(request, response): results.append(request) results.append(response) class DummyView(HTTPMethodView): def get(self, request): return text("I am get method") app.add_route(DummyView.as_view(), "/") request, response = app.test_client.get("/") assert response.text == "I am get method" assert type(results[0]) is Request assert type(results[1]) is Request assert isinstance(results[2], HTTPResponse) def test_with_custom_class_methods(app): class DummyView(HTTPMethodView): global_var = 0 def _iternal_method(self): self.global_var += 10 def get(self, request): self._iternal_method() return text( f"I am get method and global var " f"is {self.global_var}" ) app.add_route(DummyView.as_view(), "/") request, response = app.test_client.get("/") assert response.text == "I am get method and global var is 10" def test_with_decorator(app): results = [] def stupid_decorator(view): def decorator(*args, **kwargs): results.append(1) return view(*args, **kwargs) return decorator class DummyView(HTTPMethodView): decorators = [stupid_decorator] def get(self, request): return text("I am get method") app.add_route(DummyView.as_view(), "/") request, response = app.test_client.get("/") assert response.text == "I am get method" assert results[0] == 1