X-Git-Url: https://git.squeep.com/?a=blobdiff_plain;f=benchmarks%2Fload_testing%2Fhelper.ex;h=cab60acb422e7472b277b51f8ac678720e1b184d;hb=f2134e605b44c300505f81adab01354a1652b2e5;hp=47b25c65fb7fb499e4b8eb2073779473a3b4a412;hpb=2473702be22a44070fcff439ac901f5b9bb0586a;p=akkoma

diff --git a/benchmarks/load_testing/helper.ex b/benchmarks/load_testing/helper.ex
index 47b25c65f..cab60acb4 100644
--- a/benchmarks/load_testing/helper.ex
+++ b/benchmarks/load_testing/helper.ex
@@ -1,11 +1,14 @@
 defmodule Pleroma.LoadTesting.Helper do
-  defmacro __using__(_) do
-    quote do
-      import Ecto.Query
-      alias Pleroma.Repo
-      alias Pleroma.User
+  alias Ecto.Adapters.SQL
+  alias Pleroma.Repo
 
-      defp to_sec(microseconds), do: microseconds / 1_000_000
-    end
+  def to_sec(microseconds), do: microseconds / 1_000_000
+
+  def 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