forked from AkkomaGang/akkoma
Add EnsureUserKeyPlug, smaller fixes
This commit is contained in:
parent
32465b9939
commit
12bc73dd28
4 changed files with 42 additions and 1 deletions
|
@ -37,4 +37,6 @@ def call(
|
||||||
Pbkdf2.dummy_checkpw()
|
Pbkdf2.dummy_checkpw()
|
||||||
conn
|
conn
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def call(conn, _), do: conn
|
||||||
end
|
end
|
||||||
|
|
14
lib/pleroma/plugs/ensure_user_key_plug.ex
Normal file
14
lib/pleroma/plugs/ensure_user_key_plug.ex
Normal file
|
@ -0,0 +1,14 @@
|
||||||
|
defmodule Pleroma.Plugs.EnsureUserKeyPlug do
|
||||||
|
import Plug.Conn
|
||||||
|
|
||||||
|
def init(opts) do
|
||||||
|
opts
|
||||||
|
end
|
||||||
|
|
||||||
|
def call(%{assigns: %{user: _}} = conn, _), do: conn
|
||||||
|
|
||||||
|
def call(conn, _) do
|
||||||
|
conn
|
||||||
|
|> assign(:user, nil)
|
||||||
|
end
|
||||||
|
end
|
|
@ -1,4 +1,4 @@
|
||||||
defmodule Pleroma.Plugs.AuthenticationPlugTest do
|
defmodule Pleroma.Plugs.BasicAuthDecoderPlugTest do
|
||||||
use Pleroma.Web.ConnCase, async: true
|
use Pleroma.Web.ConnCase, async: true
|
||||||
|
|
||||||
alias Pleroma.Plugs.BasicAuthDecoderPlug
|
alias Pleroma.Plugs.BasicAuthDecoderPlug
|
||||||
|
|
25
test/plugs/ensure_user_key_plug_test.exs
Normal file
25
test/plugs/ensure_user_key_plug_test.exs
Normal file
|
@ -0,0 +1,25 @@
|
||||||
|
defmodule Pleroma.Plugs.EnsureUserKeyPlugTest do
|
||||||
|
use Pleroma.Web.ConnCase, async: true
|
||||||
|
|
||||||
|
alias Pleroma.Plugs.EnsureUserKeyPlug
|
||||||
|
|
||||||
|
test "if the conn has a user key set, it does nothing", %{conn: conn} do
|
||||||
|
conn =
|
||||||
|
conn
|
||||||
|
|> assign(:user, 1)
|
||||||
|
|
||||||
|
ret_conn =
|
||||||
|
conn
|
||||||
|
|> EnsureUserKeyPlug.call(%{})
|
||||||
|
|
||||||
|
assert conn == ret_conn
|
||||||
|
end
|
||||||
|
|
||||||
|
test "if the conn has no key set, it sets it to nil", %{conn: conn} do
|
||||||
|
conn =
|
||||||
|
conn
|
||||||
|
|> EnsureUserKeyPlug.call(%{})
|
||||||
|
|
||||||
|
assert Map.has_key?(conn.assigns, :user)
|
||||||
|
end
|
||||||
|
end
|
Loading…
Reference in a new issue