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