fix: Add async context manager return types (#11448)

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
Co-authored-by: Anthonios Partheniou <[email protected]>

Author: gcf-owl-bot[bot]

packages/google-cloud-migrationcenter/google/cloud/migrationcenter/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.1.0"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-migrationcenter/google/cloud/migrationcenter_v1/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.1.0"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-migrationcenter/google/cloud/migrationcenter_v1/services/migration_center/async_client.py

@@ -6158,7 +6158,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MigrationCenterAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

packages/google-cloud-migrationcenter/google/cloud/migrationcenter_v1/services/migration_center/transports/rest.py

@@ -6870,7 +6870,7 @@ def __call__(
             request_kwargs = json_format.MessageToDict(request)
             transcoded_request = path_template.transcode(http_options, **request_kwargs)
 
-            body = json.loads(json.dumps(transcoded_request["body"]))
+            body = json.dumps(transcoded_request["body"])
             uri = transcoded_request["uri"]
             method = transcoded_request["method"]
 

packages/google-cloud-migrationcenter/google/cloud/migrationcenter_v1/types/migrationcenter.py

@@ -5467,7 +5467,7 @@ class Aggregation(proto.Message):
             Count the number of matching objects.
 
             This field is a member of `oneof`_ ``aggregation_function``.
-        sum_ (google.cloud.migrationcenter_v1.types.Aggregation.Sum):
+        sum (google.cloud.migrationcenter_v1.types.Aggregation.Sum):
             Sum over a numeric field.
 
             This field is a member of `oneof`_ ``aggregation_function``.
@@ -5521,7 +5521,7 @@ class Frequency(proto.Message):
         oneof="aggregation_function",
         message=Count,
     )
-    sum_: Sum = proto.Field(
+    sum: Sum = proto.Field(
         proto.MESSAGE,
         number=3,
         oneof="aggregation_function",
@@ -5557,7 +5557,7 @@ class AggregationResult(proto.Message):
         count (google.cloud.migrationcenter_v1.types.AggregationResult.Count):
 
             This field is a member of `oneof`_ ``result``.
-        sum_ (google.cloud.migrationcenter_v1.types.AggregationResult.Sum):
+        sum (google.cloud.migrationcenter_v1.types.AggregationResult.Sum):
 
             This field is a member of `oneof`_ ``result``.
         histogram (google.cloud.migrationcenter_v1.types.AggregationResult.Histogram):
@@ -5666,7 +5666,7 @@ class Frequency(proto.Message):
         oneof="result",
         message=Count,
     )
-    sum_: Sum = proto.Field(
+    sum: Sum = proto.Field(
         proto.MESSAGE,
         number=3,
         oneof="result",

packages/google-cloud-migrationcenter/noxfile.py

@@ -305,10 +305,9 @@ def docfx(session):
 
     session.install("-e", ".")
     session.install(
-        "sphinx==4.0.1",
+        "gcp-sphinx-docfx-yaml",
         "alabaster",
         "recommonmark",
-        "gcp-sphinx-docfx-yaml",
     )
 
     shutil.rmtree(os.path.join("docs", "_build"), ignore_errors=True)

packages/google-cloud-migrationcenter/tests/unit/gapic/migrationcenter_v1/test_migration_center.py

@@ -1158,9 +1158,11 @@ async def test_list_assets_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_assets(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -1190,7 +1192,6 @@ def test_get_asset(request_type, transport: str = "grpc"):
             name="name_value",
             sources=["sources_value"],
             assigned_groups=["assigned_groups_value"],
-            machine_details=migrationcenter.MachineDetails(uuid="uuid_value"),
         )
         response = client.get_asset(request)
 
@@ -1431,7 +1432,6 @@ def test_update_asset(request_type, transport: str = "grpc"):
             name="name_value",
             sources=["sources_value"],
             assigned_groups=["assigned_groups_value"],
-            machine_details=migrationcenter.MachineDetails(uuid="uuid_value"),
         )
         response = client.update_asset(request)
 
@@ -3384,9 +3384,11 @@ async def test_list_import_jobs_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_import_jobs(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -3417,11 +3419,6 @@ def test_get_import_job(request_type, transport: str = "grpc"):
             display_name="display_name_value",
             state=migrationcenter.ImportJob.ImportJobState.IMPORT_JOB_STATE_PENDING,
             asset_source="asset_source_value",
-            validation_report=migrationcenter.ValidationReport(
-                file_validations=[
-                    migrationcenter.FileValidationReport(file_name="file_name_value")
-                ]
-            ),
         )
         response = client.get_import_job(request)
 
@@ -4631,9 +4628,6 @@ def test_get_import_data_file(request_type, transport: str = "grpc"):
             display_name="display_name_value",
             format_=migrationcenter.ImportJobFormat.IMPORT_JOB_FORMAT_RVTOOLS_XLSX,
             state=migrationcenter.ImportDataFile.State.CREATING,
-            upload_file_info=migrationcenter.UploadFileInfo(
-                signed_uri="signed_uri_value"
-            ),
         )
         response = client.get_import_data_file(request)
 
@@ -5311,9 +5305,11 @@ async def test_list_import_data_files_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_import_data_files(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -6237,9 +6233,11 @@ async def test_list_groups_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_groups(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -8113,9 +8111,11 @@ async def test_list_error_frames_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_error_frames(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -8769,9 +8769,11 @@ async def test_list_sources_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_sources(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -10188,9 +10190,11 @@ async def test_list_preference_sets_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_preference_sets(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -12630,9 +12634,11 @@ async def test_list_report_configs_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_report_configs(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -13789,9 +13795,11 @@ async def test_list_reports_async_pages():
             RuntimeError,
         )
         pages = []
-        async for page_ in (
+        # Workaround issue in python 3.9 related to code coverage by adding `# pragma: no branch`
+        # See https://github.com/googleapis/gapic-generator-python/pull/1174#issuecomment-1025132372
+        async for page_ in (  # pragma: no branch
             await client.list_reports(request={})
-        ).pages:  # pragma: no branch
+        ).pages:
             pages.append(page_)
         for page_, token in zip(pages, ["abc", "def", "ghi", ""]):
             assert page_.raw_page.next_page_token == token
@@ -14394,7 +14402,6 @@ def test_get_asset_rest(request_type):
             name="name_value",
             sources=["sources_value"],
             assigned_groups=["assigned_groups_value"],
-            machine_details=migrationcenter.MachineDetails(uuid="uuid_value"),
         )
 
         # Wrap the value into a proper Response obj
@@ -14925,7 +14932,6 @@ def test_update_asset_rest(request_type):
             name="name_value",
             sources=["sources_value"],
             assigned_groups=["assigned_groups_value"],
-            machine_details=migrationcenter.MachineDetails(uuid="uuid_value"),
         )
 
         # Wrap the value into a proper Response obj
@@ -17915,11 +17921,6 @@ def test_get_import_job_rest(request_type):
             display_name="display_name_value",
             state=migrationcenter.ImportJob.ImportJobState.IMPORT_JOB_STATE_PENDING,
             asset_source="asset_source_value",
-            validation_report=migrationcenter.ValidationReport(
-                file_validations=[
-                    migrationcenter.FileValidationReport(file_name="file_name_value")
-                ]
-            ),
         )
 
         # Wrap the value into a proper Response obj
@@ -19372,9 +19373,6 @@ def test_get_import_data_file_rest(request_type):
             display_name="display_name_value",
             format_=migrationcenter.ImportJobFormat.IMPORT_JOB_FORMAT_RVTOOLS_XLSX,
             state=migrationcenter.ImportDataFile.State.CREATING,
-            upload_file_info=migrationcenter.UploadFileInfo(
-                signed_uri="signed_uri_value"
-            ),
         )
 
         # Wrap the value into a proper Response obj

packages/google-cloud-monitoring-metrics-scopes/google/cloud/monitoring_metrics_scope/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.4.2"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-monitoring-metrics-scopes/google/cloud/monitoring_metrics_scope_v1/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "1.4.2"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-monitoring-metrics-scopes/google/cloud/monitoring_metrics_scope_v1/services/metrics_scopes/async_client.py

@@ -677,7 +677,7 @@ async def sample_delete_monitored_project():
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MetricsScopesAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):