107 lines
3.2 KiB
Elixir
107 lines
3.2 KiB
Elixir
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
|