fix: async operation samples

Author: daniel-sanche

gapic/templates/examples/feature_fragments.j2

@@ -230,7 +230,7 @@ request=request
 {# Note: this doesn't deal with enums or unions #}
 {# LROs return operation objects and paged requests return pager objects #}
 {% if transport == "grpc-async" and calling_form not in
-[calling_form_enum.LongRunningRequestPromise, calling_form_enum.RequestPagedAll] %}
+[calling_form_enum.RequestPaged, calling_form_enum.RequestPagedAll] %}
 await{{ " "}}
 {%- endif -%}
 {% if calling_form in [calling_form_enum.RequestStreamingBidi,
@@ -242,13 +242,6 @@ client.{{ render_method_name(sample)|trim }}({{ render_request_params_unary(samp
 {% endif %}
 {% endmacro %}
 
-{% macro operation_text(transport) %}
-{% if transport == "grpc-async" %}
-(await operation)
-{% else %}
-operation
-{% endif %}
-{% endmacro %}
 
 {# Setting up the method invocation is the responsibility of the caller: #}
 {# it's just easier to set up client side streaming and other things from outside this macro. #}
@@ -294,9 +287,7 @@ operation = {{ method_invocation_text|trim }}
 
 print("Waiting for operation to complete...")
 
-{% with operation_text = operation_text(transport) %}
-response = {{ operation_text|trim }}.result()
-{% endwith %}
+response = {% if transport == "grpc-async" %}await {% endif %}operation.result()
 
 # Handle the response
 {% for statement in response_statements %}

tests/unit/samplegen/test_template.py

@@ -957,7 +957,7 @@ def test_render_calling_form_longrunning_async():
 
                    print("Waiting for operation to complete...")
 
-                   response = (await operation).result()
+                   response = await operation.result()
 
                    # Handle the response
                    print("Test print statement")
@@ -1020,6 +1020,84 @@ def test_render_method_call_basic_async():
     )
 
 
+def test_render_method_call_lro_async():
+    check_template(
+        """
+        {% import "feature_fragments.j2" as frags %}
+        {{ frags.render_method_call({"rpc": "CategorizeMollusc",
+                                     "request": request,
+                                     "is_internal": False,
+                                    },
+                                  calling_form, calling_form_enum, transport) }}
+        """,
+        """
+        await client.categorize_mollusc(request=request)
+        """,
+        request=samplegen.FullRequest(
+            request_list=[
+                samplegen.TransformedRequest(base="video", body=True, single=None),
+                samplegen.TransformedRequest(base="audio", body=True, single=None),
+                samplegen.TransformedRequest(base="guess", body=True, single=None),
+            ],
+        ),
+        calling_form_enum=CallingForm,
+        calling_form=CallingForm.LongRunningRequestPromise,
+        transport="grpc-async",
+    )
+
+
+def test_render_method_call_paged_async():
+    check_template(
+        """
+        {% import "feature_fragments.j2" as frags %}
+        {{ frags.render_method_call({"rpc": "CategorizeMollusc",
+                                     "request": request,
+                                     "is_internal": False,
+                                    },
+                                  calling_form, calling_form_enum, transport) }}
+        """,
+        """
+        client.categorize_mollusc(request=request)
+        """,
+        request=samplegen.FullRequest(
+            request_list=[
+                samplegen.TransformedRequest(base="video", body=True, single=None),
+                samplegen.TransformedRequest(base="audio", body=True, single=None),
+                samplegen.TransformedRequest(base="guess", body=True, single=None),
+            ],
+        ),
+        calling_form_enum=CallingForm,
+        calling_form=CallingForm.RequestPaged,
+        transport="grpc-async",
+    )
+
+
+def test_render_method_call_paged_all_async():
+    check_template(
+        """
+        {% import "feature_fragments.j2" as frags %}
+        {{ frags.render_method_call({"rpc": "CategorizeMollusc",
+                                     "request": request,
+                                     "is_internal": False,
+                                    },
+                                  calling_form, calling_form_enum, transport) }}
+        """,
+        """
+        client.categorize_mollusc(request=request)
+        """,
+        request=samplegen.FullRequest(
+            request_list=[
+                samplegen.TransformedRequest(base="video", body=True, single=None),
+                samplegen.TransformedRequest(base="audio", body=True, single=None),
+                samplegen.TransformedRequest(base="guess", body=True, single=None),
+            ],
+        ),
+        calling_form_enum=CallingForm,
+        calling_form=CallingForm.RequestPagedAll,
+        transport="grpc-async",
+    )
+
+
 def test_render_method_call_basic_flattenable():
     check_template(
         """