Merges pull request #14 v3.9.6

gr user add name, no sub

1 files changed, 18 insertions, 57 deletions

diff --git a/tests/managers/gr/test_authentication.py b/tests/managers/gr/test_authentication.py
index 1310c79..324bfa6 100644
--- a/tests/managers/gr/test_authentication.py
+++ b/tests/managers/gr/test_authentication.py
@@ -1,11 +1,9 @@
-import logging
 from collections.abc import Callable
 from uuid import uuid4
 
 import pytest
 
 from generalresearch.managers.gr.authentication import GRTokenManager, GRUserManager
-from generalresearch.managers.gr.team import TeamManager
 from generalresearch.models.gr.authentication import GRToken, GRUser
 from generalresearch.pg_helper import PostgresConfig
 from generalresearch.redis_helper import RedisConfig
@@ -14,12 +12,13 @@ SSO_ISSUER = ""
 
 
 class TestGRUserManager:
-
     def test_create(self, gr_user: GRUser, gr_user_manager: GRUserManager):
+        assert gr_user.id is not None
         instance = gr_user_manager.get_by_id(gr_user.id)
         assert isinstance(instance, GRUser)
         assert gr_user.id == instance.id
 
+        assert gr_user.id is not None
         instance2 = gr_user_manager.get_by_id(gr_user.id)
         assert isinstance(instance2, GRUser)
         assert gr_user.model_dump_json() == instance2.model_dump_json()
@@ -28,32 +27,25 @@ class TestGRUserManager:
         with pytest.raises(expected_exception=ValueError) as cm:
             gr_user_manager.get_by_id(gr_user_id=999_999_999)
         assert "GRUser not found" in str(cm.value)
-
+        assert gr_user.id is not None
         instance = gr_user_manager.get_by_id(gr_user_id=gr_user.id)
         assert isinstance(instance, GRUser)
-        assert instance.sub == gr_user.sub
-
-    def test_get_by_sub(self, gr_user: GRUser, gr_user_manager: GRUserManager):
-        with pytest.raises(expected_exception=ValueError) as cm:
-            gr_user_manager.get_by_sub(sub=uuid4().hex)
-        assert "GRUser not found" in str(cm.value)
-
-        instance = gr_user_manager.get_by_sub(sub=gr_user.sub)
-        assert isinstance(instance, GRUser)
-        assert instance.id == gr_user.id
+        assert instance.email == gr_user.email
 
-    def test_get_by_sub_or_create(
-        self, gr_user: GRUser, gr_user_manager: GRUserManager
+    def test_get_by_email(
+        self,
+        gr_user_factory: Callable[..., GRUser],
+        gr_user_manager: GRUserManager,
     ):
-        sub = f"{uuid4().hex}-{uuid4().hex}"
+        email = f"{uuid4().hex}@example.com"
+        gr_user = gr_user_factory(email=email)
 
-        with pytest.raises(expected_exception=ValueError) as cm:
-            gr_user_manager.get_by_sub(sub=sub)
-        assert "GRUser not found" in str(cm.value)
+        with pytest.raises(expected_exception=ValueError, match="GRUser not found"):
+            gr_user_manager.get_by_email(email="missing@example.com")
 
-        instance = gr_user_manager.get_by_sub_or_create(sub=sub)
+        instance = gr_user_manager.get_by_email(email=email)
         assert isinstance(instance, GRUser)
-        assert instance.sub == sub
+        assert instance.id == gr_user.id
 
     def test_get_all(
         self, gr_user_factory: Callable[..., GRUser], gr_user_manager: GRUserManager
@@ -70,21 +62,10 @@ class TestGRUserManager:
         assert isinstance(res, list)
         assert res == []
 
-    def test_list_product_uuids(
-        self,
-        caplog,
-        gr_user: GRUser,
-        gr_user_manager: GRUserManager,
-        thl_web_rr: PostgresConfig,
-    ):
-        with caplog.at_level(logging.WARNING):
-            gr_user_manager.list_product_uuids(user=gr_user, thl_pg_config=thl_web_rr)
-        assert "prefetch not run" in caplog.text
-
 
 class TestGRTokenManager:
-
     def test_create(self, gr_user: GRUser, gr_token_manager: GRTokenManager):
+        assert gr_user.id is not None
         assert gr_token_manager.create(user_id=gr_user.id) is None
 
         token = gr_token_manager.get_by_user_id(user_id=gr_user.id)
@@ -92,6 +73,7 @@ class TestGRTokenManager:
         assert gr_user.id == token.user_id
 
     def test_get_by_user_id(self, gr_user: GRUser, gr_token_manager: GRTokenManager):
+        assert gr_user.id is not None
         assert gr_token_manager.create(user_id=gr_user.id) is None
 
         token = gr_token_manager.get_by_user_id(user_id=gr_user.id)
@@ -105,7 +87,7 @@ class TestGRTokenManager:
         gr_db: PostgresConfig,
         gr_redis_config: RedisConfig,
     ):
-
+        assert gr_user.id is not None
         gr_token_manager.create(user_id=gr_user.id)
 
         token: GRToken | None = gr_token_manager.get_by_user_id(user_id=gr_user.id)
@@ -120,6 +102,7 @@ class TestGRTokenManager:
         gr_user: GRUser,
         gr_token_manager: GRTokenManager,
     ):
+        assert gr_user.id is not None
         gr_token_manager.create(user_id=gr_user.id)
         token = gr_token_manager.get_by_user_id(user_id=gr_user.id)
         assert isinstance(token, GRToken)
@@ -131,25 +114,3 @@ class TestGRTokenManager:
         with pytest.raises(expected_exception=Exception) as cm:
             gr_token_manager.get_by_key(api_key=uuid4().hex)
         assert "No GRUser with token of " in str(cm.value)
-
-    @pytest.mark.skip(reason="no idea how to actually test this...")
-    def test_get_by_sso_key(
-        self,
-        gr_team_manager: TeamManager,
-        gr_redis_config: RedisConfig,
-    ):
-
-        api_key = "..."
-        jwks = {
-            # ...
-        }
-
-        instance = gr_team_manager.get_by_key(
-            api_key=api_key,
-            jwks=jwks,
-            audience="...",
-            issuer=SSO_ISSUER,
-            gr_redis_config=gr_redis_config,
-        )
-
-        assert isinstance(instance, GRToken)