git.squeep.com Git - akkoma/blob - test/pleroma/web/pleroma_api/controllers/mascot_controller_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 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
   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