Git of dcc - anni/blob - 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