1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 defmodule Pleroma.Web.Plugs.CacheTest do
9 alias Pleroma.Web.Plugs.Cache
13 {"cache-control", "max-age=0, private, must-revalidate"},
14 {"content-type", "cofe/hot; charset=utf-8"},
15 {"x-cache", "MISS from Pleroma"}
20 {"cache-control", "max-age=0, private, must-revalidate"},
21 {"content-type", "cofe/hot; charset=utf-8"},
22 {"x-cache", "HIT from Pleroma"}
27 test "caches a response" do
30 |> Cache.call(%{query_params: false, ttl: nil})
31 |> put_resp_content_type("cofe/hot")
32 |> send_resp(:ok, "cofe")
35 assert_raise(Plug.Conn.AlreadySentError, fn ->
37 |> Cache.call(%{query_params: false, ttl: nil})
38 |> put_resp_content_type("cofe/hot")
39 |> send_resp(:ok, "cofe")
45 |> Cache.call(%{query_params: false, ttl: nil})
52 |> Cache.call(%{query_params: false, ttl: @ttl})
53 |> put_resp_content_type("cofe/hot")
54 |> send_resp(:ok, "cofe")
59 |> Cache.call(%{query_params: false, ttl: @ttl})
62 :timer.sleep(@ttl + 1)
66 |> Cache.call(%{query_params: false, ttl: @ttl})
67 |> put_resp_content_type("cofe/hot")
68 |> send_resp(:ok, "cofe")
72 test "set ttl via conn.assigns" do
75 |> Cache.call(%{query_params: false, ttl: nil})
76 |> put_resp_content_type("cofe/hot")
77 |> assign(:cache_ttl, @ttl)
78 |> send_resp(:ok, "cofe")
83 |> Cache.call(%{query_params: false, ttl: nil})
86 :timer.sleep(@ttl + 1)
90 |> Cache.call(%{query_params: false, ttl: nil})
91 |> put_resp_content_type("cofe/hot")
92 |> send_resp(:ok, "cofe")
96 test "ignore query string when `query_params` is false" do
99 |> Cache.call(%{query_params: false, ttl: nil})
100 |> put_resp_content_type("cofe/hot")
101 |> send_resp(:ok, "cofe")
105 conn(:get, "/?cofefe")
106 |> Cache.call(%{query_params: false, ttl: nil})
110 test "take query string into account when `query_params` is true" do
113 |> Cache.call(%{query_params: true, ttl: nil})
114 |> put_resp_content_type("cofe/hot")
115 |> send_resp(:ok, "cofe")
119 conn(:get, "/?cofefe")
120 |> Cache.call(%{query_params: true, ttl: nil})
121 |> put_resp_content_type("cofe/hot")
122 |> send_resp(:ok, "cofe")
126 test "take specific query params into account when `query_params` is list" do
128 conn(:get, "/?a=1&b=2&c=3&foo=bar")
129 |> fetch_query_params()
130 |> Cache.call(%{query_params: ["a", "b", "c"], ttl: nil})
131 |> put_resp_content_type("cofe/hot")
132 |> send_resp(:ok, "cofe")
136 conn(:get, "/?bar=foo&c=3&b=2&a=1")
137 |> fetch_query_params()
138 |> Cache.call(%{query_params: ["a", "b", "c"], ttl: nil})
142 conn(:get, "/?bar=foo&c=3&b=2&a=2")
143 |> fetch_query_params()
144 |> Cache.call(%{query_params: ["a", "b", "c"], ttl: nil})
145 |> put_resp_content_type("cofe/hot")
146 |> send_resp(:ok, "cofe")
150 test "ignore not GET requests" do
154 {"cache-control", "max-age=0, private, must-revalidate"},
155 {"content-type", "cofe/hot; charset=utf-8"}
160 |> Cache.call(%{query_params: true, ttl: nil})
161 |> put_resp_content_type("cofe/hot")
162 |> send_resp(:ok, "cofe")
166 test "ignore non-successful responses" do
170 {"cache-control", "max-age=0, private, must-revalidate"},
171 {"content-type", "tea/iced; charset=utf-8"}
176 |> Cache.call(%{query_params: true, ttl: nil})
177 |> put_resp_content_type("tea/iced")
178 |> send_resp(:im_a_teapot, "🥤")