def test_set_cookies(): get = RequestFactory().get("/foo") get.waffles = {"foo": [True, False], "bar": [False, False]} resp = HttpResponse() assert not "dwf_foo" in resp.cookies assert not "dwf_bar" in resp.cookies resp = WaffleMiddleware().process_response(get, resp) assert "dwf_foo" in resp.cookies assert "dwf_bar" in resp.cookies eq_("True", resp.cookies["dwf_foo"].value) eq_("False", resp.cookies["dwf_bar"].value)
def test_rollout_cookies(): get = RequestFactory().get("/foo") get.waffles = {"foo": [True, True], "bar": [False, True], "baz": [True, False], "qux": [False, False]} resp = HttpResponse() resp = WaffleMiddleware().process_response(get, resp) for k in get.waffles: cookie = "dwf_%s" % k assert cookie in resp.cookies eq_(str(get.waffles[k][0]), resp.cookies[cookie].value) if get.waffles[k][1]: assert bool(resp.cookies[cookie]["max-age"]) == get.waffles[k][0] else: assert resp.cookies[cookie]["max-age"]