git.squeep.com Git - akkoma/blob - test/web/rich_media/helpers_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.RichMedia.HelpersTest do
   6   use Pleroma.DataCase
   7
   8   alias Pleroma.Config
   9   alias Pleroma.Object
  10   alias Pleroma.Web.CommonAPI
  11   alias Pleroma.Web.RichMedia.Helpers
  12
  13   import Pleroma.Factory
  14   import Tesla.Mock
  15
  16   setup do
  17     mock(fn env -> apply(HttpRequestMock, :request, [env]) end)
  18     rich_media = Config.get([:rich_media, :enabled])
  19     on_exit(fn -> Config.put([:rich_media, :enabled], rich_media) end)
  20
  21     :ok
  22   end
  23
  24   test "refuses to crawl incomplete URLs" do
  25     user = insert(:user)
  26
  27     {:ok, activity} =
  28       CommonAPI.post(user, %{
  29         "status" => "[test](example.com/ogp)",
  30         "content_type" => "text/markdown"
  31       })
  32
  33     Config.put([:rich_media, :enabled], true)
  34
  35     assert %{} == Pleroma.Web.RichMedia.Helpers.fetch_data_for_activity(activity)
  36   end
  37
  38   test "refuses to crawl malformed URLs" do
  39     user = insert(:user)
  40
  41     {:ok, activity} =
  42       CommonAPI.post(user, %{
  43         "status" => "[test](example.com[]/ogp)",
  44         "content_type" => "text/markdown"
  45       })
  46
  47     Config.put([:rich_media, :enabled], true)
  48
  49     assert %{} == Pleroma.Web.RichMedia.Helpers.fetch_data_for_activity(activity)
  50   end
  51
  52   test "crawls valid, complete URLs" do
  53     user = insert(:user)
  54
  55     {:ok, activity} =
  56       CommonAPI.post(user, %{
  57         "status" => "[test](https://example.com/ogp)",
  58         "content_type" => "text/markdown"
  59       })
  60
  61     Config.put([:rich_media, :enabled], true)
  62
  63     assert %{page_url: "https://example.com/ogp", rich_media: _} =
  64              Pleroma.Web.RichMedia.Helpers.fetch_data_for_activity(activity)
  65   end
  66
  67   test "refuses to crawl URLs from posts marked sensitive" do
  68     user = insert(:user)
  69
  70     {:ok, activity} =
  71       CommonAPI.post(user, %{
  72         "status" => "http://example.com/ogp",
  73         "sensitive" => true
  74       })
  75
  76     %Object{} = object = Object.normalize(activity)
  77
  78     assert object.data["sensitive"]
  79
  80     Config.put([:rich_media, :enabled], true)
  81
  82     assert %{} = Pleroma.Web.RichMedia.Helpers.fetch_data_for_activity(activity)
  83   end
  84
  85   test "refuses to crawl URLs from posts tagged NSFW" do
  86     user = insert(:user)
  87
  88     {:ok, activity} =
  89       CommonAPI.post(user, %{
  90         "status" => "http://example.com/ogp #nsfw"
  91       })
  92
  93     %Object{} = object = Object.normalize(activity)
  94
  95     assert object.data["sensitive"]
  96
  97     Config.put([:rich_media, :enabled], true)
  98
  99     assert %{} = Pleroma.Web.RichMedia.Helpers.fetch_data_for_activity(activity)
 100   end
 101
 102   test "refuses to crawl URLs of private network from posts" do
 103     user = insert(:user)
 104
 105     {:ok, activity} =
 106       CommonAPI.post(user, %{"status" => "http://127.0.0.1:4000/notice/9kCP7VNyPJXFOXDrgO"})
 107
 108     {:ok, activity2} = CommonAPI.post(user, %{"status" => "https://10.111.10.1/notice/9kCP7V"})
 109     {:ok, activity3} = CommonAPI.post(user, %{"status" => "https://172.16.32.40/notice/9kCP7V"})
 110     {:ok, activity4} = CommonAPI.post(user, %{"status" => "https://192.168.10.40/notice/9kCP7V"})
 111     {:ok, activity5} = CommonAPI.post(user, %{"status" => "https://pleroma.local/notice/9kCP7V"})
 112
 113     Config.put([:rich_media, :enabled], true)
 114
 115     assert %{} = Helpers.fetch_data_for_activity(activity)
 116     assert %{} = Helpers.fetch_data_for_activity(activity2)
 117     assert %{} = Helpers.fetch_data_for_activity(activity3)
 118     assert %{} = Helpers.fetch_data_for_activity(activity4)
 119     assert %{} = Helpers.fetch_data_for_activity(activity5)
 120   end
 121 end