fresco 3.3.1__py3-none-any.whl → 3.3.3__py3-none-any.whl

fresco/tests/test_response.py

@@ -0,0 +1,224 @@
+# encoding=UTF-8
+# Copyright 2015 Oliver Cope
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+#     Unless required by applicable law or agreed to in writing, software
+#     distributed under the License is distributed on an "AS IS" BASIS,
+#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#     See the License for the specific language governing permissions and
+#     limitations under the License.
+#
+from unittest.mock import Mock
+import pytest
+
+from fresco import FrescoApp
+from fresco.cookie import Cookie
+from fresco.response import Response
+from fresco.routing import GET
+from fresco.routing import RouteNotFound
+
+
+class TestCannedResponses(object):
+    def test_redirect(self):
+        with FrescoApp().requestcontext():
+            r = Response.redirect("/new_location")
+        assert b"http://localhost/new_location" in b"".join(r.content_iterator)
+        assert r.get_header("Location") == "http://localhost/new_location"
+        assert r.status == "302 Found"
+
+    def test_redirect_temporary(self):
+        with FrescoApp().requestcontext():
+            r = Response.redirect_temporary("/new_location")
+        assert b"http://localhost/new_location" in b"".join(r.content_iterator)
+        assert r.get_header("Location") == "http://localhost/new_location"
+        assert r.status == "302 Found"
+
+    def test_redirect_permanent(self):
+        with FrescoApp().requestcontext():
+            r = Response.redirect_permanent("/new_location")
+        assert b"http://localhost/new_location" in b"".join(r.content_iterator)
+        assert r.get_header("Location") == "http://localhost/new_location"
+        assert r.status == "301 Moved Permanently"
+
+    def test_redirect_to_view(self):
+        app = FrescoApp()
+        view = Mock(return_value=Response())
+        app.route("/arfle/<barfle:int>", GET, view)
+        with app.requestcontext():
+            r = Response.redirect(view, barfle=42)
+        assert r.get_header("Location") == "http://localhost/arfle/42"
+
+    def test_redirect_to_view_spec(self):
+        app = FrescoApp()
+        view = Mock(return_value=Response())
+        app.route("/arfle/<barfle:int>", GET, view, name="gloop")
+        with app.requestcontext():
+            r = Response.redirect("gloop", barfle=42)
+        assert r.get_header("Location") == "http://localhost/arfle/42"
+
+    def test_redirect_to_simple_string(self):
+        with FrescoApp().requestcontext():
+            r = Response.redirect("gloop")
+        assert r.get_header("Location") == "http://localhost/gloop"
+
+    def test_redirect_raises_RouteNotFound(self):
+        view = Mock()
+        with FrescoApp().requestcontext():
+            with pytest.raises(RouteNotFound):
+                Response.redirect("gloop", arfle="a")
+            with pytest.raises(RouteNotFound):
+                Response.redirect(view)
+
+    def test_redirect_to_unsafe_url(self):
+        with pytest.raises(ValueError):
+            with FrescoApp().requestcontext():
+                Response.redirect("http://bad.example.com/")
+
+    def test_redirect_uses_fallback_for_unsafe_url(self):
+        app = FrescoApp()
+        view = Mock(return_value=Response())
+        app.route("/arfle/<barfle:int>", GET, view, name="gloop")
+        with app.requestcontext():
+            r = Response.redirect("http://bad.example.com/", "gloop", barfle=23)
+        assert r.get_header("Location") == "http://localhost/arfle/23"
+
+    def test_not_found(self):
+        with FrescoApp().requestcontext():
+            r = Response.not_found()
+        assert b"not found" in b"".join(r.content_iterator).lower()
+        assert r.status_code == 404
+
+    def test_error(self):
+        with FrescoApp().requestcontext():
+            r = Response.error()
+        assert b"internal server error" in b"".join(r.content_iterator).lower()
+        assert r.status_code == 500
+
+    def test_method_not_allowed(self):
+        with FrescoApp().requestcontext():
+            r = Response.method_not_allowed(valid_methods=("PUT", "DELETE"))
+        assert b"not allowed" in b"".join(r.content_iterator).lower()
+        assert r.status == "405 Method Not Allowed"
+        assert r.get_header("Allow") == "PUT,DELETE"
+
+    def test_unauthorized(self):
+        with FrescoApp().requestcontext():
+            r = Response.unauthorized(authenticate='myscheme realm="spaghetti"')
+        assert r.status == "401 Unauthorized"
+        assert r.get_header("WWW-Authenticate") == 'myscheme realm="spaghetti"'
+
+    def test_unauthorized_basic(self):
+        with FrescoApp().requestcontext():
+            r = Response.unauthorized_basic(realm="spaghetti")
+        assert r.status == "401 Unauthorized"
+        assert r.get_header("WWW-Authenticate") == 'Basic realm="spaghetti"'
+
+    def test_meta_refresh(self):
+        with FrescoApp().requestcontext():
+            r = Response.meta_refresh("/next_page")
+        content = b"".join(r.content_iterator)
+        assert b'<a href="http://localhost/next_page">' in content
+        assert (
+            b'<meta http-equiv="refresh" content="0;'
+            b' url=http://localhost/next_page">' in content
+        )
+        assert r.status == "200 OK"
+        assert r.get_header("Content-Type") == "text/html"
+
+    def test_json(self):
+        with FrescoApp().requestcontext():
+            r = Response.json(
+                {"arfle": "barfle"},
+                status=201,
+                indent=4,
+                separators=(",", ":\t"),
+            )
+        assert r.status == "201 Created"
+        assert r.get_header("Content-Type") == "application/json"
+        assert b"".join(r.content_iterator) == b'{\n    "arfle":\t"barfle"\n}'
+
+
+class TestResponse(object):
+    def test_unknown_response_header_from_python_symbol(self):
+        r = Response(x_myheader="Foo", content_type=None)
+        assert r.headers == [("X-Myheader", "Foo")]
+
+    def test_standard_headers_capitalized_from_symbol(self):
+        r = Response(etag="foo")
+        assert r.headers == [("ETag", "foo")]
+
+        r = Response(x_ua_compatible="foo")
+        assert r.headers == [("X-UA-Compatible", "foo")]
+
+    def test_content_type_added_only_if_content(self):
+        r = Response(content=[])
+        assert "Content-Type" in dict(r.headers)
+
+        r = Response()
+        assert "Content-Type" not in dict(r.headers)
+
+        r = Response(content=None)
+        assert "Content-Type" not in dict(r.headers)
+
+    def test_add_response_header(self):
+        r = Response(x_myheader="Foo")
+        r = r.add_header("X-Myheader", "Bar")
+        assert r.headers == [("X-Myheader", "Foo"), ("X-Myheader", "Bar")]
+
+    def test_set_content_type_header(self):
+        r = Response(content_type="text/rtf; charset=ISO-8859-1")
+        assert r.get_header("Content-Type") == "text/rtf; charset=ISO-8859-1"
+
+    def test_set_cookie(self):
+        r = Response(set_cookie=Cookie(name="fruit", value="banana"))
+        assert r.get_header("Set-Cookie") == "fruit=banana;Path=/;SameSite=Lax"
+
+    def test_delete_cookie(self):
+        r = Response().delete_cookie(name="fruit")
+        assert r.get_header("Set-Cookie") == (
+            "fruit=;Expires=Thu, 01 Jan 1970 00:00:00 GMT;Max-Age=0;Path=/;SameSite=Lax"
+        )
+
+    def test_response_get_headers(self):
+        r = Response(
+            ["whoa nelly!"],
+            content_type="text/plain",
+            x_test_header=["1", "2"],
+        )
+        assert r.get_header("content-type") == "text/plain"
+        assert r.get_header("x-test-header") == "1,2"
+        assert r.get_header("X-Test-Header") == "1,2"
+        assert r.get_header("x-does-not-exist", "boo!") == "boo!"
+
+    def test_response_buffered_sets_content_length(self):
+        r = Response(["whoa nelly!"]).buffered()
+        assert r.get_header("content-length") == "11"
+
+        r = Response(["whoa nélly!"]).buffered()
+        assert r.get_header("content-length") == "12"
+
+    def test_it_encodes_a_unicode_string_according_to_content_type(self):
+        r = Response("café", content_type="text/plain; charset=utf-8")
+        assert list(r({}, Mock())) == ["café".encode("utf-8")]
+        r = Response("café", content_type="text/plain; charset=latin-1")
+        assert list(r({}, Mock())) == ["café".encode("latin-1")]
+
+
+class TestAddVary(object):
+    def test_it_adds_a_vary_header(self):
+        r = Response(content_type="text/plain").add_vary("Accept-Encoding")
+        assert ("Vary", "Accept-Encoding") in r.headers
+
+    def test_it_extends_an_existing_header(self):
+        r = (
+            Response(content_type="text/plain")
+            .add_vary("Accept-Encoding")
+            .add_vary("Cookie")
+        )
+        assert "Cookie" in r.get_header("vary")
+        assert "Accept-Encoding" in r.get_header("vary")

fresco/tests/test_routeargs.py

@@ -0,0 +1,223 @@
+# Copyright 2015 Oliver Cope
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+#     Unless required by applicable law or agreed to in writing, software
+#     distributed under the License is distributed on an "AS IS" BASIS,
+#     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#     See the License for the specific language governing permissions and
+#     limitations under the License.
+#
+from unittest.mock import Mock, call
+
+from fresco import routeargs
+from fresco.core import FrescoApp
+from fresco.response import Response
+from fresco.routing import GET
+from fresco.routing import POST
+from fresco.routing import Route
+
+
+class TestRouteArg(object):
+    def test_routekwarg_configured(self):
+        A = routeargs.RouteArg()
+        route = Route("/", GET, lambda r: None, x=A)
+        assert A.route is route
+        assert A.name == "x"
+
+    def test_routekwarg_value_passed(self):
+        view = Mock(return_value=Response())
+        routekwarg = Mock(spec=routeargs.RouteArg)
+        routekwarg.return_value = "xyzzy"
+
+        app = FrescoApp()
+        app.route("/", GET, view, x=routekwarg)
+        with app.requestcontext("/") as c:
+            app.view()
+            assert routekwarg.call_args_list == [call(c.request)]
+            assert view.call_args_list == [call(x="xyzzy")]
+
+    def test_queryarg_value_passed(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.QueryArg())
+        with app.requestcontext("/?x=foo"):
+            app.view()
+            assert view.call_args_list == [call(x="foo")]
+
+    def test_formarg_value_passed(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormArg())
+        with app.requestcontext("/?x=foo"):
+            app.view()
+            assert view.call_args_list == [call(x="foo")]
+
+    def test_sessionarg_value_passed(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.SessionArg())
+        with app.requestcontext("/?x=foo") as c:
+            c.request.environ[c.request.SESSION_ENV_KEY] = {"x": "foo"}
+            app.view()
+            assert view.call_args_list == [call(x="foo")]
+
+    def test_cookiearg_value_passed(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.CookieArg())
+        with app.requestcontext("/", HTTP_COOKIE="x=foo"):
+            app.view()
+            assert view.call_args_list == [call(x="foo")]
+
+    def test_cookiearg_listvalue_passed(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.CookieArg([str]))
+        with app.requestcontext("/", HTTP_COOKIE="x=foo;x=bar"):
+            app.view()
+            assert view.call_args_list == [call(x=["foo", "bar"])]
+
+    def test_requestarg_value_converted(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormArg(float))
+        with app.requestcontext("/?x=0"):
+            app.view()
+            assert view.call_args_list == [call(x=0.0)]
+
+    def test_it_converts_lists(self):
+        arg = routeargs.RequestArg(converter=[str])
+        arg.configure(None, "foo")
+        with FrescoApp().requestcontext("/?foo=a&foo=b") as r:
+            assert arg(r.request) == ["a", "b"]
+
+    def test_requestarg_default_value(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormArg(default="d"))
+        with app.requestcontext("/"):
+            app.view()
+            assert view.call_args_list == [call(x="d")]
+
+    def test_it_doesnt_convert_default_values(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormArg(int, default=None))
+        with app.requestcontext("/"):
+            app.view()
+            assert view.call_args_list == [call(x=None)]
+
+    def test_access_to_missing_requestarg_returns_badrequest(self):
+        def view(x):
+            x == 0
+
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormArg())
+        with app.requestcontext("/"):
+            assert "Bad Request" in app.view().status
+
+    def test_missing_requestarg_with_conversion_returns_badrequest(self):
+        def view(x):
+            x == 0
+
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormArg(int))
+        with app.requestcontext("/"):
+            assert "Bad Request" in app.view().status
+
+    def test_routeargs_work_as_positional_arguments(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, args=[routeargs.FormArg(key="x")])
+        with app.requestcontext("/?x=foo"):
+            app.view()
+            assert view.call_args_list == [call("foo")]
+
+    def test_routearg_classes_are_auto_instantiated(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormArg)
+        with app.requestcontext("/?x=foo"):
+            app.view()
+            assert view.call_args_list == [call(x="foo")]
+
+
+class TestRequestObject(object):
+    def test_it_passes_the_request(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.RequestObject())
+        with app.requestcontext("/") as c:
+            app.view()
+            x = view.call_args[1]["x"]
+            assert x is c.request
+
+
+class Test_routearg(object):
+    def test_it_calls_func(object):
+        view = Mock(return_value=Response())
+        argfunc = Mock()
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.routearg(argfunc))
+        with app.requestcontext("/"):
+            app.view()
+            assert argfunc.call_count == 1, argfunc.call_count
+            assert view.call_args[1]["x"] is argfunc()
+
+    def test_it_passes_additional_args(object):
+        view = Mock(return_value=Response())
+        argfunc = Mock()
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.routearg(argfunc, "x", y=42))
+        with app.requestcontext("/") as c:
+            app.view()
+            assert argfunc.call_args == ((c.request, "x"), {"y": 42})
+            assert view.call_args[1]["x"] is argfunc()
+
+
+class TestFormData(object):
+    def test_it_passes_the_form_dict(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", GET, view, x=routeargs.FormData())
+        with app.requestcontext("/") as c:
+            app.view()
+            x = view.call_args[1]["x"]
+            assert x is c.request.form
+
+
+class TestJSONPayload(object):
+    def test_it_passes_json_data(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", POST, view, x=routeargs.JSONPayload())
+        with app.requestcontext_post(
+            "/", data=b'{"foo":"bar"}', CONTENT_TYPE="application/json"
+        ):
+            app.view()
+            x = view.call_args[1]["x"]
+            assert x == {"foo": "bar"}
+
+    def test_it_returns_bad_request_on_invalid_payload(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", POST, view, x=routeargs.JSONPayload())
+        with app.requestcontext_post(
+            "/", data=b"invalid json!", CONTENT_TYPE="application/json"
+        ):
+            r = app.view()
+            assert r.status_code == 400
+
+    def test_it_returns_a_default_value(self):
+        view = Mock(return_value=Response())
+        app = FrescoApp()
+        app.route("/", POST, view, x=routeargs.JSONPayload(default="default value"))
+        with app.requestcontext_post("/"):
+            app.view()
+        x = view.call_args[1]["x"]
+        assert x == "default value"