56 lines
2.4 KiB
Python
56 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""RPA 失败结果与 artifacts 单元测试。"""
|
||
from __future__ import annotations
|
||
|
||
import unittest
|
||
from unittest.mock import MagicMock, patch
|
||
|
||
from service.adapter import BatchItem, SimulatorBrowserRpaAdapter
|
||
from service.adapter.simulator_rpa import RpaError
|
||
|
||
|
||
class TestBatchSubmitFailureResult(unittest.TestCase):
|
||
@patch("service.adapter.simulator_rpa.find_chrome_executable", return_value="C:/Chrome/chrome.exe")
|
||
@patch("service.adapter.simulator_rpa.browser_session")
|
||
def test_rpa_error_converts_to_failed_result(self, mock_session: MagicMock, _chrome: MagicMock) -> None:
|
||
page = MagicMock()
|
||
mock_cm = MagicMock()
|
||
mock_cm.__enter__.return_value = page
|
||
mock_cm.__exit__.return_value = False
|
||
mock_session.return_value = mock_cm
|
||
|
||
adapter = SimulatorBrowserRpaAdapter(artifacts_dir="/tmp/art")
|
||
items = [BatchItem(row_index=1, name="A", account="1", amount=1.0)]
|
||
|
||
with patch.object(adapter, "_login", side_effect=RpaError("登录第一步失败：timeout", screenshot_path="/tmp/art/x.png")):
|
||
result = adapter.submit_batch("acct-demo-001", items)
|
||
|
||
self.assertFalse(result.ok)
|
||
self.assertIn("登录第一步失败", result.error_msg or "")
|
||
self.assertEqual(result.artifacts.get("adapter"), "simulator_browser_rpa")
|
||
self.assertEqual(result.artifacts.get("screenshot"), "/tmp/art/x.png")
|
||
|
||
@patch("service.adapter.simulator_rpa.find_chrome_executable", return_value="C:/Chrome/chrome.exe")
|
||
@patch("service.adapter.simulator_rpa.browser_session")
|
||
def test_unexpected_error_includes_adapter_name(self, mock_session: MagicMock, _chrome: MagicMock) -> None:
|
||
page = MagicMock()
|
||
mock_cm = MagicMock()
|
||
mock_cm.__enter__.return_value = page
|
||
mock_cm.__exit__.return_value = False
|
||
mock_session.return_value = mock_cm
|
||
|
||
adapter = SimulatorBrowserRpaAdapter(artifacts_dir=None)
|
||
items = [BatchItem(row_index=1, name="A", account="1", amount=1.0)]
|
||
|
||
with patch.object(adapter, "_login", side_effect=RuntimeError("boom")):
|
||
result = adapter.submit_batch("acct-demo-001", items)
|
||
|
||
self.assertFalse(result.ok)
|
||
self.assertIn("未预期异常", result.error_msg or "")
|
||
self.assertEqual(result.artifacts.get("adapter"), "simulator_browser_rpa")
|
||
self.assertNotIn("screenshot", result.artifacts)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|