X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=benchmarks%2Fload_testing%2Fhelper.ex;h=cab60acb422e7472b277b51f8ac678720e1b184d;hb=7f692343c80ddf353712490edfbcdb14866f5685;hp=23bbb1cec22f1288f6ce6555971e01846742364f;hpb=eb9744cadea7191b088ddaadfbd5fa4d4fd45090;p=akkoma

diff --git a/benchmarks/load_testing/helper.ex b/benchmarks/load_testing/helper.ex
index 23bbb1cec..cab60acb4 100644
--- a/benchmarks/load_testing/helper.ex
+++ b/benchmarks/load_testing/helper.ex
@@ -1,3 +1,14 @@
 defmodule Pleroma.LoadTesting.Helper do
+  alias Ecto.Adapters.SQL
+  alias Pleroma.Repo
+
   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