From a86c5484d42d569f8d82320354adafe281a5904e Mon Sep 17 00:00:00 2001 From: Rafael Sarmiento Date: Thu, 12 Sep 2024 15:20:28 +0200 Subject: [PATCH] fix unit tests --- tests/test_spawner.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/test_spawner.py b/tests/test_spawner.py index acefef6..628fdf3 100644 --- a/tests/test_spawner.py +++ b/tests/test_spawner.py @@ -29,6 +29,25 @@ async def refresh_user(self, user, handler=None): return {"auth_state": auth_state} +async def get_firecrest_client(spawner): + auth_state_refreshed = await spawner.user.authenticator.refresh_user(spawner.user) # noqa E501 + access_token = auth_state_refreshed['auth_state']['access_token'] + + client = firecrest.AsyncFirecrest( + firecrest_url=spawner.firecrest_url, + authorization=FirecrestAccessTokenAuth(access_token) + ) + + return client + + +# FIXME: Setup the auth state in the unit tests +# Since the auth state is not setup for the unit tests, +# the spawner's get_firecrest_client method will fail +# when trying to get a key from a none `auth_state` +SlurmSpawner.get_firecrest_client = get_firecrest_client + + def new_spawner(db, spawner_class=SlurmSpawner, **kwargs): user = db.query(orm.User).first() hub = Hub()