total rebase
[anni] / test / pleroma / web / plugs / uploaded_media_plug_test.exs
1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
4
5 defmodule Pleroma.Web.Plugs.UploadedMediaPlugTest do
6   use Pleroma.Web.ConnCase, async: true
7
8   alias Pleroma.UnstubbedConfigMock, as: ConfigMock
9   alias Pleroma.Upload
10
11   import Mox
12
13   defp upload_file(context) do
14     ConfigMock
15     |> stub_with(Pleroma.Test.StaticConfig)
16
17     Pleroma.DataCase.ensure_local_uploader(context)
18
19     File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
20
21     file = %Plug.Upload{
22       content_type: "image/jpeg",
23       path: Path.absname("test/fixtures/image_tmp.jpg"),
24       filename: "nice_tf.jpg"
25     }
26
27     {:ok, data} = Upload.store(file)
28     [%{"href" => attachment_url} | _] = data["url"]
29     [attachment_url: attachment_url]
30   end
31
32   setup_all :upload_file
33
34   setup do
35     ConfigMock
36     |> stub_with(Pleroma.Test.StaticConfig)
37
38     :ok
39   end
40
41   test "does not send Content-Disposition header when name param is not set", %{
42     attachment_url: attachment_url
43   } do
44     conn = get(build_conn(), attachment_url)
45     refute Enum.any?(conn.resp_headers, &(elem(&1, 0) == "content-disposition"))
46   end
47
48   test "sends Content-Disposition header when name param is set", %{
49     attachment_url: attachment_url
50   } do
51     conn = get(build_conn(), attachment_url <> ~s[?name="cofe".gif])
52
53     assert Enum.any?(
54              conn.resp_headers,
55              &(&1 == {"content-disposition", ~s[inline; filename="\\"cofe\\".gif"]})
56            )
57   end
58 end