git.squeep.com Git - akkoma/blob - test/plugs/oauth_plug_test.exs

   1 defmodule Pleroma.Plugs.OAuthPlugTest do
   2   use Pleroma.Web.ConnCase, async: true
   3
   4   alias Pleroma.Plugs.OAuthPlug
   5   import Pleroma.Factory
   6
   7   @session_opts [
   8     store: :cookie,
   9     key: "_test",
  10     signing_salt: "cooldude"
  11   ]
  12
  13   setup %{conn: conn} do
  14     user = insert(:user)
  15     {:ok, %{token: token}} = Pleroma.Web.OAuth.Token.create_token(insert(:oauth_app), user)
  16     %{user: user, token: token, conn: conn}
  17   end
  18
  19   test "with valid token(uppercase), it assigns the user", %{conn: conn} = opts do
  20     conn =
  21       conn
  22       |> put_req_header("authorization", "BEARER #{opts[:token]}")
  23       |> OAuthPlug.call(%{})
  24
  25     assert conn.assigns[:user] == opts[:user]
  26   end
  27
  28   test "with valid token(downcase), it assigns the user", %{conn: conn} = opts do
  29     conn =
  30       conn
  31       |> put_req_header("authorization", "bearer #{opts[:token]}")
  32       |> OAuthPlug.call(%{})
  33
  34     assert conn.assigns[:user] == opts[:user]
  35   end
  36
  37   test "with invalid token, it not assigns the user", %{conn: conn} do
  38     conn =
  39       conn
  40       |> put_req_header("authorization", "bearer TTTTT")
  41       |> OAuthPlug.call(%{})
  42
  43     refute conn.assigns[:user]
  44   end
  45
  46   test "when token is missed but token in session, it assigns the user", %{conn: conn} = opts do
  47     conn =
  48       conn
  49       |> Plug.Session.call(Plug.Session.init(@session_opts))
  50       |> fetch_session()
  51       |> put_session(:oauth_token, opts[:token])
  52       |> OAuthPlug.call(%{})
  53
  54     assert conn.assigns[:user] == opts[:user]
  55   end
  56 end