Пример #1
0
def test_uses_session():
    view = uses_session(lambda app, request: Response())

    app = pretend.stub()
    request = pretend.stub(_session=pretend.stub())
    response = view(app, request)

    assert request.session is request._session
    assert response.vary.as_set() == {"cookie"}
Пример #2
0
def test_uses_session(monkeypatch):
    vary_wrapper = pretend.call_recorder(lambda fn: fn)
    add_vary = pretend.call_recorder(lambda *varies: vary_wrapper)
    monkeypatch.setattr(warehouse.sessions, "add_vary", add_vary)

    @pretend.call_recorder
    def view(context, request):
        pass

    context = pretend.stub()
    request = pretend.stub()

    wrapped = uses_session(view)
    wrapped(context, request)

    assert view.calls == [pretend.call(context, request)]
    assert request._allow_session
    assert add_vary.calls == [pretend.call("Cookie")]
    assert vary_wrapper.calls == [pretend.call(mock.ANY)]
Пример #3
0
def test_uses_session(monkeypatch):
    vary_wrapper = pretend.call_recorder(lambda fn: fn)
    add_vary = pretend.call_recorder(lambda *varies: vary_wrapper)
    monkeypatch.setattr(warehouse.sessions, "add_vary", add_vary)

    @pretend.call_recorder
    def view(context, request):
        pass

    context = pretend.stub()
    request = pretend.stub()

    wrapped = uses_session(view)
    wrapped(context, request)

    assert view.calls == [pretend.call(context, request)]
    assert request._allow_session
    assert add_vary.calls == [pretend.call("Cookie")]
    assert vary_wrapper.calls == [pretend.call(mock.ANY)]