1 defmodule Pleroma.LoadTesting.Generator do
2 use Pleroma.LoadTesting.Helper
3 alias Pleroma.Web.CommonAPI
5 def generate_users(opts) do
6 IO.puts("Starting generating #{opts[:users_max]} users...")
7 {time, _} = :timer.tc(fn -> do_generate_users(opts) end)
9 IO.puts("Inserting users take #{to_sec(time)} sec.\n")
12 defp do_generate_users(opts) do
13 max = Keyword.get(opts, :users_max)
17 &generate_user_data(&1),
24 defp generate_user_data(i) do
25 remote = Enum.random([true, false])
28 name: "Test ใในใ User #{i}",
29 email: "user#{i}@example.com",
32 "$pbkdf2-sha512$160000$bU.OSFI7H/yqWb5DPEqyjw$uKp/2rmXw12QqnRRTqTtuk2DTwZfF8VR4MYW2xMeIlqPR/UX1nT1CEKVUx2CowFMZ5JON8aDvURrZpJjSgqXrg",
33 bio: "Tester Number #{i}",
41 Enum.random(["https://domain1.com", "https://domain2.com", "https://domain3.com"])
43 ap_id = "#{base_url}/users/#{user.nickname}"
47 follower_address: ap_id <> "/followers",
48 following_address: ap_id <> "/following",
53 ap_id: User.ap_id(user),
54 follower_address: User.ap_followers(user),
55 following_address: User.ap_following(user),
56 following: [User.ap_id(user)]
60 user = Map.merge(user, user_urls)
65 def generate_activities(user, users) do
66 do_generate_activities(user, users)
69 defp do_generate_activities(user, users) do
70 IO.puts("Starting generating 20000 common activities...")
77 do_generate_activity([user | users])
85 IO.puts("Inserting common activities take #{to_sec(time)} sec.\n")
87 IO.puts("Starting generating 20000 activities with mentions...")
94 do_generate_activity_with_mention(user, users)
102 IO.puts("Inserting activities with menthions take #{to_sec(time)} sec.\n")
104 IO.puts("Starting generating 10000 activities with threads...")
111 do_generate_threads([user | users])
119 IO.puts("Inserting activities with threads take #{to_sec(time)} sec.\n")
122 defp do_generate_activity(users) do
124 "status" => "Some status without mention with random user"
127 CommonAPI.post(Enum.random(users), post)
130 defp do_generate_activity_with_mention(user, users) do
131 mentions_cnt = Enum.random([2, 3, 4, 5])
132 with_user = Enum.random([true, false])
133 users = Enum.shuffle(users)
134 mentions_users = Enum.take(users, mentions_cnt)
135 mentions_users = if with_user, do: [user | mentions_users], else: mentions_users
138 Enum.map(mentions_users, fn user -> "@" <> user.nickname end) |> Enum.join(", ")
141 "status" => mentions_str <> "some status with mentions random users"
144 CommonAPI.post(Enum.random(users), post)
147 defp do_generate_threads(users) do
148 thread_length = Enum.random([2, 3, 4, 5])
149 actor = Enum.random(users)
152 "status" => "Start of the thread"
155 {:ok, activity} = CommonAPI.post(actor, post)
157 Enum.each(1..thread_length, fn _ ->
158 user = Enum.random(users)
161 "status" => "@#{actor.nickname} reply to thread",
162 "in_reply_to_status_id" => activity.id
165 CommonAPI.post(user, post)
169 def generate_remote_activities(user, users) do
170 do_generate_remote_activities(user, users)
173 defp do_generate_remote_activities(user, users) do
174 IO.puts("Starting generating 10000 remote activities...")
181 do_generate_remote_activity(i, user, users)
189 IO.puts("Inserting remote activities take #{to_sec(time)} sec.\n")
192 defp do_generate_remote_activity(i, user, users) do
193 actor = Enum.random(users)
194 %{host: host} = URI.parse(actor.ap_id)
195 date = Date.utc_today()
196 datetime = DateTime.utc_now()
199 "actor" => actor.ap_id,
200 "cc" => [actor.follower_address, user.ap_id],
201 "context" => "tag:mastodon.example.org,#{date}:objectId=#{i}:objectType=Conversation",
202 "id" => actor.ap_id <> "/statuses/#{i}/activity",
204 "actor" => actor.ap_id,
205 "atomUri" => actor.ap_id <> "/statuses/#{i}",
207 "attributedTo" => actor.ap_id,
210 "cc" => [actor.follower_address, user.ap_id],
212 "<p><span class=\"h-card\"><a href=\"" <>
214 "\" class=\"u-url mention\">@<span>" <> user.nickname <> "</span></a></span></p>",
215 "context" => "tag:mastodon.example.org,#{date}:objectId=#{i}:objectType=Conversation",
217 "tag:mastodon.example.org,#{date}:objectId=#{i}:objectType=Conversation",
219 "id" => actor.ap_id <> "/statuses/#{i}",
221 "inReplyToAtomUri" => nil,
222 "published" => datetime,
227 "href" => user.ap_id,
228 "name" => "@#{user.nickname}@#{host}",
232 "to" => ["https://www.w3.org/ns/activitystreams#Public"],
234 "url" => "http://#{host}/@#{actor.nickname}/#{i}"
236 "published" => datetime,
237 "to" => ["https://www.w3.org/ns/activitystreams#Public"],
241 Pleroma.Web.ActivityPub.ActivityPub.insert(map, false)
244 def generate_dms(user, users, opts) do
245 IO.puts("Starting generating #{opts[:dms_max]} DMs")
246 {time, _} = :timer.tc(fn -> do_generate_dms(user, users, opts) end)
247 IO.puts("Inserting dms take #{to_sec(time)} sec.\n")
250 defp do_generate_dms(user, users, opts) do
254 do_generate_dm(user, users)
262 defp do_generate_dm(user, users) do
264 "status" => "@#{user.nickname} some direct message",
265 "visibility" => "direct"
268 CommonAPI.post(Enum.random(users), post)
271 def generate_long_thread(user, users, opts) do
272 IO.puts("Starting generating long thread with #{opts[:thread_length]} replies")
273 {time, activity} = :timer.tc(fn -> do_generate_long_thread(user, users, opts) end)
274 IO.puts("Inserting long thread replies take #{to_sec(time)} sec.\n")
278 defp do_generate_long_thread(user, users, opts) do
279 {:ok, %{id: id} = activity} = CommonAPI.post(user, %{"status" => "Start of long thread"})
282 1..opts[:thread_length],
283 fn _ -> do_generate_thread(users, id) end,
292 defp do_generate_thread(users, activity_id) do
293 CommonAPI.post(Enum.random(users), %{
294 "status" => "reply to main post",
295 "in_reply_to_status_id" => activity_id
299 def generate_non_visible_message(user, users) do
300 IO.puts("Starting generating 1000 non visible posts")
304 do_generate_non_visible_posts(user, users)
307 IO.puts("Inserting non visible posts take #{to_sec(time)} sec.\n")
310 defp do_generate_non_visible_posts(user, users) do
311 [not_friend | users] = users
313 make_friends(user, users)
315 Task.async_stream(1..1000, fn _ -> do_generate_non_visible_post(not_friend, users) end,
322 defp make_friends(_user, []), do: nil
324 defp make_friends(user, [friend | users]) do
325 {:ok, _} = User.follow(user, friend)
326 {:ok, _} = User.follow(friend, user)
327 make_friends(user, users)
330 defp do_generate_non_visible_post(not_friend, users) do
332 "status" => "some non visible post",
333 "visibility" => "private"
336 {:ok, activity} = CommonAPI.post(not_friend, post)
338 thread_length = Enum.random([2, 3, 4, 5])
340 Enum.each(1..thread_length, fn _ ->
341 user = Enum.random(users)
344 "status" => "@#{not_friend.nickname} reply to non visible post",
345 "in_reply_to_status_id" => activity.id,
346 "visibility" => "private"
349 CommonAPI.post(user, post)