summaryrefslogtreecommitdiff
path: root/test/mix/tasks/pleroma/count_statuses_test.exs
blob: 80ec206b82665990a66b280c5c9e770bb440800e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# Pleroma: A lightweight social networking server
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only

defmodule Mix.Tasks.Pleroma.CountStatusesTest do
  # Uses log capture, has to stay synchronous
  use Pleroma.DataCase

  alias Pleroma.User
  alias Pleroma.Web.CommonAPI

  import ExUnit.CaptureIO, only: [capture_io: 1]
  import Pleroma.Factory

  test "counts statuses" do
    user = insert(:user)
    {:ok, _} = CommonAPI.post(user, %{status: "test"})
    {:ok, _} = CommonAPI.post(user, %{status: "test2"})

    user2 = insert(:user)
    {:ok, _} = CommonAPI.post(user2, %{status: "test3"})

    user = refresh_record(user)
    user2 = refresh_record(user2)

    assert %{note_count: 2} = user
    assert %{note_count: 1} = user2

    {:ok, user} = User.update_note_count(user, 0)
    {:ok, user2} = User.update_note_count(user2, 0)

    assert %{note_count: 0} = user
    assert %{note_count: 0} = user2

    assert capture_io(fn -> Mix.Tasks.Pleroma.CountStatuses.run([]) end) == "Done\n"

    assert %{note_count: 2} = refresh_record(user)
    assert %{note_count: 1} = refresh_record(user2)
  end
end