Disabled "Collapse" button in "Flat" view

Author: AdwaitSalankar

src/vorta/views/diff_result.py

@@ -190,10 +190,13 @@ def change_display_mode(self, selection: int):
         """
         if selection == 0:
             mode = FileTreeModel.DisplayMode.TREE
+            self.bCollapseAll.setEnabled(True)
         elif selection == 1:
             mode = FileTreeModel.DisplayMode.SIMPLIFIED_TREE
+            self.bCollapseAll.setEnabled(True)
         elif selection == 2:
             mode = FileTreeModel.DisplayMode.FLAT
+            self.bCollapseAll.setEnabled(False)
         else:
             raise Exception("Unknown item in comboBoxDisplayMode with index {}".format(selection))
 

src/vorta/views/extract_dialog.py

@@ -185,8 +185,10 @@ def change_display_mode(self, selection: int):
         """
         if selection == 0:
             mode = FileTreeModel.DisplayMode.TREE
+            self.bCollapseAll.setEnabled(True)
         elif selection == 1:
             mode = FileTreeModel.DisplayMode.SIMPLIFIED_TREE
+            self.bCollapseAll.setEnabled(True)
         else:
             raise Exception("Unknown item in comboBoxDisplayMode with index {}".format(selection))
 

tests/unit/test_diff.py

@@ -6,14 +6,17 @@
 import vorta.views.archive_tab
 from PyQt6.QtCore import QDateTime, QItemSelectionModel, Qt
 from PyQt6.QtWidgets import QMenu
+from vorta.store.models import ArchiveModel
 from vorta.views.diff_result import (
     ChangeType,
     DiffData,
+    DiffResultDialog,
     DiffTree,
     FileType,
     parse_diff_json,
     parse_diff_lines,
 )
+from vorta.views.partials.treemodel import FileTreeModel
 
 
 def setup_diff_result_window(qtbot, mocker, tab, borg_json_output, json_mock_file="diff_archives"):
@@ -457,3 +460,19 @@ def test_archive_diff_json_parser(line, expected):
 
     assert item.path == PurePath(expected[0]).parts
     assert item.data == DiffData(*expected[1:])
+
+
+@pytest.mark.parametrize(
+    "selection, expected_mode, expected_bCollapseAllEnabled",
+    [
+        (0, FileTreeModel.DisplayMode.TREE, True),
+        (1, FileTreeModel.DisplayMode.SIMPLIFIED_TREE, True),
+        (2, FileTreeModel.DisplayMode.FLAT, False),
+    ],
+)
+def test_change_display_mode(selection: int, expected_mode, expected_bCollapseAllEnabled):
+    dialog = DiffResultDialog(ArchiveModel(), ArchiveModel(), DiffTree())
+    dialog.change_display_mode(selection)
+
+    assert dialog.model.mode == expected_mode
+    assert dialog.bCollapseAll.isEnabled() == expected_bCollapseAllEnabled

tests/unit/test_extract.py

@@ -1,7 +1,15 @@
+import pytest
 import vorta.borg
 from PyQt6.QtCore import QModelIndex, Qt
-from vorta.views.extract_dialog import ExtractTree, FileData, FileType, parse_json_lines
-from vorta.views.partials.treemodel import FileSystemItem
+from vorta.store.models import ArchiveModel
+from vorta.views.extract_dialog import (
+    ExtractDialog,
+    ExtractTree,
+    FileData,
+    FileType,
+    parse_json_lines,
+)
+from vorta.views.partials.treemodel import FileSystemItem, FileTreeModel
 
 
 def prepare_borg(mocker, borg_json_output):
@@ -177,3 +185,15 @@ def test_selection():
 
     select(model, iab)
     assert a.data.checkstate == Qt.CheckState(1)
+
+
+@pytest.mark.parametrize(
+    "selection, expected_mode, expected_bCollapseAllEnabled",
+    [(0, FileTreeModel.DisplayMode.TREE, True), (1, FileTreeModel.DisplayMode.SIMPLIFIED_TREE, True)],
+)
+def test_change_display_mode(selection: int, expected_mode, expected_bCollapseAllEnabled):
+    dialog = ExtractDialog(ArchiveModel(), ExtractTree())
+    dialog.change_display_mode(selection)
+
+    assert dialog.model.mode == expected_mode
+    assert dialog.bCollapseAll.isEnabled() == expected_bCollapseAllEnabled