From e7a92b4c14ec25344d86edf13180ce52c673f1f7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonatan=20K=C5=82osko?= <jonatanklosko@gmail.com>
Date: Wed, 25 Oct 2023 21:04:20 +0700
Subject: [PATCH] Make setting default hub a sync test

---
 .../livebook_teams/web/hub/edit_live_test.exs | 106 +++++++-----------
 1 file changed, 40 insertions(+), 66 deletions(-)

diff --git a/test/livebook_teams/web/hub/edit_live_test.exs b/test/livebook_teams/web/hub/edit_live_test.exs
index 95a195579..5fe5d1f47 100644
--- a/test/livebook_teams/web/hub/edit_live_test.exs
+++ b/test/livebook_teams/web/hub/edit_live_test.exs
@@ -187,72 +187,6 @@ defmodule LivebookWeb.Integration.Hub.EditLiveTest do
       end
     end
 
-    test "mark hub as default", %{conn: conn, hub: hub} do
-      {:ok, view, _html} = live(conn, ~p"/hub/#{hub.id}")
-
-      assert view
-             |> element("a", "Mark as default")
-             |> has_element?()
-
-      refute view
-             |> element("a", "Remove as default")
-             |> has_element?()
-
-      assert {:ok, _view, _html} =
-               view
-               |> element("a", "Mark as default")
-               |> render_click()
-               |> follow_redirect(conn)
-
-      {:ok, view, _html} = live(conn, ~p"/hub/#{hub.id}")
-
-      assert view
-             |> element("a", "Remove as default")
-             |> has_element?()
-
-      refute view
-             |> element("a", "Mark as default")
-             |> has_element?()
-
-      assert view
-             |> element("span", "Default")
-             |> has_element?()
-    end
-
-    test "remove hub as default", %{conn: conn, hub: hub} do
-      Hubs.set_default_hub(hub.id)
-
-      {:ok, view, _html} = live(conn, ~p"/hub/#{hub.id}")
-
-      assert view
-             |> element("a", "Remove as default")
-             |> has_element?()
-
-      refute view
-             |> element("a", "Mark as default")
-             |> has_element?()
-
-      assert {:ok, _view, _html} =
-               view
-               |> element("a", "Remove as default")
-               |> render_click()
-               |> follow_redirect(conn)
-
-      {:ok, view, _html} = live(conn, ~p"/hub/#{hub.id}")
-
-      assert view
-             |> element("a", "Mark as default")
-             |> has_element?()
-
-      refute view
-             |> element("a", "Remove as default")
-             |> has_element?()
-
-      refute view
-             |> element("span", "Default")
-             |> has_element?()
-    end
-
     test "creates a file system", %{conn: conn, hub: hub} do
       {:ok, view, _html} = live(conn, ~p"/hub/#{hub.id}")
 
@@ -383,4 +317,44 @@ defmodule LivebookWeb.Integration.Hub.EditLiveTest do
       """)
     end)
   end
+
+  defmodule Global do
+    use Livebook.TeamsIntegrationCase, async: false
+
+    setup %{user: user, node: node} do
+      Livebook.Hubs.subscribe([:crud, :connection, :secrets, :file_systems])
+      hub = create_team_hub(user, node)
+      id = hub.id
+
+      assert_receive {:hub_connected, ^id}
+
+      {:ok, hub: hub}
+    end
+
+    test "marking and unmarking hub as default", %{conn: conn, hub: hub} do
+      {:ok, view, _html} = live(conn, ~p"/hub/#{hub.id}")
+
+      assert {:ok, view, _html} =
+               view
+               |> element("a", "Mark as default")
+               |> render_click()
+               |> follow_redirect(conn)
+
+      assert view
+             |> element("span", "Default")
+             |> has_element?()
+
+      assert Hubs.get_default_hub().id == hub.id
+
+      assert {:ok, view, _html} =
+               view
+               |> element("a", "Remove as default")
+               |> render_click()
+               |> follow_redirect(conn)
+
+      refute view
+             |> element("span", "Default")
+             |> has_element?()
+    end
+  end
 end