X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=benchmarks%2Fload_testing%2Fusers.ex;h=1a8c6e22f5b2a02ff7f0a4eeaaf1242305d786a7;hb=7f692343c80ddf353712490edfbcdb14866f5685;hp=b73ac8651017b63224afd483b0e8071cf62358d3;hpb=56503c385e8412a1189748bcf3fdfd4090be9f56;p=akkoma

diff --git a/benchmarks/load_testing/users.ex b/benchmarks/load_testing/users.ex
index b73ac8651..1a8c6e22f 100644
--- a/benchmarks/load_testing/users.ex
+++ b/benchmarks/load_testing/users.ex
@@ -33,7 +33,7 @@ defmodule Pleroma.LoadTesting.Users do
   def generate_users(max) do
     IO.puts("Starting generating #{max} users...")
 
-    {time, _} =
+    {time, users} =
       :timer.tc(fn ->
         Task.async_stream(
           1..max,
@@ -41,10 +41,11 @@ defmodule Pleroma.LoadTesting.Users do
           max_concurrency: @max_concurrency,
           timeout: 30_000
         )
-        |> Stream.run()
+        |> Enum.to_list()
       end)
 
-    IO.puts("Generating users take #{to_sec(time)} sec.\n")
+    IO.puts("Generating users took #{to_sec(time)} sec.\n")
+    users
   end
 
   defp generate_user(i) do
@@ -104,7 +105,7 @@ defmodule Pleroma.LoadTesting.Users do
         |> run_stream(main_user)
       end)
 
-    IO.puts("Making friends take #{to_sec(time)} sec.\n")
+    IO.puts("Making friends took #{to_sec(time)} sec.\n")
   end
 
   def make_friends(%User{} = main_user, %User{} = user) do