from unittest.mock import MagicMock, patch

from backend.pashub_fetcher.token_getter import get_token_from_local_storage


def _configure_playwright_mock(mock_sync_playwright: MagicMock) -> None:
    mock_page = MagicMock()
    mock_page.url = "https://pashub.net/dashboard"
    mock_page.evaluate.return_value = "fake-token"

    mock_context = MagicMock()
    mock_context.new_page.return_value = mock_page

    mock_browser = MagicMock()
    mock_browser.new_context.return_value = mock_context

    mock_p = MagicMock()
    mock_p.chromium.launch.return_value = mock_browser

    mock_sync_playwright.return_value.__enter__.return_value = mock_p


@patch("backend.pashub_fetcher.token_getter.shutil.rmtree")
@patch("backend.pashub_fetcher.token_getter.glob.glob")
@patch("backend.pashub_fetcher.token_getter.sync_playwright")
def test_playwright_tmp_dirs_are_cleaned_up_after_browser_close(
    mock_sync_playwright: MagicMock,
    mock_glob: MagicMock,
    mock_rmtree: MagicMock,
) -> None:
    # Arrange
    fake_artifacts = ["/tmp/playwright-artifacts-abc12"]
    fake_profiles = ["/tmp/playwright_chromiumdev_profile-xyz99"]

    def glob_side_effect(pattern: str) -> list[str]:
        if "playwright-artifacts-*" in pattern:
            return fake_artifacts
        if "playwright_chromiumdev_profile-*" in pattern:
            return fake_profiles
        return []

    mock_glob.side_effect = glob_side_effect
    _configure_playwright_mock(mock_sync_playwright)

    # Act
    get_token_from_local_storage("user@example.com", "secret")

    # Assert
    mock_rmtree.assert_any_call("/tmp/playwright-artifacts-abc12", ignore_errors=True)
    mock_rmtree.assert_any_call("/tmp/playwright_chromiumdev_profile-xyz99", ignore_errors=True)