|
- defmodule Pleroma.Web.FederatorTest do
- alias Pleroma.Web.Federator
- alias Pleroma.Web.CommonAPI
- use Pleroma.DataCase
- import Pleroma.Factory
- import Mock
-
- setup_all do
- Tesla.Mock.mock_global(fn env -> apply(HttpRequestMock, :request, [env]) end)
- :ok
- end
-
- test "enqueues an element according to priority" do
- queue = [%{item: 1, priority: 2}]
-
- new_queue = Federator.enqueue_sorted(queue, 2, 1)
- assert new_queue == [%{item: 2, priority: 1}, %{item: 1, priority: 2}]
-
- new_queue = Federator.enqueue_sorted(queue, 2, 3)
- assert new_queue == [%{item: 1, priority: 2}, %{item: 2, priority: 3}]
- end
-
- test "pop first item" do
- queue = [%{item: 2, priority: 1}, %{item: 1, priority: 2}]
-
- assert {2, [%{item: 1, priority: 2}]} = Federator.queue_pop(queue)
- end
-
- describe "Publish an activity" do
- setup do
- user = insert(:user)
- {:ok, activity} = CommonAPI.post(user, %{"status" => "HI"})
-
- relay_mock = {
- Pleroma.Web.ActivityPub.Relay,
- [],
- [publish: fn _activity -> send(self(), :relay_publish) end]
- }
-
- %{activity: activity, relay_mock: relay_mock}
- end
-
- test "with relays active, it publishes to the relay", %{
- activity: activity,
- relay_mock: relay_mock
- } do
- with_mocks([relay_mock]) do
- Federator.handle(:publish, activity)
- end
-
- assert_received :relay_publish
- end
-
- test "with relays deactivated, it does not publish to the relay", %{
- activity: activity,
- relay_mock: relay_mock
- } do
- Pleroma.Config.put([:instance, :allow_relay], false)
-
- with_mocks([relay_mock]) do
- Federator.handle(:publish, activity)
- end
-
- refute_received :relay_publish
-
- Pleroma.Config.put([:instance, :allow_relay], true)
- end
- end
-
- describe "Receive an activity" do
- test "successfully processes incoming AP docs with correct origin" do
- params = %{
- "@context" => "https://www.w3.org/ns/activitystreams",
- "actor" => "http://mastodon.example.org/users/admin",
- "type" => "Create",
- "id" => "http://mastodon.example.org/users/admin/activities/1",
- "object" => %{
- "type" => "Note",
- "content" => "hi world!",
- "id" => "http://mastodon.example.org/users/admin/objects/1",
- "attributedTo" => "http://mastodon.example.org/users/admin"
- },
- "to" => ["https://www.w3.org/ns/activitystreams#Public"]
- }
-
- {:ok, _activity} = Federator.handle(:incoming_ap_doc, params)
- end
-
- test "rejects incoming AP docs with incorrect origin" do
- params = %{
- "@context" => "https://www.w3.org/ns/activitystreams",
- "actor" => "https://niu.moe/users/rye",
- "type" => "Create",
- "id" => "http://mastodon.example.org/users/admin/activities/1",
- "object" => %{
- "type" => "Note",
- "content" => "hi world!",
- "id" => "http://mastodon.example.org/users/admin/objects/1",
- "attributedTo" => "http://mastodon.example.org/users/admin"
- },
- "to" => ["https://www.w3.org/ns/activitystreams#Public"]
- }
-
- :error = Federator.handle(:incoming_ap_doc, params)
- end
- end
- end
|