From 42cd2af31d17d809a9ec5651ce7026496e1a97a9 Mon Sep 17 00:00:00 2001
From: Alexandre de Souza <alexandre@aledsz.com.br>
Date: Mon, 17 Feb 2025 10:56:52 -0300
Subject: [PATCH] Fix logout button with disabled "Authenticate with Livebook
 Teams" (#2935)

---
 lib/livebook_web/components/layout_components.ex | 2 +-
 test/livebook/runtime/evaluator_test.exs         | 4 ++--
 test/support/hub_helpers.ex                      | 4 +++-
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/lib/livebook_web/components/layout_components.ex b/lib/livebook_web/components/layout_components.ex
index e2805b9d6..c6f9cbaf5 100644
--- a/lib/livebook_web/components/layout_components.ex
+++ b/lib/livebook_web/components/layout_components.ex
@@ -130,7 +130,7 @@ defmodule LivebookWeb.LayoutComponents do
         </div>
         <div class="flex flex-col">
           <button
-            :if={Livebook.Config.logout_enabled?()}
+            :if={Livebook.Config.logout_enabled?() and @current_user.email != nil}
             class="h-7 flex items-center text-gray-400 hover:text-white border-l-4 border-transparent hover:border-white"
             aria-label="logout"
             phx-click="logout"
diff --git a/test/livebook/runtime/evaluator_test.exs b/test/livebook/runtime/evaluator_test.exs
index c41d74b16..5c5e982f2 100644
--- a/test/livebook/runtime/evaluator_test.exs
+++ b/test/livebook/runtime/evaluator_test.exs
@@ -1560,12 +1560,12 @@ defmodule Livebook.Runtime.EvaluatorTest do
       Evaluator.evaluate_code(evaluator, :erlang, "list_to_binary(1).", :code_4, [])
       assert_receive {:runtime_evaluation_response, :code_4, error(message), metadata()}
 
-      assert clean_message(message) === """
+      assert clean_message(message) =~ """
              exception error: bad argument
                in function  list_to_binary/1
                   called as list_to_binary(1)
                   *** argument 1: not an iolist term
-               in call from erl_eval:do_apply/7 (erl_eval.erl, line 915)\
+               in call from erl_eval:do_apply/7 (erl_eval.erl, line\
              """
     end
   end
diff --git a/test/support/hub_helpers.ex b/test/support/hub_helpers.ex
index 0a36699bc..ff070d1a7 100644
--- a/test/support/hub_helpers.ex
+++ b/test/support/hub_helpers.ex
@@ -57,6 +57,7 @@ defmodule Livebook.HubHelpers do
     org_key = erpc_call(node, :create_org_key, [[org: org, key_hash: key_hash]])
     org_key_pair = erpc_call(node, :create_org_key_pair, [[org: org]])
     token = erpc_call(node, :associate_user_with_org, [user, org])
+    erpc_call(node, :create_billing_subscription, [org])
 
     build(:team,
       id: "team-#{org.name}",
@@ -66,7 +67,8 @@ defmodule Livebook.HubHelpers do
       org_key_id: org_key.id,
       org_public_key: org_key_pair.public_key,
       session_token: token,
-      teams_key: teams_key
+      teams_key: teams_key,
+      billing_status: %{disabled: false, type: nil}
     )
   end