Probably fixes the issue hj had, where polls would have different counters between endpoints.
		
			
				
	
	
		
			217 lines
		
	
	
	
		
			7.4 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			217 lines
		
	
	
	
		
			7.4 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
# Pleroma: A lightweight social networking server
 | 
						|
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
 | 
						|
# SPDX-License-Identifier: AGPL-3.0-only
 | 
						|
 | 
						|
defmodule Pleroma.ObjectTest do
 | 
						|
  use Pleroma.DataCase
 | 
						|
  import ExUnit.CaptureLog
 | 
						|
  import Pleroma.Factory
 | 
						|
  import Tesla.Mock
 | 
						|
  alias Pleroma.Activity
 | 
						|
  alias Pleroma.Object
 | 
						|
  alias Pleroma.Repo
 | 
						|
  alias Pleroma.Web.CommonAPI
 | 
						|
 | 
						|
  setup do
 | 
						|
    mock(fn env -> apply(HttpRequestMock, :request, [env]) end)
 | 
						|
    :ok
 | 
						|
  end
 | 
						|
 | 
						|
  test "returns an object by it's AP id" do
 | 
						|
    object = insert(:note)
 | 
						|
    found_object = Object.get_by_ap_id(object.data["id"])
 | 
						|
 | 
						|
    assert object == found_object
 | 
						|
  end
 | 
						|
 | 
						|
  describe "generic changeset" do
 | 
						|
    test "it ensures uniqueness of the id" do
 | 
						|
      object = insert(:note)
 | 
						|
      cs = Object.change(%Object{}, %{data: %{id: object.data["id"]}})
 | 
						|
      assert cs.valid?
 | 
						|
 | 
						|
      {:error, _result} = Repo.insert(cs)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "deletion function" do
 | 
						|
    test "deletes an object" do
 | 
						|
      object = insert(:note)
 | 
						|
      found_object = Object.get_by_ap_id(object.data["id"])
 | 
						|
 | 
						|
      assert object == found_object
 | 
						|
 | 
						|
      Object.delete(found_object)
 | 
						|
 | 
						|
      found_object = Object.get_by_ap_id(object.data["id"])
 | 
						|
 | 
						|
      refute object == found_object
 | 
						|
 | 
						|
      assert found_object.data["type"] == "Tombstone"
 | 
						|
    end
 | 
						|
 | 
						|
    test "ensures cache is cleared for the object" do
 | 
						|
      object = insert(:note)
 | 
						|
      cached_object = Object.get_cached_by_ap_id(object.data["id"])
 | 
						|
 | 
						|
      assert object == cached_object
 | 
						|
 | 
						|
      Cachex.put(:web_resp_cache, URI.parse(object.data["id"]).path, "cofe")
 | 
						|
 | 
						|
      Object.delete(cached_object)
 | 
						|
 | 
						|
      {:ok, nil} = Cachex.get(:object_cache, "object:#{object.data["id"]}")
 | 
						|
      {:ok, nil} = Cachex.get(:web_resp_cache, URI.parse(object.data["id"]).path)
 | 
						|
 | 
						|
      cached_object = Object.get_cached_by_ap_id(object.data["id"])
 | 
						|
 | 
						|
      refute object == cached_object
 | 
						|
 | 
						|
      assert cached_object.data["type"] == "Tombstone"
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "normalizer" do
 | 
						|
    test "fetches unknown objects by default" do
 | 
						|
      %Object{} =
 | 
						|
        object = Object.normalize("http://mastodon.example.org/@admin/99541947525187367")
 | 
						|
 | 
						|
      assert object.data["url"] == "http://mastodon.example.org/@admin/99541947525187367"
 | 
						|
    end
 | 
						|
 | 
						|
    test "fetches unknown objects when fetch_remote is explicitly true" do
 | 
						|
      %Object{} =
 | 
						|
        object = Object.normalize("http://mastodon.example.org/@admin/99541947525187367", true)
 | 
						|
 | 
						|
      assert object.data["url"] == "http://mastodon.example.org/@admin/99541947525187367"
 | 
						|
    end
 | 
						|
 | 
						|
    test "does not fetch unknown objects when fetch_remote is false" do
 | 
						|
      assert is_nil(
 | 
						|
               Object.normalize("http://mastodon.example.org/@admin/99541947525187367", false)
 | 
						|
             )
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe "get_by_id_and_maybe_refetch" do
 | 
						|
    setup do
 | 
						|
      mock(fn
 | 
						|
        %{method: :get, url: "https://patch.cx/objects/9a172665-2bc5-452d-8428-2361d4c33b1d"} ->
 | 
						|
          %Tesla.Env{status: 200, body: File.read!("test/fixtures/tesla_mock/poll_original.json")}
 | 
						|
 | 
						|
        env ->
 | 
						|
          apply(HttpRequestMock, :request, [env])
 | 
						|
      end)
 | 
						|
 | 
						|
      mock_modified = fn resp ->
 | 
						|
        mock(fn
 | 
						|
          %{method: :get, url: "https://patch.cx/objects/9a172665-2bc5-452d-8428-2361d4c33b1d"} ->
 | 
						|
            resp
 | 
						|
 | 
						|
          env ->
 | 
						|
            apply(HttpRequestMock, :request, [env])
 | 
						|
        end)
 | 
						|
      end
 | 
						|
 | 
						|
      on_exit(fn -> mock(fn env -> apply(HttpRequestMock, :request, [env]) end) end)
 | 
						|
 | 
						|
      [mock_modified: mock_modified]
 | 
						|
    end
 | 
						|
 | 
						|
    test "refetches if the time since the last refetch is greater than the interval", %{
 | 
						|
      mock_modified: mock_modified
 | 
						|
    } do
 | 
						|
      %Object{} =
 | 
						|
        object = Object.normalize("https://patch.cx/objects/9a172665-2bc5-452d-8428-2361d4c33b1d")
 | 
						|
 | 
						|
      Object.set_cache(object)
 | 
						|
 | 
						|
      assert Enum.at(object.data["oneOf"], 0)["replies"]["totalItems"] == 4
 | 
						|
      assert Enum.at(object.data["oneOf"], 1)["replies"]["totalItems"] == 0
 | 
						|
 | 
						|
      mock_modified.(%Tesla.Env{
 | 
						|
        status: 200,
 | 
						|
        body: File.read!("test/fixtures/tesla_mock/poll_modified.json")
 | 
						|
      })
 | 
						|
 | 
						|
      updated_object = Object.get_by_id_and_maybe_refetch(object.id, interval: -1)
 | 
						|
      object_in_cache = Object.get_cached_by_ap_id(object.data["id"])
 | 
						|
      assert updated_object == object_in_cache
 | 
						|
      assert Enum.at(updated_object.data["oneOf"], 0)["replies"]["totalItems"] == 8
 | 
						|
      assert Enum.at(updated_object.data["oneOf"], 1)["replies"]["totalItems"] == 3
 | 
						|
    end
 | 
						|
 | 
						|
    test "returns the old object if refetch fails", %{mock_modified: mock_modified} do
 | 
						|
      %Object{} =
 | 
						|
        object = Object.normalize("https://patch.cx/objects/9a172665-2bc5-452d-8428-2361d4c33b1d")
 | 
						|
 | 
						|
      Object.set_cache(object)
 | 
						|
 | 
						|
      assert Enum.at(object.data["oneOf"], 0)["replies"]["totalItems"] == 4
 | 
						|
      assert Enum.at(object.data["oneOf"], 1)["replies"]["totalItems"] == 0
 | 
						|
 | 
						|
      assert capture_log(fn ->
 | 
						|
               mock_modified.(%Tesla.Env{status: 404, body: ""})
 | 
						|
 | 
						|
               updated_object = Object.get_by_id_and_maybe_refetch(object.id, interval: -1)
 | 
						|
               object_in_cache = Object.get_cached_by_ap_id(object.data["id"])
 | 
						|
               assert updated_object == object_in_cache
 | 
						|
               assert Enum.at(updated_object.data["oneOf"], 0)["replies"]["totalItems"] == 4
 | 
						|
               assert Enum.at(updated_object.data["oneOf"], 1)["replies"]["totalItems"] == 0
 | 
						|
             end) =~
 | 
						|
               "[error] Couldn't refresh https://patch.cx/objects/9a172665-2bc5-452d-8428-2361d4c33b1d"
 | 
						|
    end
 | 
						|
 | 
						|
    test "does not refetch if the time since the last refetch is greater than the interval", %{
 | 
						|
      mock_modified: mock_modified
 | 
						|
    } do
 | 
						|
      %Object{} =
 | 
						|
        object = Object.normalize("https://patch.cx/objects/9a172665-2bc5-452d-8428-2361d4c33b1d")
 | 
						|
 | 
						|
      Object.set_cache(object)
 | 
						|
 | 
						|
      assert Enum.at(object.data["oneOf"], 0)["replies"]["totalItems"] == 4
 | 
						|
      assert Enum.at(object.data["oneOf"], 1)["replies"]["totalItems"] == 0
 | 
						|
 | 
						|
      mock_modified.(%Tesla.Env{
 | 
						|
        status: 200,
 | 
						|
        body: File.read!("test/fixtures/tesla_mock/poll_modified.json")
 | 
						|
      })
 | 
						|
 | 
						|
      updated_object = Object.get_by_id_and_maybe_refetch(object.id, interval: 100)
 | 
						|
      object_in_cache = Object.get_cached_by_ap_id(object.data["id"])
 | 
						|
      assert updated_object == object_in_cache
 | 
						|
      assert Enum.at(updated_object.data["oneOf"], 0)["replies"]["totalItems"] == 4
 | 
						|
      assert Enum.at(updated_object.data["oneOf"], 1)["replies"]["totalItems"] == 0
 | 
						|
    end
 | 
						|
 | 
						|
    test "preserves internal fields on refetch", %{mock_modified: mock_modified} do
 | 
						|
      %Object{} =
 | 
						|
        object = Object.normalize("https://patch.cx/objects/9a172665-2bc5-452d-8428-2361d4c33b1d")
 | 
						|
 | 
						|
      Object.set_cache(object)
 | 
						|
 | 
						|
      assert Enum.at(object.data["oneOf"], 0)["replies"]["totalItems"] == 4
 | 
						|
      assert Enum.at(object.data["oneOf"], 1)["replies"]["totalItems"] == 0
 | 
						|
 | 
						|
      user = insert(:user)
 | 
						|
      activity = Activity.get_create_by_object_ap_id(object.data["id"])
 | 
						|
      {:ok, _activity, object} = CommonAPI.favorite(activity.id, user)
 | 
						|
 | 
						|
      assert object.data["like_count"] == 1
 | 
						|
 | 
						|
      mock_modified.(%Tesla.Env{
 | 
						|
        status: 200,
 | 
						|
        body: File.read!("test/fixtures/tesla_mock/poll_modified.json")
 | 
						|
      })
 | 
						|
 | 
						|
      updated_object = Object.get_by_id_and_maybe_refetch(object.id, interval: -1)
 | 
						|
      object_in_cache = Object.get_cached_by_ap_id(object.data["id"])
 | 
						|
      assert updated_object == object_in_cache
 | 
						|
      assert Enum.at(updated_object.data["oneOf"], 0)["replies"]["totalItems"] == 8
 | 
						|
      assert Enum.at(updated_object.data["oneOf"], 1)["replies"]["totalItems"] == 3
 | 
						|
 | 
						|
      assert updated_object.data["like_count"] == 1
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 |