135 lines
		
	
	
	
		
			4.9 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			135 lines
		
	
	
	
		
			4.9 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.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
 | 
