git.squeep.com Git - akkoma/blob - test/web/retry_queue_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule MockActivityPub do
   6   def publish_one({ret, waiter}) do
   7     send(waiter, :complete)
   8     {ret, "success"}
   9   end
  10 end
  11
  12 defmodule Pleroma.Web.Federator.RetryQueueTest do
  13   use Pleroma.DataCase
  14   alias Pleroma.Web.Federator.RetryQueue
  15
  16   @small_retry_count 0
  17   @hopeless_retry_count 10
  18
  19   setup do
  20     RetryQueue.reset_stats()
  21   end
  22
  23   test "RetryQueue responds to stats request" do
  24     assert %{delivered: 0, dropped: 0} == RetryQueue.get_stats()
  25   end
  26
  27   test "failed posts are retried" do
  28     {:retry, _timeout} = RetryQueue.get_retry_params(@small_retry_count)
  29
  30     wait_task =
  31       Task.async(fn ->
  32         receive do
  33           :complete -> :ok
  34         end
  35       end)
  36
  37     RetryQueue.enqueue({:ok, wait_task.pid}, MockActivityPub, @small_retry_count)
  38     Task.await(wait_task)
  39     assert %{delivered: 1, dropped: 0} == RetryQueue.get_stats()
  40   end
  41
  42   test "posts that have been tried too many times are dropped" do
  43     {:drop, _timeout} = RetryQueue.get_retry_params(@hopeless_retry_count)
  44
  45     RetryQueue.enqueue({:ok, nil}, MockActivityPub, @hopeless_retry_count)
  46     assert %{delivered: 0, dropped: 1} == RetryQueue.get_stats()
  47   end
  48 end