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

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