git.squeep.com Git - akkoma/blob - test/jobs_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.JobsTest do
   6   use ExUnit.Case, async: true
   7
   8   alias Pleroma.Jobs
   9   alias Jobs.WorkerMock
  10
  11   setup do
  12     state = %{
  13       queues: Enum.into([Jobs.create_queue(:testing)], %{}),
  14       refs: %{}
  15     }
  16
  17     [state: state]
  18   end
  19
  20   test "creates queue" do
  21     queue = Jobs.create_queue(:foobar)
  22
  23     assert {:foobar, set} = queue
  24     assert :set == elem(set, 0) |> elem(0)
  25   end
  26
  27   test "enqueues an element according to priority" do
  28     queue = [%{item: 1, priority: 2}]
  29
  30     new_queue = Jobs.enqueue_sorted(queue, 2, 1)
  31     assert new_queue == [%{item: 2, priority: 1}, %{item: 1, priority: 2}]
  32
  33     new_queue = Jobs.enqueue_sorted(queue, 2, 3)
  34     assert new_queue == [%{item: 1, priority: 2}, %{item: 2, priority: 3}]
  35   end
  36
  37   test "pop first item" do
  38     queue = [%{item: 2, priority: 1}, %{item: 1, priority: 2}]
  39
  40     assert {2, [%{item: 1, priority: 2}]} = Jobs.queue_pop(queue)
  41   end
  42
  43   test "enqueue a job", %{state: state} do
  44     assert {:noreply, new_state} =
  45              Jobs.handle_cast({:enqueue, :testing, WorkerMock, [:test_job, :foo, :bar], 3}, state)
  46
  47     assert %{queues: %{testing: {running_jobs, []}}, refs: _} = new_state
  48     assert :sets.size(running_jobs) == 1
  49     assert [ref] = :sets.to_list(running_jobs)
  50     assert %{refs: %{^ref => :testing}} = new_state
  51   end
  52
  53   test "max jobs setting", %{state: state} do
  54     max_jobs = Pleroma.Config.get([Jobs, :testing, :max_jobs])
  55
  56     {:noreply, state} =
  57       Enum.reduce(1..(max_jobs + 1), {:noreply, state}, fn _, {:noreply, state} ->
  58         Jobs.handle_cast({:enqueue, :testing, WorkerMock, [:test_job, :foo, :bar], 3}, state)
  59       end)
  60
  61     assert %{
  62              queues: %{
  63                testing:
  64                  {running_jobs, [%{item: {WorkerMock, [:test_job, :foo, :bar]}, priority: 3}]}
  65              }
  66            } = state
  67
  68     assert :sets.size(running_jobs) == max_jobs
  69   end
  70
  71   test "remove job after it finished", %{state: state} do
  72     {:noreply, new_state} =
  73       Jobs.handle_cast({:enqueue, :testing, WorkerMock, [:test_job, :foo, :bar], 3}, state)
  74
  75     %{queues: %{testing: {running_jobs, []}}} = new_state
  76     [ref] = :sets.to_list(running_jobs)
  77
  78     assert {:noreply, %{queues: %{testing: {running_jobs, []}}, refs: %{}}} =
  79              Jobs.handle_info({:DOWN, ref, :process, nil, nil}, new_state)
  80
  81     assert :sets.size(running_jobs) == 0
  82   end
  83 end