Pleroma/test/support/builders/activity_builder.ex

49 lines
1.3 KiB
Elixir
Raw Normal View History

2017-03-21 17:53:20 +01:00
defmodule Pleroma.Builders.ActivityBuilder do
alias Pleroma.Builders.UserBuilder
alias Pleroma.Web.ActivityPub.ActivityPub
2017-03-21 20:22:05 +01:00
def build(data \\ %{}, opts \\ %{}) do
2017-04-16 15:28:28 +02:00
user = opts[:user] || Pleroma.Factory.insert(:user)
2017-03-21 20:22:05 +01:00
activity = %{
2017-05-16 15:31:11 +02:00
"id" => Pleroma.Web.ActivityPub.Utils.generate_object_id,
2017-03-21 17:53:20 +01:00
"actor" => user.ap_id,
"to" => ["https://www.w3.org/ns/activitystreams#Public"],
2018-02-25 17:48:31 +01:00
"type" => "Create",
2017-03-21 17:53:20 +01:00
"object" => %{
"type" => "Note",
2018-02-25 17:48:31 +01:00
"content" => "test",
"to" => ["https://www.w3.org/ns/activitystreams#Public"],
2017-03-21 17:53:20 +01:00
}
}
2017-03-21 20:22:05 +01:00
Map.merge(activity, data)
end
2017-03-21 17:53:20 +01:00
2017-03-21 20:22:05 +01:00
def insert(data \\ %{}, opts \\ %{}) do
activity = build(data, opts)
ActivityPub.insert(activity)
end
def insert_list(times, data \\ %{}, opts \\ %{}) do
Enum.map(1..times, fn (n) ->
2018-02-25 17:48:31 +01:00
{:ok, activity} = insert(data, opts)
2017-03-21 20:22:05 +01:00
activity
end)
end
def public_and_non_public do
2017-04-16 15:28:28 +02:00
user = Pleroma.Factory.insert(:user)
2017-03-21 20:22:05 +01:00
public = build(%{"id" => 1}, %{user: user})
2018-02-25 17:48:31 +01:00
non_public = build(%{"id" => 2, "to" => [user.follower_address]}, %{user: user})
2017-03-21 17:53:20 +01:00
{:ok, public} = ActivityPub.insert(public)
{:ok, non_public} = ActivityPub.insert(non_public)
%{
public: public,
non_public: non_public,
user: user
}
end
end