From 404a9ccb9a220f3f52ee03bd69bd3746d95794cc Mon Sep 17 00:00:00 2001
From: Maxim Filippov <colixer@gmail.com>
Date: Wed, 18 Dec 2019 23:11:42 +0300
Subject: Stats: return status counts by scope

---
 test/stats_test.exs                                | 52 ++++++++++++++++++++++
 .../controllers/instance_controller_test.exs       | 10 ++++-
 2 files changed, 61 insertions(+), 1 deletion(-)
 create mode 100644 test/stats_test.exs

(limited to 'test')

diff --git a/test/stats_test.exs b/test/stats_test.exs
new file mode 100644
index 000000000..31c2f8db3
--- /dev/null
+++ b/test/stats_test.exs
@@ -0,0 +1,52 @@
+# Pleroma: A lightweight social networking server
+# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
+# SPDX-License-Identifier: AGPL-3.0-only
+
+defmodule Pleroma.StatsTest do
+  use Pleroma.DataCase
+
+  import Pleroma.Factory
+
+  alias Pleroma.Web.CommonAPI
+
+  describe "statuses count" do
+    setup do
+      user = insert(:user)
+      other_user = insert(:user)
+
+      CommonAPI.post(user, %{"visibility" => "public", "status" => "hey"})
+
+      Enum.each(0..1, fn _ ->
+        CommonAPI.post(user, %{
+          "visibility" => "unlisted",
+          "status" => "hey"
+        })
+      end)
+
+      Enum.each(0..2, fn _ ->
+        CommonAPI.post(user, %{
+          "visibility" => "direct",
+          "status" => "hey @#{other_user.nickname}"
+        })
+      end)
+
+      Enum.each(0..3, fn _ ->
+        CommonAPI.post(user, %{
+          "visibility" => "private",
+          "status" => "hey"
+        })
+      end)
+
+      :ok
+    end
+
+    test "it returns total number of statuses" do
+      data = Pleroma.Stats.get_stat_data()
+
+      assert data.stats.status_count.public == 1
+      assert data.stats.status_count.unlisted == 2
+      assert data.stats.status_count.direct == 3
+      assert data.stats.status_count.private == 4
+    end
+  end
+end
diff --git a/test/web/mastodon_api/controllers/instance_controller_test.exs b/test/web/mastodon_api/controllers/instance_controller_test.exs
index e00de6b18..7aa7c8648 100644
--- a/test/web/mastodon_api/controllers/instance_controller_test.exs
+++ b/test/web/mastodon_api/controllers/instance_controller_test.exs
@@ -58,7 +58,15 @@ defmodule Pleroma.Web.MastodonAPI.InstanceControllerTest do
 
     assert stats
     assert stats["user_count"] == 1
-    assert stats["status_count"] == 1
+
+    assert stats["status_count"] == %{
+             "all" => 1,
+             "direct" => 0,
+             "private" => 0,
+             "public" => 1,
+             "unlisted" => 0
+           }
+
     assert stats["domain_count"] == 2
   end
 
-- 
cgit v1.2.3