From 45d66cd40d409a8a05c8dde7193f1e8469058fff Mon Sep 17 00:00:00 2001
From: stuppie
Date: Thu, 22 Oct 2026 18:29:27 -0600
Subject: trying to fix some tests

---
 tests/models/gr/test_team.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

(limited to 'tests/models/gr')

diff --git a/tests/models/gr/test_team.py b/tests/models/gr/test_team.py
index 8cc00c8..c6ca451 100644
--- a/tests/models/gr/test_team.py
+++ b/tests/models/gr/test_team.py
@@ -60,23 +60,29 @@ class TestTeam:
     def test_memberships(
         self,
         gr_team: Team,
-        gr_user: GRUser,
         gr_membership,
         gr_user_factory: Callable[..., GRUser],
         gr_membership_manager: MembershipManager,
     ):
+        gr_user = gr_user_factory(email="member@example.com")
+        gr_membership_manager.create(team=gr_team, gr_user=gr_user)
+
         assert gr_team.memberships is None
 
         gr_team.prefetch_memberships(gr_membership_manager=gr_membership_manager)
         assert isinstance(gr_team.memberships, list)
-        assert len(gr_team.memberships) == 1
-        assert gr_team.memberships[0].user_id == gr_user.id
+        assert len(gr_team.memberships) == 2
+        membership = next(
+            item for item in gr_team.memberships if item.user_id == gr_user.id
+        )
+        assert membership.user.id == gr_user.id
+        assert membership.user.email == gr_user.email
 
         # Create another new Membership
         gr_membership_manager.create(team=gr_team, gr_user=gr_user_factory())
-        assert len(gr_team.memberships) == 1
-        gr_team.prefetch_memberships(gr_membership_manager=gr_membership_manager)
         assert len(gr_team.memberships) == 2
+        gr_team.prefetch_memberships(gr_membership_manager=gr_membership_manager)
+        assert len(gr_team.memberships) == 3
 
     def test_gr_users(
         self,
-- 
cgit v1.2.3