fix: updated upsert to flush and refresh the dao before returning

src/entities/models/dao.py

@@ -20,7 +20,7 @@ class FinancialInstitutionDao(AuditMixin, Base):
     name: Mapped[str] = mapped_column(index=True)
     is_active: Mapped[bool] = mapped_column(index=True)
     domains: Mapped[List["FinancialInstitutionDomainDao"]] = relationship(
-        "FinancialInstitutionDomainDao", back_populates="fi"
+        "FinancialInstitutionDomainDao", back_populates="fi", lazy="selectin"
     )
     tax_id: Mapped[str] = mapped_column(String(9), unique=True, nullable=True)
     rssd_id: Mapped[int] = mapped_column(unique=True, nullable=True)

src/entities/repos/institutions_repo.py

@@ -80,7 +80,8 @@ async def upsert_institution(session: AsyncSession, fi: FinancialInstitutionDto)
         else:
             for key, value in fi_data.items():
                 setattr(db_fi, key, value)
-        await session.commit()
+        await session.flush([db_fi])
+        await session.refresh(db_fi)
         return db_fi
 
 

tests/entities/repos/test_institutions_repo.py

@@ -188,7 +188,7 @@ async def test_get_institutions_by_lei_list_item_not_existing(self, query_sessio
         assert len(res) == 0
 
     async def test_add_institution(self, transaction_session: AsyncSession):
-        await repo.upsert_institution(
+        db_fi = await repo.upsert_institution(
             transaction_session,
             FinancialInstitutionDao(
                 name="New Bank 123",
@@ -212,6 +212,7 @@ async def test_add_institution(self, transaction_session: AsyncSession):
                 top_holder_rssd_id=876543,
             ),
         )
+        assert db_fi.domains == []
         res = await repo.get_institutions(transaction_session)
         assert len(res) == 4