mirror of https://github.com/borgbase/vorta
180 lines
4.7 KiB
Python
180 lines
4.7 KiB
Python
from PyQt5.QtCore import QModelIndex, Qt
|
|
import vorta.borg
|
|
from vorta.views.extract_dialog import ExtractTree, FileData, FileType, parse_json_lines
|
|
from vorta.views.partials.treemodel import FileSystemItem
|
|
|
|
|
|
def prepare_borg(mocker, borg_json_output):
|
|
stdout, stderr = borg_json_output("list_archive")
|
|
popen_result = mocker.MagicMock(stdout=stdout, stderr=stderr, returncode=0)
|
|
mocker.patch.object(vorta.borg.borg_job, "Popen", return_value=popen_result)
|
|
|
|
|
|
default = {
|
|
"mode": "drwxrwxr-x",
|
|
"user": "theuser",
|
|
"group": "theuser",
|
|
"uid": 1000,
|
|
"gid": 1000,
|
|
"path": "",
|
|
"healthy": True,
|
|
"source": "",
|
|
"linktarget": "",
|
|
"flags": None,
|
|
"mtime": "2022-05-13T14:33:57.305797",
|
|
"size": 0,
|
|
}
|
|
|
|
|
|
def updated(path, values):
|
|
d = default.copy()
|
|
d.update(values)
|
|
d["path"] = path
|
|
return d
|
|
|
|
|
|
def test_parser():
|
|
"""Test creating a tree with correct data from json lines."""
|
|
|
|
lines = [
|
|
updated("a", {}),
|
|
updated("a/b", {"mode": "-rwxrwxr-x"}),
|
|
updated("a/b/c", {}),
|
|
updated("a/b/d", {}),
|
|
updated("a/a", {}),
|
|
updated("a/a/a", {}),
|
|
updated("a/a/b", {}),
|
|
updated("a/a/c", {"healthy": False}),
|
|
updated("a/a/d", {}),
|
|
updated("a/a/e", {}),
|
|
]
|
|
|
|
model = ExtractTree()
|
|
parse_json_lines(lines, model)
|
|
|
|
index = model.indexPath(("a", "b"))
|
|
assert index != QModelIndex()
|
|
assert index.internalPointer().data.file_type == FileType.FILE
|
|
|
|
item = model.getItem(("a", "a", "c"))
|
|
assert item
|
|
assert item.data.health is False
|
|
|
|
|
|
def select(model, index):
|
|
model.setData(index, Qt.CheckState.Checked, Qt.ItemDataRole.CheckStateRole)
|
|
|
|
|
|
def deselect(model, index):
|
|
model.setData(index, Qt.CheckState.Unchecked, Qt.ItemDataRole.CheckStateRole)
|
|
|
|
|
|
def test_selection():
|
|
"""Test selecting items for extraction."""
|
|
lines = [
|
|
updated("a", {}),
|
|
updated("a/b", {}),
|
|
updated("a/b/c", {}),
|
|
updated("a/b/d", {}),
|
|
updated("a/a", {}),
|
|
updated("a/a/a", {}),
|
|
updated("a/a/b", {}),
|
|
updated("a/a/c", {}),
|
|
updated("a/a/d", {}),
|
|
updated("a/a/e", {}),
|
|
updated("c", {}),
|
|
]
|
|
model = ExtractTree()
|
|
parse_json_lines(lines, model)
|
|
|
|
# Test select
|
|
ic = model.index(1, 0, QModelIndex())
|
|
c: FileSystemItem[FileData] = ic.internalPointer()
|
|
|
|
select(model, ic)
|
|
assert c.data.checkstate == 2
|
|
assert c.data.checked_children == 0
|
|
|
|
# Test deselect
|
|
deselect(model, ic)
|
|
assert c.data.checkstate == 0
|
|
assert c.data.checked_children == 0
|
|
|
|
# Test select parent as well as children
|
|
ia = model.index(0, 0, QModelIndex())
|
|
a: FileSystemItem[FileData] = ia.internalPointer()
|
|
aa = model.getItem(("a", "a"))
|
|
aab = model.getItem(("a", "a", "b"))
|
|
ab = model.getItem(("a", "b"))
|
|
abc = model.getItem(("a", "b", "c"))
|
|
|
|
select(model, ia)
|
|
assert a.data.checkstate
|
|
assert a.data.checked_children == 2
|
|
assert a.data.checked_children == 2
|
|
assert aab.data.checkstate
|
|
assert ab.data.checkstate
|
|
assert aa.data.checkstate
|
|
assert aa.data.checked_children == 5
|
|
|
|
# Test deselect item as well as children with selected parent
|
|
iab = model.indexPath(("a", "b"))
|
|
deselect(model, iab)
|
|
|
|
assert a.data.checkstate == 1
|
|
assert aa.data.checkstate == 2
|
|
assert ab.data.checkstate == 0
|
|
assert abc.data.checkstate == 0
|
|
assert a.data.checked_children == 1
|
|
assert ab.data.checked_children == 0
|
|
|
|
# Test deselect item and children
|
|
deselect(model, ia)
|
|
|
|
assert a.data.checkstate == 0
|
|
assert aa.data.checkstate == 0
|
|
assert ab.data.checkstate == 0
|
|
assert a.data.checked_children == 0
|
|
assert aa.data.checked_children == 0
|
|
|
|
# Test select child with partially selected parent
|
|
iaac = model.indexPath(("a", "a", "c"))
|
|
|
|
select(model, ia)
|
|
deselect(model, iab)
|
|
deselect(model, iaac)
|
|
select(model, iab)
|
|
select(model, iaac)
|
|
|
|
assert a.data.checkstate == 1
|
|
assert aa.data.checkstate == 1
|
|
assert ab.data.checkstate == 2
|
|
|
|
assert a.data.checked_children == 2
|
|
assert ab.data.checked_children == 2
|
|
assert aa.data.checked_children == 5
|
|
|
|
# Test deselect all children with selected parent
|
|
iaa = model.indexPath(("a", "a"))
|
|
deselect(model, iaa)
|
|
deselect(model, iab)
|
|
|
|
assert a.data.checkstate == 0
|
|
assert a.data.checked_children == 0
|
|
|
|
# Test select child with deselected parent
|
|
select(model, iaac)
|
|
|
|
assert a.data.checkstate == 1
|
|
assert ab.data.checkstate == 0
|
|
assert aa.data.checkstate == 1
|
|
assert a.data.checked_children == 1
|
|
assert ab.data.checked_children == 0
|
|
assert aa.data.checked_children == 1
|
|
|
|
select(model, iaa)
|
|
assert a.data.checkstate == 1
|
|
|
|
select(model, iab)
|
|
assert a.data.checkstate == 1
|