mirror of
https://github.com/borgbase/vorta
synced 2024-12-31 20:27:00 +00:00
386 lines
13 KiB
Python
386 lines
13 KiB
Python
|
"""
|
||
|
These tests compare the output of the diff command with the expected output.
|
||
|
"""
|
||
|
|
||
|
import pytest
|
||
|
import vorta.borg
|
||
|
import vorta.utils
|
||
|
import vorta.views.archive_tab
|
||
|
from pkg_resources import parse_version
|
||
|
from vorta.borg.diff import BorgDiffJob
|
||
|
from vorta.views.diff_result import (
|
||
|
ChangeType,
|
||
|
DiffTree,
|
||
|
FileType,
|
||
|
ParseThread,
|
||
|
)
|
||
|
|
||
|
|
||
|
@pytest.mark.parametrize(
|
||
|
'archive_name_1, archive_name_2, expected',
|
||
|
[
|
||
|
(
|
||
|
'test-archive1',
|
||
|
'test-archive2',
|
||
|
[
|
||
|
{
|
||
|
'subpath': 'dir',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.MODIFIED,
|
||
|
'modified': None,
|
||
|
},
|
||
|
'min_version': '1.2.4',
|
||
|
'max_version': '1.2.4',
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'file',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.MODIFIED,
|
||
|
'modified': (0, 0),
|
||
|
},
|
||
|
'min_version': '1.2.4',
|
||
|
'max_version': '1.2.4',
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'chrdev',
|
||
|
'data': {
|
||
|
'file_type': FileType.CHRDEV,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
'modified': None,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'fifo',
|
||
|
'data': {
|
||
|
'file_type': FileType.FIFO,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
'modified': None,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'hardlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
'modified': None,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'symlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.LINK,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
'modified': None,
|
||
|
},
|
||
|
},
|
||
|
],
|
||
|
),
|
||
|
(
|
||
|
'test-archive2',
|
||
|
'test-archive3',
|
||
|
[
|
||
|
{
|
||
|
'subpath': 'borg_src',
|
||
|
'match_startsWith': True,
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.MODIFIED,
|
||
|
'modified': None,
|
||
|
},
|
||
|
'min_version': '1.2.4',
|
||
|
'max_version': '1.2.4',
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'dir',
|
||
|
'data': {
|
||
|
'file_type': FileType.DIRECTORY,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
'modified': None,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'chrdev',
|
||
|
'data': {
|
||
|
'file_type': FileType.CHRDEV,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'fifo',
|
||
|
'data': {
|
||
|
'file_type': FileType.FIFO,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'file',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'hardlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'symlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.LINK,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'dir1',
|
||
|
'data': {
|
||
|
'file_type': FileType.DIRECTORY,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'chrdev',
|
||
|
'data': {
|
||
|
'file_type': FileType.CHRDEV,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'fifo',
|
||
|
'data': {
|
||
|
'file_type': FileType.FIFO,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'file',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'hardlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'symlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.LINK,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
],
|
||
|
),
|
||
|
(
|
||
|
'test-archive3',
|
||
|
'test-archive4',
|
||
|
[
|
||
|
{
|
||
|
'subpath': 'dir1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.MODIFIED,
|
||
|
},
|
||
|
'min_version': '1.2.4',
|
||
|
'max_version': '1.2.4',
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'chrdev',
|
||
|
'data': {
|
||
|
'file_type': FileType.CHRDEV,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'chrdev1',
|
||
|
'data': {
|
||
|
'file_type': FileType.CHRDEV,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'fifo',
|
||
|
'data': {
|
||
|
'file_type': FileType.FIFO,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'fifo1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FIFO,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'file',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'file1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'hardlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'hardlink1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'symlink',
|
||
|
'data': {
|
||
|
'file_type': FileType.LINK,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'symlink1',
|
||
|
'data': {
|
||
|
'file_type': FileType.LINK,
|
||
|
'change_type': ChangeType.ADDED,
|
||
|
},
|
||
|
},
|
||
|
],
|
||
|
),
|
||
|
(
|
||
|
'test-archive4',
|
||
|
'test-archive5',
|
||
|
[
|
||
|
{
|
||
|
'subpath': 'dir1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.MODIFIED,
|
||
|
},
|
||
|
'min_version': '1.2.4',
|
||
|
'max_version': '1.2.4',
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'chrdev1',
|
||
|
'data': {
|
||
|
'file_type': FileType.CHRDEV,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'fifo1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FIFO,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'file1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'hardlink1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
'subpath': 'symlink1',
|
||
|
'data': {
|
||
|
'file_type': FileType.LINK,
|
||
|
'change_type': ChangeType.REMOVED,
|
||
|
},
|
||
|
},
|
||
|
],
|
||
|
),
|
||
|
(
|
||
|
'test-archive5',
|
||
|
'test-archive6',
|
||
|
[
|
||
|
{
|
||
|
'subpath': 'dir1',
|
||
|
'data': {
|
||
|
'file_type': FileType.FILE,
|
||
|
'change_type': ChangeType.MODIFIED,
|
||
|
},
|
||
|
'min_version': '1.2.4',
|
||
|
'max_version': '1.2.4',
|
||
|
},
|
||
|
],
|
||
|
),
|
||
|
],
|
||
|
)
|
||
|
def test_archive_diff_lines(qapp, qtbot, borg_version, create_test_repo, archive_name_1, archive_name_2, expected):
|
||
|
"""Test that the diff lines are parsed correctly for supported borg versions"""
|
||
|
parsed_borg_version = borg_version[1]
|
||
|
supports_fifo = parsed_borg_version > parse_version('1.1.18')
|
||
|
supports_chrdev = create_test_repo[2]
|
||
|
|
||
|
params = BorgDiffJob.prepare(vorta.store.models.BackupProfileModel.select().first(), archive_name_1, archive_name_2)
|
||
|
thread = BorgDiffJob(params['cmd'], params, qapp)
|
||
|
|
||
|
with qtbot.waitSignal(thread.result, **pytest._wait_defaults) as blocker:
|
||
|
blocker.connect(thread.updated)
|
||
|
thread.run()
|
||
|
|
||
|
diff_lines = blocker.args[0]['data']
|
||
|
json_lines = blocker.args[0]['params']['json_lines']
|
||
|
|
||
|
model = DiffTree()
|
||
|
model.setMode(model.DisplayMode.FLAT)
|
||
|
|
||
|
# Use ParseThread to parse the diff lines
|
||
|
parse_thread = ParseThread(diff_lines, json_lines, model)
|
||
|
parse_thread.start()
|
||
|
qtbot.waitUntil(lambda: parse_thread.isFinished(), **pytest._wait_defaults)
|
||
|
|
||
|
expected = [
|
||
|
item
|
||
|
for item in expected
|
||
|
if (
|
||
|
('min_version' not in item or parse_version(item['min_version']) <= parsed_borg_version)
|
||
|
and ('max_version' not in item or parse_version(item['max_version']) >= parsed_borg_version)
|
||
|
and (item['data']['file_type'] != FileType.FIFO or supports_fifo)
|
||
|
and (item['data']['file_type'] != FileType.CHRDEV or supports_chrdev)
|
||
|
)
|
||
|
]
|
||
|
|
||
|
# diff versions of borg produce inconsistent ordering of diff lines so we sort the expected and model
|
||
|
expected = sorted(expected, key=lambda item: item['subpath'])
|
||
|
sorted_model = sorted(
|
||
|
[model.index(index, 0).internalPointer() for index in range(model.rowCount())],
|
||
|
key=lambda item: item.subpath,
|
||
|
)
|
||
|
|
||
|
assert len(sorted_model) == len(expected)
|
||
|
|
||
|
for index, item in enumerate(expected):
|
||
|
if 'match_startsWith' in item and item['match_startsWith']:
|
||
|
assert sorted_model[index].subpath.startswith(item['subpath'])
|
||
|
else:
|
||
|
assert sorted_model[index].subpath == item['subpath']
|
||
|
|
||
|
for key, value in item['data'].items():
|
||
|
assert getattr(sorted_model[index].data, key) == value
|