First
[anni] / test / pleroma / web / pleroma_api / controllers / mascot_controller_test.exs
1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
4
5 defmodule Pleroma.Web.PleromaAPI.MascotControllerTest do
6   use Pleroma.Web.ConnCase, async: true
7
8   alias Pleroma.User
9
10   test "mascot upload" do
11     %{conn: conn} = oauth_access(["write:accounts"])
12
13     non_image_file = %Plug.Upload{
14       content_type: "audio/mpeg",
15       path: Path.absname("test/fixtures/sound.mp3"),
16       filename: "sound.mp3"
17     }
18
19     ret_conn =
20       conn
21       |> put_req_header("content-type", "multipart/form-data")
22       |> put("/api/v1/pleroma/mascot", %{"file" => non_image_file})
23
24     assert json_response_and_validate_schema(ret_conn, 415)
25
26     file = %Plug.Upload{
27       content_type: "image/jpeg",
28       path: Path.absname("test/fixtures/image.jpg"),
29       filename: "an_image.jpg"
30     }
31
32     conn =
33       conn
34       |> put_req_header("content-type", "multipart/form-data")
35       |> put("/api/v1/pleroma/mascot", %{"file" => file})
36
37     assert %{"id" => _, "type" => _image} = json_response_and_validate_schema(conn, 200)
38   end
39
40   test "mascot retrieving" do
41     %{user: user, conn: conn} = oauth_access(["read:accounts", "write:accounts"])
42
43     # When user hasn't set a mascot, we should just get pleroma tan back
44     ret_conn = get(conn, "/api/v1/pleroma/mascot")
45
46     assert %{"url" => url} = json_response_and_validate_schema(ret_conn, 200)
47     assert url =~ "pleroma-fox-tan-smol"
48
49     # When a user sets their mascot, we should get that back
50     file = %Plug.Upload{
51       content_type: "image/jpeg",
52       path: Path.absname("test/fixtures/image.jpg"),
53       filename: "an_image.jpg"
54     }
55
56     ret_conn =
57       conn
58       |> put_req_header("content-type", "multipart/form-data")
59       |> put("/api/v1/pleroma/mascot", %{"file" => file})
60
61     assert json_response_and_validate_schema(ret_conn, 200)
62
63     user = User.get_cached_by_id(user.id)
64
65     conn =
66       conn
67       |> assign(:user, user)
68       |> get("/api/v1/pleroma/mascot")
69
70     assert %{"url" => url, "type" => "image"} = json_response_and_validate_schema(conn, 200)
71     assert url =~ "an_image"
72   end
73 end