test_session_vary_cookie() — flask Function Reference

Architecture documentation for the test_session_vary_cookie() function in test_basic.py from the flask codebase.

Function python ApplicationCore ExtensionRegistry

Entity Profile

ApplicationCore→ ExtensionRegistry→ test_session_vary_cookie() — flask Function Reference

Dependency Diagram

graph TD
  0dc72432_1c1d_850e_3e42_a4c4a8eba22e["test_session_vary_cookie()"]
  85bc4fb5_d1d7_a135_020d_69e052c12c0f["test_basic.py"]
  0dc72432_1c1d_850e_3e42_a4c4a8eba22e -->|defined in| 85bc4fb5_d1d7_a135_020d_69e052c12c0f
  style 0dc72432_1c1d_850e_3e42_a4c4a8eba22e fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

tests/test_basic.py lines 520–578

def test_session_vary_cookie(app, client):
    @app.route("/set")
    def set_session():
        flask.session["test"] = "test"
        return ""

    @app.route("/get")
    def get():
        return flask.session.get("test")

    @app.route("/getitem")
    def getitem():
        return flask.session["test"]

    @app.route("/setdefault")
    def setdefault():
        return flask.session.setdefault("test", "default")

    @app.route("/clear")
    def clear():
        flask.session.clear()
        return ""

    @app.route("/vary-cookie-header-set")
    def vary_cookie_header_set():
        response = flask.Response()
        response.vary.add("Cookie")
        flask.session["test"] = "test"
        return response

    @app.route("/vary-header-set")
    def vary_header_set():
        response = flask.Response()
        response.vary.update(("Accept-Encoding", "Accept-Language"))
        flask.session["test"] = "test"
        return response

    @app.route("/no-vary-header")
    def no_vary_header():
        return ""

    def expect(path, header_value="Cookie"):
        rv = client.get(path)

        if header_value:
            # The 'Vary' key should exist in the headers only once.
            assert len(rv.headers.get_all("Vary")) == 1
            assert rv.headers["Vary"] == header_value
        else:
            assert "Vary" not in rv.headers

    expect("/set")
    expect("/get")
    expect("/getitem")
    expect("/setdefault")
    expect("/clear")
    expect("/vary-cookie-header-set")
    expect("/vary-header-set", "Accept-Encoding, Accept-Language, Cookie")
    expect("/no-vary-header", None)