conftest respects relationships

letta-ai · Jun 19, 2024 · eb50aff · eb50aff
1 parent f34d7ec
commit eb50aff
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/orm/test_bases.py b/tests/orm/test_bases.py
@@ -20,5 +20,6 @@ def test_prefixed_ids(self, db_session):
         assert user.id.startswith('user-')
         assert str(user._id) in user.id
         with db_session as session:
-            assert user.organization.id.startswith('organization-')
-            assert str(user.organization._id) in user.organization.id
+            session.add(user)
+            assert user.organization.id.startswith('organization-'), "Organization id is prefixed incorrectly"
+            assert str(user.organization._id) in user.organization.id, "Organization id is not using the correct uuid"