1
0
Fork 0
mirror of https://github.com/borgbase/vorta synced 2024-12-23 08:17:08 +00:00
vorta/tests/unit/test_profile.py

87 lines
4.1 KiB
Python
Raw Normal View History

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
2020-10-30 04:15:38 +00:00
def test_profile_add_delete(qapp, qtbot, mocker):
"""Tests adding and deleting profiles."""
2020-10-30 04:15:38 +00:00
main = qapp.main_window
# add profile and ensure it is created as intended
main.profile_add_action()
add_profile_window = main.window
2020-10-30 04:15:38 +00:00
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)
2020-10-30 04:15:38 +00:00
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'
2020-10-30 04:15:38 +00:00
def test_profile_edit(qapp, qtbot):
"""Tests editing/renaming a profile"""
2020-10-30 04:15:38 +00:00
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
2020-10-30 04:15:38 +00:00
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.
2020-10-30 04:15:38 +00:00
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)