X-Git-Url: https://git.squeep.com/?a=blobdiff_plain;f=test%2Fsupport%2Foban_helpers.ex;h=9b6e5256e54ca7f981a15a240f0c6b0971a06375;hb=39127f15eb36c4fb41838a232f9d80c50416838d;hp=54b5a95662314a943fa0eecf9e34551235e310f9;hpb=b7fad8d395c2bd1afe445a370e539571f5ec0c18;p=akkoma

diff --git a/test/support/oban_helpers.ex b/test/support/oban_helpers.ex
index 54b5a9566..9b6e5256e 100644
--- a/test/support/oban_helpers.ex
+++ b/test/support/oban_helpers.ex
@@ -1,14 +1,27 @@
 # Pleroma: A lightweight social networking server
-# Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
+# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
 # SPDX-License-Identifier: AGPL-3.0-only
 
-defmodule Pleroma.ObanHelpers do
+defmodule Pleroma.Tests.ObanHelpers do
   @moduledoc """
   Oban test helpers.
   """
 
+  require Ecto.Query
+
   alias Pleroma.Repo
 
+  def wipe_all do
+    Repo.delete_all(Oban.Job)
+  end
+
+  def perform_all do
+    Oban.Job
+    |> Ecto.Query.where(state: "available")
+    |> Repo.all()
+    |> perform()
+  end
+
   def perform(%Oban.Job{} = job) do
     res = apply(String.to_existing_atom("Elixir." <> job.worker), :perform, [job])
     Repo.delete(job)