git.squeep.com Git - akkoma/blob - test/web/streamer/streamer_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.StreamerTest do
   6   use Pleroma.DataCase
   7
   8   import Pleroma.Factory
   9
  10   alias Pleroma.Conversation.Participation
  11   alias Pleroma.List
  12   alias Pleroma.Object
  13   alias Pleroma.User
  14   alias Pleroma.Web.CommonAPI
  15   alias Pleroma.Web.Streamer
  16   alias Pleroma.Web.StreamerView
  17
  18   @moduletag needs_streamer: true, capture_log: true
  19
  20   setup do: clear_config([:instance, :skip_thread_containment])
  21
  22   describe "get_topic without an user" do
  23     test "allows public" do
  24       assert {:ok, "public"} = Streamer.get_topic("public", nil)
  25       assert {:ok, "public:local"} = Streamer.get_topic("public:local", nil)
  26       assert {:ok, "public:media"} = Streamer.get_topic("public:media", nil)
  27       assert {:ok, "public:local:media"} = Streamer.get_topic("public:local:media", nil)
  28     end
  29
  30     test "allows hashtag streams" do
  31       assert {:ok, "hashtag:cofe"} = Streamer.get_topic("hashtag", nil, %{"tag" => "cofe"})
  32     end
  33
  34     test "disallows user streams" do
  35       assert {:error, _} = Streamer.get_topic("user", nil)
  36       assert {:error, _} = Streamer.get_topic("user:notification", nil)
  37       assert {:error, _} = Streamer.get_topic("direct", nil)
  38     end
  39
  40     test "disallows list streams" do
  41       assert {:error, _} = Streamer.get_topic("list", nil, %{"list" => 42})
  42     end
  43   end
  44
  45   describe "get_topic with an user" do
  46     setup do
  47       user = insert(:user)
  48       {:ok, %{user: user}}
  49     end
  50
  51     test "allows public streams", %{user: user} do
  52       assert {:ok, "public"} = Streamer.get_topic("public", user)
  53       assert {:ok, "public:local"} = Streamer.get_topic("public:local", user)
  54       assert {:ok, "public:media"} = Streamer.get_topic("public:media", user)
  55       assert {:ok, "public:local:media"} = Streamer.get_topic("public:local:media", user)
  56     end
  57
  58     test "allows user streams", %{user: user} do
  59       expected_user_topic = "user:#{user.id}"
  60       expected_notif_topic = "user:notification:#{user.id}"
  61       expected_direct_topic = "direct:#{user.id}"
  62       assert {:ok, ^expected_user_topic} = Streamer.get_topic("user", user)
  63       assert {:ok, ^expected_notif_topic} = Streamer.get_topic("user:notification", user)
  64       assert {:ok, ^expected_direct_topic} = Streamer.get_topic("direct", user)
  65     end
  66
  67     test "allows hashtag streams", %{user: user} do
  68       assert {:ok, "hashtag:cofe"} = Streamer.get_topic("hashtag", user, %{"tag" => "cofe"})
  69     end
  70
  71     test "disallows registering to an user stream", %{user: user} do
  72       another_user = insert(:user)
  73       assert {:error, _} = Streamer.get_topic("user:#{another_user.id}", user)
  74       assert {:error, _} = Streamer.get_topic("user:notification:#{another_user.id}", user)
  75       assert {:error, _} = Streamer.get_topic("direct:#{another_user.id}", user)
  76     end
  77
  78     test "allows list stream that are owned by the user", %{user: user} do
  79       {:ok, list} = List.create("Test", user)
  80       assert {:error, _} = Streamer.get_topic("list:#{list.id}", user)
  81       assert {:ok, _} = Streamer.get_topic("list", user, %{"list" => list.id})
  82     end
  83
  84     test "disallows list stream that are not owned by the user", %{user: user} do
  85       another_user = insert(:user)
  86       {:ok, list} = List.create("Test", another_user)
  87       assert {:error, _} = Streamer.get_topic("list:#{list.id}", user)
  88       assert {:error, _} = Streamer.get_topic("list", user, %{"list" => list.id})
  89     end
  90   end
  91
  92   describe "user streams" do
  93     setup do
  94       user = insert(:user)
  95       notify = insert(:notification, user: user, activity: build(:note_activity))
  96       {:ok, %{user: user, notify: notify}}
  97     end
  98
  99     test "it streams the user's post in the 'user' stream", %{user: user} do
 100       Streamer.get_topic_and_add_socket("user", user)
 101       {:ok, activity} = CommonAPI.post(user, %{status: "hey"})
 102       assert_receive {:render_with_user, _, _, ^activity}
 103       refute Streamer.filtered_by_user?(user, activity)
 104     end
 105
 106     test "it streams boosts of the user in the 'user' stream", %{user: user} do
 107       Streamer.get_topic_and_add_socket("user", user)
 108
 109       other_user = insert(:user)
 110       {:ok, activity} = CommonAPI.post(other_user, %{status: "hey"})
 111       {:ok, announce} = CommonAPI.repeat(activity.id, user)
 112
 113       assert_receive {:render_with_user, Pleroma.Web.StreamerView, "update.json", ^announce}
 114       refute Streamer.filtered_by_user?(user, announce)
 115     end
 116
 117     test "it sends notify to in the 'user' stream", %{user: user, notify: notify} do
 118       Streamer.get_topic_and_add_socket("user", user)
 119       Streamer.stream("user", notify)
 120       assert_receive {:render_with_user, _, _, ^notify}
 121       refute Streamer.filtered_by_user?(user, notify)
 122     end
 123
 124     test "it sends notify to in the 'user:notification' stream", %{user: user, notify: notify} do
 125       Streamer.get_topic_and_add_socket("user:notification", user)
 126       Streamer.stream("user:notification", notify)
 127       assert_receive {:render_with_user, _, _, ^notify}
 128       refute Streamer.filtered_by_user?(user, notify)
 129     end
 130
 131     test "it sends chat messages to the 'user:pleroma_chat' stream", %{user: user} do
 132       other_user = insert(:user)
 133
 134       {:ok, create_activity} = CommonAPI.post_chat_message(other_user, user, "hey")
 135       object = Object.normalize(create_activity, false)
 136       Streamer.get_topic_and_add_socket("user:pleroma_chat", user)
 137       Streamer.stream("user:pleroma_chat", object)
 138       text = StreamerView.render("chat_update.json", object, user, [user.ap_id, other_user.ap_id])
 139       assert_receive {:text, ^text}
 140     end
 141
 142     test "it sends chat messages to the 'user' stream", %{user: user} do
 143       other_user = insert(:user)
 144
 145       {:ok, create_activity} = CommonAPI.post_chat_message(other_user, user, "hey")
 146       object = Object.normalize(create_activity, false)
 147       Streamer.get_topic_and_add_socket("user", user)
 148       Streamer.stream("user", object)
 149       text = StreamerView.render("chat_update.json", object, user, [user.ap_id, other_user.ap_id])
 150       assert_receive {:text, ^text}
 151     end
 152
 153     test "it sends chat message notifications to the 'user:notification' stream", %{user: user} do
 154       other_user = insert(:user)
 155
 156       {:ok, create_activity} = CommonAPI.post_chat_message(other_user, user, "hey")
 157
 158       notify =
 159         Repo.get_by(Pleroma.Notification, user_id: user.id, activity_id: create_activity.id)
 160         |> Repo.preload(:activity)
 161
 162       Streamer.get_topic_and_add_socket("user:notification", user)
 163       Streamer.stream("user:notification", notify)
 164       assert_receive {:render_with_user, _, _, ^notify}
 165       refute Streamer.filtered_by_user?(user, notify)
 166     end
 167
 168     test "it doesn't send notify to the 'user:notification' stream when a user is blocked", %{
 169       user: user
 170     } do
 171       blocked = insert(:user)
 172       {:ok, _user_relationship} = User.block(user, blocked)
 173
 174       Streamer.get_topic_and_add_socket("user:notification", user)
 175
 176       {:ok, activity} = CommonAPI.post(user, %{status: ":("})
 177       {:ok, _} = CommonAPI.favorite(blocked, activity.id)
 178
 179       refute_receive _
 180     end
 181
 182     test "it doesn't send notify to the 'user:notification' stream when a thread is muted", %{
 183       user: user
 184     } do
 185       user2 = insert(:user)
 186
 187       {:ok, activity} = CommonAPI.post(user, %{status: "super hot take"})
 188       {:ok, _} = CommonAPI.add_mute(user, activity)
 189
 190       Streamer.get_topic_and_add_socket("user:notification", user)
 191
 192       {:ok, favorite_activity} = CommonAPI.favorite(user2, activity.id)
 193
 194       refute_receive _
 195       assert Streamer.filtered_by_user?(user, favorite_activity)
 196     end
 197
 198     test "it sends favorite to 'user:notification' stream'", %{
 199       user: user
 200     } do
 201       user2 = insert(:user, %{ap_id: "https://hecking-lewd-place.com/user/meanie"})
 202
 203       {:ok, activity} = CommonAPI.post(user, %{status: "super hot take"})
 204       Streamer.get_topic_and_add_socket("user:notification", user)
 205       {:ok, favorite_activity} = CommonAPI.favorite(user2, activity.id)
 206
 207       assert_receive {:render_with_user, _, "notification.json", notif}
 208       assert notif.activity.id == favorite_activity.id
 209       refute Streamer.filtered_by_user?(user, notif)
 210     end
 211
 212     test "it doesn't send the 'user:notification' stream' when a domain is blocked", %{
 213       user: user
 214     } do
 215       user2 = insert(:user, %{ap_id: "https://hecking-lewd-place.com/user/meanie"})
 216
 217       {:ok, user} = User.block_domain(user, "hecking-lewd-place.com")
 218       {:ok, activity} = CommonAPI.post(user, %{status: "super hot take"})
 219       Streamer.get_topic_and_add_socket("user:notification", user)
 220       {:ok, favorite_activity} = CommonAPI.favorite(user2, activity.id)
 221
 222       refute_receive _
 223       assert Streamer.filtered_by_user?(user, favorite_activity)
 224     end
 225
 226     test "it sends follow activities to the 'user:notification' stream", %{
 227       user: user
 228     } do
 229       user_url = user.ap_id
 230       user2 = insert(:user)
 231
 232       body =
 233         File.read!("test/fixtures/users_mock/localhost.json")
 234         |> String.replace("{{nickname}}", user.nickname)
 235         |> Jason.encode!()
 236
 237       Tesla.Mock.mock_global(fn
 238         %{method: :get, url: ^user_url} ->
 239           %Tesla.Env{status: 200, body: body}
 240       end)
 241
 242       Streamer.get_topic_and_add_socket("user:notification", user)
 243       {:ok, _follower, _followed, follow_activity} = CommonAPI.follow(user2, user)
 244
 245       assert_receive {:render_with_user, _, "notification.json", notif}
 246       assert notif.activity.id == follow_activity.id
 247       refute Streamer.filtered_by_user?(user, notif)
 248     end
 249   end
 250
 251   test "it sends to public authenticated" do
 252     user = insert(:user)
 253     other_user = insert(:user)
 254
 255     Streamer.get_topic_and_add_socket("public", other_user)
 256
 257     {:ok, activity} = CommonAPI.post(user, %{status: "Test"})
 258     assert_receive {:render_with_user, _, _, ^activity}
 259     refute Streamer.filtered_by_user?(user, activity)
 260   end
 261
 262   test "works for deletions" do
 263     user = insert(:user)
 264     other_user = insert(:user)
 265     {:ok, activity} = CommonAPI.post(other_user, %{status: "Test"})
 266
 267     Streamer.get_topic_and_add_socket("public", user)
 268
 269     {:ok, _} = CommonAPI.delete(activity.id, other_user)
 270     activity_id = activity.id
 271     assert_receive {:text, event}
 272     assert %{"event" => "delete", "payload" => ^activity_id} = Jason.decode!(event)
 273   end
 274
 275   test "it sends to public unauthenticated" do
 276     user = insert(:user)
 277
 278     Streamer.get_topic_and_add_socket("public", nil)
 279
 280     {:ok, activity} = CommonAPI.post(user, %{status: "Test"})
 281     activity_id = activity.id
 282     assert_receive {:text, event}
 283     assert %{"event" => "update", "payload" => payload} = Jason.decode!(event)
 284     assert %{"id" => ^activity_id} = Jason.decode!(payload)
 285
 286     {:ok, _} = CommonAPI.delete(activity.id, user)
 287     assert_receive {:text, event}
 288     assert %{"event" => "delete", "payload" => ^activity_id} = Jason.decode!(event)
 289   end
 290
 291   describe "thread_containment" do
 292     test "it filters to user if recipients invalid and thread containment is enabled" do
 293       Pleroma.Config.put([:instance, :skip_thread_containment], false)
 294       author = insert(:user)
 295       user = insert(:user)
 296       User.follow(user, author, :follow_accept)
 297
 298       activity =
 299         insert(:note_activity,
 300           note:
 301             insert(:note,
 302               user: author,
 303               data: %{"to" => ["TEST-FFF"]}
 304             )
 305         )
 306
 307       Streamer.get_topic_and_add_socket("public", user)
 308       Streamer.stream("public", activity)
 309       assert_receive {:render_with_user, _, _, ^activity}
 310       assert Streamer.filtered_by_user?(user, activity)
 311     end
 312
 313     test "it sends message if recipients invalid and thread containment is disabled" do
 314       Pleroma.Config.put([:instance, :skip_thread_containment], true)
 315       author = insert(:user)
 316       user = insert(:user)
 317       User.follow(user, author, :follow_accept)
 318
 319       activity =
 320         insert(:note_activity,
 321           note:
 322             insert(:note,
 323               user: author,
 324               data: %{"to" => ["TEST-FFF"]}
 325             )
 326         )
 327
 328       Streamer.get_topic_and_add_socket("public", user)
 329       Streamer.stream("public", activity)
 330
 331       assert_receive {:render_with_user, _, _, ^activity}
 332       refute Streamer.filtered_by_user?(user, activity)
 333     end
 334
 335     test "it sends message if recipients invalid and thread containment is enabled but user's thread containment is disabled" do
 336       Pleroma.Config.put([:instance, :skip_thread_containment], false)
 337       author = insert(:user)
 338       user = insert(:user, skip_thread_containment: true)
 339       User.follow(user, author, :follow_accept)
 340
 341       activity =
 342         insert(:note_activity,
 343           note:
 344             insert(:note,
 345               user: author,
 346               data: %{"to" => ["TEST-FFF"]}
 347             )
 348         )
 349
 350       Streamer.get_topic_and_add_socket("public", user)
 351       Streamer.stream("public", activity)
 352
 353       assert_receive {:render_with_user, _, _, ^activity}
 354       refute Streamer.filtered_by_user?(user, activity)
 355     end
 356   end
 357
 358   describe "blocks" do
 359     test "it filters messages involving blocked users" do
 360       user = insert(:user)
 361       blocked_user = insert(:user)
 362       {:ok, _user_relationship} = User.block(user, blocked_user)
 363
 364       Streamer.get_topic_and_add_socket("public", user)
 365       {:ok, activity} = CommonAPI.post(blocked_user, %{status: "Test"})
 366       assert_receive {:render_with_user, _, _, ^activity}
 367       assert Streamer.filtered_by_user?(user, activity)
 368     end
 369
 370     test "it filters messages transitively involving blocked users" do
 371       blocker = insert(:user)
 372       blockee = insert(:user)
 373       friend = insert(:user)
 374
 375       Streamer.get_topic_and_add_socket("public", blocker)
 376
 377       {:ok, _user_relationship} = User.block(blocker, blockee)
 378
 379       {:ok, activity_one} = CommonAPI.post(friend, %{status: "hey! @#{blockee.nickname}"})
 380
 381       assert_receive {:render_with_user, _, _, ^activity_one}
 382       assert Streamer.filtered_by_user?(blocker, activity_one)
 383
 384       {:ok, activity_two} = CommonAPI.post(blockee, %{status: "hey! @#{friend.nickname}"})
 385
 386       assert_receive {:render_with_user, _, _, ^activity_two}
 387       assert Streamer.filtered_by_user?(blocker, activity_two)
 388
 389       {:ok, activity_three} = CommonAPI.post(blockee, %{status: "hey! @#{blocker.nickname}"})
 390
 391       assert_receive {:render_with_user, _, _, ^activity_three}
 392       assert Streamer.filtered_by_user?(blocker, activity_three)
 393     end
 394   end
 395
 396   describe "lists" do
 397     test "it doesn't send unwanted DMs to list" do
 398       user_a = insert(:user)
 399       user_b = insert(:user)
 400       user_c = insert(:user)
 401
 402       {:ok, user_a} = User.follow(user_a, user_b)
 403
 404       {:ok, list} = List.create("Test", user_a)
 405       {:ok, list} = List.follow(list, user_b)
 406
 407       Streamer.get_topic_and_add_socket("list", user_a, %{"list" => list.id})
 408
 409       {:ok, _activity} =
 410         CommonAPI.post(user_b, %{
 411           status: "@#{user_c.nickname} Test",
 412           visibility: "direct"
 413         })
 414
 415       refute_receive _
 416     end
 417
 418     test "it doesn't send unwanted private posts to list" do
 419       user_a = insert(:user)
 420       user_b = insert(:user)
 421
 422       {:ok, list} = List.create("Test", user_a)
 423       {:ok, list} = List.follow(list, user_b)
 424
 425       Streamer.get_topic_and_add_socket("list", user_a, %{"list" => list.id})
 426
 427       {:ok, _activity} =
 428         CommonAPI.post(user_b, %{
 429           status: "Test",
 430           visibility: "private"
 431         })
 432
 433       refute_receive _
 434     end
 435
 436     test "it sends wanted private posts to list" do
 437       user_a = insert(:user)
 438       user_b = insert(:user)
 439
 440       {:ok, user_a} = User.follow(user_a, user_b)
 441
 442       {:ok, list} = List.create("Test", user_a)
 443       {:ok, list} = List.follow(list, user_b)
 444
 445       Streamer.get_topic_and_add_socket("list", user_a, %{"list" => list.id})
 446
 447       {:ok, activity} =
 448         CommonAPI.post(user_b, %{
 449           status: "Test",
 450           visibility: "private"
 451         })
 452
 453       assert_receive {:render_with_user, _, _, ^activity}
 454       refute Streamer.filtered_by_user?(user_a, activity)
 455     end
 456   end
 457
 458   describe "muted reblogs" do
 459     test "it filters muted reblogs" do
 460       user1 = insert(:user)
 461       user2 = insert(:user)
 462       user3 = insert(:user)
 463       CommonAPI.follow(user1, user2)
 464       CommonAPI.hide_reblogs(user1, user2)
 465
 466       {:ok, create_activity} = CommonAPI.post(user3, %{status: "I'm kawen"})
 467
 468       Streamer.get_topic_and_add_socket("user", user1)
 469       {:ok, announce_activity} = CommonAPI.repeat(create_activity.id, user2)
 470       assert_receive {:render_with_user, _, _, ^announce_activity}
 471       assert Streamer.filtered_by_user?(user1, announce_activity)
 472     end
 473
 474     test "it filters reblog notification for reblog-muted actors" do
 475       user1 = insert(:user)
 476       user2 = insert(:user)
 477       CommonAPI.follow(user1, user2)
 478       CommonAPI.hide_reblogs(user1, user2)
 479
 480       {:ok, create_activity} = CommonAPI.post(user1, %{status: "I'm kawen"})
 481       Streamer.get_topic_and_add_socket("user", user1)
 482       {:ok, _announce_activity} = CommonAPI.repeat(create_activity.id, user2)
 483
 484       assert_receive {:render_with_user, _, "notification.json", notif}
 485       assert Streamer.filtered_by_user?(user1, notif)
 486     end
 487
 488     test "it send non-reblog notification for reblog-muted actors" do
 489       user1 = insert(:user)
 490       user2 = insert(:user)
 491       CommonAPI.follow(user1, user2)
 492       CommonAPI.hide_reblogs(user1, user2)
 493
 494       {:ok, create_activity} = CommonAPI.post(user1, %{status: "I'm kawen"})
 495       Streamer.get_topic_and_add_socket("user", user1)
 496       {:ok, _favorite_activity} = CommonAPI.favorite(user2, create_activity.id)
 497
 498       assert_receive {:render_with_user, _, "notification.json", notif}
 499       refute Streamer.filtered_by_user?(user1, notif)
 500     end
 501   end
 502
 503   test "it filters posts from muted threads" do
 504     user = insert(:user)
 505     user2 = insert(:user)
 506     Streamer.get_topic_and_add_socket("user", user2)
 507     {:ok, user2, user, _activity} = CommonAPI.follow(user2, user)
 508     {:ok, activity} = CommonAPI.post(user, %{status: "super hot take"})
 509     {:ok, _} = CommonAPI.add_mute(user2, activity)
 510     assert_receive {:render_with_user, _, _, ^activity}
 511     assert Streamer.filtered_by_user?(user2, activity)
 512   end
 513
 514   describe "direct streams" do
 515     setup do
 516       :ok
 517     end
 518
 519     test "it sends conversation update to the 'direct' stream", %{} do
 520       user = insert(:user)
 521       another_user = insert(:user)
 522
 523       Streamer.get_topic_and_add_socket("direct", user)
 524
 525       {:ok, _create_activity} =
 526         CommonAPI.post(another_user, %{
 527           status: "hey @#{user.nickname}",
 528           visibility: "direct"
 529         })
 530
 531       assert_receive {:text, received_event}
 532
 533       assert %{"event" => "conversation", "payload" => received_payload} =
 534                Jason.decode!(received_event)
 535
 536       assert %{"last_status" => last_status} = Jason.decode!(received_payload)
 537       [participation] = Participation.for_user(user)
 538       assert last_status["pleroma"]["direct_conversation_id"] == participation.id
 539     end
 540
 541     test "it doesn't send conversation update to the 'direct' stream when the last message in the conversation is deleted" do
 542       user = insert(:user)
 543       another_user = insert(:user)
 544
 545       Streamer.get_topic_and_add_socket("direct", user)
 546
 547       {:ok, create_activity} =
 548         CommonAPI.post(another_user, %{
 549           status: "hi @#{user.nickname}",
 550           visibility: "direct"
 551         })
 552
 553       create_activity_id = create_activity.id
 554       assert_receive {:render_with_user, _, _, ^create_activity}
 555       assert_receive {:text, received_conversation1}
 556       assert %{"event" => "conversation", "payload" => _} = Jason.decode!(received_conversation1)
 557
 558       {:ok, _} = CommonAPI.delete(create_activity_id, another_user)
 559
 560       assert_receive {:text, received_event}
 561
 562       assert %{"event" => "delete", "payload" => ^create_activity_id} =
 563                Jason.decode!(received_event)
 564
 565       refute_receive _
 566     end
 567
 568     test "it sends conversation update to the 'direct' stream when a message is deleted" do
 569       user = insert(:user)
 570       another_user = insert(:user)
 571       Streamer.get_topic_and_add_socket("direct", user)
 572
 573       {:ok, create_activity} =
 574         CommonAPI.post(another_user, %{
 575           status: "hi @#{user.nickname}",
 576           visibility: "direct"
 577         })
 578
 579       {:ok, create_activity2} =
 580         CommonAPI.post(another_user, %{
 581           status: "hi @#{user.nickname} 2",
 582           in_reply_to_status_id: create_activity.id,
 583           visibility: "direct"
 584         })
 585
 586       assert_receive {:render_with_user, _, _, ^create_activity}
 587       assert_receive {:render_with_user, _, _, ^create_activity2}
 588       assert_receive {:text, received_conversation1}
 589       assert %{"event" => "conversation", "payload" => _} = Jason.decode!(received_conversation1)
 590       assert_receive {:text, received_conversation1}
 591       assert %{"event" => "conversation", "payload" => _} = Jason.decode!(received_conversation1)
 592
 593       {:ok, _} = CommonAPI.delete(create_activity2.id, another_user)
 594
 595       assert_receive {:text, received_event}
 596       assert %{"event" => "delete", "payload" => _} = Jason.decode!(received_event)
 597
 598       assert_receive {:text, received_event}
 599
 600       assert %{"event" => "conversation", "payload" => received_payload} =
 601                Jason.decode!(received_event)
 602
 603       assert %{"last_status" => last_status} = Jason.decode!(received_payload)
 604       assert last_status["id"] == to_string(create_activity.id)
 605     end
 606   end
 607 end