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

   1 defmodule Pleroma.Web.FederatorTest do
   2   alias Pleroma.Web.Federator
   3   alias Pleroma.Web.CommonAPI
   4   use Pleroma.DataCase
   5   import Pleroma.Factory
   6   import Mock
   7
   8   test "enqueues an element according to priority" do
   9     queue = [%{item: 1, priority: 2}]
  10
  11     new_queue = Federator.enqueue_sorted(queue, 2, 1)
  12     assert new_queue == [%{item: 2, priority: 1}, %{item: 1, priority: 2}]
  13
  14     new_queue = Federator.enqueue_sorted(queue, 2, 3)
  15     assert new_queue == [%{item: 1, priority: 2}, %{item: 2, priority: 3}]
  16   end
  17
  18   test "pop first item" do
  19     queue = [%{item: 2, priority: 1}, %{item: 1, priority: 2}]
  20
  21     assert {2, [%{item: 1, priority: 2}]} = Federator.queue_pop(queue)
  22   end
  23
  24   describe "Publish an activity" do
  25     setup do
  26       user = insert(:user)
  27       {:ok, activity} = CommonAPI.post(user, %{"status" => "HI"})
  28
  29       relay_mock = {
  30         Pleroma.Web.ActivityPub.Relay,
  31         [],
  32         [publish: fn _activity -> send(self(), :relay_publish) end]
  33       }
  34
  35       %{activity: activity, relay_mock: relay_mock}
  36     end
  37
  38     test "with relays active, it publishes to the relay", %{
  39       activity: activity,
  40       relay_mock: relay_mock
  41     } do
  42       with_mocks([relay_mock]) do
  43         Federator.handle(:publish, activity)
  44       end
  45
  46       assert_received :relay_publish
  47     end
  48
  49     test "with relays deactivated, it does not publish to the relay", %{
  50       activity: activity,
  51       relay_mock: relay_mock
  52     } do
  53       Pleroma.Config.put([:instance, :allow_relay], false)
  54
  55       with_mocks([relay_mock]) do
  56         Federator.handle(:publish, activity)
  57       end
  58
  59       refute_received :relay_publish
  60
  61       Pleroma.Config.put([:instance, :allow_relay], true)
  62     end
  63   end
  64 end