| 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
 | # 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.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
  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
    clear_config([:instances_favicons, :enabled], true)
    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
end
 |