@typespec/http-client-python 0.9.0-dev.1 → 0.9.0

package/generator/build/lib/pygen/codegen/templates/validation.py.jinja2

@@ -1,4 +1,6 @@
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 import functools
 
 def api_version_validation(**kwargs):

package/generator/build/lib/pygen/codegen/templates/vendor.py.jinja2

@@ -1,5 +1,7 @@
 {% import 'keywords.jinja2' as keywords with context %}
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 {{ imports }}
 

package/generator/build/lib/pygen/codegen/templates/version.py.jinja2

@@ -1,4 +1,6 @@
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 VERSION = "{{ code_model.options['package_version'] }}"

package/generator/pygen/codegen/__init__.py

@@ -13,7 +13,7 @@ from .. import Plugin
 from ..utils import parse_args
 from .models.code_model import CodeModel
 from .serializers import JinjaSerializer
-from ._utils import DEFAULT_HEADER_TEXT, VALID_PACKAGE_MODE, TYPESPEC_PACKAGE_MODE
+from ._utils import VALID_PACKAGE_MODE, TYPESPEC_PACKAGE_MODE
 
 
 def _default_pprint(package_name: str) -> str:
@@ -53,24 +53,6 @@ class OptionsRetriever:
         key = prop.replace("_", "-")
         return self.options.get(key, self.OPTIONS_TO_DEFAULT.get(key))
 
-    @property
-    def company_name(self) -> str:
-        return self.options.get("company-name", "Microsoft" if self.is_azure_flavor else "")
-
-    @property
-    def license_header(self) -> str:
-        license_header = self.options.get(
-            "header-text",
-            (DEFAULT_HEADER_TEXT.format(company_name=self.company_name) if self.company_name else ""),
-        )
-        if license_header:
-            license_header = license_header.replace("\n", "\n# ")
-            license_header = (
-                "# --------------------------------------------------------------------------\n# " + license_header
-            )
-            license_header += "\n# --------------------------------------------------------------------------"
-        return license_header
-
     @property
     def show_operations(self) -> bool:
         return self.options.get("show-operations", not self.low_level_client)
@@ -168,6 +150,11 @@ class OptionsRetriever:
     def package_version(self) -> Optional[str]:
         return str(self.options.get("package-version", ""))
 
+    
+    @property
+    def header_text(self) -> Optional[str]:
+        return self.options.get("header-text")
+
 
 class CodeGenerator(Plugin):
     def __init__(self, *args, **kwargs: Any) -> None:
@@ -286,7 +273,7 @@ class CodeGenerator(Plugin):
         flags = [
             "azure_arm",
             "head_as_boolean",
-            "license_header",
+            "header_text",
             "keep_version_file",
             "no_async",
             "no_namespace_folders",
@@ -314,7 +301,6 @@ class CodeGenerator(Plugin):
             "default_api_version",
             "from_typespec",
             "flavor",
-            "company_name",
             "emit_cross_language_definition_file",
         ]
         return {f: getattr(self.options_retriever, f) for f in flags}

package/generator/pygen/codegen/_utils.py

@@ -5,12 +5,36 @@
 # --------------------------------------------------------------------------
 
 DEFAULT_HEADER_TEXT = (
-    "Copyright (c) {company_name} Corporation. All rights reserved.\n"
+    "Copyright (c) Microsoft Corporation. All rights reserved.\n"
     "Licensed under the MIT License. See License.txt in the project root for license information.\n"
-    "Code generated by {company_name} (R) Python Code Generator.\n"
+    "Code generated by Microsoft (R) Python Code Generator.\n"
     "Changes may cause incorrect behavior and will be lost if the code is regenerated."
 )
 
+DEFAULT_LICENSE_DESCRIPTION = (
+    "Copyright (c) Microsoft Corporation.\n"
+    "\n"
+    "MIT License\n"
+    "\n"
+    "Permission is hereby granted, free of charge, to any person obtaining a copy\n"
+    "of this software and associated documentation files (the \"Software\"), to deal\n"
+    "in the Software without restriction, including without limitation the rights\n"
+    "to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n"
+    "copies of the Software, and to permit persons to whom the Software is\n"
+    "furnished to do so, subject to the following conditions:\n"
+    "\n"
+    "The above copyright notice and this permission notice shall be included in all\n"
+    "copies or substantial portions of the Software.\n"
+    "\n"
+    "THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n"
+    "IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n"
+    "FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n"
+    "AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n"
+    "LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n"
+    "OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n"
+    "SOFTWARE."
+)
+
 SWAGGER_PACKAGE_MODE = ["mgmtplane", "dataplane"]  # for backward compatibility
 TYPESPEC_PACKAGE_MODE = ["azure-mgmt", "azure-dataplane", "generic"]
 VALID_PACKAGE_MODE = SWAGGER_PACKAGE_MODE + TYPESPEC_PACKAGE_MODE

package/generator/pygen/codegen/models/code_model.py

@@ -13,6 +13,7 @@ from .client import Client
 from .request_builder import RequestBuilder, OverloadedRequestBuilder
 from .operation_group import OperationGroup
 from .utils import NamespaceType
+from .._utils import DEFAULT_HEADER_TEXT, DEFAULT_LICENSE_DESCRIPTION
 
 
 def _is_legacy(options) -> bool:
@@ -406,3 +407,35 @@ class CodeModel:  # pylint: disable=too-many-public-methods, disable=too-many-in
     @property
     def is_tsp(self) -> bool:
         return self.options.get("tsp_file") is not None
+
+    @property
+    def license_header(self) -> str:
+        if self.yaml_data.get("licenseInfo") or not self.is_azure_flavor:
+            # typespec unbranded case and azure case with custom license
+            license_header = self.yaml_data.get("licenseInfo", {}).get("header", "")
+        else:
+            # typespec azure case without custom license and swagger case
+            license_header = self.options.get("header_text", DEFAULT_HEADER_TEXT)
+        if license_header:
+            license_header = license_header.replace("\n", "\n# ")
+            license_header = (
+                "# --------------------------------------------------------------------------\n# " + license_header
+            )
+            license_header += "\n# --------------------------------------------------------------------------"
+        return license_header
+
+    @property
+    def license_description(self) -> str:
+        if self.yaml_data.get("licenseInfo") or not self.is_azure_flavor:
+            # typespec unbranded case and azure case with custom license
+            return self.yaml_data.get("licenseInfo", {}).get("description", "")
+        # typespec azure case without custom license and swagger case
+        return DEFAULT_LICENSE_DESCRIPTION
+
+    @property
+    def company_name(self) -> str:
+        if self.yaml_data.get("licenseInfo") or not self.is_azure_flavor:
+            # typespec unbranded case and azure case with custom license
+            return self.yaml_data.get("licenseInfo", {}).get("company", "")
+        # typespec azure case without custom license and swagger case
+        return "Microsoft Corporation"

package/generator/pygen/codegen/models/credential_types.py

@@ -48,17 +48,20 @@ class BearerTokenCredentialPolicyType(_CredentialPolicyBaseType):
         yaml_data: Dict[str, Any],
         code_model: "CodeModel",
         credential_scopes: List[str],
+        flows: Optional[Dict[str, Any]] = None,
     ) -> None:
         super().__init__(yaml_data, code_model)
         self.credential_scopes = credential_scopes
+        self.flows = flows
 
     def call(self, async_mode: bool) -> str:
         policy_name = f"{'Async' if async_mode else ''}BearerTokenCredentialPolicy"
-        return f"policies.{policy_name}(self.credential, *self.credential_scopes, **kwargs)"
+        auth_flows = f"auth_flows={self.flows}, " if self.flows else ""
+        return f"policies.{policy_name}(self.credential, *self.credential_scopes, {auth_flows}**kwargs)"
 
     @classmethod
     def from_yaml(cls, yaml_data: Dict[str, Any], code_model: "CodeModel") -> "BearerTokenCredentialPolicyType":
-        return cls(yaml_data, code_model, yaml_data["credentialScopes"])
+        return cls(yaml_data, code_model, yaml_data["credentialScopes"], yaml_data.get("flows"))
 
 
 class ARMChallengeAuthenticationPolicyType(BearerTokenCredentialPolicyType):

package/generator/pygen/codegen/serializers/__init__.py

@@ -202,6 +202,8 @@ class JinjaSerializer(ReaderAndWriter):
             )
 
             package_files = _PACKAGE_FILES
+            if not self.code_model.license_description:
+                package_files.remove("LICENSE.jinja2")
         elif Path(self.code_model.options["package_mode"]).exists():
             env = Environment(
                 loader=FileSystemLoader(str(Path(self.code_model.options["package_mode"]))),
@@ -273,7 +275,7 @@ class JinjaSerializer(ReaderAndWriter):
         if not "" in group_names:
             self.write_file(
                 rest_path / Path("__init__.py"),
-                self.code_model.options["license_header"],
+                self.code_model.license_header,
             )
 
     def _serialize_and_write_single_rest_layer(

package/generator/pygen/codegen/templates/client_container.py.jinja2

@@ -2,7 +2,9 @@
 {% import 'operation_tools.jinja2' as op_tools %}
 {# actual template starts here #}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 {{ imports }}
 

package/generator/pygen/codegen/templates/config_container.py.jinja2

@@ -2,7 +2,9 @@
 {% import 'operation_tools.jinja2' as op_tools %}
 {# actual template starts here #}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 {{ imports }}
 

package/generator/pygen/codegen/templates/conftest.py.jinja2

@@ -1,5 +1,7 @@
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 import os
 import pytest
 from dotenv import load_dotenv

package/generator/pygen/codegen/templates/enum_container.py.jinja2

@@ -1,6 +1,8 @@
 {% import 'operation_tools.jinja2' as op_tools %}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 from enum import Enum
 from {{ code_model.core_library }}{{ "" if code_model.is_azure_flavor else ".utils" }} import CaseInsensitiveEnumMeta

package/generator/pygen/codegen/templates/init.py.jinja2

@@ -1,6 +1,8 @@
 {% import 'keywords.jinja2' as keywords %}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 {{ keywords.path_type_checking_imports() }}
 {% if clients %}
     {% for client in clients %}

package/generator/pygen/codegen/templates/model_base.py.jinja2

@@ -1,9 +1,7 @@
 # coding=utf-8
-# --------------------------------------------------------------------------
-# Copyright (c) {{ code_model.options["company_name"] }} Corporation. All rights reserved.
-# Licensed under the MIT License. See License.txt in the project root for
-# license information.
-# --------------------------------------------------------------------------
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 # pylint: disable=protected-access, broad-except
 
 import copy

package/generator/pygen/codegen/templates/model_container.py.jinja2

@@ -1,6 +1,8 @@
 {% import 'operation_tools.jinja2' as op_tools %}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 {% if serializer.global_pylint_disables() %}
 {{ serializer.global_pylint_disables() }}
 {% endif %}

package/generator/pygen/codegen/templates/model_init.py.jinja2

@@ -1,6 +1,8 @@
 {% import 'keywords.jinja2' as keywords %}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 {{ keywords.path_type_checking_imports() }}
 {% if schemas %}
 

package/generator/pygen/codegen/templates/operation_groups_container.py.jinja2

@@ -2,7 +2,9 @@
 {% set operations_description = "async operations" if async_mode else "operations" %}
 {% set return_none_type_annotation = " -> None" if async_mode else "" %}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 {{ imports }}
 {{ unset }}
 {% if code_model.options["builders_visibility"] == "embedded" and not async_mode %}

package/generator/pygen/codegen/templates/operations_folder_init.py.jinja2

@@ -2,7 +2,9 @@
 {% import 'keywords.jinja2' as keywords %}
 {# actual template starts here #}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 {{ keywords.path_type_checking_imports() }}
 {{ op_tools.serialize(operation_group_imports()) }}
 {{ keywords.patch_imports() }}

package/generator/pygen/codegen/templates/packaging_templates/LICENSE.jinja2

@@ -1,21 +1 @@
-Copyright (c) {{ code_model.options["company_name"] }} Corporation.
-
-MIT License
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+{{ code_model.license_description }}

package/generator/pygen/codegen/templates/packaging_templates/setup.py.jinja2

@@ -1,6 +1,8 @@
 # coding=utf-8
-{{ license_header }}
-# coding: utf-8
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
+
 {% if package_mode %}
 
 import os
@@ -32,7 +34,7 @@ with open(os.path.join(package_folder_path, "_version.py"), "r") as fd:
 
 if not version:
     raise RuntimeError("Cannot find version information")
-{% set description = "\"" + code_model.options["company_name"] + " {} Client Library for Python\".format(PACKAGE_PPRINT_NAME)" %}
+{% set description = "\"" + code_model.company_name + " {} Client Library for Python\".format(PACKAGE_PPRINT_NAME)" %}
 {% set author_email = "azpysdkhelp@microsoft.com" %}
 {% set url = "https://github.com/Azure/azure-sdk-for-python/tree/main/sdk" %}
 {% else %}
@@ -52,7 +54,7 @@ setup(
     long_description=open("README.md", "r").read(),
     long_description_content_type="text/markdown",
     license="MIT License",
-    author="{{ code_model.options["company_name"] }} Corporation",
+    author="{{ code_model.company_name }}",
     {% endif %}
     {% if code_model.is_azure_flavor %}
     author_email="{{ author_email }}",

package/generator/pygen/codegen/templates/patch.py.jinja2

@@ -1,7 +1,12 @@
-# ------------------------------------
-# Copyright (c) {{ code_model.options["company_name"] }} Corporation.
-# Licensed under the MIT License.
-# ------------------------------------
+# coding=utf-8
+{% if not code_model.is_azure_flavor %}
+{{ code_model.license_header }}
+{% else %}
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# --------------------------------------------------------------------------
+{% endif %}
 """Customize generated code here.
 
 Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize

package/generator/pygen/codegen/templates/request_builders.py.jinja2

@@ -1,6 +1,8 @@
 {% import 'operation_tools.jinja2' as op_tools %}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 {{ imports }}
 
 {{ op_tools.declare_serializer(code_model) }}

package/generator/pygen/codegen/templates/rest_init.py.jinja2

@@ -1,5 +1,7 @@
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 {% for request_builder in request_builders %}
 from ._request_builders import {{ request_builder.name }}

package/generator/pygen/codegen/templates/sample.py.jinja2

@@ -1,7 +1,9 @@
 # coding=utf-8
 {% set aad_token = "DefaultAzureCredential" %}
 {% set azure_key = "AzureKeyCredential" %}
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 {{ imports }}
 """

package/generator/pygen/codegen/templates/serialization.py.jinja2

@@ -1,28 +1,7 @@
-# --------------------------------------------------------------------------
-#
-# Copyright (c) {{ code_model.options["company_name"] }} Corporation. All rights reserved.
-#
-# The MIT License (MIT)
-#
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the ""Software""), to
-# deal in the Software without restriction, including without limitation the
-# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
-# sell copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-#
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
-# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
-# IN THE SOFTWARE.
-#
-# --------------------------------------------------------------------------
+# coding=utf-8
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 # pyright: reportUnnecessaryTypeIgnoreComment=false
 

package/generator/pygen/codegen/templates/test.py.jinja2

@@ -3,7 +3,9 @@
 {% set async = "async " if test.async_mode else "" %}
 {% set async_suffix = "_async" if test.async_mode else "" %}
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 import pytest
 {{ imports }}
 

package/generator/pygen/codegen/templates/testpreparer.py.jinja2

@@ -1,5 +1,7 @@
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 {{ imports }}
 
 {% for test_name in test_names %}

package/generator/pygen/codegen/templates/types.py.jinja2

@@ -1,5 +1,7 @@
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 {{ imports }}
 {% for nu in code_model.named_unions %}

package/generator/pygen/codegen/templates/validation.py.jinja2

@@ -1,4 +1,6 @@
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 import functools
 
 def api_version_validation(**kwargs):

package/generator/pygen/codegen/templates/vendor.py.jinja2

@@ -1,5 +1,7 @@
 {% import 'keywords.jinja2' as keywords with context %}
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 {{ imports }}
 

package/generator/pygen/codegen/templates/version.py.jinja2

@@ -1,4 +1,6 @@
 # coding=utf-8
-{{ code_model.options['license_header'] }}
+{% if code_model.license_header %}
+{{ code_model.license_header }}
+{% endif %}
 
 VERSION = "{{ code_model.options['package_version'] }}"

package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_lro_rpc_async.py

@@ -4,8 +4,8 @@
 # license information.
 # --------------------------------------------------------------------------
 import pytest
-from azurecore.lro.rpc.aio import RpcClient
-from azurecore.lro.rpc import models
+from specs.azure.core.lro.rpc.aio import RpcClient
+from specs.azure.core.lro.rpc import models
 
 
 @pytest.fixture

package/generator/test/azure/mock_api_tests/test_azure_core_lro_rpc.py

@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 import pytest
-from azurecore.lro.rpc import RpcClient, models
+from specs.azure.core.lro.rpc import RpcClient, models
 
 
 @pytest.fixture

package/generator/test/azure/requirements.txt

@@ -9,7 +9,7 @@ azure-mgmt-core==1.3.2
 -e ./generated/azure-client-generator-core-flatten-property
 -e ./generated/azure-core-basic
 -e ./generated/azure-core-scalar
--e ./generated/azurecore-lro-rpc
+-e ./generated/azure-core-lro-rpc
 -e ./generated/azure-core-lro-standard
 -e ./generated/azure-core-model
 -e ./generated/azure-core-traits
@@ -46,6 +46,7 @@ azure-mgmt-core==1.3.2
 -e ./generated/server-versions-versioned
 -e ./generated/server-versions-not-versioned
 -e ./generated/special-words
+-e ./generated/streaming-jsonl
 -e ./generated/typetest-array
 -e ./generated/typetest-dictionary
 -e ./generated/typetest-enum-extensible

package/generator/test/generic_mock_api_tests/asynctests/test_authentication_async.py

@@ -33,6 +33,10 @@ def token_credential(core_library):
         async def get_token(*scopes):
             return core_library.credentials.AccessToken(token="".join(scopes), expires_on=1800)
 
+        @staticmethod
+        async def get_token_info(*scopes, **kwargs):
+            return core_library.credentials.AccessTokenInfo(token="".join(scopes), expires_on=1800)
+
     return FakeCredential()
 
 

package/generator/test/generic_mock_api_tests/asynctests/test_headasboolean_async.py

@@ -25,11 +25,11 @@ async def client_false():
 
 @pytest.mark.asyncio
 async def test_head_true(client_true):
-    body = models_true.VisibilityModel(query_prop=123)
-    assert await client_true.head_model(body) == True
+    body = models_true.VisibilityModel()
+    assert await client_true.head_model(body, query_prop=123) == True
 
 
 @pytest.mark.asyncio
 async def test_head_false(client_false):
-    body = models_false.VisibilityModel(query_prop=123)
-    assert await client_false.head_model(body) is None
+    body = models_false.VisibilityModel()
+    assert await client_false.head_model(body, query_prop=123) is None

package/generator/test/generic_mock_api_tests/asynctests/test_streaming_jsonl_async.py

@@ -0,0 +1,27 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import pytest
+
+from streaming.jsonl.aio import JsonlClient
+
+
+@pytest.fixture
+async def client():
+    async with JsonlClient(endpoint="http://localhost:3000") as client:
+        yield client
+
+
+JSONL = b'{"desc": "one"}\n{"desc": "two"}\n{"desc": "three"}'
+
+
+@pytest.mark.asyncio
+async def test_basic_send(client: JsonlClient):
+    await client.basic.send(JSONL)
+
+
+@pytest.mark.asyncio
+async def test_basic_recv(client: JsonlClient):
+    assert b"".join([d async for d in (await client.basic.receive())]) == JSONL

package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_visibility_async.py

@@ -16,7 +16,7 @@ async def client():
 
 @pytest.mark.asyncio
 async def test_get_model(client):
-    result = await client.get_model(models.VisibilityModel(query_prop=123))
+    result = await client.get_model(models.VisibilityModel(), query_prop=123)
     assert result == models.VisibilityModel(read_prop="abc")