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

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.FederatorTest do
   6   alias Pleroma.Web.{CommonAPI, Federator}
   7   alias Pleroma.Instances
   8   use Pleroma.DataCase
   9   import Pleroma.Factory
  10   import Mock
  11
  12   setup_all do
  13     Tesla.Mock.mock_global(fn env -> apply(HttpRequestMock, :request, [env]) end)
  14     :ok
  15   end
  16
  17   describe "Publish an activity" do
  18     setup do
  19       user = insert(:user)
  20       {:ok, activity} = CommonAPI.post(user, %{"status" => "HI"})
  21
  22       relay_mock = {
  23         Pleroma.Web.ActivityPub.Relay,
  24         [],
  25         [publish: fn _activity -> send(self(), :relay_publish) end]
  26       }
  27
  28       %{activity: activity, relay_mock: relay_mock}
  29     end
  30
  31     test "with relays active, it publishes to the relay", %{
  32       activity: activity,
  33       relay_mock: relay_mock
  34     } do
  35       with_mocks([relay_mock]) do
  36         Federator.publish(activity)
  37       end
  38
  39       assert_received :relay_publish
  40     end
  41
  42     test "with relays deactivated, it does not publish to the relay", %{
  43       activity: activity,
  44       relay_mock: relay_mock
  45     } do
  46       Pleroma.Config.put([:instance, :allow_relay], false)
  47
  48       with_mocks([relay_mock]) do
  49         Federator.publish(activity)
  50       end
  51
  52       refute_received :relay_publish
  53
  54       Pleroma.Config.put([:instance, :allow_relay], true)
  55     end
  56   end
  57
  58   describe "Targets reachability filtering in `publish`" do
  59     test_with_mock "it federates only to reachable instances via AP",
  60                    Federator,
  61                    [:passthrough],
  62                    [] do
  63       user = insert(:user)
  64
  65       {inbox1, inbox2} =
  66         {"https://domain.com/users/nick1/inbox", "https://domain2.com/users/nick2/inbox"}
  67
  68       insert(:user, %{
  69         local: false,
  70         nickname: "nick1@domain.com",
  71         ap_id: "https://domain.com/users/nick1",
  72         info: %{ap_enabled: true, source_data: %{"inbox" => inbox1}}
  73       })
  74
  75       insert(:user, %{
  76         local: false,
  77         nickname: "nick2@domain2.com",
  78         ap_id: "https://domain2.com/users/nick2",
  79         info: %{ap_enabled: true, source_data: %{"inbox" => inbox2}}
  80       })
  81
  82       dt = NaiveDateTime.utc_now()
  83       Instances.set_unreachable(inbox1, dt)
  84
  85       Instances.set_consistently_unreachable(URI.parse(inbox2).host)
  86
  87       {:ok, _activity} =
  88         CommonAPI.post(user, %{"status" => "HI @nick1@domain.com, @nick2@domain2.com!"})
  89
  90       assert called(
  91                Federator.enqueue(:publish_single_ap, %{inbox: inbox1, unreachable_since: dt})
  92              )
  93
  94       refute called(Federator.enqueue(:publish_single_ap, %{inbox: inbox2}))
  95     end
  96
  97     test_with_mock "it federates only to reachable instances via Websub",
  98                    Federator,
  99                    [:passthrough],
 100                    [] do
 101       user = insert(:user)
 102       websub_topic = Pleroma.Web.OStatus.feed_path(user)
 103
 104       sub1 =
 105         insert(:websub_subscription, %{
 106           topic: websub_topic,
 107           state: "active",
 108           callback: "http://pleroma.soykaf.com/cb"
 109         })
 110
 111       sub2 =
 112         insert(:websub_subscription, %{
 113           topic: websub_topic,
 114           state: "active",
 115           callback: "https://pleroma2.soykaf.com/cb"
 116         })
 117
 118       dt = NaiveDateTime.utc_now()
 119       Instances.set_unreachable(sub2.callback, dt)
 120
 121       Instances.set_consistently_unreachable(sub1.callback)
 122
 123       {:ok, _activity} = CommonAPI.post(user, %{"status" => "HI"})
 124
 125       assert called(
 126                Federator.enqueue(:publish_single_websub, %{
 127                  callback: sub2.callback,
 128                  unreachable_since: dt
 129                })
 130              )
 131
 132       refute called(Federator.enqueue(:publish_single_websub, %{callback: sub1.callback}))
 133     end
 134
 135     test_with_mock "it federates only to reachable instances via Salmon",
 136                    Federator,
 137                    [:passthrough],
 138                    [] do
 139       user = insert(:user)
 140
 141       remote_user1 =
 142         insert(:user, %{
 143           local: false,
 144           nickname: "nick1@domain.com",
 145           ap_id: "https://domain.com/users/nick1",
 146           info: %{salmon: "https://domain.com/salmon"}
 147         })
 148
 149       remote_user2 =
 150         insert(:user, %{
 151           local: false,
 152           nickname: "nick2@domain2.com",
 153           ap_id: "https://domain2.com/users/nick2",
 154           info: %{salmon: "https://domain2.com/salmon"}
 155         })
 156
 157       dt = NaiveDateTime.utc_now()
 158       Instances.set_unreachable(remote_user2.ap_id, dt)
 159
 160       Instances.set_consistently_unreachable("domain.com")
 161
 162       {:ok, _activity} =
 163         CommonAPI.post(user, %{"status" => "HI @nick1@domain.com, @nick2@domain2.com!"})
 164
 165       assert called(
 166                Federator.enqueue(:publish_single_salmon, %{
 167                  recipient: remote_user2,
 168                  unreachable_since: dt
 169                })
 170              )
 171
 172       refute called(Federator.enqueue(:publish_single_websub, %{recipient: remote_user1}))
 173     end
 174   end
 175
 176   describe "Receive an activity" do
 177     test "successfully processes incoming AP docs with correct origin" do
 178       params = %{
 179         "@context" => "https://www.w3.org/ns/activitystreams",
 180         "actor" => "http://mastodon.example.org/users/admin",
 181         "type" => "Create",
 182         "id" => "http://mastodon.example.org/users/admin/activities/1",
 183         "object" => %{
 184           "type" => "Note",
 185           "content" => "hi world!",
 186           "id" => "http://mastodon.example.org/users/admin/objects/1",
 187           "attributedTo" => "http://mastodon.example.org/users/admin"
 188         },
 189         "to" => ["https://www.w3.org/ns/activitystreams#Public"]
 190       }
 191
 192       {:ok, _activity} = Federator.incoming_ap_doc(params)
 193     end
 194
 195     test "rejects incoming AP docs with incorrect origin" do
 196       params = %{
 197         "@context" => "https://www.w3.org/ns/activitystreams",
 198         "actor" => "https://niu.moe/users/rye",
 199         "type" => "Create",
 200         "id" => "http://mastodon.example.org/users/admin/activities/1",
 201         "object" => %{
 202           "type" => "Note",
 203           "content" => "hi world!",
 204           "id" => "http://mastodon.example.org/users/admin/objects/1",
 205           "attributedTo" => "http://mastodon.example.org/users/admin"
 206         },
 207         "to" => ["https://www.w3.org/ns/activitystreams#Public"]
 208       }
 209
 210       :error = Federator.incoming_ap_doc(params)
 211     end
 212   end
 213 end