91 lines
2.5 KiB
Elixir
91 lines
2.5 KiB
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.Uploaders.S3Test do
|
|
use Pleroma.DataCase
|
|
|
|
alias Pleroma.Config
|
|
alias Pleroma.Uploaders.S3
|
|
|
|
import Mock
|
|
import ExUnit.CaptureLog
|
|
|
|
setup do
|
|
config = Config.get([Pleroma.Uploaders.S3])
|
|
|
|
Config.put([Pleroma.Uploaders.S3],
|
|
bucket: "test_bucket",
|
|
public_endpoint: "https://s3.amazonaws.com"
|
|
)
|
|
|
|
on_exit(fn ->
|
|
Config.put([Pleroma.Uploaders.S3], config)
|
|
end)
|
|
|
|
:ok
|
|
end
|
|
|
|
describe "get_file/1" do
|
|
test "it returns path to local folder for files" do
|
|
assert S3.get_file("test_image.jpg") == {
|
|
:ok,
|
|
{:url, "https://s3.amazonaws.com/test_bucket/test_image.jpg"}
|
|
}
|
|
end
|
|
|
|
test "it returns path without bucket when truncated_namespace set to ''" do
|
|
Config.put([Pleroma.Uploaders.S3],
|
|
bucket: "test_bucket",
|
|
public_endpoint: "https://s3.amazonaws.com",
|
|
truncated_namespace: ""
|
|
)
|
|
|
|
assert S3.get_file("test_image.jpg") == {
|
|
:ok,
|
|
{:url, "https://s3.amazonaws.com/test_image.jpg"}
|
|
}
|
|
end
|
|
|
|
test "it returns path with bucket namespace when namespace is set" do
|
|
Config.put([Pleroma.Uploaders.S3],
|
|
bucket: "test_bucket",
|
|
public_endpoint: "https://s3.amazonaws.com",
|
|
bucket_namespace: "family"
|
|
)
|
|
|
|
assert S3.get_file("test_image.jpg") == {
|
|
:ok,
|
|
{:url, "https://s3.amazonaws.com/family:test_bucket/test_image.jpg"}
|
|
}
|
|
end
|
|
end
|
|
|
|
describe "put_file/1" do
|
|
setup do
|
|
file_upload = %Pleroma.Upload{
|
|
name: "image-tet.jpg",
|
|
content_type: "image/jpg",
|
|
path: "test_folder/image-tet.jpg",
|
|
tempfile: Path.absname("test/fixtures/image_tmp.jpg")
|
|
}
|
|
|
|
[file_upload: file_upload]
|
|
end
|
|
|
|
test "save file", %{file_upload: file_upload} do
|
|
with_mock ExAws, request: fn _ -> {:ok, :ok} end do
|
|
assert S3.put_file(file_upload) == {:ok, {:file, "test_folder/image-tet.jpg"}}
|
|
end
|
|
end
|
|
|
|
test "returns error", %{file_upload: file_upload} do
|
|
with_mock ExAws, request: fn _ -> {:error, "S3 Upload failed"} end do
|
|
assert capture_log(fn ->
|
|
assert S3.put_file(file_upload) == {:error, "S3 Upload failed"}
|
|
end) =~ "Elixir.Pleroma.Uploaders.S3: {:error, \"S3 Upload failed\"}"
|
|
end
|
|
end
|
|
end
|
|
end
|