1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 defmodule Pleroma.Web.FedSockets.FedRegistryTest do
8 alias Pleroma.Web.FedSockets
9 alias Pleroma.Web.FedSockets.FedRegistry
10 alias Pleroma.Web.FedSockets.SocketInfo
12 @good_domain "http://good.domain"
13 @good_domain_origin "good.domain:80"
16 start_supervised({Pleroma.Web.FedSockets.Supervisor, []})
17 build_test_socket(@good_domain)
23 describe "add_fed_socket/1 without conflicting sockets" do
24 test "can be added" do
26 assert {:ok, %SocketInfo{origin: origin}} = FedRegistry.get_fed_socket(@good_domain_origin)
27 assert origin == "good.domain:80"
30 test "multiple origins can be added" do
31 build_test_socket("http://anothergood.domain")
34 assert {:ok, %SocketInfo{origin: origin_1}} =
35 FedRegistry.get_fed_socket(@good_domain_origin)
37 assert {:ok, %SocketInfo{origin: origin_2}} =
38 FedRegistry.get_fed_socket("anothergood.domain:80")
40 assert origin_1 == "good.domain:80"
41 assert origin_2 == "anothergood.domain:80"
42 assert FedRegistry.list_all() |> Enum.count() == 2
46 describe "add_fed_socket/1 when duplicate sockets conflict" do
48 build_test_socket(@good_domain)
49 build_test_socket(@good_domain)
54 test "will be ignored" do
55 assert {:ok, %SocketInfo{origin: origin, pid: pid_one}} =
56 FedRegistry.get_fed_socket(@good_domain_origin)
58 assert origin == "good.domain:80"
60 assert FedRegistry.list_all() |> Enum.count() == 1
63 test "the newer process will be closed" do
64 pid_two = build_test_socket(@good_domain)
66 assert {:ok, %SocketInfo{origin: origin, pid: pid_one}} =
67 FedRegistry.get_fed_socket(@good_domain_origin)
69 assert origin == "good.domain:80"
72 refute Process.alive?(pid_two)
74 assert FedRegistry.list_all() |> Enum.count() == 1
78 describe "get_fed_socket/1" do
79 test "returns missing for unknown hosts" do
80 assert {:error, :missing} = FedRegistry.get_fed_socket("not_a_dmoain")
83 test "returns rejected for hosts previously rejected" do
85 |> FedSockets.uri_for_origin()
86 |> FedRegistry.set_host_rejected()
88 assert {:error, :rejected} = FedRegistry.get_fed_socket("rejected.domain:80")
91 test "can retrieve a previously added SocketInfo" do
92 build_test_socket(@good_domain)
94 assert {:ok, %SocketInfo{origin: origin}} = FedRegistry.get_fed_socket(@good_domain_origin)
95 assert origin == "good.domain:80"
98 test "removes references to SocketInfos when the process crashes" do
99 assert {:ok, %SocketInfo{origin: origin, pid: pid}} =
100 FedRegistry.get_fed_socket(@good_domain_origin)
102 assert origin == "good.domain:80"
104 Process.exit(pid, :testing)
106 assert {:error, :missing} = FedRegistry.get_fed_socket(@good_domain_origin)
110 def build_test_socket(uri) do
111 Kernel.spawn(fn -> fed_socket_almost(uri) end)
114 def fed_socket_almost(origin) do
115 FedRegistry.add_fed_socket(origin)