Add a test for List.get_lists_account_belongs

1 files changed, 20 insertions, 0 deletions

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 @@ defmodule Pleroma.ListTest 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