git.squeep.com Git - akkoma/blob - test/integration/mastodon_websocket_test.exs

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Integration.MastodonWebsocketTest do
   6   use Pleroma.DataCase
   7
   8   import Pleroma.Factory
   9
  10   alias Pleroma.Integration.WebsocketClient
  11   alias Pleroma.Web.CommonAPI
  12   alias Pleroma.Web.OAuth
  13
  14   @path Pleroma.Web.Endpoint.url()
  15         |> URI.parse()
  16         |> Map.put(:scheme, "ws")
  17         |> Map.put(:path, "/api/v1/streaming")
  18         |> URI.to_string()
  19
  20   def start_socket(qs \\ nil, headers \\ []) do
  21     path =
  22       case qs do
  23         nil -> @path
  24         qs -> @path <> qs
  25       end
  26
  27     WebsocketClient.start_link(self(), path, headers)
  28   end
  29
  30   test "refuses invalid requests" do
  31     assert {:error, {400, _}} = start_socket()
  32     assert {:error, {404, _}} = start_socket("?stream=ncjdk")
  33   end
  34
  35   test "requires authentication and a valid token for protected streams" do
  36     assert {:error, {403, _}} = start_socket("?stream=user&access_token=aaaaaaaaaaaa")
  37     assert {:error, {403, _}} = start_socket("?stream=user")
  38   end
  39
  40   @tag needs_streamer: true
  41   test "allows public streams without authentication" do
  42     assert {:ok, _} = start_socket("?stream=public")
  43     assert {:ok, _} = start_socket("?stream=public:local")
  44     assert {:ok, _} = start_socket("?stream=hashtag&tag=lain")
  45   end
  46
  47   @tag needs_streamer: true
  48   test "receives well formatted events" do
  49     user = insert(:user)
  50     {:ok, _} = start_socket("?stream=public")
  51     {:ok, activity} = CommonAPI.post(user, %{"status" => "nice echo chamber"})
  52
  53     assert_receive {:text, raw_json}, 1_000
  54     assert {:ok, json} = Jason.decode(raw_json)
  55
  56     assert "update" == json["event"]
  57     assert json["payload"]
  58     assert {:ok, json} = Jason.decode(json["payload"])
  59
  60     view_json =
  61       Pleroma.Web.MastodonAPI.StatusView.render("status.json", activity: activity, for: nil)
  62       |> Jason.encode!()
  63       |> Jason.decode!()
  64
  65     assert json == view_json
  66   end
  67
  68   describe "with a valid user token" do
  69     setup do
  70       {:ok, app} =
  71         Pleroma.Repo.insert(
  72           OAuth.App.register_changeset(%OAuth.App{}, %{
  73             client_name: "client",
  74             scopes: ["scope"],
  75             redirect_uris: "url"
  76           })
  77         )
  78
  79       user = insert(:user)
  80
  81       {:ok, auth} = OAuth.Authorization.create_authorization(app, user)
  82
  83       {:ok, token} = OAuth.Token.exchange_token(app, auth)
  84
  85       %{user: user, token: token}
  86     end
  87
  88     test "accepts valid tokens", state do
  89       assert {:ok, _} = start_socket("?stream=user&access_token=#{state.token.token}")
  90     end
  91
  92     @tag needs_streamer: true
  93     test "accepts the 'user' stream", %{token: token} = _state do
  94       assert {:ok, _} = start_socket("?stream=user&access_token=#{token.token}")
  95       assert {:error, {403, "Forbidden"}} = start_socket("?stream=user")
  96     end
  97
  98     @tag needs_streamer: true
  99     test "accepts the 'user:notification' stream", %{token: token} = _state do
 100       assert {:ok, _} = start_socket("?stream=user:notification&access_token=#{token.token}")
 101       assert {:error, {403, "Forbidden"}} = start_socket("?stream=user:notification")
 102     end
 103
 104     @tag needs_streamer: true
 105     test "accepts valid token on Sec-WebSocket-Protocol header", %{token: token} do
 106       assert {:ok, _} = start_socket("?stream=user", [{"Sec-WebSocket-Protocol", token.token}])
 107
 108       assert {:error, {403, "Forbidden"}} =
 109                start_socket("?stream=user", [{"Sec-WebSocket-Protocol", "I am a friend"}])
 110     end
 111   end
 112 end