fix wait command problem

src/stream-analytics/azext_stream_analytics/_help.py

@@ -95,6 +95,15 @@
                az stream-analytics job stop --resource-group MyResourceGroup --name MyJobName
 """
 
+helps['stream-analytics job wait'] = """
+type: command
+short-summary: Place the CLI in a waiting state until a condition of the streaming job is met.
+examples:
+  - name: Pause executing next line of CLI script until the streaming job is successfully provisioned.
+    text: |
+        az stream-analytics job wait --resource-group MyResourceGroup --name MyJobName --created
+"""
+
 helps['stream-analytics input'] = """
     type: group
     short-summary: Commands to manage stream-analytics input.

src/stream-analytics/azext_stream_analytics/_params.py

@@ -74,6 +74,9 @@ def load_arguments(self, _):
     with self.argument_context('stream-analytics job stop') as c:
         c.argument('name', options_list=['--name', '-n'], help='The name of the streaming job.')
 
+    with self.argument_context('stream-analytics job wait') as c:
+        c.argument('job_name', options_list=['--name', '-n'], help='The name of the streaming job.')
+
     with self.argument_context('stream-analytics input create') as c:
         c.argument('job_name', id_part='name', help='The name of the streaming job.')
         c.argument('name', id_part='child_name_1', options_list=['--name', '-n'], help='The name of the input.')

src/stream-analytics/azext_stream_analytics/commands.py

@@ -37,7 +37,6 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'get_stream_analytics_input')
         g.custom_command('list', 'list_stream_analytics_input')
         g.custom_command('test', 'test_stream_analytics_input', supports_no_wait=True)
-        g.wait_command('wait')
 
     from ._client_factory import cf_outputs
     stream_analytics_outputs = CliCommandType(
@@ -49,7 +48,6 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'get_stream_analytics_output')
         g.custom_command('list', 'list_stream_analytics_output')
         g.custom_command('test', 'test_stream_analytics_output', supports_no_wait=True)
-        g.wait_command('wait')
 
     from ._client_factory import cf_transformations
     stream_analytics_transformations = CliCommandType(
@@ -70,7 +68,6 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'get_stream_analytics_function')
         g.custom_command('list', 'list_stream_analytics_function')
         g.custom_command('test', 'test_stream_analytics_function', supports_no_wait=True)
-        g.wait_command('wait')
 
     from ._client_factory import cf_subscriptions
     stream_analytics_subscriptions = CliCommandType(