git.squeep.com Git - akkoma/blob - test/pleroma/web/activity_pub/transmogrifier/add_remove_handling_test.exs

   1 defmodule Pleroma.Web.ActivityPub.Transmogrifier.AddRemoveHandlingTest do
   2   use Oban.Testing, repo: Pleroma.Repo
   3   use Pleroma.DataCase, async: true
   4
   5   require Pleroma.Constants
   6
   7   import Pleroma.Factory
   8
   9   alias Pleroma.User
  10   alias Pleroma.Web.ActivityPub.Transmogrifier
  11
  12   test "it accepts Add/Remove activities" do
  13     user =
  14       "test/fixtures/users_mock/user.json"
  15       |> File.read!()
  16       |> String.replace("{{nickname}}", "lain")
  17
  18     object_id = "c61d6733-e256-4fe1-ab13-1e369789423f"
  19
  20     object =
  21       "test/fixtures/statuses/note.json"
  22       |> File.read!()
  23       |> String.replace("{{nickname}}", "lain")
  24       |> String.replace("{{object_id}}", object_id)
  25
  26     object_url = "https://example.com/objects/#{object_id}"
  27
  28     actor = "https://example.com/users/lain"
  29
  30     Tesla.Mock.mock(fn
  31       %{
  32         method: :get,
  33         url: ^actor
  34       } ->
  35         %Tesla.Env{
  36           status: 200,
  37           body: user,
  38           headers: [{"content-type", "application/activity+json"}]
  39         }
  40
  41       %{
  42         method: :get,
  43         url: ^object_url
  44       } ->
  45         %Tesla.Env{
  46           status: 200,
  47           body: object,
  48           headers: [{"content-type", "application/activity+json"}]
  49         }
  50
  51       %{method: :get, url: "https://example.com/users/lain/collections/featured"} ->
  52         %Tesla.Env{
  53           status: 200,
  54           body:
  55             "test/fixtures/users_mock/masto_featured.json"
  56             |> File.read!()
  57             |> String.replace("{{domain}}", "example.com")
  58             |> String.replace("{{nickname}}", "lain"),
  59           headers: [{"content-type", "application/activity+json"}]
  60         }
  61     end)
  62
  63     message = %{
  64       "id" => "https://example.com/objects/d61d6733-e256-4fe1-ab13-1e369789423f",
  65       "actor" => actor,
  66       "object" => object_url,
  67       "target" => "https://example.com/users/lain/collections/featured",
  68       "type" => "Add",
  69       "to" => [Pleroma.Constants.as_public()],
  70       "cc" => ["https://example.com/users/lain/followers"]
  71     }
  72
  73     assert {:ok, activity} = Transmogrifier.handle_incoming(message)
  74     assert activity.data == message
  75     user = User.get_cached_by_ap_id(actor)
  76     assert user.pinned_objects[object_url]
  77
  78     remove = %{
  79       "id" => "http://localhost:400/objects/d61d6733-e256-4fe1-ab13-1e369789423d",
  80       "actor" => actor,
  81       "object" => object_url,
  82       "target" => "https://example.com/users/lain/collections/featured",
  83       "type" => "Remove",
  84       "to" => [Pleroma.Constants.as_public()],
  85       "cc" => ["https://example.com/users/lain/followers"]
  86     }
  87
  88     assert {:ok, activity} = Transmogrifier.handle_incoming(remove)
  89     assert activity.data == remove
  90
  91     user = refresh_record(user)
  92     refute user.pinned_objects[object_url]
  93   end
  94
  95   test "Add/Remove activities for remote users without featured address" do
  96     user = insert(:user, local: false, domain: "example.com")
  97
  98     user =
  99       user
 100       |> Ecto.Changeset.change(featured_address: nil)
 101       |> Repo.update!()
 102
 103     %{host: host} = URI.parse(user.ap_id)
 104
 105     user_data =
 106       "test/fixtures/users_mock/user.json"
 107       |> File.read!()
 108       |> String.replace("{{nickname}}", user.nickname)
 109
 110     object_id = "c61d6733-e256-4fe1-ab13-1e369789423f"
 111
 112     object =
 113       "test/fixtures/statuses/note.json"
 114       |> File.read!()
 115       |> String.replace("{{nickname}}", user.nickname)
 116       |> String.replace("{{object_id}}", object_id)
 117
 118     object_url = "https://#{host}/objects/#{object_id}"
 119
 120     actor = "https://#{host}/users/#{user.nickname}"
 121
 122     featured = "https://#{host}/users/#{user.nickname}/collections/featured"
 123
 124     Tesla.Mock.mock(fn
 125       %{
 126         method: :get,
 127         url: ^actor
 128       } ->
 129         %Tesla.Env{
 130           status: 200,
 131           body: user_data,
 132           headers: [{"content-type", "application/activity+json"}]
 133         }
 134
 135       %{
 136         method: :get,
 137         url: ^object_url
 138       } ->
 139         %Tesla.Env{
 140           status: 200,
 141           body: object,
 142           headers: [{"content-type", "application/activity+json"}]
 143         }
 144
 145       %{method: :get, url: ^featured} ->
 146         %Tesla.Env{
 147           status: 200,
 148           body:
 149             "test/fixtures/users_mock/masto_featured.json"
 150             |> File.read!()
 151             |> String.replace("{{domain}}", "#{host}")
 152             |> String.replace("{{nickname}}", user.nickname),
 153           headers: [{"content-type", "application/activity+json"}]
 154         }
 155     end)
 156
 157     message = %{
 158       "id" => "https://#{host}/objects/d61d6733-e256-4fe1-ab13-1e369789423f",
 159       "actor" => actor,
 160       "object" => object_url,
 161       "target" => "https://#{host}/users/#{user.nickname}/collections/featured",
 162       "type" => "Add",
 163       "to" => [Pleroma.Constants.as_public()],
 164       "cc" => ["https://#{host}/users/#{user.nickname}/followers"]
 165     }
 166
 167     assert {:ok, activity} = Transmogrifier.handle_incoming(message)
 168     assert activity.data == message
 169     user = User.get_cached_by_ap_id(actor)
 170     assert user.pinned_objects[object_url]
 171   end
 172 end