79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
|
import pytest
|
|||
|
|
from unittest.mock import MagicMock, patch
|
|||
|
|
from io import BytesIO
|
|||
|
|
from openpyxl import Workbook
|
|||
|
|
from app.services.agent_assets import AgentAssetService
|
|||
|
|
from app.schemas.agent_asset import AgentAssetOnlyOfficeCallbackWrite
|
|||
|
|
|
|||
|
|
def test_onlyoffice_callback_generates_summary_note():
|
|||
|
|
# Setup mock DB and repository
|
|||
|
|
db = MagicMock()
|
|||
|
|
service = AgentAssetService(db)
|
|||
|
|
service.repository = MagicMock()
|
|||
|
|
service.spreadsheet_manager = MagicMock()
|
|||
|
|
service._ensure_ready = MagicMock()
|
|||
|
|
|
|||
|
|
# Mock asset and metadata
|
|||
|
|
asset = MagicMock()
|
|||
|
|
asset.id = "test-asset"
|
|||
|
|
asset.name = "测试规则"
|
|||
|
|
service._require_spreadsheet_rule = MagicMock(return_value=asset)
|
|||
|
|
service._resolve_working_version = MagicMock(return_value="v1")
|
|||
|
|
|
|||
|
|
base_meta = MagicMock()
|
|||
|
|
base_meta.file_name = "test.xlsx"
|
|||
|
|
base_meta.storage_key = "old-key"
|
|||
|
|
base_meta.checksum = "old-checksum"
|
|||
|
|
service._resolve_spreadsheet_version_meta = MagicMock(return_value=("v1", base_meta))
|
|||
|
|
|
|||
|
|
# Create base workbook
|
|||
|
|
base_wb = Workbook()
|
|||
|
|
base_ws = base_wb.active
|
|||
|
|
base_ws["A1"] = "old value"
|
|||
|
|
|
|||
|
|
# Mock loading base workbook
|
|||
|
|
service._load_spreadsheet_for_compare = MagicMock(return_value=base_wb)
|
|||
|
|
service.spreadsheet_manager.resolve_storage_path = MagicMock()
|
|||
|
|
|
|||
|
|
# Create new content (modified)
|
|||
|
|
new_wb = Workbook()
|
|||
|
|
new_ws = new_wb.active
|
|||
|
|
new_ws["A1"] = "new value" # 1 cell changed
|
|||
|
|
new_ws["B2"] = "added" # 1 more cell changed
|
|||
|
|
|
|||
|
|
# Mock URL open to return new content
|
|||
|
|
new_content_bio = BytesIO()
|
|||
|
|
new_wb.save(new_content_bio)
|
|||
|
|
new_content = new_content_bio.getvalue()
|
|||
|
|
|
|||
|
|
with patch("app.services.agent_assets.urlopen") as mock_urlopen:
|
|||
|
|
mock_response = MagicMock()
|
|||
|
|
mock_response.read.return_value = new_content
|
|||
|
|
mock_response.__enter__.return_value = mock_response
|
|||
|
|
mock_urlopen.return_value = mock_response
|
|||
|
|
|
|||
|
|
# Mock upload_rule_spreadsheet
|
|||
|
|
service.upload_rule_spreadsheet = MagicMock()
|
|||
|
|
|
|||
|
|
# Execute callback handler
|
|||
|
|
payload = {
|
|||
|
|
"status": 2,
|
|||
|
|
"url": "http://onlyoffice/download",
|
|||
|
|
"users": ["test_user"]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
service.handle_rule_spreadsheet_onlyoffice_callback(
|
|||
|
|
"test-asset",
|
|||
|
|
version="v1",
|
|||
|
|
payload=payload
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Verify upload_rule_spreadsheet was called with correct change_note
|
|||
|
|
service.upload_rule_spreadsheet.assert_called_once()
|
|||
|
|
call_args = service.upload_rule_spreadsheet.call_args[1]
|
|||
|
|
assert "涉及 1 个 Sheet，共 2 处改动" in call_args["change_note"]
|
|||
|
|
assert call_args["actor"] == "test_user"
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
pytest.main([__file__])
|