Allow top level commands to be renamed

jamesls · jamesls · commit 48b2b45b9efa · 2013-08-21T16:04:54.000-07:00
Also adding tests to verify this.
diff --git a/awscli/clidriver.py b/awscli/clidriver.py
@@ -321,7 +321,9 @@ def create_help_command(self):
         return ServiceHelpCommand(session=self.session,
                                   obj=service_object,
                                   command_table=command_table,
-                                  arg_table=None)
+                                  arg_table=None,
+                                  event_class='Operation',
+                                  name=self._name)
 
     def _create_parser(self):
         command_table = self._get_command_table()
diff --git a/awscli/help.py b/awscli/help.py
@@ -237,13 +237,20 @@ class ServiceHelpCommand(HelpCommand):
 
     EventHandlerClass = ServiceDocumentEventHandler
 
+    def __init__(self, session, obj, command_table, arg_table, name,
+                 event_class):
+        super(ServiceHelpCommand, self).__init__(session, obj, command_table,
+                                                 arg_table)
+        self._name = name
+        self._event_class = event_class
+
     @property
     def event_class(self):
-        return 'Service'
+        return self._event_class
 
     @property
     def name(self):
-        return self.obj.endpoint_prefix
+        return self._name
 
 
 class OperationHelpCommand(HelpCommand):
@@ -263,7 +270,6 @@ def __init__(self, session, service, operation, arg_table, name,
         self._name = name
         self._event_class = event_class
 
-
     @property
     def event_class(self):
         return self._event_class
diff --git a/tests/unit/customizations/test_renames.py b/tests/unit/customizations/test_renames.py
@@ -0,0 +1,33 @@
+# Copyright 2013 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+from tests import unittest
+from tests import BaseAWSHelpOutputTest
+
+import mock
+
+from awscli.customizations import utils
+
+
+class TestCommandTableRenames(BaseAWSHelpOutputTest):
+
+    def test_rename_command_table(self):
+        handler = lambda command_table, **kwargs: utils.rename_command(
+            command_table, 'ec2', 'fooec2')
+        # Verify that we can rename a top level command.
+        self.session.register('building-command-table.main', handler)
+        self.driver.main(['fooec2', 'help'])
+        self.assert_contains('fooec2')
+
+        # We can also see subcommands help as well.
+        self.driver.main(['fooec2', 'run-instances', 'help'])
+        self.assert_contains('run-instances')
