diff --git a/src/db/client/async_.py b/src/db/client/async_.py index b4311733..d4368dd7 100644 --- a/src/db/client/async_.py +++ b/src/db/client/async_.py @@ -901,6 +901,7 @@ async def insert_url(self, session: AsyncSession, url_info: URLInfo) -> int: batch_id=url_info.batch_id, url_id=url_entry.id ) + session.add(link) return url_entry.id @session_manager diff --git a/tests/automated/integration/db/client/test_insert_urls.py b/tests/automated/integration/db/client/test_insert_urls.py index a9aaf1fe..9fd65eed 100644 --- a/tests/automated/integration/db/client/test_insert_urls.py +++ b/tests/automated/integration/db/client/test_insert_urls.py @@ -2,7 +2,9 @@ from src.core.enums import BatchStatus from src.db.models.instantiations.batch.pydantic import BatchInfo +from src.db.models.instantiations.link.batch_url import LinkBatchURL from src.db.models.instantiations.url.core.pydantic import URLInfo +from src.db.models.instantiations.url.core.sqlalchemy import URL @pytest.mark.asyncio @@ -46,3 +48,11 @@ async def test_insert_urls( assert insert_urls_info.original_count == 2 assert insert_urls_info.duplicate_count == 1 + + urls = await adb_client_test.get_all(URL) + assert len(urls) == 2 + + links: list[LinkBatchURL] = await adb_client_test.get_all(LinkBatchURL) + assert len(links) == 2 + for link in links: + assert link.batch_id == batch_id