git.squeep.com Git - akkoma/blob - test/web/activity_pub/mrf/steal_emoji_policy_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.ActivityPub.MRF.StealEmojiPolicyTest do
   6   use Pleroma.DataCase
   7
   8   alias Pleroma.Config
   9   alias Pleroma.Web.ActivityPub.MRF.StealEmojiPolicy
  10
  11   setup_all do
  12     Tesla.Mock.mock_global(fn env -> apply(HttpRequestMock, :request, [env]) end)
  13     :ok
  14   end
  15
  16   setup do
  17     clear_config(:mrf_steal_emoji)
  18
  19     emoji_path = Path.join(Config.get([:instance, :static_dir]), "emoji/stolen")
  20     File.rm_rf!(emoji_path)
  21     File.mkdir!(emoji_path)
  22
  23     Pleroma.Emoji.reload()
  24   end
  25
  26   test "does nothing by default" do
  27     installed_emoji = Pleroma.Emoji.get_all() |> Enum.map(fn {k, _} -> k end)
  28     refute "firedfox" in installed_emoji
  29
  30     message = %{
  31       "type" => "Create",
  32       "object" => %{
  33         "emoji" => [{"firedfox", "https://example.org/emoji/firedfox.png"}],
  34         "actor" => "https://example.org/users/admin"
  35       }
  36     }
  37
  38     assert {:ok, message} == StealEmojiPolicy.filter(message)
  39
  40     installed_emoji = Pleroma.Emoji.get_all() |> Enum.map(fn {k, _} -> k end)
  41     refute "firedfox" in installed_emoji
  42   end
  43
  44   test "Steals emoji on unknown shortcode from allowed remote host" do
  45     installed_emoji = Pleroma.Emoji.get_all() |> Enum.map(fn {k, _} -> k end)
  46     refute "firedfox" in installed_emoji
  47
  48     message = %{
  49       "type" => "Create",
  50       "object" => %{
  51         "emoji" => [{"firedfox", "https://example.org/emoji/firedfox.png"}],
  52         "actor" => "https://example.org/users/admin"
  53       }
  54     }
  55
  56     Config.put([:mrf_steal_emoji, :hosts], ["example.org"])
  57     Config.put([:mrf_steal_emoji, :size_limit], 284_468)
  58
  59     assert {:ok, message} == StealEmojiPolicy.filter(message)
  60
  61     installed_emoji = Pleroma.Emoji.get_all() |> Enum.map(fn {k, _} -> k end)
  62     assert "firedfox" in installed_emoji
  63   end
  64 end