livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2025-09-04 03:54:24 +08:00
Code Issues Projects Releases Packages Wiki Activity

Override ZTA name from test

Browse source
This commit is contained in:
Alexandre de Souza 2025-07-15 16:59:33 -03:00
parent 44429f4eb7
commit cf21a3f72d
No known key found for this signature in database
GPG key ID: E39228FFBA346545
2 changed files with 21 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
lib/livebook_web/plugs/user_plug.ex
View file

@ -35,7 +35,7 @@ defmodule LivebookWeb.UserPlug do
defp ensure_user_identity(conn) do
{_type, module, _key} = identity_provider(conn)
{conn, identity_data} = module.authenticate(LivebookWeb.ZTA, conn, [])
{conn, identity_data} = authenticate(module, conn, [])
cond do
conn.halted ->
@ -138,12 +138,26 @@ defmodule LivebookWeb.UserPlug do
"""
@spec identity_provider(Plug.Conn.t() | map()) :: {atom(), module, binary}
if Mix.env() == :test do
def identity_provider(%Plug.Conn{} = conn), do: identity_provider(get_session(conn))
def identity_provider(%{} = session) do
def identity_provider(%Plug.Conn{} = conn) do
session = get_session(conn)
session["identity_provider_test_override"] || Livebook.Config.identity_provider()
end
else
def identity_provider(_), do: Livebook.Config.identity_provider()
end
@zta_name LivebookWeb.ZTA
@spec authenticate(module(), Plug.Conn.t() | map(), keyword()) ::
{Plug.Conn.t(), Livebook.ZTA.metadata()}
if Mix.env() == :test do
def authenticate(module, %Plug.Conn{} = conn, opts) do
session = get_session(conn)
name = session["zta_name_test_override"] || @zta_name
module.authenticate(name, conn, opts)
end
else
def identity_provider(module, conn, opts), do: module.authenticate(@zta_name, conn, opts)
end
end

5
test/support/conn_case.ex
View file

@ -33,9 +33,10 @@ defmodule LivebookWeb.ConnCase do
Plug.Test.init_test_session(conn, %{authentication_test_override: authentication})
end
def with_authorization(conn, id) do
def with_authorization(conn, id, name) do
Plug.Test.init_test_session(conn, %{
identity_provider_test_override: {:zta, Livebook.ZTA.LivebookTeams, id}
identity_provider_test_override: {:zta, Livebook.ZTA.LivebookTeams, id},
zta_name_test_override: name
})
end
end