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

   1 defmodule MockActivityPub do
   2   def publish_one(ret) do
   3     {ret, "success"}
   4   end
   5 end
   6
   7 defmodule Pleroma.Web.Federator.RetryQueueTest do
   8   use Pleroma.DataCase
   9   alias Pleroma.Web.Federator.RetryQueue
  10
  11   @small_retry_count 0
  12   @hopeless_retry_count 10
  13
  14   test "failed posts are retried" do
  15     {:retry, _timeout} = RetryQueue.get_retry_params(@small_retry_count)
  16
  17     assert {:noreply, %{delivered: 1}} ==
  18              RetryQueue.handle_info({:send, :ok, MockActivityPub, @small_retry_count}, %{
  19                delivered: 0
  20              })
  21   end
  22
  23   test "posts that have been tried too many times are dropped" do
  24     {:drop, _timeout} = RetryQueue.get_retry_params(@hopeless_retry_count)
  25
  26     assert {:noreply, %{dropped: 1}} ==
  27              RetryQueue.handle_cast({:maybe_enqueue, %{}, nil, @hopeless_retry_count}, %{
  28                dropped: 0
  29              })
  30   end
  31 end