import sys
from pathlib import Path

sys.path.append(str(Path(__file__).resolve().parents[1]))

from seedpass.core.manager import PasswordManager, EncryptionMode
import queue


class FakeBackupManager:
    def __init__(self, calls):
        self.calls = calls

    def create_backup(self):
        self.calls["create"] += 1

    def restore_latest_backup(self):
        self.calls["restore"] += 1


def _make_pm():
    pm = PasswordManager.__new__(PasswordManager)
    pm.encryption_mode = EncryptionMode.SEED_ONLY
    pm.notifications = queue.Queue()
    return pm


def test_handle_verify_checksum_success(monkeypatch, tmp_path, capsys):
    pm = _make_pm()
    chk_file = tmp_path / "chk.txt"
    chk_file.write_text("abc")
    monkeypatch.setattr("seedpass.core.manager.SCRIPT_CHECKSUM_FILE", chk_file)
    monkeypatch.setattr("seedpass.core.manager.calculate_checksum", lambda _: "abc")
    pm.handle_verify_checksum()
    out = capsys.readouterr().out
    assert "Checksum verification passed." in out


def test_handle_verify_checksum_failure(monkeypatch, tmp_path, capsys):
    pm = _make_pm()
    chk_file = tmp_path / "chk.txt"
    chk_file.write_text("xyz")
    monkeypatch.setattr("seedpass.core.manager.SCRIPT_CHECKSUM_FILE", chk_file)
    monkeypatch.setattr("seedpass.core.manager.calculate_checksum", lambda _: "abc")
    pm.handle_verify_checksum()
    out = capsys.readouterr().out
    assert "Checksum verification failed" in out


def test_handle_verify_checksum_missing(monkeypatch, tmp_path, capsys):
    pm = _make_pm()
    chk_file = tmp_path / "chk.txt"
    monkeypatch.setattr("seedpass.core.manager.SCRIPT_CHECKSUM_FILE", chk_file)
    monkeypatch.setattr("seedpass.core.manager.calculate_checksum", lambda _: "abc")

    def raise_missing(*_args, **_kwargs):
        raise FileNotFoundError

    monkeypatch.setattr("seedpass.core.manager.verify_checksum", raise_missing)
    pm.handle_verify_checksum()
    note = pm.notifications.get_nowait()
    assert note.level == "WARNING"
    assert "generate script checksum" in note.message.lower()


def test_backup_and_restore_database(monkeypatch, capsys):
    pm = _make_pm()
    calls = {"create": 0, "restore": 0}
    pm.backup_manager = FakeBackupManager(calls)
    pm.backup_database()
    out1 = capsys.readouterr().out
    pm.restore_database()
    out2 = capsys.readouterr().out
    assert calls["create"] == 1
    assert calls["restore"] == 1
    assert "Backup created successfully." in out1
    assert "Database restored from the latest backup successfully." in out2