git.squeep.com Git - akkoma/blob - test/web/federator_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.FederatorTest do
   6   alias Pleroma.Instances
   7   alias Pleroma.Tests.ObanHelpers
   8   alias Pleroma.Web.CommonAPI
   9   alias Pleroma.Web.Federator
  10   alias Pleroma.Workers.PublisherWorker
  11
  12   use Pleroma.DataCase
  13   use Oban.Testing, repo: Pleroma.Repo
  14
  15   import Pleroma.Factory
  16   import Mock
  17
  18   setup_all do
  19     Tesla.Mock.mock_global(fn env -> apply(HttpRequestMock, :request, [env]) end)
  20
  21     :ok
  22   end
  23
  24   clear_config_all([:instance, :federating]) do
  25     Pleroma.Config.put([:instance, :federating], true)
  26   end
  27
  28   clear_config([:instance, :allow_relay])
  29   clear_config([:instance, :rewrite_policy])
  30   clear_config([:mrf_keyword])
  31
  32   describe "Publish an activity" do
  33     setup do
  34       user = insert(:user)
  35       {:ok, activity} = CommonAPI.post(user, %{"status" => "HI"})
  36
  37       relay_mock = {
  38         Pleroma.Web.ActivityPub.Relay,
  39         [],
  40         [publish: fn _activity -> send(self(), :relay_publish) end]
  41       }
  42
  43       %{activity: activity, relay_mock: relay_mock}
  44     end
  45
  46     test "with relays active, it publishes to the relay", %{
  47       activity: activity,
  48       relay_mock: relay_mock
  49     } do
  50       with_mocks([relay_mock]) do
  51         Federator.publish(activity)
  52         ObanHelpers.perform(all_enqueued(worker: PublisherWorker))
  53       end
  54
  55       assert_received :relay_publish
  56     end
  57
  58     test "with relays deactivated, it does not publish to the relay", %{
  59       activity: activity,
  60       relay_mock: relay_mock
  61     } do
  62       Pleroma.Config.put([:instance, :allow_relay], false)
  63
  64       with_mocks([relay_mock]) do
  65         Federator.publish(activity)
  66         ObanHelpers.perform(all_enqueued(worker: PublisherWorker))
  67       end
  68
  69       refute_received :relay_publish
  70     end
  71   end
  72
  73   describe "Targets reachability filtering in `publish`" do
  74     test "it federates only to reachable instances via AP" do
  75       user = insert(:user)
  76
  77       {inbox1, inbox2} =
  78         {"https://domain.com/users/nick1/inbox", "https://domain2.com/users/nick2/inbox"}
  79
  80       insert(:user, %{
  81         local: false,
  82         nickname: "nick1@domain.com",
  83         ap_id: "https://domain.com/users/nick1",
  84         source_data: %{"inbox" => inbox1},
  85         ap_enabled: true
  86       })
  87
  88       insert(:user, %{
  89         local: false,
  90         nickname: "nick2@domain2.com",
  91         ap_id: "https://domain2.com/users/nick2",
  92         source_data: %{"inbox" => inbox2},
  93         ap_enabled: true
  94       })
  95
  96       dt = NaiveDateTime.utc_now()
  97       Instances.set_unreachable(inbox1, dt)
  98
  99       Instances.set_consistently_unreachable(URI.parse(inbox2).host)
 100
 101       {:ok, _activity} =
 102         CommonAPI.post(user, %{"status" => "HI @nick1@domain.com, @nick2@domain2.com!"})
 103
 104       expected_dt = NaiveDateTime.to_iso8601(dt)
 105
 106       ObanHelpers.perform(all_enqueued(worker: PublisherWorker))
 107
 108       assert ObanHelpers.member?(
 109                %{
 110                  "op" => "publish_one",
 111                  "params" => %{"inbox" => inbox1, "unreachable_since" => expected_dt}
 112                },
 113                all_enqueued(worker: PublisherWorker)
 114              )
 115     end
 116   end
 117
 118   describe "Receive an activity" do
 119     test "successfully processes incoming AP docs with correct origin" do
 120       params = %{
 121         "@context" => "https://www.w3.org/ns/activitystreams",
 122         "actor" => "http://mastodon.example.org/users/admin",
 123         "type" => "Create",
 124         "id" => "http://mastodon.example.org/users/admin/activities/1",
 125         "object" => %{
 126           "type" => "Note",
 127           "content" => "hi world!",
 128           "id" => "http://mastodon.example.org/users/admin/objects/1",
 129           "attributedTo" => "http://mastodon.example.org/users/admin"
 130         },
 131         "to" => ["https://www.w3.org/ns/activitystreams#Public"]
 132       }
 133
 134       assert {:ok, job} = Federator.incoming_ap_doc(params)
 135       assert {:ok, _activity} = ObanHelpers.perform(job)
 136     end
 137
 138     test "rejects incoming AP docs with incorrect origin" do
 139       params = %{
 140         "@context" => "https://www.w3.org/ns/activitystreams",
 141         "actor" => "https://niu.moe/users/rye",
 142         "type" => "Create",
 143         "id" => "http://mastodon.example.org/users/admin/activities/1",
 144         "object" => %{
 145           "type" => "Note",
 146           "content" => "hi world!",
 147           "id" => "http://mastodon.example.org/users/admin/objects/1",
 148           "attributedTo" => "http://mastodon.example.org/users/admin"
 149         },
 150         "to" => ["https://www.w3.org/ns/activitystreams#Public"]
 151       }
 152
 153       assert {:ok, job} = Federator.incoming_ap_doc(params)
 154       assert :error = ObanHelpers.perform(job)
 155     end
 156
 157     test "it does not crash if MRF rejects the post" do
 158       Pleroma.Config.put([:mrf_keyword, :reject], ["lain"])
 159
 160       Pleroma.Config.put(
 161         [:instance, :rewrite_policy],
 162         Pleroma.Web.ActivityPub.MRF.KeywordPolicy
 163       )
 164
 165       params =
 166         File.read!("test/fixtures/mastodon-post-activity.json")
 167         |> Poison.decode!()
 168
 169       assert {:ok, job} = Federator.incoming_ap_doc(params)
 170       assert :error = ObanHelpers.perform(job)
 171     end
 172   end
 173 end