# -*- 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()