upload_test.exs 4.06 KB
Newer Older
lain's avatar
lain committed
1
2
3
4
defmodule Pleroma.UploadTest do
  alias Pleroma.Upload
  use Pleroma.DataCase

href's avatar
href committed
5
  describe "Storing a file with the Local uploader" do
6
    setup [:ensure_local_uploader]
href's avatar
href committed
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

    test "returns a media url" do
      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

      file = %Plug.Upload{
        content_type: "image/jpg",
        path: Path.absname("test/fixtures/image_tmp.jpg"),
        filename: "image.jpg"
      }

      {:ok, data} = Upload.store(file)

      assert %{"url" => [%{"href" => url}]} = data

      assert String.starts_with?(url, Pleroma.Web.base_url() <> "/media/")
    end

href's avatar
href committed
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
    test "returns a media url with configured base_url" do
      base_url = "https://cache.pleroma.social"

      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

      file = %Plug.Upload{
        content_type: "image/jpg",
        path: Path.absname("test/fixtures/image_tmp.jpg"),
        filename: "image.jpg"
      }

      {:ok, data} = Upload.store(file, base_url: base_url)

      assert %{"url" => [%{"href" => url}]} = data

      assert String.starts_with?(url, base_url <> "/media/")
    end

Sir_Boops's avatar
Sir_Boops committed
42
43
44
    test "copies the file to the configured folder with deduping" do
      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

lain's avatar
lain committed
45
46
      file = %Plug.Upload{
        content_type: "image/jpg",
Sir_Boops's avatar
Sir_Boops committed
47
        path: Path.absname("test/fixtures/image_tmp.jpg"),
lain's avatar
lain committed
48
49
50
        filename: "an [image.jpg"
      }

href's avatar
href committed
51
      {:ok, data} = Upload.store(file, filters: [Pleroma.Upload.Filter.Dedupe])
lain's avatar
lain committed
52

href's avatar
href committed
53
54
55
      assert List.first(data["url"])["href"] ==
               Pleroma.Web.base_url() <>
                 "/media/e7a6d0cf595bff76f14c9a98b6c199539559e8b844e02e51e5efcfd1f614a2df.jpg"
lain's avatar
lain committed
56
    end
eal's avatar
eal committed
57

Sir_Boops's avatar
Sir_Boops committed
58
59
60
    test "copies the file to the configured folder without deduping" do
      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

lain's avatar
lain committed
61
      file = %Plug.Upload{
Sir_Boops's avatar
Sir_Boops committed
62
63
        content_type: "image/jpg",
        path: Path.absname("test/fixtures/image_tmp.jpg"),
lain's avatar
lain committed
64
65
66
        filename: "an [image.jpg"
      }

href's avatar
href committed
67
      {:ok, data} = Upload.store(file)
Sir_Boops's avatar
Sir_Boops committed
68
      assert data["name"] == "an [image.jpg"
eal's avatar
eal committed
69
70
    end

Sir_Boops's avatar
Sir_Boops committed
71
72
73
    test "fixes incorrect content type" do
      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

lain's avatar
lain committed
74
      file = %Plug.Upload{
Sir_Boops's avatar
Sir_Boops committed
75
76
        content_type: "application/octet-stream",
        path: Path.absname("test/fixtures/image_tmp.jpg"),
lain's avatar
lain committed
77
78
79
        filename: "an [image.jpg"
      }

href's avatar
href committed
80
      {:ok, data} = Upload.store(file, filters: [Pleroma.Upload.Filter.Dedupe])
Sir_Boops's avatar
Sir_Boops committed
81
      assert hd(data["url"])["mediaType"] == "image/jpeg"
eal's avatar
eal committed
82
    end
83
84
85
86
87
88
89
90
91
92

    test "adds missing extension" do
      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

      file = %Plug.Upload{
        content_type: "image/jpg",
        path: Path.absname("test/fixtures/image_tmp.jpg"),
        filename: "an [image"
      }

href's avatar
href committed
93
      {:ok, data} = Upload.store(file)
94
95
      assert data["name"] == "an [image.jpg"
    end
96
97
98
99
100
101
102
103
104
105

    test "fixes incorrect file extension" do
      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

      file = %Plug.Upload{
        content_type: "image/jpg",
        path: Path.absname("test/fixtures/image_tmp.jpg"),
        filename: "an [image.blah"
      }

href's avatar
href committed
106
      {:ok, data} = Upload.store(file)
107
108
109
110
111
112
113
114
115
116
117
118
      assert data["name"] == "an [image.jpg"
    end

    test "don't modify filename of an unknown type" do
      File.cp("test/fixtures/test.txt", "test/fixtures/test_tmp.txt")

      file = %Plug.Upload{
        content_type: "text/plain",
        path: Path.absname("test/fixtures/test_tmp.txt"),
        filename: "test.txt"
      }

href's avatar
href committed
119
      {:ok, data} = Upload.store(file)
120
121
      assert data["name"] == "test.txt"
    end
href's avatar
href committed
122
123
124
125
126
127
128
129
130
131
132
133
134
135

    test "copies the file to the configured folder with anonymizing filename" do
      File.cp!("test/fixtures/image.jpg", "test/fixtures/image_tmp.jpg")

      file = %Plug.Upload{
        content_type: "image/jpg",
        path: Path.absname("test/fixtures/image_tmp.jpg"),
        filename: "an [image.jpg"
      }

      {:ok, data} = Upload.store(file, filters: [Pleroma.Upload.Filter.AnonymizeFilename])

      refute data["name"] == "an [image.jpg"
    end
lain's avatar
lain committed
136
137
  end
end