from pathlib import Path from typing import Optional import pytest from pytest import MonkeyPatch from pytest_mock import MockerFixture from dangerzone import util from dangerzone.gui import Application from dangerzone.gui.logic import DangerzoneGui from dangerzone.gui.updater import UpdaterThread from dangerzone.isolation_provider.dummy import Dummy def get_qt_app() -> Application: if Application.instance() is None: # type: ignore [call-arg] return Application() else: return Application.instance() # type: ignore [call-arg] def generate_isolated_updater( tmp_path: Path, monkeypatch: MonkeyPatch, app_mocker: Optional[MockerFixture] = None, ) -> UpdaterThread: """Generate an Updater class with its own settings.""" if app_mocker: app = app_mocker.MagicMock() else: app = get_qt_app() dummy = Dummy() # XXX: We can monkey-patch global state without wrapping it in a context manager, or # worrying that it will leak between tests, for two reasons: # # 1. Parallel tests in PyTest take place in different processes. # 2. The monkeypatch fixture tears down the monkey-patch after each test ends. monkeypatch.setattr(util, "get_config_dir", lambda: tmp_path) dangerzone = DangerzoneGui(app, isolation_provider=dummy) updater = UpdaterThread(dangerzone) return updater @pytest.fixture def updater( tmp_path: Path, monkeypatch: MonkeyPatch, mocker: MockerFixture ) -> UpdaterThread: return generate_isolated_updater(tmp_path, monkeypatch, mocker) @pytest.fixture def qt_updater(tmp_path: Path, monkeypatch: MonkeyPatch) -> UpdaterThread: return generate_isolated_updater(tmp_path, monkeypatch)