From a1712042bd198831a604447753e25915a8ba3b35 Mon Sep 17 00:00:00 2001
From: smokestacklightnin
 <125844868+smokestacklightnin@users.noreply.github.com>
Date: Fri, 3 Jan 2025 00:50:00 -0800
Subject: [PATCH] Import `get_keycloak_admin` from `upgrade`

---
 tests/tests_unit/test_upgrade.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/tests_unit/test_upgrade.py b/tests/tests_unit/test_upgrade.py
index a3c7fbb8be..70acdac065 100644
--- a/tests/tests_unit/test_upgrade.py
+++ b/tests/tests_unit/test_upgrade.py
@@ -154,13 +154,13 @@ class MonkeypatchApiResponse:
         monkey_patch_list_namespaced_daemon_set,
     )
 
-    from _nebari import keycloak as _keycloak
+    from _nebari import upgrade as _upgrade
 
     def monkey_patch_get_keycloak_admin(*args, **kwargs):
         return MockKeycloakAdmin()
 
     monkeypatch.setattr(
-        _keycloak,
+        _upgrade,
         "get_keycloak_admin",
         monkey_patch_get_keycloak_admin,
     )