53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
"""
|
|
Integration test for batch generation (stub)
|
|
"""
|
|
|
|
import pytest
|
|
from unittest.mock import Mock, MagicMock
|
|
from src.generation.batch_processor import BatchProcessor
|
|
from src.generation.service import ContentGenerator
|
|
|
|
|
|
def test_batch_processor_initialization():
|
|
"""Test BatchProcessor can be initialized"""
|
|
mock_generator = Mock(spec=ContentGenerator)
|
|
mock_content_repo = Mock()
|
|
mock_project_repo = Mock()
|
|
|
|
processor = BatchProcessor(
|
|
content_generator=mock_generator,
|
|
content_repo=mock_content_repo,
|
|
project_repo=mock_project_repo
|
|
)
|
|
|
|
assert processor is not None
|
|
assert processor.stats["total_jobs"] == 0
|
|
assert processor.stats["processed_jobs"] == 0
|
|
|
|
|
|
def test_batch_processor_stats_initialization():
|
|
"""Test BatchProcessor initializes stats correctly"""
|
|
mock_generator = Mock(spec=ContentGenerator)
|
|
mock_content_repo = Mock()
|
|
mock_project_repo = Mock()
|
|
|
|
processor = BatchProcessor(
|
|
content_generator=mock_generator,
|
|
content_repo=mock_content_repo,
|
|
project_repo=mock_project_repo
|
|
)
|
|
|
|
expected_keys = [
|
|
"total_jobs",
|
|
"processed_jobs",
|
|
"total_articles",
|
|
"generated_articles",
|
|
"augmented_articles",
|
|
"failed_articles"
|
|
]
|
|
|
|
for key in expected_keys:
|
|
assert key in processor.stats
|
|
assert processor.stats[key] == 0
|
|
|