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     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     Map.merge(activity, data)
  19   end
  20
  21   def insert(data \\ %{}, opts \\ %{}) do
  22     activity = build(data, opts)
  23     ActivityPub.insert(activity)
  24   end
  25
  26   def insert_list(times, data \\ %{}, opts \\ %{}) do
  27     Enum.map(1..times, fn (n) ->
  28       {:ok, activity} = insert(data, opts)
  29       activity
  30     end)
  31   end
  32
  33   def public_and_non_public do
  34     user = Pleroma.Factory.insert(:user)
  35
  36     public = build(%{"id" => 1}, %{user: user})
  37     non_public = build(%{"id" => 2, "to" => [user.follower_address]}, %{user: user})
  38
  39     {:ok, public} = ActivityPub.insert(public)
  40     {:ok, non_public} = ActivityPub.insert(non_public)
  41
  42     %{
  43       public: public,
  44       non_public: non_public,
  45       user: user
  46     }
  47   end
  48 end