Add some tests
[akkoma] / test / upload_test.exs
1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
4
5 defmodule Pleroma.UploadTest do
6 alias Pleroma.Upload
7 use Pleroma.DataCase
8
9 describe "Storing a file with the Local uploader" do
10 setup [:ensure_local_uploader]
11
12 test "returns a media url" do
13 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
14
15 file = %Plug.Upload{
16 content_type: "image/jpg",
17 path: Path.absname("test/fixtures/image_tmp.jpg"),
18 filename: "image.jpg"
19 }
20
21 {:ok, data} = Upload.store(file)
22
23 assert %{"url" => [%{"href" => url}]} = data
24
25 assert String.starts_with?(url, Pleroma.Web.base_url() <> "/media/")
26 end
27
28 test "returns a media url with configured base_url" do
29 base_url = "https://cache.pleroma.social"
30
31 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
32
33 file = %Plug.Upload{
34 content_type: "image/jpg",
35 path: Path.absname("test/fixtures/image_tmp.jpg"),
36 filename: "image.jpg"
37 }
38
39 {:ok, data} = Upload.store(file, base_url: base_url)
40
41 assert %{"url" => [%{"href" => url}]} = data
42
43 assert String.starts_with?(url, base_url <> "/media/")
44 end
45
46 test "copies the file to the configured folder with deduping" do
47 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
48
49 file = %Plug.Upload{
50 content_type: "image/jpg",
51 path: Path.absname("test/fixtures/image_tmp.jpg"),
52 filename: "an [image.jpg"
53 }
54
55 {:ok, data} = Upload.store(file, filters: [Pleroma.Upload.Filter.Dedupe])
56
57 assert List.first(data["url"])["href"] ==
58 Pleroma.Web.base_url() <>
59 "/media/e7a6d0cf595bff76f14c9a98b6c199539559e8b844e02e51e5efcfd1f614a2df.jpg"
60 end
61
62 test "copies the file to the configured folder without deduping" do
63 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
64
65 file = %Plug.Upload{
66 content_type: "image/jpg",
67 path: Path.absname("test/fixtures/image_tmp.jpg"),
68 filename: "an [image.jpg"
69 }
70
71 {:ok, data} = Upload.store(file)
72 assert data["name"] == "an [image.jpg"
73 end
74
75 test "fixes incorrect content type" do
76 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
77
78 file = %Plug.Upload{
79 content_type: "application/octet-stream",
80 path: Path.absname("test/fixtures/image_tmp.jpg"),
81 filename: "an [image.jpg"
82 }
83
84 {:ok, data} = Upload.store(file, filters: [Pleroma.Upload.Filter.Dedupe])
85 assert hd(data["url"])["mediaType"] == "image/jpeg"
86 end
87
88 test "adds missing extension" do
89 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
90
91 file = %Plug.Upload{
92 content_type: "image/jpg",
93 path: Path.absname("test/fixtures/image_tmp.jpg"),
94 filename: "an [image"
95 }
96
97 {:ok, data} = Upload.store(file)
98 assert data["name"] == "an [image.jpg"
99 end
100
101 test "fixes incorrect file extension" do
102 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
103
104 file = %Plug.Upload{
105 content_type: "image/jpg",
106 path: Path.absname("test/fixtures/image_tmp.jpg"),
107 filename: "an [image.blah"
108 }
109
110 {:ok, data} = Upload.store(file)
111 assert data["name"] == "an [image.jpg"
112 end
113
114 test "don't modify filename of an unknown type" do
115 File.cp("test/fixtures/test.txt", "test/fixtures/test_tmp.txt")
116
117 file = %Plug.Upload{
118 content_type: "text/plain",
119 path: Path.absname("test/fixtures/test_tmp.txt"),
120 filename: "test.txt"
121 }
122
123 {:ok, data} = Upload.store(file)
124 assert data["name"] == "test.txt"
125 end
126
127 test "copies the file to the configured folder with anonymizing filename" do
128 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
129
130 file = %Plug.Upload{
131 content_type: "image/jpg",
132 path: Path.absname("test/fixtures/image_tmp.jpg"),
133 filename: "an [image.jpg"
134 }
135
136 {:ok, data} = Upload.store(file, filters: [Pleroma.Upload.Filter.AnonymizeFilename])
137
138 refute data["name"] == "an [image.jpg"
139 end
140
141 test "escapes invalid characters in url" do
142 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
143
144 file = %Plug.Upload{
145 content_type: "image/jpg",
146 path: Path.absname("test/fixtures/image_tmp.jpg"),
147 filename: "an… image.jpg"
148 }
149
150 {:ok, data} = Upload.store(file)
151 [attachment_url | _] = data["url"]
152
153 assert Path.basename(attachment_url["href"]) ==
154 "an%E2%80%A6%20image.jpg"
155 end
156
157 test "escapes reserved uri characters" do
158 File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")
159
160 file = %Plug.Upload{
161 content_type: "image/jpg",
162 path: Path.absname("test/fixtures/image_tmp.jpg"),
163 filename: ":?#[]@!$&\\'()*+,;=.jpg"
164 }
165
166 {:ok, data} = Upload.store(file)
167 [attachment_url | _] = data["url"]
168
169 assert Path.basename(attachment_url["href"]) ==
170 "%3A%3F%23%5B%5D%40%21%24%26%5C%27%28%29%2A%2B%2C%3B%3D.jpg"
171 end
172 end
173 end