82 lines
		
	
	
	
		
			2.4 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			82 lines
		
	
	
	
		
			2.4 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
 | |
| 
 | |
|   clear_config([Pleroma.Uploaders.S3]) do
 | |
|     Config.put([Pleroma.Uploaders.S3],
 | |
|       bucket: "test_bucket",
 | |
|       public_endpoint: "https://s3.amazonaws.com"
 | |
|     )
 | |
|   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
 | 
