mirror of
https://github.com/borgbase/vorta
synced 2024-12-22 15:57:34 +00:00
86 lines
4.1 KiB
Python
86 lines
4.1 KiB
Python
import tempfile
|
|
|
|
from PyQt6 import QtCore
|
|
from PyQt6.QtWidgets import QDialogButtonBox, QFileDialog, QMessageBox, QToolTip
|
|
|
|
from vorta.store.models import BackupProfileModel, SourceFileModel
|
|
from vorta.views.export_window import ExportWindow
|
|
|
|
|
|
def test_profile_add_delete(qapp, qtbot, mocker):
|
|
"""Tests adding and deleting profiles."""
|
|
main = qapp.main_window
|
|
|
|
# add profile and ensure it is created as intended
|
|
main.profile_add_action()
|
|
add_profile_window = main.window
|
|
qtbot.keyClicks(add_profile_window.profileNameField, 'Test Profile')
|
|
save_button = add_profile_window.buttonBox.button(QDialogButtonBox.StandardButton.Save)
|
|
qtbot.mouseClick(save_button, QtCore.Qt.MouseButton.LeftButton)
|
|
assert BackupProfileModel.get_or_none(name='Test Profile') is not None
|
|
assert main.profileSelector.currentItem().text() == 'Test Profile'
|
|
|
|
# delete the new profile and ensure it is no longer available.
|
|
mocker.patch.object(QMessageBox, 'question', return_value=QMessageBox.StandardButton.Yes)
|
|
qtbot.mouseClick(main.profileDeleteButton, QtCore.Qt.MouseButton.LeftButton)
|
|
assert BackupProfileModel.get_or_none(name='Test Profile') is None
|
|
assert main.profileSelector.currentItem().text() == 'Default'
|
|
|
|
# attempt to delete the last remaining profile
|
|
# see that it cannot be deleted, a warning is displayed, and the profile remains
|
|
warning = mocker.patch.object(QToolTip, 'showText')
|
|
qtbot.mouseClick(main.profileDeleteButton, QtCore.Qt.MouseButton.LeftButton)
|
|
assert "Cannot delete the last profile." in warning.call_args[0][1]
|
|
assert BackupProfileModel.get_or_none(name='Default') is not None
|
|
assert main.profileSelector.currentItem().text() == 'Default'
|
|
|
|
|
|
def test_profile_edit(qapp, qtbot):
|
|
"""Tests editing/renaming a profile"""
|
|
main = qapp.main_window
|
|
|
|
# click to rename profile, clear the name field, type new profile name
|
|
qtbot.mouseClick(main.profileRenameButton, QtCore.Qt.MouseButton.LeftButton)
|
|
edit_profile_window = main.window
|
|
edit_profile_window.profileNameField.setText("")
|
|
qtbot.keyClicks(edit_profile_window.profileNameField, 'Test Profile')
|
|
save_button = edit_profile_window.buttonBox.button(QDialogButtonBox.StandardButton.Save)
|
|
qtbot.mouseClick(save_button, QtCore.Qt.MouseButton.LeftButton)
|
|
|
|
# assert a profile by the old name no longer exists, and the newly named profile does exist and is selected.
|
|
assert BackupProfileModel.get_or_none(name='Default') is None
|
|
assert BackupProfileModel.get_or_none(name='Test Profile') is not None
|
|
assert main.profileSelector.currentItem().text() == 'Test Profile'
|
|
|
|
|
|
def test_profile_import_no_duplicate_sources(qapp, qtbot, mocker):
|
|
"""Tests importing an existing profile and choosing to overwrite does not add duplicate sources."""
|
|
main = qapp.main_window
|
|
|
|
# Create a new profile and add sources
|
|
profile = BackupProfileModel.create(name='Test Profile')
|
|
sources = ['/path/to/source1', '/path/to/source2']
|
|
for source_path in sources:
|
|
SourceFileModel.create(dir=source_path, path=source_path, profile=profile)
|
|
|
|
# Export the profile
|
|
with tempfile.NamedTemporaryFile(suffix=".json", delete=False) as temp_file:
|
|
temp_file_path = temp_file.name
|
|
|
|
export_window = ExportWindow(profile)
|
|
mocker.patch.object(QFileDialog, 'getSaveFileName', return_value=(temp_file_path, 'JSON (*.json)'))
|
|
export_window.run()
|
|
|
|
# Mock the QMessageBox to return 'Yes' to overwrite the profile
|
|
mocker.patch.object(QMessageBox, 'question', return_value=QMessageBox.StandardButton.Yes)
|
|
|
|
# Mock the QFileDialog to return the name of the temporary file
|
|
mocker.patch.object(QFileDialog, 'getOpenFileName', return_value=(temp_file_path, 'JSON (*.json)'))
|
|
|
|
# Import the profile and choose to overwrite
|
|
main.profile_import_action()
|
|
|
|
# Check that the sources in the profile are the same as before and that there are no duplicates
|
|
imported_sources = [source.dir for source in SourceFileModel.select().where(SourceFileModel.profile == profile)]
|
|
assert set(imported_sources) == set(sources)
|
|
assert len(imported_sources) == len(sources)
|