Added new test

1 files changed, 41 insertions, 2 deletions

diff --git a/tests/unit/mainwindow/test_prompt.py b/tests/unit/mainwindow/test_prompt.py
index e67ab1702..8f32d6338 100644
--- a/tests/unit/mainwindow/test_prompt.py
+++ b/tests/unit/mainwindow/test_prompt.py
@@ -85,8 +85,6 @@ class TestFileCompletion:
 
         prompt.item_focus('next')
 
-        # qtbot.keyPress(prompt._lineedit, Qt.Key_Tab)
-        # qtbot.keyRelease(prompt._lineedit, Qt.Key_Tab)
         assert prompt._lineedit.text() == str(testdir / 'foo')
 
         # Deleting /[foo]
@@ -97,6 +95,47 @@ class TestFileCompletion:
         prompt.item_focus('next')
         prompt.item_focus('next')
         assert prompt._lineedit.text() == str(testdir / 'bar')
+    
+    def test_filtering_path(self, qtbot, tmp_path, get_prompt):
+        testdir = tmp_path / 'test'
+
+        for directory in ['bar', 'foo', 'bat']:
+            (testdir / directory).mkdir(parents=True)
+        
+        prompt = get_prompt(str(testdir) + os.sep)
+
+        # Make sure all directories are shown
+        num_rows = prompt._file_model.rowCount(prompt._root_index)
+        visible = []
+        for row in range(num_rows):
+            index = prompt._file_model.index(row, 0, prompt._file_model.index(os.path.basename(prompt._lineedit.text())))
+            if prompt._file_view.isRowHidden(index.row(), index.parent()):
+                visible.append(index.data())
+        
+        assert visible.sort() == ['bar', 'foo', 'bat'].sort()
+
+        # Only foo should be completed with f
+        qtbot.keyPress(prompt._lineedit, Qt.Key_F)
+        num_rows = prompt._file_model.rowCount(prompt._root_index)
+        visible = []
+        for row in range(num_rows):
+            index = prompt._file_model.index(row, 0, prompt._file_model.index(os.path.basename(prompt._lineedit.text())))
+            if prompt._file_view.isRowHidden(index.row(), index.parent()):
+                visible.append(index.data())
+        
+        assert visible.sort() == ['foo'].sort()
+
+        # bat and bar should show up with a typed
+        qtbot.keyPress(prompt._lineedit, Qt.Key_Backspace)
+        qtbot.keyPress(prompt._lineedit, Qt.Key_A)
+        num_rows = prompt._file_model.rowCount(prompt._root_index)
+        visible = []
+        for row in range(num_rows):
+            index = prompt._file_model.index(row, 0, prompt._file_model.index(os.path.basename(prompt._lineedit.text())))
+            if prompt._file_view.isRowHidden(index.row(), index.parent()):
+                visible.append(index.data())
+        
+        assert visible.sort() == ['bar', 'bat'].sort()
 
     @pytest.mark.linux
     def test_root_path(self, get_prompt):