From a249cbf18724084d6bcb7fee0ba577241c1f6adb Mon Sep 17 00:00:00 2001
From: AkiraFukushima <h3.poteto@gmail.com>
Date: Fri, 19 Oct 2018 21:24:15 +0900
Subject: [PATCH] Add a test for List.get_lists_account_belongs

---
 lib/pleroma/list.ex | 10 +++++++++-
 test/list_test.exs  | 20 ++++++++++++++++++++
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/lib/pleroma/list.ex b/lib/pleroma/list.ex
index a99e3245b..891c73f5a 100644
--- a/lib/pleroma/list.ex
+++ b/lib/pleroma/list.ex
@@ -72,11 +72,19 @@ def get_lists_from_activity(%Activity{actor: ap_id}) do
   # Get lists to which the account belongs.
   def get_lists_account_belongs(%User{} = owner, account_id) do
     user = Repo.get(User, account_id)
+
     query =
       from(
         l in Pleroma.List,
-        where: l.user_id == ^owner.id and fragment("? = ANY(?)", ^user.follower_address, l.following)
+        where:
+          l.user_id == ^owner.id and
+            fragment(
+              "? = ANY(?)",
+              ^user.follower_address,
+              l.following
+            )
       )
+
     Repo.all(query)
   end
 
diff --git a/test/list_test.exs b/test/list_test.exs
index da3b88024..3c07e56b2 100644
--- a/test/list_test.exs
+++ b/test/list_test.exs
@@ -90,4 +90,24 @@ test "getting all lists the user is a member of" do
     assert list_two in lists
     refute list_three in lists
   end
+
+  test "getting onwered lists the user belongs" do
+    owner = insert(:user)
+    not_owner = insert(:user)
+    member_1 = insert(:user)
+    member_2 = insert(:user)
+    {:ok, ownered_list} = Pleroma.List.create("ownered", owner)
+    {:ok, not_ownered_list} = Pleroma.List.create("not ownered", not_owner)
+    {:ok, ownered_list} = Pleroma.List.follow(ownered_list, member_1)
+    {:ok, ownered_list} = Pleroma.List.follow(ownered_list, member_2)
+    {:ok, not_ownered_list} = Pleroma.List.follow(not_ownered_list, member_1)
+    {:ok, not_ownered_list} = Pleroma.List.follow(not_ownered_list, member_2)
+
+    lists_1 = Pleroma.List.get_lists_account_belongs(owner, member_1.id)
+    assert ownered_list in lists_1
+    refute not_ownered_list in lists_1
+    lists_2 = Pleroma.List.get_lists_account_belongs(owner, member_2.id)
+    assert ownered_list in lists_2
+    refute not_ownered_list in lists_2
+  end
 end