git.squeep.com Git - akkoma/blob - test/pleroma/web/rich_media/helpers_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.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
  19     :ok
  20   end
  21
  22   setup do: clear_config([:rich_media, :enabled])
  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 of private network from posts" do
  68     user = insert(:user)
  69
  70     {:ok, activity} =
  71       CommonAPI.post(user, %{status: "http://127.0.0.1:4000/notice/9kCP7VNyPJXFOXDrgO"})
  72
  73     {:ok, activity2} = CommonAPI.post(user, %{status: "https://10.111.10.1/notice/9kCP7V"})
  74     {:ok, activity3} = CommonAPI.post(user, %{status: "https://172.16.32.40/notice/9kCP7V"})
  75     {:ok, activity4} = CommonAPI.post(user, %{status: "https://192.168.10.40/notice/9kCP7V"})
  76     {:ok, activity5} = CommonAPI.post(user, %{status: "https://pleroma.local/notice/9kCP7V"})
  77
  78     Config.put([:rich_media, :enabled], true)
  79
  80     assert %{} = Helpers.fetch_data_for_activity(activity)
  81     assert %{} = Helpers.fetch_data_for_activity(activity2)
  82     assert %{} = Helpers.fetch_data_for_activity(activity3)
  83     assert %{} = Helpers.fetch_data_for_activity(activity4)
  84     assert %{} = Helpers.fetch_data_for_activity(activity5)
  85   end
  86 end