mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-08 07:18:47 +00:00
Add tests for checksum and backup handlers
This commit is contained in:
58
src/tests/test_manager_checksum_backup.py
Normal file
58
src/tests/test_manager_checksum_backup.py
Normal file
@@ -0,0 +1,58 @@
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
||||
|
||||
from password_manager.manager import PasswordManager
|
||||
|
||||
|
||||
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)
|
||||
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("password_manager.manager.SCRIPT_CHECKSUM_FILE", chk_file)
|
||||
monkeypatch.setattr("password_manager.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("password_manager.manager.SCRIPT_CHECKSUM_FILE", chk_file)
|
||||
monkeypatch.setattr("password_manager.manager.calculate_checksum", lambda _: "abc")
|
||||
pm.handle_verify_checksum()
|
||||
out = capsys.readouterr().out
|
||||
assert "Checksum verification failed" in out
|
||||
|
||||
|
||||
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
|
Reference in New Issue
Block a user