kai/invoice-master-poc-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

re-structure

Browse Source
This commit is contained in:
Yaojia Wang
2026-02-01 22:55:31 +01:00
parent 400b12a967
commit b602d0a340
176 changed files with 856 additions and 853 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
tests/data/repositories/test_batch_upload_repository.py
View File

@@ -9,8 +9,8 @@ from datetime import datetime, timezone
from unittest.mock import MagicMock, patch
from uuid import uuid4, UUID
from inference.data.admin_models import BatchUpload, BatchUploadFile
from inference.data.repositories.batch_upload_repository import BatchUploadRepository
from backend.data.admin_models import BatchUpload, BatchUploadFile
from backend.data.repositories.batch_upload_repository import BatchUploadRepository
class TestBatchUploadRepository:
@@ -54,7 +54,7 @@ class TestBatchUploadRepository:
def test_create_returns_batch(self, repo):
"""Test create returns created batch upload."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
mock_ctx.return_value.__exit__ = MagicMock(return_value=False)
@@ -70,7 +70,7 @@ class TestBatchUploadRepository:
def test_create_with_upload_source(self, repo):
"""Test create with custom upload source."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
mock_ctx.return_value.__exit__ = MagicMock(return_value=False)
@@ -87,7 +87,7 @@ class TestBatchUploadRepository:
def test_create_default_upload_source(self, repo):
"""Test create uses default upload source."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
mock_ctx.return_value.__exit__ = MagicMock(return_value=False)
@@ -107,7 +107,7 @@ class TestBatchUploadRepository:
def test_get_returns_batch(self, repo, sample_batch):
"""Test get returns batch when exists."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = sample_batch
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -121,7 +121,7 @@ class TestBatchUploadRepository:
def test_get_returns_none_when_not_found(self, repo):
"""Test get returns None when batch not found."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = None
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -138,7 +138,7 @@ class TestBatchUploadRepository:
def test_update_updates_batch(self, repo, sample_batch):
"""Test update updates batch fields."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = sample_batch
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -156,7 +156,7 @@ class TestBatchUploadRepository:
def test_update_ignores_unknown_fields(self, repo, sample_batch):
"""Test update ignores unknown fields."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = sample_batch
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -171,7 +171,7 @@ class TestBatchUploadRepository:
def test_update_not_found(self, repo):
"""Test update does nothing when batch not found."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = None
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -183,7 +183,7 @@ class TestBatchUploadRepository:
def test_update_multiple_fields(self, repo, sample_batch):
"""Test update can update multiple fields."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = sample_batch
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -206,7 +206,7 @@ class TestBatchUploadRepository:
def test_create_file_returns_file(self, repo):
"""Test create_file returns created file record."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
mock_ctx.return_value.__exit__ = MagicMock(return_value=False)
@@ -221,7 +221,7 @@ class TestBatchUploadRepository:
def test_create_file_with_kwargs(self, repo):
"""Test create_file with additional kwargs."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
mock_ctx.return_value.__exit__ = MagicMock(return_value=False)
@@ -242,7 +242,7 @@ class TestBatchUploadRepository:
def test_update_file_updates_file(self, repo, sample_file):
"""Test update_file updates file fields."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = sample_file
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -258,7 +258,7 @@ class TestBatchUploadRepository:
def test_update_file_ignores_unknown_fields(self, repo, sample_file):
"""Test update_file ignores unknown fields."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = sample_file
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -273,7 +273,7 @@ class TestBatchUploadRepository:
def test_update_file_not_found(self, repo):
"""Test update_file does nothing when file not found."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = None
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -285,7 +285,7 @@ class TestBatchUploadRepository:
def test_update_file_multiple_fields(self, repo, sample_file):
"""Test update_file can update multiple fields."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.get.return_value = sample_file
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -304,7 +304,7 @@ class TestBatchUploadRepository:
def test_get_files_returns_list(self, repo, sample_file):
"""Test get_files returns list of files."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.exec.return_value.all.return_value = [sample_file]
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -317,7 +317,7 @@ class TestBatchUploadRepository:
def test_get_files_returns_empty_list(self, repo):
"""Test get_files returns empty list when no files."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.exec.return_value.all.return_value = []
mock_ctx.return_value.__enter__ = MagicMock(return_value=mock_session)
@@ -333,7 +333,7 @@ class TestBatchUploadRepository:
def test_get_paginated_returns_batches_and_total(self, repo, sample_batch):
"""Test get_paginated returns list of batches and total count."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.exec.return_value.one.return_value = 1
mock_session.exec.return_value.all.return_value = [sample_batch]
@@ -347,7 +347,7 @@ class TestBatchUploadRepository:
def test_get_paginated_with_pagination(self, repo, sample_batch):
"""Test get_paginated with limit and offset."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.exec.return_value.one.return_value = 100
mock_session.exec.return_value.all.return_value = [sample_batch]
@@ -360,7 +360,7 @@ class TestBatchUploadRepository:
def test_get_paginated_empty_results(self, repo):
"""Test get_paginated with no results."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.exec.return_value.one.return_value = 0
mock_session.exec.return_value.all.return_value = []
@@ -374,7 +374,7 @@ class TestBatchUploadRepository:
def test_get_paginated_with_admin_token(self, repo, sample_batch):
"""Test get_paginated with admin_token parameter (deprecated, ignored)."""
with patch("inference.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
with patch("backend.data.repositories.batch_upload_repository.get_session_context") as mock_ctx:
mock_session = MagicMock()
mock_session.exec.return_value.one.return_value = 1
mock_session.exec.return_value.all.return_value = [sample_batch]