git.squeep.com Git - akkoma/blob - test/pleroma/uploaders/s3_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Uploaders.S3Test do
   6   use Pleroma.DataCase
   7
   8   alias Pleroma.Config
   9   alias Pleroma.Uploaders.S3
  10
  11   import Mock
  12   import ExUnit.CaptureLog
  13
  14   setup do
  15     clear_config([Pleroma.Upload, :uploader], Pleroma.Uploaders.S3)
  16     clear_config([Pleroma.Upload, :base_url], "https://s3.amazonaws.com")
  17     clear_config([Pleroma.Uploaders.S3])
  18     clear_config([Pleroma.Uploaders.S3, :bucket], "test_bucket")
  19   end
  20
  21   describe "get_file/1" do
  22     test "it returns path to local folder for files" do
  23       assert S3.get_file("test_image.jpg") == {
  24                :ok,
  25                {:url, "https://s3.amazonaws.com/test_bucket/test_image.jpg"}
  26              }
  27     end
  28
  29     test "it returns path without bucket when truncated_namespace set to ''" do
  30       Config.put([Pleroma.Uploaders.S3],
  31         bucket: "test_bucket",
  32         bucket_namespace: "myaccount",
  33         truncated_namespace: ""
  34       )
  35
  36       Config.put([Pleroma.Upload, :base_url], "https://s3.amazonaws.com")
  37
  38       assert S3.get_file("test_image.jpg") == {
  39                :ok,
  40                {:url, "https://s3.amazonaws.com/test_image.jpg"}
  41              }
  42     end
  43
  44     test "it returns path with bucket namespace when namespace is set" do
  45       Config.put([Pleroma.Uploaders.S3],
  46         bucket: "test_bucket",
  47         bucket_namespace: "family"
  48       )
  49
  50       assert S3.get_file("test_image.jpg") == {
  51                :ok,
  52                {:url, "https://s3.amazonaws.com/family:test_bucket/test_image.jpg"}
  53              }
  54     end
  55   end
  56
  57   describe "put_file/1" do
  58     setup do
  59       file_upload = %Pleroma.Upload{
  60         name: "image-tet.jpg",
  61         content_type: "image/jpeg",
  62         path: "test_folder/image-tet.jpg",
  63         tempfile: Path.absname("test/instance_static/add/shortcode.png")
  64       }
  65
  66       [file_upload: file_upload]
  67     end
  68
  69     test "save file", %{file_upload: file_upload} do
  70       with_mock ExAws, request: fn _ -> {:ok, :ok} end do
  71         assert S3.put_file(file_upload) == {:ok, {:file, "test_folder/image-tet.jpg"}}
  72       end
  73     end
  74
  75     test "returns error", %{file_upload: file_upload} do
  76       with_mock ExAws, request: fn _ -> {:error, "S3 Upload failed"} end do
  77         assert capture_log(fn ->
  78                  assert S3.put_file(file_upload) == {:error, "S3 Upload failed"}
  79                end) =~ "Elixir.Pleroma.Uploaders.S3: {:error, \"S3 Upload failed\"}"
  80       end
  81     end
  82   end
  83
  84   describe "delete_file/1" do
  85     test_with_mock "deletes file", ExAws, request: fn _req -> {:ok, %{status_code: 204}} end do
  86       assert :ok = S3.delete_file("image.jpg")
  87       assert_called(ExAws.request(:_))
  88     end
  89   end
  90 end