From dce8972ce034a2d93bffbe3d6b47bb884479bd3a Mon Sep 17 00:00:00 2001 From: Noah Gorny Date: Fri, 11 Jun 2021 04:23:47 +0300 Subject: [PATCH] tests: unit: Fix test_command.py after new command addition --- tests/unit/test_commands.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_commands.py b/tests/unit/test_commands.py index f34f7e5387b..cb144c5f6da 100644 --- a/tests/unit/test_commands.py +++ b/tests/unit/test_commands.py @@ -11,7 +11,8 @@ # These are the expected names of the commands whose classes inherit from # IndexGroupCommand. -EXPECTED_INDEX_GROUP_COMMANDS = ['download', 'install', 'list', 'wheel'] +EXPECTED_INDEX_GROUP_COMMANDS = [ + 'download', 'index', 'install', 'list', 'wheel'] def check_commands(pred, expected): @@ -49,7 +50,9 @@ def test_session_commands(): def is_session_command(command): return isinstance(command, SessionCommandMixin) - expected = ['download', 'install', 'list', 'search', 'uninstall', 'wheel'] + expected = [ + 'download', 'index', 'install', 'list', 'search', 'uninstall', 'wheel' + ] check_commands(is_session_command, expected)