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"}
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)]