1 files changed, 10 insertions, 4 deletions
diff --git a/tests/completions/completions.py b/tests/completions/completions.py
index f810c4f08..372ed7840 100644
--- a/tests/completions/completions.py
+++ b/tests/completions/completions.py
@@ -9,15 +9,13 @@ MAIN_COMMANDS = [
     'artifact ',
     'build ',
     'checkout ',
-    'fetch ',
     'help ',
     'init ',
     'pull ',
     'push ',
     'shell ',
     'show ',
-    'source-checkout ',
-    'track ',
+    'source ',
     'workspace '
 ]
 
@@ -50,6 +48,12 @@ MAIN_OPTIONS = [
     "--version ",
 ]
 
+SOURCE_COMMANDS = [
+    'checkout ',
+    'fetch ',
+    'track ',
+]
+
 WORKSPACE_COMMANDS = [
     'close ',
     'list ',
@@ -115,6 +119,7 @@ def assert_completion_failed(cli, cmd, word_idx, expected, cwd=None):
     ('bst ', 1, MAIN_COMMANDS),
     ('bst pu', 1, ['pull ', 'push ']),
     ('bst pul', 1, ['pull ']),
+    ('bst source ', 2, SOURCE_COMMANDS),
     ('bst w ', 1, ['workspace ']),
     ('bst workspace ', 2, WORKSPACE_COMMANDS),
 ])
@@ -267,9 +272,10 @@ def test_argument_element_invalid(datafiles, cli, project, cmd, word_idx, expect
 @pytest.mark.parametrize("cmd,word_idx,expected", [
     ('bst he', 1, ['help ']),
     ('bst help ', 2, MAIN_COMMANDS),
-    ('bst help fe', 2, ['fetch ']),
+    ('bst help in', 2, ['init ']),
     ('bst help p', 2, ['pull ', 'push ']),
     ('bst help p', 2, ['pull ', 'push ']),
+    ('bst help source ', 3, SOURCE_COMMANDS),
     ('bst help w', 2, ['workspace ']),
     ('bst help workspace ', 3, WORKSPACE_COMMANDS),
 ])