feat: add explorer widget

2026-05-11 09:05:43 +02:00 · 2025-07-26 19:45:24 +02:00
parent 8b013d5dce
commit 1bec9bd9b2
11 changed files with 1195 additions and 0 deletions
@@ -0,0 +1,118 @@
+from pathlib import Path
+
+import pytest
+from qtpy.QtCore import QEvent, Qt
+from qtpy.QtGui import QMouseEvent
+
+from bec_widgets.widgets.containers.explorer.script_tree_widget import ScriptTreeWidget
+
+
+@pytest.fixture
+def script_tree(qtbot, tmpdir):
+    """Create a ScriptTreeWidget with the tmpdir directory"""
+    # Create test files and directories
+    (Path(tmpdir) / "test_file.py").touch()
+    (Path(tmpdir) / "test_dir").mkdir()
+    (Path(tmpdir) / "test_dir" / "nested_file.py").touch()
+
+    widget = ScriptTreeWidget()
+    widget.set_directory(str(tmpdir))
+    qtbot.addWidget(widget)
+    qtbot.waitExposed(widget)
+    yield widget
+
+
+def test_script_tree_set_directory(script_tree, tmpdir):
+    """Test setting the directory"""
+    assert script_tree.directory == str(tmpdir)
+
+
+def test_script_tree_hover_events(script_tree, qtbot):
+    """Test mouse hover events and actions button visibility"""
+
+    # Get the tree view and its viewport
+    tree_view = script_tree.tree
+    viewport = tree_view.viewport()
+
+    # Find the position of the first item (test_file.py)
+    index = script_tree.proxy_model.index(0, 0)  # first item
+    rect = tree_view.visualRect(index)
+    pos = rect.center()
+
+    # Initially, no item should be hovered
+    assert script_tree.delegate.hovered_index.isValid() == False
+
+    # Simulate a mouse move event over the item
+    mouse_event = QMouseEvent(
+        QEvent.Type.MouseMove,
+        pos,
+        tree_view.mapToGlobal(pos),
+        Qt.MouseButton.NoButton,
+        Qt.MouseButton.NoButton,
+        Qt.KeyboardModifier.NoModifier,
+    )
+
+    # Send the event to the viewport (the event filter is installed on the viewport)
+    script_tree.eventFilter(viewport, mouse_event)
+
+    qtbot.wait(100)  # Allow time for the hover to be processed
+
+    # Now, the hover index should be set to the first item
+    assert script_tree.delegate.hovered_index.isValid() == True
+    assert script_tree.delegate.hovered_index.row() == index.row()
+
+    # Simulate mouse leaving the viewport
+    leave_event = QEvent(QEvent.Type.Leave)
+    script_tree.eventFilter(viewport, leave_event)
+
+    qtbot.wait(100)  # Allow time for the leave event to be processed
+
+    # After leaving, no item should be hovered
+    assert script_tree.delegate.hovered_index.isValid() == False
+
+
+@pytest.mark.timeout(10)
+def test_script_tree_on_item_clicked(script_tree, qtbot, tmpdir):
+    """Test that _on_item_clicked emits file_selected signal only for Python files"""
+
+    file_selected_signals = []
+    file_open_requested_signals = []
+
+    def on_file_selected(file_path):
+        file_selected_signals.append(file_path)
+
+    def on_file_open_requested(file_path):
+        file_open_requested_signals.append(file_path)
+
+    # Connect to the signal
+    script_tree.file_selected.connect(on_file_selected)
+    script_tree.file_open_requested.connect(on_file_open_requested)
+
+    # Wait until the model sees test_file.py
+    def has_py_file():
+        nonlocal py_file_index
+        root_index = script_tree.tree.rootIndex()
+        for i in range(script_tree.proxy_model.rowCount(root_index)):
+            index = script_tree.proxy_model.index(i, 0, root_index)
+            source_index = script_tree.proxy_model.mapToSource(index)
+            if script_tree.model.fileName(source_index) == "test_file.py":
+                py_file_index = index
+                return True
+        return False
+
+    py_file_index = None
+    qtbot.waitUntil(has_py_file)
+
+    # Simulate clicking on the center of the item
+    script_tree._on_item_clicked(py_file_index)
+    qtbot.wait(100)  # Allow time for the click to be processed
+
+    py_file_index = None
+    qtbot.waitUntil(has_py_file)
+
+    script_tree._on_item_double_clicked(py_file_index)
+    qtbot.wait(100)
+
+    # Verify the signal was emitted with the correct path
+    assert len(file_selected_signals) == 1
+    assert Path(file_selected_signals[0]).name == "test_file.py"