import io
import pytest
import peewee
import restatic
from restatic.application import RestaticApp
from restatic.models import RepoModel, SourceDirModel
@pytest.fixture()
def app(tmpdir, qtbot):
tmp_db = tmpdir.join("settings.sqlite")
mock_db = peewee.SqliteDatabase(str(tmp_db))
restatic.models.init_db(mock_db)
app = RestaticApp([])
qtbot.addWidget(app.main_window)
return app
@pytest.fixture()
def app_with_repo(app):
profile = app.main_window.current_profile
new_repo = RepoModel(url="sftp:user@host.de:/srv/restic-repo")
new_repo.save()
profile.repo = new_repo
profile.save()
source_dir = SourceDirModel(dir="/tmp", repo=new_repo)
source_dir.save()
return app
@pytest.fixture
def restic_json_output():
def _read_json(subcommand):
stdout = open(f"tests/restic_json_output/{subcommand}_stdout.json").read()
stderr = open(f"tests/restic_json_output/{subcommand}_stderr.json").read()
return io.StringIO(stdout), io.StringIO(stderr)
return _read_json
