git.squeep.com Git - akkoma/blob - test/web/activity_pub/mrf/mediaproxy_warming_policy_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.ActivityPub.MRF.MediaProxyWarmingPolicyTest do
   6   use Pleroma.DataCase
   7
   8   alias Pleroma.HTTP
   9   alias Pleroma.Tests.ObanHelpers
  10   alias Pleroma.Web.ActivityPub.MRF.MediaProxyWarmingPolicy
  11
  12   import Mock
  13
  14   @message %{
  15     "type" => "Create",
  16     "object" => %{
  17       "type" => "Note",
  18       "content" => "content",
  19       "attachment" => [
  20         %{"url" => [%{"href" => "http://example.com/image.jpg"}]}
  21       ]
  22     }
  23   }
  24
  25   test "it prefetches media proxy URIs" do
  26     with_mock HTTP, get: fn _, _, _ -> {:ok, []} end do
  27       MediaProxyWarmingPolicy.filter(@message)
  28
  29       ObanHelpers.perform_all()
  30       # Performing jobs which has been just enqueued
  31       ObanHelpers.perform_all()
  32
  33       assert called(HTTP.get(:_, :_, :_))
  34     end
  35   end
  36
  37   test "it does nothing when no attachments are present" do
  38     object =
  39       @message["object"]
  40       |> Map.delete("attachment")
  41
  42     message =
  43       @message
  44       |> Map.put("object", object)
  45
  46     with_mock HTTP, get: fn _, _, _ -> {:ok, []} end do
  47       MediaProxyWarmingPolicy.filter(message)
  48       refute called(HTTP.get(:_, :_, :_))
  49     end
  50   end
  51 end