Git of dcc - anni/blob - test/support/builders/activity_builder.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Builders.ActivityBuilder do
   6   alias Pleroma.Web.ActivityPub.ActivityPub
   7
   8   def build(data \\ %{}, opts \\ %{}) do
   9     user = opts[:user] || Pleroma.Factory.insert(:user)
  10
  11     activity = %{
  12       "id" => Pleroma.Web.ActivityPub.Utils.generate_object_id(),
  13       "actor" => user.ap_id,
  14       "to" => ["https://www.w3.org/ns/activitystreams#Public"],
  15       "type" => "Create",
  16       "object" => %{
  17         "type" => "Note",
  18         "content" => "test",
  19         "to" => ["https://www.w3.org/ns/activitystreams#Public"]
  20       }
  21     }
  22
  23     Map.merge(activity, data)
  24   end
  25
  26   def insert(data \\ %{}, opts \\ %{}) do
  27     activity = build(data, opts)
  28
  29     case ActivityPub.insert(activity) do
  30       ok = {:ok, activity} ->
  31         ActivityPub.notify_and_stream(activity)
  32         ok
  33
  34       error ->
  35         error
  36     end
  37   end
  38
  39   def insert_list(times, data \\ %{}, opts \\ %{}) do
  40     Enum.map(1..times, fn _n ->
  41       {:ok, activity} = insert(data, opts)
  42       activity
  43     end)
  44   end
  45
  46   def public_and_non_public do
  47     user = Pleroma.Factory.insert(:user)
  48
  49     public = build(%{"id" => 1}, %{user: user})
  50     non_public = build(%{"id" => 2, "to" => [user.follower_address]}, %{user: user})
  51
  52     {:ok, public} = ActivityPub.insert(public)
  53     {:ok, non_public} = ActivityPub.insert(non_public)
  54
  55     %{
  56       public: public,
  57       non_public: non_public,
  58       user: user
  59     }
  60   end
  61 end