63 lines
		
	
	
	
		
			1.2 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			63 lines
		
	
	
	
		
			1.2 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
defmodule Pleroma.Web.StreamerTest do
 | 
						|
  use Pleroma.DataCase
 | 
						|
 | 
						|
  alias Pleroma.Web.Streamer
 | 
						|
  alias Pleroma.User
 | 
						|
  alias Pleroma.Web.CommonAPI
 | 
						|
  import Pleroma.Factory
 | 
						|
 | 
						|
  test "it sends to public" do
 | 
						|
    user = insert(:user)
 | 
						|
    other_user = insert(:user)
 | 
						|
 | 
						|
    task =
 | 
						|
      Task.async(fn ->
 | 
						|
        assert_receive {:text, _}, 4_000
 | 
						|
      end)
 | 
						|
 | 
						|
    fake_socket = %{
 | 
						|
      transport_pid: task.pid,
 | 
						|
      assigns: %{
 | 
						|
        user: user
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    {:ok, activity} = CommonAPI.post(other_user, %{"status" => "Test"})
 | 
						|
 | 
						|
    topics = %{
 | 
						|
      "public" => [fake_socket]
 | 
						|
    }
 | 
						|
 | 
						|
    Streamer.push_to_socket(topics, "public", activity)
 | 
						|
 | 
						|
    Task.await(task)
 | 
						|
  end
 | 
						|
 | 
						|
  test "it doesn't send to blocked users" do
 | 
						|
    user = insert(:user)
 | 
						|
    blocked_user = insert(:user)
 | 
						|
    {:ok, user} = User.block(user, blocked_user)
 | 
						|
 | 
						|
    task =
 | 
						|
      Task.async(fn ->
 | 
						|
        refute_receive {:text, _}, 1_000
 | 
						|
      end)
 | 
						|
 | 
						|
    fake_socket = %{
 | 
						|
      transport_pid: task.pid,
 | 
						|
      assigns: %{
 | 
						|
        user: user
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    {:ok, activity} = CommonAPI.post(blocked_user, %{"status" => "Test"})
 | 
						|
 | 
						|
    topics = %{
 | 
						|
      "public" => [fake_socket]
 | 
						|
    }
 | 
						|
 | 
						|
    Streamer.push_to_socket(topics, "public", activity)
 | 
						|
 | 
						|
    Task.await(task)
 | 
						|
  end
 | 
						|
end
 |