test_visit_usecase.py 732 B

123456789101112131415161718
  1. import pytest
  2. from unittest.mock import AsyncMock, MagicMock, patch
  3. from app.use_cases.camoufox.visit import VisitUseCase
  4. class TestVisitUseCase:
  5. @patch("app.use_cases.camoufox.visit.CamoufoxWrapper")
  6. def test_execute_calls_wrapper_visit(self, mock_wrapper_class):
  7. mock_wrapper = MagicMock()
  8. mock_wrapper.visit = AsyncMock(return_value={"html": "<html>", "screenshot": None})
  9. mock_wrapper_class.return_value = mock_wrapper
  10. use_case = VisitUseCase()
  11. import asyncio
  12. result = asyncio.run(use_case.execute("https://example.com", {}))
  13. mock_wrapper.visit.assert_called_once_with("https://example.com", {})
  14. assert result == {"html": "<html>", "screenshot": None}