162 lines
6 KiB
Elixir
162 lines
6 KiB
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.Instances.InstanceTest do
|
|
alias Pleroma.Instances
|
|
alias Pleroma.Instances.Instance
|
|
alias Pleroma.Repo
|
|
|
|
use Pleroma.DataCase
|
|
|
|
import ExUnit.CaptureLog
|
|
import Pleroma.Factory
|
|
|
|
setup_all do: clear_config([:instance, :federation_reachability_timeout_days], 1)
|
|
|
|
describe "set_reachable/1" do
|
|
test "clears `unreachable_since` of existing matching Instance record having non-nil `unreachable_since`" do
|
|
unreachable_since = NaiveDateTime.to_iso8601(NaiveDateTime.utc_now())
|
|
instance = insert(:instance, unreachable_since: unreachable_since)
|
|
|
|
assert {:ok, instance} = Instance.set_reachable(instance.host)
|
|
refute instance.unreachable_since
|
|
end
|
|
|
|
test "keeps nil `unreachable_since` of existing matching Instance record having nil `unreachable_since`" do
|
|
instance = insert(:instance, unreachable_since: nil)
|
|
|
|
assert {:ok, instance} = Instance.set_reachable(instance.host)
|
|
refute instance.unreachable_since
|
|
end
|
|
|
|
test "does NOT create an Instance record in case of no existing matching record" do
|
|
host = "domain.org"
|
|
assert nil == Instance.set_reachable(host)
|
|
|
|
assert [] = Repo.all(Ecto.Query.from(i in Instance))
|
|
assert Instance.reachable?(host)
|
|
end
|
|
end
|
|
|
|
describe "set_unreachable/1" do
|
|
test "creates new record having `unreachable_since` to current time if record does not exist" do
|
|
assert {:ok, instance} = Instance.set_unreachable("https://domain.com/path")
|
|
|
|
instance = Repo.get(Instance, instance.id)
|
|
assert instance.unreachable_since
|
|
assert "domain.com" == instance.host
|
|
end
|
|
|
|
test "sets `unreachable_since` of existing record having nil `unreachable_since`" do
|
|
instance = insert(:instance, unreachable_since: nil)
|
|
refute instance.unreachable_since
|
|
|
|
assert {:ok, _} = Instance.set_unreachable(instance.host)
|
|
|
|
instance = Repo.get(Instance, instance.id)
|
|
assert instance.unreachable_since
|
|
end
|
|
|
|
test "does NOT modify `unreachable_since` value of existing record in case it's present" do
|
|
instance =
|
|
insert(:instance, unreachable_since: NaiveDateTime.add(NaiveDateTime.utc_now(), -10))
|
|
|
|
assert instance.unreachable_since
|
|
initial_value = instance.unreachable_since
|
|
|
|
assert {:ok, _} = Instance.set_unreachable(instance.host)
|
|
|
|
instance = Repo.get(Instance, instance.id)
|
|
assert initial_value == instance.unreachable_since
|
|
end
|
|
end
|
|
|
|
describe "set_unreachable/2" do
|
|
test "sets `unreachable_since` value of existing record in case it's newer than supplied value" do
|
|
instance =
|
|
insert(:instance, unreachable_since: NaiveDateTime.add(NaiveDateTime.utc_now(), -10))
|
|
|
|
assert instance.unreachable_since
|
|
|
|
past_value = NaiveDateTime.add(NaiveDateTime.utc_now(), -100)
|
|
assert {:ok, _} = Instance.set_unreachable(instance.host, past_value)
|
|
|
|
instance = Repo.get(Instance, instance.id)
|
|
assert past_value == instance.unreachable_since
|
|
end
|
|
|
|
test "does NOT modify `unreachable_since` value of existing record in case it's equal to or older than supplied value" do
|
|
instance =
|
|
insert(:instance, unreachable_since: NaiveDateTime.add(NaiveDateTime.utc_now(), -10))
|
|
|
|
assert instance.unreachable_since
|
|
initial_value = instance.unreachable_since
|
|
|
|
assert {:ok, _} = Instance.set_unreachable(instance.host, NaiveDateTime.utc_now())
|
|
|
|
instance = Repo.get(Instance, instance.id)
|
|
assert initial_value == instance.unreachable_since
|
|
end
|
|
end
|
|
|
|
describe "get_or_update_favicon/1" do
|
|
test "Scrapes favicon URLs" do
|
|
Tesla.Mock.mock(fn %{url: "https://favicon.example.org/"} ->
|
|
%Tesla.Env{
|
|
status: 200,
|
|
body: ~s[<html><head><link rel="icon" href="/favicon.png"></head></html>]
|
|
}
|
|
end)
|
|
|
|
assert "https://favicon.example.org/favicon.png" ==
|
|
Instance.get_or_update_favicon(URI.parse("https://favicon.example.org/"))
|
|
end
|
|
|
|
test "Returns nil on too long favicon URLs" do
|
|
long_favicon_url =
|
|
"https://Lorem.ipsum.dolor.sit.amet/consecteturadipiscingelit/Praesentpharetrapurusutaliquamtempus/Mauriseulaoreetarcu/atfacilisisorci/Nullamporttitor/nequesedfeugiatmollis/dolormagnaefficiturlorem/nonpretiumsapienorcieurisus/Nullamveleratsem/Maecenassedaccumsanexnam/favicon.png"
|
|
|
|
Tesla.Mock.mock(fn %{url: "https://long-favicon.example.org/"} ->
|
|
%Tesla.Env{
|
|
status: 200,
|
|
body:
|
|
~s[<html><head><link rel="icon" href="] <> long_favicon_url <> ~s["></head></html>]
|
|
}
|
|
end)
|
|
|
|
assert capture_log(fn ->
|
|
assert nil ==
|
|
Instance.get_or_update_favicon(
|
|
URI.parse("https://long-favicon.example.org/")
|
|
)
|
|
end) =~
|
|
"Instance.get_or_update_favicon(\"long-favicon.example.org\") error: %Postgrex.Error{"
|
|
end
|
|
|
|
test "Handles not getting a favicon URL properly" do
|
|
Tesla.Mock.mock(fn %{url: "https://no-favicon.example.org/"} ->
|
|
%Tesla.Env{
|
|
status: 200,
|
|
body: ~s[<html><head><h1>I wil look down and whisper "GNO.."</h1></head></html>]
|
|
}
|
|
end)
|
|
|
|
refute capture_log(fn ->
|
|
assert nil ==
|
|
Instance.get_or_update_favicon(
|
|
URI.parse("https://no-favicon.example.org/")
|
|
)
|
|
end) =~ "Instance.scrape_favicon(\"https://no-favicon.example.org/\") error: "
|
|
end
|
|
|
|
test "Doesn't scrapes unreachable instances" do
|
|
instance = insert(:instance, unreachable_since: Instances.reachability_datetime_threshold())
|
|
url = "https://" <> instance.host
|
|
|
|
assert capture_log(fn -> assert nil == Instance.get_or_update_favicon(URI.parse(url)) end) =~
|
|
"Instance.scrape_favicon(\"#{url}\") ignored unreachable host"
|
|
end
|
|
end
|
|
end
|