sync with develop

[akkoma] / benchmarks / mix / tasks / pleroma / load_testing.ex

diff --git a/benchmarks/mix/tasks/pleroma/load_testing.ex b/benchmarks/mix/tasks/pleroma/load_testing.ex
index 2623009906892f4d58d49f35792fc7bc70a2d102..72b225f09e330f872c99eca21d90db4e2d5a2a28 100644 (file)
--- a/benchmarks/mix/tasks/pleroma/load_testing.ex
+++ b/benchmarks/mix/tasks/pleroma/load_testing.ex
@@ -1,8 +1,8 @@
 defmodule Mix.Tasks.Pleroma.LoadTesting do
   use Mix.Task
   import Ecto.Query
+  import Pleroma.LoadTesting.Helper, only: [clean_tables: 0]
 
-  alias Ecto.Adapters.SQL
   alias Pleroma.Repo
   alias Pleroma.User
 
@@ -63,12 +63,4 @@ defmodule Mix.Tasks.Pleroma.LoadTesting do
 
     Pleroma.LoadTesting.Fetcher.run_benchmarks(user)
   end
-
-  defp clean_tables do
-    IO.puts("Deleting old data...\n")
-    SQL.query!(Repo, "TRUNCATE users CASCADE;")
-    SQL.query!(Repo, "TRUNCATE activities CASCADE;")
-    SQL.query!(Repo, "TRUNCATE objects CASCADE;")
-    SQL.query!(Repo, "TRUNCATE oban_jobs CASCADE;")
-  end
 end