git.squeep.com Git - akkoma/blob - test/support/builders/activity_builder.ex

   1 defmodule Pleroma.Builders.ActivityBuilder do
   2   alias Pleroma.Web.ActivityPub.ActivityPub
   3
   4   def build(data \\ %{}, opts \\ %{}) do
   5     user = opts[:user] || Pleroma.Factory.insert(:user)
   6
   7     activity = %{
   8       "id" => Pleroma.Web.ActivityPub.Utils.generate_object_id(),
   9       "actor" => user.ap_id,
  10       "to" => ["https://www.w3.org/ns/activitystreams#Public"],
  11       "type" => "Create",
  12       "object" => %{
  13         "type" => "Note",
  14         "content" => "test",
  15         "to" => ["https://www.w3.org/ns/activitystreams#Public"]
  16       }
  17     }
  18
  19     Map.merge(activity, data)
  20   end
  21
  22   def insert(data \\ %{}, opts \\ %{}) do
  23     activity = build(data, opts)
  24     ActivityPub.insert(activity)
  25   end
  26
  27   def insert_list(times, data \\ %{}, opts \\ %{}) do
  28     Enum.map(1..times, fn _n ->
  29       {:ok, activity} = insert(data, opts)
  30       activity
  31     end)
  32   end
  33
  34   def public_and_non_public do
  35     user = Pleroma.Factory.insert(:user)
  36
  37     public = build(%{"id" => 1}, %{user: user})
  38     non_public = build(%{"id" => 2, "to" => [user.follower_address]}, %{user: user})
  39
  40     {:ok, public} = ActivityPub.insert(public)
  41     {:ok, non_public} = ActivityPub.insert(non_public)
  42
  43     %{
  44       public: public,
  45       non_public: non_public,
  46       user: user
  47     }
  48   end
  49 end