Added unit tests for index based completion

1 files changed, 48 insertions, 4 deletions

diff --git a/tests/test_completion.py b/tests/test_completion.py
index c07deeaa..bdeae2e2 100644
--- a/tests/test_completion.py
+++ b/tests/test_completion.py
@@ -389,13 +389,12 @@ def test_path_completion_directories_only(request):
     assert path_complete(text, line, begidx, endidx, dir_only=True) == ['scripts' + os.path.sep]
 
 
-# List of strings used with flag and index based complete functions
+# List of strings used with flag and index based completion functions
 food_item_strs = ['Pizza', 'Hamburger', 'Ham', 'Potato']
-sports_equipment_strs = ['Soccer', 'Socks', 'Basketball', 'Football']
+sports_equipment_strs = ['Bat', 'Basket', 'Basketball', 'Football']
 
-# Dictionaries used with flag and index based complete functions
+# Dictionary used with flag based completion functions
 flag_dict = {'-f': food_item_strs, '-s': sports_equipment_strs}
-position_dict = {1: food_item_strs, 2: sports_equipment_strs}
 
 def test_flag_based_completion_single_end():
     text = 'Pi'
@@ -439,6 +438,51 @@ def test_flag_based_default_completer(request):
 
     assert flag_based_complete(text, line, begidx, endidx, flag_dict, path_complete) == ['conftest.py ']
 
+# Dictionary used with index based completion functions
+index_dict = {1: food_item_strs, 2: sports_equipment_strs}
+
+def test_index_based_completion_single_end():
+    text = 'Foo'
+    line = 'command Pizza Foo'
+    endidx = len(line)
+    begidx = endidx - len(text)
+
+    assert index_based_complete(text, line, begidx, endidx, index_dict) == ['Football ']
+
+def test_index_based_completion_single_mid():
+    text = 'Foo'
+    line = 'command Pizza Foo'
+    begidx = len(line) - len(text)
+    endidx = begidx + 1
+
+    assert index_based_complete(text, line, begidx, endidx, index_dict) == ['Football']
+
+def test_index_based_completion_multiple():
+    text = ''
+    line = 'command Pizza '
+    endidx = len(line)
+    begidx = endidx - len(text)
+    assert index_based_complete(text, line, begidx, endidx, index_dict) == sorted(sports_equipment_strs)
+
+def test_index_based_completion_nomatch():
+    text = 'q'
+    line = 'command q'
+    endidx = len(line)
+    begidx = endidx - len(text)
+    assert index_based_complete(text, line, begidx, endidx, index_dict) == []
+
+def test_index_based_default_completer(request):
+    test_dir = os.path.dirname(request.module.__file__)
+
+    text = 'c'
+    path = os.path.join(test_dir, text)
+    line = 'command Pizza Bat {}'.format(path)
+
+    endidx = len(line)
+    begidx = endidx - len(text)
+
+    assert flag_based_complete(text, line, begidx, endidx, flag_dict, path_complete) == ['conftest.py ']
+
 
 def test_parseline_command_and_args(cmd2_app):
     line = 'help history'