summaryrefslogtreecommitdiff
path: root/test/mix/tasks/pleroma/count_statuses_test.exs
diff options
context:
space:
mode:
Diffstat (limited to 'test/mix/tasks/pleroma/count_statuses_test.exs')
-rw-r--r--test/mix/tasks/pleroma/count_statuses_test.exs39
1 files changed, 39 insertions, 0 deletions
diff --git a/test/mix/tasks/pleroma/count_statuses_test.exs b/test/mix/tasks/pleroma/count_statuses_test.exs
new file mode 100644
index 000000000..c5cd16960
--- /dev/null
+++ b/test/mix/tasks/pleroma/count_statuses_test.exs
@@ -0,0 +1,39 @@
+# Pleroma: A lightweight social networking server
+# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
+# SPDX-License-Identifier: AGPL-3.0-only
+
+defmodule Mix.Tasks.Pleroma.CountStatusesTest do
+ 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