[api] OPSAPS-23455 add option to associated case, upload yarn application support bundle to Cloudera

Added a new function create_yarn_application_diagnostics_bundle to yarn service.
The new function adds the support for new arguments ticketNumber and comments.

The signature of the function create_yarn_application_diagnostics_bundle is not
compatible with the existing function collect_yarn_application_diagnostics and
that is why, I added a new function and marked the older one as deprecated.

Author: amitsrivastava

python/src/cm_api/endpoints/services.py

@@ -468,16 +468,36 @@ def create_yarn_node_manager_remote_app_log_dir(self):
 
   def collect_yarn_application_diagnostics(self, *application_ids):
     """
+    DEPRECATED: use create_yarn_application_diagnostics_bundle on the Yarn service. Deprecated since v10.
+
     Collects the Diagnostics data for Yarn applications.
 
-    @param application_ids: An array of strings containing the id's of the
+    @param application_ids: An array of strings containing the ids of the
                             yarn applications.
     @return: Reference to the submitted command.
     @since: API v8
     """
     args = dict(applicationIds = application_ids)
     return self._cmd('yarnApplicationDiagnosticsCollection', api_version=8, data=args)
 
+  def create_yarn_application_diagnostics_bundle(self, application_ids, ticket_number=None, comments=None):
+    """
+    Collects the Diagnostics data for Yarn applications.
+
+    @param application_ids: An array of strings containing the ids of the
+                            yarn applications.
+    @param ticket_number: If applicable, the support ticket number of the issue
+                          being experienced on the cluster.
+    @param comments: Additional comments
+    @return: Reference to the submitted command.
+    @since: API v10
+    """
+    args = dict(applicationIds = application_ids,
+                ticketNumber = ticket_number,
+                comments = comments)
+
+    return self._cmd('yarnApplicationDiagnosticsCollection', api_version=10, data=args)
+
   def get_config(self, view = None):
     """
     Retrieve the service's configuration.

python/src/cm_api_tests/test_yarn.py

@@ -71,3 +71,13 @@ def test_collect_yarn_application_diagnostics(self):
     resp = service.collect_yarn_application_diagnostics('randomId-1', 'randomId-2', 'randomId-3')
 
     self.assertEquals('YarnApplicationDiagnosticsCollection', resp.name)
+
+  def test_create_yarn_application_diagnostics_bundle(self):
+    resource = utils.MockResource(self)
+    service = ApiService(resource, name="bar")
+
+    resource.expect("POST", "/cm/service/commands/yarnApplicationDiagnosticsCollection",
+        retdata={ 'name' : 'YarnApplicationDiagnosticsCollection' })
+    resp = service.create_yarn_application_diagnostics_bundle(['randomId-1', 'randomId-2', 'randomId-3'], 'test_ticket', 'test comment')
+
+    self.assertEquals('YarnApplicationDiagnosticsCollection', resp.name)