From 17062154d5815ed432879fc27442c555e0a8f67d Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 08:29:59 +0000 Subject: [PATCH 1/8] Update dependencies (2024-12-06 08:29:59) --- packages/autorest.python/package.json | 4 +-- packages/typespec-python/package.json | 8 ++--- pnpm-lock.yaml | 43 ++++++++++++++------------- 3 files changed, 28 insertions(+), 27 deletions(-) diff --git a/packages/autorest.python/package.json b/packages/autorest.python/package.json index 0ff3e054799..ef030eddca6 100644 --- a/packages/autorest.python/package.json +++ b/packages/autorest.python/package.json @@ -29,7 +29,7 @@ }, "homepage": "https://github.com/Azure/autorest.python/blob/main/README.md", "dependencies": { - "@typespec/http-client-python": "~0.3.12", + "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", "@autorest/system-requirements": "~1.0.2", "fs-extra": "~11.2.0", "tsx": "~4.19.1" @@ -47,4 +47,4 @@ "requirements.txt", "generator/" ] -} +} \ No newline at end of file diff --git a/packages/typespec-python/package.json b/packages/typespec-python/package.json index b0103b083ed..61a1e0d39f1 100644 --- a/packages/typespec-python/package.json +++ b/packages/typespec-python/package.json @@ -54,13 +54,13 @@ "@azure-tools/typespec-azure-resource-manager": ">=0.48.0 <1.0.0", "@azure-tools/typespec-autorest": ">=0.48.0 <1.0.0", "@azure-tools/typespec-azure-rulesets": ">=0.48.0 <3.0.0", - "@azure-tools/typespec-client-generator-core": ">=0.48.0 <1.0.0" + "@azure-tools/typespec-client-generator-core": ">=0.48.1 <1.0.0" }, "dependencies": { "js-yaml": "~4.1.0", "semver": "~7.6.2", "tsx": "~4.19.1", - "@typespec/http-client-python": "~0.3.12", + "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", "fs-extra": "~11.2.0" }, "devDependencies": { @@ -73,7 +73,7 @@ "@azure-tools/typespec-azure-core": "~0.48.0", "@azure-tools/typespec-azure-rulesets": "~0.48.0", "@azure-tools/typespec-autorest": "~0.48.0", - "@azure-tools/typespec-client-generator-core": "~0.48.0", + "@azure-tools/typespec-client-generator-core": "~0.48.1", "@azure-tools/cadl-ranch-expect": "~0.15.6", "@azure-tools/cadl-ranch-specs": "~0.39.1", "@azure-tools/cadl-ranch": "~0.16.1", @@ -90,4 +90,4 @@ "chalk": "5.3.0", "@types/fs-extra": "11.0.4" } -} +} \ No newline at end of file diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml index 77ea4fde1aa..d5e70613a06 100644 --- a/pnpm-lock.yaml +++ b/pnpm-lock.yaml @@ -54,8 +54,8 @@ importers: specifier: ~1.0.2 version: 1.0.2 '@typespec/http-client-python': - specifier: ~0.3.12 - version: 0.3.12(adr36ipppdbqiaonm57cbaqjli) + specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz + version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) fs-extra: specifier: ~11.2.0 version: 11.2.0 @@ -76,8 +76,8 @@ importers: packages/typespec-python: dependencies: '@typespec/http-client-python': - specifier: ~0.3.12 - version: 0.3.12(adr36ipppdbqiaonm57cbaqjli) + specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz + version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) fs-extra: specifier: ~11.2.0 version: 11.2.0 @@ -102,7 +102,7 @@ importers: version: 0.39.1(@azure-tools/cadl-ranch-expect@0.15.6(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)))(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0))(@typespec/xml@0.59.0(@typespec/compiler@0.62.0)) '@azure-tools/typespec-autorest': specifier: ~0.48.0 - version: 0.48.0(eltkenvcfknbgsjpk2r2c3erra) + version: 0.48.0(2qbrnlizjwf7gxxublducwg43e) '@azure-tools/typespec-azure-core': specifier: ~0.48.0 version: 0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))) @@ -111,10 +111,10 @@ importers: version: 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) '@azure-tools/typespec-azure-rulesets': specifier: ~0.48.0 - version: 0.48.0(vtbpfuptnbf2g743tnkevc4ay4) + version: 0.48.0(e3ngzrairqtkdxikevgks3pfwe) '@azure-tools/typespec-client-generator-core': - specifier: ~0.48.0 - version: 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) + specifier: ~0.48.1 + version: 0.48.4(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) '@types/fs-extra': specifier: 11.0.4 version: 11.0.4 @@ -254,8 +254,8 @@ packages: '@azure-tools/typespec-client-generator-core': ~0.48.0 '@typespec/compiler': ~0.62.0 - '@azure-tools/typespec-client-generator-core@0.48.0': - resolution: {integrity: sha512-+fmKjapz0kP7ONPZap8dgcIKIdQw+YBSrf89csbIyhPTcLnVAk/BKljo8FoNypKXwqKHenslLm0njBKPllkopg==} + '@azure-tools/typespec-client-generator-core@0.48.4': + resolution: {integrity: sha512-TvX84FiQ3rax0e838m6kpVj8F24OzKAbyLgUXXZ/TjfxhvZb1u0ojMjSKAvmcal2klROJqRlj4d9tImidPYpgA==} engines: {node: '>=18.0.0'} peerDependencies: '@azure-tools/typespec-azure-core': ~0.48.0 @@ -1233,8 +1233,9 @@ packages: engines: {node: '>=18.0.0'} hasBin: true - '@typespec/http-client-python@0.3.12': - resolution: {integrity: sha512-S4BvoANekP4RgT9Uqpj+K2DmrW3ceUqJ4qqZwolipJBGbjHQMYrbMT8oLTkaSlCGb0dFphibnfLu7AVqkmZ8vQ==} + '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz': + resolution: {tarball: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz} + version: 0.3.12 engines: {node: '>=18.0.0'} peerDependencies: '@azure-tools/typespec-autorest': '>=0.48.0 <1.0.0' @@ -3825,11 +3826,11 @@ snapshots: - debug - supports-color - '@azure-tools/typespec-autorest@0.48.0(eltkenvcfknbgsjpk2r2c3erra)': + '@azure-tools/typespec-autorest@0.48.0(2qbrnlizjwf7gxxublducwg43e)': dependencies: '@azure-tools/typespec-azure-core': 0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))) '@azure-tools/typespec-azure-resource-manager': 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) - '@azure-tools/typespec-client-generator-core': 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) + '@azure-tools/typespec-client-generator-core': 0.48.4(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) '@typespec/compiler': 0.62.0 '@typespec/http': 0.62.0(@typespec/compiler@0.62.0) '@typespec/openapi': 0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)) @@ -3853,14 +3854,14 @@ snapshots: change-case: 5.4.4 pluralize: 8.0.0 - '@azure-tools/typespec-azure-rulesets@0.48.0(vtbpfuptnbf2g743tnkevc4ay4)': + '@azure-tools/typespec-azure-rulesets@0.48.0(e3ngzrairqtkdxikevgks3pfwe)': dependencies: '@azure-tools/typespec-azure-core': 0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))) '@azure-tools/typespec-azure-resource-manager': 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) - '@azure-tools/typespec-client-generator-core': 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) + '@azure-tools/typespec-client-generator-core': 0.48.4(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) '@typespec/compiler': 0.62.0 - '@azure-tools/typespec-client-generator-core@0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0))': + '@azure-tools/typespec-client-generator-core@0.48.4(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0))': dependencies: '@azure-tools/typespec-azure-core': 0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))) '@typespec/compiler': 0.62.0 @@ -4918,13 +4919,13 @@ snapshots: yaml: 2.5.1 yargs: 17.7.2 - '@typespec/http-client-python@0.3.12(adr36ipppdbqiaonm57cbaqjli)': + '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi)': dependencies: - '@azure-tools/typespec-autorest': 0.48.0(eltkenvcfknbgsjpk2r2c3erra) + '@azure-tools/typespec-autorest': 0.48.0(2qbrnlizjwf7gxxublducwg43e) '@azure-tools/typespec-azure-core': 0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))) '@azure-tools/typespec-azure-resource-manager': 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) - '@azure-tools/typespec-azure-rulesets': 0.48.0(vtbpfuptnbf2g743tnkevc4ay4) - '@azure-tools/typespec-client-generator-core': 0.48.0(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) + '@azure-tools/typespec-azure-rulesets': 0.48.0(e3ngzrairqtkdxikevgks3pfwe) + '@azure-tools/typespec-client-generator-core': 0.48.4(@azure-tools/typespec-azure-core@0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))))(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/openapi@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)))(@typespec/versioning@0.62.0(@typespec/compiler@0.62.0)) '@typespec/compiler': 0.62.0 '@typespec/http': 0.62.0(@typespec/compiler@0.62.0) '@typespec/openapi': 0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0)) From f1226f55a372f4d809d9796ab4ddee6e3de2339a Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 08:34:20 +0000 Subject: [PATCH 2/8] Regenerate for typespec-python (2024-12-06 08:34:20) --- .../apikey}/apiview_mapping_python.json | 0 .../authentication/apikey/models/_models.py | 1 + .../generated_tests/conftest.py | 35 -- .../generated_tests/test_api_key.py | 31 -- .../generated_tests/test_api_key_async.py | 32 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../http/custom}/apiview_mapping_python.json | 0 .../http/custom/models/_models.py | 1 + .../generated_tests/conftest.py | 35 -- .../generated_tests/test_custom.py | 31 -- .../generated_tests/test_custom_async.py | 32 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../oauth2}/apiview_mapping_python.json | 0 .../authentication/oauth2/models/_models.py | 1 + .../generated_tests/conftest.py | 35 -- .../generated_tests/test_oauth2.py | 31 -- .../generated_tests/test_oauth2_async.py | 32 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../union}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_union.py | 31 -- .../generated_tests/test_union_async.py | 32 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../generated_tests/conftest.py | 35 -- ...test_access_public_operation_operations.py | 35 -- ...ccess_public_operation_operations_async.py | 36 -- ...ss_shared_model_in_operation_operations.py | 24 -- ...red_model_in_operation_operations_async.py | 25 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../core/access}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- .../generated_tests/test_flatten_property.py | 35 -- .../test_flatten_property_async.py | 36 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- ...est_usage_model_in_operation_operations.py | 44 -- ...age_model_in_operation_operations_async.py | 45 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../core/usage}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_basic.py | 91 ---- .../generated_tests/test_basic_async.py | 92 ----- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../core/basic}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_standard.py | 48 --- .../generated_tests/test_standard_async.py | 55 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../lro/standard}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- ..._azure_core_embedding_vector_operations.py | 44 -- ..._core_embedding_vector_operations_async.py | 45 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../core/model}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_page.py | 42 -- .../generated_tests/test_page_async.py | 43 -- ...page_two_models_as_page_item_operations.py | 31 -- ...wo_models_as_page_item_operations_async.py | 32 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../core/page}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- ...scalar_azure_location_scalar_operations.py | 67 --- ..._azure_location_scalar_operations_async.py | 68 --- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../core/scalar}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_traits.py | 37 -- .../generated_tests/test_traits_async.py | 38 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../core/traits}/apiview_mapping_python.json | 0 .../generated_samples/basic.py | 37 -- .../generated_tests/conftest.py | 37 -- .../generated_tests/test_azure_example.py | 31 -- .../test_azure_example_async.py | 32 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../basic}/apiview_mapping_python.json | 0 .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- ..._properties_managed_identity_operations.py | 96 ----- ...rties_managed_identity_operations_async.py | 97 ----- .../resources}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_resources_nested_operations.py | 106 ----- .../test_resources_nested_operations_async.py | 113 ----- .../test_resources_singleton_operations.py | 91 ---- ...st_resources_singleton_operations_async.py | 94 ----- .../test_resources_top_level_operations.py | 125 ------ ...st_resources_top_level_operations_async.py | 132 ------ .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 37 -- .../generated_tests/test_xms_request_id.py | 22 - .../test_xms_request_id_async.py | 23 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../lro/rpc}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_rpc.py | 24 -- .../generated_tests/test_rpc_async.py | 27 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../naming}/apiview_mapping_python.json | 0 .../client-naming/generated_tests/conftest.py | 35 -- .../generated_tests/test_naming.py | 86 ---- .../generated_tests/test_naming_async.py | 87 ---- .../test_naming_client_model_operations.py | 35 -- ...st_naming_client_model_operations_async.py | 36 -- .../test_naming_union_enum_operations.py | 37 -- ...test_naming_union_enum_operations_async.py | 38 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../service}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_service.py | 31 -- .../generated_tests/test_service_async.py | 32 -- .../test_service_bar_operations.py | 31 -- .../test_service_bar_operations_async.py | 32 -- .../test_service_baz_operations.py | 22 - .../test_service_baz_operations_async.py | 23 -- .../test_service_foo_operations.py | 31 -- .../test_service_foo_operations_async.py | 32 -- .../test_service_qux_operations.py | 31 -- .../test_service_qux_operations_async.py | 32 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../multiclient}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 44 -- .../generated_tests/test_a.py | 40 -- .../generated_tests/test_aasync.py | 41 -- .../generated_tests/test_b.py | 40 -- .../generated_tests/test_basync.py | 41 -- .../generated_tests/testpreparer.py | 38 -- .../generated_tests/testpreparer_async.py | 31 -- .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- .../generated_tests/test_renamed_operation.py | 40 -- .../test_renamed_operation_async.py | 41 -- ...test_renamed_operation_group_operations.py | 40 -- ...enamed_operation_group_operations_async.py | 41 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- ...t_two_operation_group_group1_operations.py | 40 -- ...operation_group_group1_operations_async.py | 41 -- ...t_two_operation_group_group2_operations.py | 40 -- ...operation_group_group2_operations_async.py | 41 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../bytes}/apiview_mapping_python.json | 0 .../encode-bytes/generated_tests/conftest.py | 35 -- .../test_bytes_header_operations.py | 57 --- .../test_bytes_header_operations_async.py | 58 --- .../test_bytes_property_operations.py | 57 --- .../test_bytes_property_operations_async.py | 58 --- .../test_bytes_query_operations.py | 57 --- .../test_bytes_query_operations_async.py | 58 --- .../test_bytes_request_body_operations.py | 73 ---- ...est_bytes_request_body_operations_async.py | 74 ---- .../test_bytes_response_body_operations.py | 58 --- ...st_bytes_response_body_operations_async.py | 59 --- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../datetime}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_datetime_header_operations.py | 68 --- .../test_datetime_header_operations_async.py | 69 ---- .../test_datetime_property_operations.py | 68 --- ...test_datetime_property_operations_async.py | 69 ---- .../test_datetime_query_operations.py | 68 --- .../test_datetime_query_operations_async.py | 69 ---- ...est_datetime_response_header_operations.py | 49 --- ...tetime_response_header_operations_async.py | 50 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../duration}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_duration_header_operations.py | 79 ---- .../test_duration_header_operations_async.py | 80 ---- .../test_duration_property_operations.py | 79 ---- ...test_duration_property_operations_async.py | 80 ---- .../test_duration_query_operations.py | 79 ---- .../test_duration_query_operations_async.py | 80 ---- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../numeric}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_numeric_property_operations.py | 46 --- .../test_numeric_property_operations_async.py | 47 --- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../generated_tests/conftest.py | 35 -- .../generated_tests/test_visibility.py | 90 ---- .../generated_tests/test_visibility_async.py | 91 ---- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_visibility.py | 90 ---- .../generated_tests/test_visibility_async.py | 91 ---- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_basic_explicit_body_operations.py | 24 -- ...st_basic_explicit_body_operations_async.py | 25 -- .../test_basic_implicit_body_operations.py | 25 -- ...st_basic_implicit_body_operations_async.py | 26 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../basic}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- .../generated_tests/test_body_optionality.py | 36 -- .../test_body_optionality_async.py | 37 -- ...ptionality_optional_explicit_operations.py | 31 -- ...lity_optional_explicit_operations_async.py | 32 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- ...est_collection_format_header_operations.py | 24 -- ...llection_format_header_operations_async.py | 25 -- ...test_collection_format_query_operations.py | 68 --- ...ollection_format_query_operations_async.py | 69 ---- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_spread_alias_operations.py | 83 ---- .../test_spread_alias_operations_async.py | 84 ---- .../test_spread_model_operations.py | 75 ---- .../test_spread_model_operations_async.py | 76 ---- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../spread}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 -- ...t_negotiation_different_body_operations.py | 35 -- ...tiation_different_body_operations_async.py | 36 -- ...ontent_negotiation_same_body_operations.py | 35 -- ..._negotiation_same_body_operations_async.py | 36 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- .../generated_tests/test_json_merge_patch.py | 61 --- .../test_json_merge_patch_async.py | 62 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_media_type_string_body_operations.py | 55 --- ...media_type_string_body_operations_async.py | 56 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../mediatype}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_multi_part_form_data_operations.py | 152 ------- ...t_multi_part_form_data_operations_async.py | 153 ------- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../multipart}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_pageable.py | 22 - .../generated_tests/test_pageable_async.py | 23 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../pageable}/apiview_mapping_python.json | 0 .../payload-xml/generated_tests/conftest.py | 35 -- ...el_with_array_of_model_value_operations.py | 34 -- ...h_array_of_model_value_operations_async.py | 35 -- ..._model_with_attributes_value_operations.py | 34 -- ..._with_attributes_value_operations_async.py | 35 -- ..._model_with_dictionary_value_operations.py | 34 -- ..._with_dictionary_value_operations_async.py | 35 -- ...model_with_empty_array_value_operations.py | 34 -- ...with_empty_array_value_operations_async.py | 35 -- ...del_with_encoded_names_value_operations.py | 34 -- ...th_encoded_names_value_operations_async.py | 35 -- ...el_with_optional_field_value_operations.py | 34 -- ...h_optional_field_value_operations_async.py | 35 -- ...el_with_renamed_arrays_value_operations.py | 34 -- ...h_renamed_arrays_value_operations_async.py | 35 -- ...el_with_renamed_fields_value_operations.py | 34 -- ...h_renamed_fields_value_operations_async.py | 35 -- ...del_with_simple_arrays_value_operations.py | 34 -- ...th_simple_arrays_value_operations_async.py | 35 -- ...st_xml_model_with_text_value_operations.py | 34 -- ..._model_with_text_value_operations_async.py | 35 -- ...l_with_unwrapped_array_value_operations.py | 34 -- ..._unwrapped_array_value_operations_async.py | 35 -- .../test_xml_simple_model_value_operations.py | 34 -- ...xml_simple_model_value_operations_async.py | 35 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../xml}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 47 --- .../test_resiliency_service_driven.py | 42 -- .../test_resiliency_service_driven_async.py | 43 -- .../generated_tests/testpreparer.py | 28 -- .../generated_tests/testpreparer_async.py | 20 - .../srv/driven1}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 47 --- .../test_resiliency_service_driven.py | 51 --- .../test_resiliency_service_driven_async.py | 52 --- .../generated_tests/testpreparer.py | 28 -- .../generated_tests/testpreparer_async.py | 20 - .../srv/driven2}/apiview_mapping_python.json | 0 .../routes/generated_tests/conftest.py | 35 -- .../routes/generated_tests/test_routes.py | 22 - .../generated_tests/test_routes_async.py | 23 -- .../test_routes_in_interface_operations.py | 22 - ...st_routes_in_interface_operations_async.py | 23 -- .../test_routes_path_parameters_operations.py | 332 --------------- ...routes_path_parameters_operations_async.py | 333 --------------- ...test_routes_query_parameters_operations.py | 178 -------- ...outes_query_parameters_operations_async.py | 179 -------- .../routes/generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../{ => routes}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_json_property_operations.py | 33 -- .../test_json_property_operations_async.py | 34 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../json}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_not_defined.py | 22 - .../generated_tests/test_not_defined_async.py | 23 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../notdefined}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_multiple.py | 33 -- .../generated_tests/test_multiple_async.py | 34 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../multiple}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_single.py | 22 - .../generated_tests/test_single_async.py | 23 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../path/single}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 37 -- .../generated_tests/test_not_versioned.py | 44 -- .../test_not_versioned_async.py | 45 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../notversioned}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_versioned.py | 49 --- .../generated_tests/test_versioned_async.py | 50 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../versioned}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 -- .../test_conditional_request.py | 49 --- .../test_conditional_request_async.py | 50 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 37 -- .../generated_tests/test_repeatability.py | 22 - .../test_repeatability_async.py | 23 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../special-words/generated_tests/conftest.py | 37 -- ...ecial_words_model_properties_operations.py | 24 -- ...words_model_properties_operations_async.py | 25 -- .../test_special_words_models_operations.py | 376 ----------------- ...t_special_words_models_operations_async.py | 377 ----------------- .../test_special_words_operations.py | 310 -------------- .../test_special_words_operations_async.py | 311 -------------- ...est_special_words_parameters_operations.py | 387 ----------------- ...ecial_words_parameters_operations_async.py | 388 ------------------ .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_array_boolean_value_operations.py | 33 -- ...st_array_boolean_value_operations_async.py | 34 -- .../test_array_datetime_value_operations.py | 33 -- ...t_array_datetime_value_operations_async.py | 34 -- .../test_array_duration_value_operations.py | 33 -- ...t_array_duration_value_operations_async.py | 34 -- .../test_array_float32_value_operations.py | 33 -- ...st_array_float32_value_operations_async.py | 34 -- .../test_array_int32_value_operations.py | 33 -- ...test_array_int32_value_operations_async.py | 34 -- .../test_array_int64_value_operations.py | 33 -- ...test_array_int64_value_operations_async.py | 34 -- .../test_array_model_value_operations.py | 33 -- ...test_array_model_value_operations_async.py | 34 -- ...array_nullable_boolean_value_operations.py | 33 -- ...nullable_boolean_value_operations_async.py | 34 -- ...t_array_nullable_float_value_operations.py | 33 -- ...y_nullable_float_value_operations_async.py | 34 -- ...t_array_nullable_int32_value_operations.py | 33 -- ...y_nullable_int32_value_operations_async.py | 34 -- ...t_array_nullable_model_value_operations.py | 33 -- ...y_nullable_model_value_operations_async.py | 34 -- ..._array_nullable_string_value_operations.py | 33 -- ..._nullable_string_value_operations_async.py | 34 -- .../test_array_string_value_operations.py | 33 -- ...est_array_string_value_operations_async.py | 34 -- .../test_array_unknown_value_operations.py | 33 -- ...st_array_unknown_value_operations_async.py | 34 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../array}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- ...est_dictionary_boolean_value_operations.py | 33 -- ...ctionary_boolean_value_operations_async.py | 34 -- ...st_dictionary_datetime_value_operations.py | 33 -- ...tionary_datetime_value_operations_async.py | 34 -- ...st_dictionary_duration_value_operations.py | 33 -- ...tionary_duration_value_operations_async.py | 34 -- ...est_dictionary_float32_value_operations.py | 33 -- ...ctionary_float32_value_operations_async.py | 34 -- .../test_dictionary_int32_value_operations.py | 33 -- ...dictionary_int32_value_operations_async.py | 34 -- .../test_dictionary_int64_value_operations.py | 33 -- ...dictionary_int64_value_operations_async.py | 34 -- .../test_dictionary_model_value_operations.py | 33 -- ...dictionary_model_value_operations_async.py | 34 -- ...tionary_nullable_float_value_operations.py | 33 -- ...y_nullable_float_value_operations_async.py | 34 -- ...ionary_recursive_model_value_operations.py | 33 -- ..._recursive_model_value_operations_async.py | 34 -- ...test_dictionary_string_value_operations.py | 33 -- ...ictionary_string_value_operations_async.py | 34 -- ...est_dictionary_unknown_value_operations.py | 33 -- ...ctionary_unknown_value_operations_async.py | 34 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../dictionary}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_extensible_string_operations.py | 55 --- ...test_extensible_string_operations_async.py | 56 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../extensible}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_fixed_string_operations.py | 46 --- .../test_fixed_string_operations_async.py | 47 --- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../enum/fixed}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_empty.py | 44 -- .../generated_tests/test_empty_async.py | 45 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../model/empty}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- .../test_enum_discriminator.py | 89 ---- .../test_enum_discriminator_async.py | 90 ---- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 -- .../test_nested_discriminator.py | 71 ---- .../test_nested_discriminator_async.py | 72 ---- .../generated_tests/testpreparer.py | 28 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- .../generated_tests/test_not_discriminated.py | 44 -- .../test_not_discriminated_async.py | 45 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_recursive.py | 33 -- .../generated_tests/test_recursive_async.py | 34 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../recursive}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 -- .../test_single_discriminator.py | 80 ---- .../test_single_discriminator_async.py | 81 ---- .../generated_tests/testpreparer.py | 28 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_usage.py | 44 -- .../generated_tests/test_usage_async.py | 45 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../model/usage}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_visibility.py | 90 ---- .../generated_tests/test_visibility_async.py | 91 ---- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../visibility}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 45 -- ...tends_different_spread_float_operations.py | 33 -- ...different_spread_float_operations_async.py | 34 -- ...different_spread_model_array_operations.py | 33 -- ...ent_spread_model_array_operations_async.py | 34 -- ...tends_different_spread_model_operations.py | 33 -- ...different_spread_model_operations_async.py | 34 -- ...ends_different_spread_string_operations.py | 33 -- ...ifferent_spread_string_operations_async.py | 34 -- ...nal_properties_extends_float_operations.py | 33 -- ...operties_extends_float_operations_async.py | 34 -- ...operties_extends_model_array_operations.py | 33 -- ...es_extends_model_array_operations_async.py | 34 -- ...nal_properties_extends_model_operations.py | 33 -- ...operties_extends_model_operations_async.py | 34 -- ...al_properties_extends_string_operations.py | 33 -- ...perties_extends_string_operations_async.py | 34 -- ...ties_extends_unknown_derived_operations.py | 33 -- ...xtends_unknown_derived_operations_async.py | 34 -- ...xtends_unknown_discriminated_operations.py | 33 -- ..._unknown_discriminated_operations_async.py | 34 -- ...l_properties_extends_unknown_operations.py | 33 -- ...erties_extends_unknown_operations_async.py | 34 -- ...ditional_properties_is_float_operations.py | 33 -- ...al_properties_is_float_operations_async.py | 34 -- ...al_properties_is_model_array_operations.py | 33 -- ...perties_is_model_array_operations_async.py | 34 -- ...ditional_properties_is_model_operations.py | 33 -- ...al_properties_is_model_operations_async.py | 34 -- ...itional_properties_is_string_operations.py | 33 -- ...l_properties_is_string_operations_async.py | 34 -- ...roperties_is_unknown_derived_operations.py | 33 -- ...ies_is_unknown_derived_operations_async.py | 34 -- ...ies_is_unknown_discriminated_operations.py | 33 -- ..._unknown_discriminated_operations_async.py | 34 -- ...tional_properties_is_unknown_operations.py | 33 -- ..._properties_is_unknown_operations_async.py | 34 -- ...l_properties_multiple_spread_operations.py | 33 -- ...erties_multiple_spread_operations_async.py | 34 -- ...rties_spread_different_float_operations.py | 33 -- ...spread_different_float_operations_async.py | 34 -- ...spread_different_model_array_operations.py | 33 -- ..._different_model_array_operations_async.py | 34 -- ...rties_spread_different_model_operations.py | 33 -- ...spread_different_model_operations_async.py | 34 -- ...ties_spread_different_string_operations.py | 33 -- ...pread_different_string_operations_async.py | 34 -- ...onal_properties_spread_float_operations.py | 33 -- ...roperties_spread_float_operations_async.py | 34 -- ...roperties_spread_model_array_operations.py | 33 -- ...ies_spread_model_array_operations_async.py | 34 -- ...onal_properties_spread_model_operations.py | 33 -- ...roperties_spread_model_operations_async.py | 34 -- ...d_record_discriminated_union_operations.py | 33 -- ...rd_discriminated_union_operations_async.py | 34 -- ...ord_non_discriminated_union2_operations.py | 33 -- ...n_discriminated_union2_operations_async.py | 36 -- ...ord_non_discriminated_union3_operations.py | 33 -- ...n_discriminated_union3_operations_async.py | 36 -- ...cord_non_discriminated_union_operations.py | 33 -- ...on_discriminated_union_operations_async.py | 34 -- ...operties_spread_record_union_operations.py | 33 -- ...es_spread_record_union_operations_async.py | 34 -- ...nal_properties_spread_string_operations.py | 33 -- ...operties_spread_string_operations_async.py | 34 -- .../generated_tests/testpreparer.py | 28 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_nullable_bytes_operations.py | 53 --- .../test_nullable_bytes_operations_async.py | 54 --- ...st_nullable_collections_byte_operations.py | 53 --- ...lable_collections_byte_operations_async.py | 54 --- ...t_nullable_collections_model_operations.py | 53 --- ...able_collections_model_operations_async.py | 54 --- ..._nullable_collections_string_operations.py | 53 --- ...ble_collections_string_operations_async.py | 54 --- .../test_nullable_datetime_operations.py | 53 --- ...test_nullable_datetime_operations_async.py | 54 --- .../test_nullable_duration_operations.py | 53 --- ...test_nullable_duration_operations_async.py | 54 --- .../test_nullable_string_operations.py | 53 --- .../test_nullable_string_operations_async.py | 54 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../nullable}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- ...est_optional_boolean_literal_operations.py | 53 --- ...tional_boolean_literal_operations_async.py | 54 --- .../test_optional_bytes_operations.py | 53 --- .../test_optional_bytes_operations_async.py | 54 --- ...st_optional_collections_byte_operations.py | 53 --- ...ional_collections_byte_operations_async.py | 54 --- ...t_optional_collections_model_operations.py | 53 --- ...onal_collections_model_operations_async.py | 54 --- .../test_optional_datetime_operations.py | 53 --- ...test_optional_datetime_operations_async.py | 54 --- .../test_optional_duration_operations.py | 53 --- ...test_optional_duration_operations_async.py | 54 --- .../test_optional_float_literal_operations.py | 53 --- ...optional_float_literal_operations_async.py | 54 --- .../test_optional_int_literal_operations.py | 53 --- ...t_optional_int_literal_operations_async.py | 54 --- .../test_optional_plain_date_operations.py | 53 --- ...st_optional_plain_date_operations_async.py | 54 --- .../test_optional_plain_time_operations.py | 53 --- ...st_optional_plain_time_operations_async.py | 54 --- ...tional_required_and_optional_operations.py | 53 --- ..._required_and_optional_operations_async.py | 54 --- ...test_optional_string_literal_operations.py | 53 --- ...ptional_string_literal_operations_async.py | 54 --- .../test_optional_string_operations.py | 53 --- .../test_optional_string_operations_async.py | 54 --- ...optional_union_float_literal_operations.py | 53 --- ...al_union_float_literal_operations_async.py | 54 --- ...t_optional_union_int_literal_operations.py | 53 --- ...onal_union_int_literal_operations_async.py | 54 --- ...ptional_union_string_literal_operations.py | 53 --- ...l_union_string_literal_operations_async.py | 54 --- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../optional}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- ..._value_types_boolean_literal_operations.py | 33 -- ..._types_boolean_literal_operations_async.py | 34 -- .../test_value_types_boolean_operations.py | 33 -- ...st_value_types_boolean_operations_async.py | 34 -- .../test_value_types_bytes_operations.py | 33 -- ...test_value_types_bytes_operations_async.py | 34 -- ..._value_types_collections_int_operations.py | 33 -- ..._types_collections_int_operations_async.py | 34 -- ...alue_types_collections_model_operations.py | 33 -- ...ypes_collections_model_operations_async.py | 34 -- ...lue_types_collections_string_operations.py | 33 -- ...pes_collections_string_operations_async.py | 34 -- .../test_value_types_datetime_operations.py | 33 -- ...t_value_types_datetime_operations_async.py | 34 -- .../test_value_types_decimal128_operations.py | 33 -- ...value_types_decimal128_operations_async.py | 34 -- .../test_value_types_decimal_operations.py | 33 -- ...st_value_types_decimal_operations_async.py | 34 -- ...alue_types_dictionary_string_operations.py | 33 -- ...ypes_dictionary_string_operations_async.py | 34 -- .../test_value_types_duration_operations.py | 33 -- ...t_value_types_duration_operations_async.py | 34 -- .../test_value_types_enum_operations.py | 33 -- .../test_value_types_enum_operations_async.py | 34 -- ..._value_types_extensible_enum_operations.py | 33 -- ..._types_extensible_enum_operations_async.py | 34 -- ...st_value_types_float_literal_operations.py | 33 -- ...ue_types_float_literal_operations_async.py | 34 -- .../test_value_types_float_operations.py | 33 -- ...test_value_types_float_operations_async.py | 34 -- ...test_value_types_int_literal_operations.py | 33 -- ...alue_types_int_literal_operations_async.py | 34 -- .../test_value_types_int_operations.py | 33 -- .../test_value_types_int_operations_async.py | 34 -- .../test_value_types_model_operations.py | 33 -- ...test_value_types_model_operations_async.py | 34 -- .../test_value_types_never_operations.py | 33 -- ...test_value_types_never_operations_async.py | 34 -- ...t_value_types_string_literal_operations.py | 33 -- ...e_types_string_literal_operations_async.py | 34 -- .../test_value_types_string_operations.py | 33 -- ...est_value_types_string_operations_async.py | 34 -- ...value_types_union_enum_value_operations.py | 33 -- ...types_union_enum_value_operations_async.py | 34 -- ...ue_types_union_float_literal_operations.py | 33 -- ...es_union_float_literal_operations_async.py | 34 -- ...alue_types_union_int_literal_operations.py | 33 -- ...ypes_union_int_literal_operations_async.py | 34 -- ...e_types_union_string_literal_operations.py | 33 -- ...s_union_string_literal_operations_async.py | 34 -- ...st_value_types_unknown_array_operations.py | 33 -- ...ue_types_unknown_array_operations_async.py | 34 -- ...est_value_types_unknown_dict_operations.py | 33 -- ...lue_types_unknown_dict_operations_async.py | 34 -- ...test_value_types_unknown_int_operations.py | 33 -- ...alue_types_unknown_int_operations_async.py | 34 -- ...t_value_types_unknown_string_operations.py | 33 -- ...e_types_unknown_string_operations_async.py | 34 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../valuetypes}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_scalar_boolean_operations.py | 34 -- .../test_scalar_boolean_operations_async.py | 35 -- .../test_scalar_decimal128_type_operations.py | 45 -- ...scalar_decimal128_type_operations_async.py | 46 --- ...est_scalar_decimal128_verify_operations.py | 34 -- ...alar_decimal128_verify_operations_async.py | 35 -- .../test_scalar_decimal_type_operations.py | 45 -- ...st_scalar_decimal_type_operations_async.py | 46 --- .../test_scalar_decimal_verify_operations.py | 34 -- ..._scalar_decimal_verify_operations_async.py | 35 -- .../test_scalar_string_operations.py | 34 -- .../test_scalar_string_operations_async.py | 35 -- .../test_scalar_unknown_operations.py | 34 -- .../test_scalar_unknown_operations_async.py | 35 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../scalar}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../test_union_enums_only_operations.py | 34 -- .../test_union_enums_only_operations_async.py | 35 -- .../test_union_floats_only_operations.py | 34 -- ...test_union_floats_only_operations_async.py | 35 -- .../test_union_ints_only_operations.py | 34 -- .../test_union_ints_only_operations_async.py | 35 -- .../test_union_mixed_literals_operations.py | 34 -- ...t_union_mixed_literals_operations_async.py | 35 -- .../test_union_mixed_types_operations.py | 48 --- ...test_union_mixed_types_operations_async.py | 49 --- .../test_union_models_only_operations.py | 34 -- ...test_union_models_only_operations_async.py | 35 -- .../test_union_string_and_array_operations.py | 34 -- ...union_string_and_array_operations_async.py | 35 -- ...nion_string_extensible_named_operations.py | 34 -- ...tring_extensible_named_operations_async.py | 35 -- ...test_union_string_extensible_operations.py | 34 -- ...nion_string_extensible_operations_async.py | 35 -- .../test_union_strings_only_operations.py | 34 -- ...est_union_strings_only_operations_async.py | 35 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../union}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_added.py | 36 -- .../generated_tests/test_added_async.py | 37 -- .../test_added_interface_v2_operations.py | 24 -- ...est_added_interface_v2_operations_async.py | 25 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../added}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 37 -- .../generated_tests/test_made_optional.py | 24 -- .../test_made_optional_async.py | 25 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../madeoptional}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_removed.py | 35 -- .../generated_tests/test_removed_async.py | 36 -- .../generated_tests/testpreparer.py | 24 -- .../generated_tests/testpreparer_async.py | 20 - .../removed}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 -- .../generated_tests/test_renamed_from.py | 25 -- .../test_renamed_from_async.py | 26 -- ...t_renamed_from_new_interface_operations.py | 24 -- ...med_from_new_interface_operations_async.py | 25 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../renamedfrom}/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 45 -- .../test_return_type_changed_from.py | 25 -- .../test_return_type_changed_from_async.py | 26 -- .../generated_tests/testpreparer.py | 28 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 -- .../generated_tests/test_type_changed_from.py | 25 -- .../test_type_changed_from_async.py | 26 -- .../generated_tests/testpreparer.py | 26 -- .../generated_tests/testpreparer_async.py | 20 - .../apiview_mapping_python.json | 0 .../authentication/apikey/models/_models.py | 1 + .../http/custom/models/_models.py | 1 + .../authentication/oauth2/models/_models.py | 1 + 781 files changed, 6 insertions(+), 29025 deletions(-) rename packages/typespec-python/test/azure/generated/authentication-api-key/{ => authentication/apikey}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/authentication-http-custom/{ => authentication/http/custom}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/authentication-oauth2/{ => authentication/oauth2}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/authentication-union/{ => authentication/union}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-client-generator-core-access/{ => specs/azure/clientgenerator/core/access}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/{ => specs/azure/clientgenerator/core/flattenproperty}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/{ => specs/azure/clientgenerator/core/usage}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-basic/{ => specs/azure/core/basic}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-lro-standard/{ => specs/azure/core/lro/standard}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-model/{ => specs/azure/core/model}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-page/{ => specs/azure/core/page}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-scalar/{ => specs/azure/core/scalar}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-traits/{ => specs/azure/core/traits}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-example-basic/{ => specs/azure/example/basic}/apiview_mapping_python.json (100%) rename packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/{ => azure/resourcemanager/commonproperties}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py rename packages/typespec-python/test/azure/generated/azure-resource-manager-resources/{ => azure/resourcemanager/resources}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py rename packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/{ => azure/specialheaders/xmsclientrequestid}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azurecore-lro-rpc/{ => azurecore/lro/rpc}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py delete mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py delete mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-naming/{ => client/naming}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-default/{ => client/structure/service}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-multiclient/{ => client/structure/multiclient}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-renamedoperation/{ => client/structure/renamedoperation}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/{ => client/structure/twooperationgroup}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-bytes/{ => encode/bytes}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-datetime/{ => encode/datetime}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-duration/{ => encode/duration}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-numeric/{ => encode/numeric}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/headasbooleanfalse/{ => headasbooleanfalse}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/headasbooleantrue/{ => headasbooleantrue}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-basic/{ => parameters/basic}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-body-optionality/{ => parameters/bodyoptionality}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-collection-format/{ => parameters/collectionformat}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-spread/{ => parameters/spread}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-content-negotiation/{ => payload/contentnegotiation}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-json-merge-patch/{ => payload/jsonmergepatch}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-media-type/{ => payload/mediatype}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-multipart/{ => payload/multipart}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-pageable/{ => payload/pageable}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-xml/{ => payload/xml}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/resiliency-srv-driven1/{ => resiliency/srv/driven1}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/resiliency-srv-driven2/{ => resiliency/srv/driven2}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/routes/{ => routes}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/serialization-encoded-name-json/{ => serialization/encodedname/json}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py delete mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py delete mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-endpoint-not-defined/{ => server/endpoint/notdefined}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-path-multiple/{ => server/path/multiple}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-path-single/{ => server/path/single}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-versions-not-versioned/{ => server/versions/notversioned}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-versions-versioned/{ => server/versions/versioned}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/special-headers-conditional-request/{ => specialheaders/conditionalrequest}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/special-headers-repeatability/{ => specialheaders/repeatability}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/special-words/{ => specialwords}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-array/{ => typetest/array}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-dictionary/{ => typetest/dictionary}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-enum-extensible/{ => typetest/enum/extensible}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-enum-fixed/{ => typetest/enum/fixed}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-empty/{ => typetest/model/empty}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/{ => typetest/model/enumdiscriminator}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/{ => typetest/model/nesteddiscriminator}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/{ => typetest/model/notdiscriminated}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-recursive/{ => typetest/model/recursive}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/{ => typetest/model/singlediscriminator}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-usage/{ => typetest/model/usage}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-visibility/{ => typetest/model/visibility}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/{ => typetest/property/additionalproperties}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-nullable/{ => typetest/property/nullable}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-optional/{ => typetest/property/optional}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-valuetypes/{ => typetest/property/valuetypes}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-scalar/{ => typetest/scalar}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-union/{ => typetest/union}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-added/{ => versioning/added}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-madeoptional/{ => versioning/madeoptional}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-removed/{ => versioning/removed}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-renamedfrom/{ => versioning/renamedfrom}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/{ => versioning/returntypechangedfrom}/apiview_mapping_python.json (100%) delete mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py delete mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-typechangedfrom/{ => versioning/typechangedfrom}/apiview_mapping_python.json (100%) diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-api-key/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/models/_models.py b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/models/_models.py index dd2935b5984..c9f6c70366a 100644 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/models/_models.py +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/models/_models.py @@ -16,6 +16,7 @@ class InvalidAuth(_model_base.Model): """InvalidAuth. + All required parameters must be populated in order to send to server. :ivar error: Required. :vartype error: str diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py deleted file mode 100644 index 8082748a92a..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - apikey_subscription_id = os.environ.get("APIKEY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - apikey_tenant_id = os.environ.get("APIKEY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - apikey_client_id = os.environ.get("APIKEY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - apikey_client_secret = os.environ.get("APIKEY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=apikey_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=apikey_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=apikey_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=apikey_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py deleted file mode 100644 index 2916e7b9498..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ApiKeyClientTestBase, ApiKeyPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestApiKey(ApiKeyClientTestBase): - @ApiKeyPreparer() - @recorded_by_proxy - def test_valid(self, apikey_endpoint): - client = self.create_client(endpoint=apikey_endpoint) - response = client.valid() - - # please add some check logic here by yourself - # ... - - @ApiKeyPreparer() - @recorded_by_proxy - def test_invalid(self, apikey_endpoint): - client = self.create_client(endpoint=apikey_endpoint) - response = client.invalid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py deleted file mode 100644 index 214be1ee608..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ApiKeyPreparer -from testpreparer_async import ApiKeyClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestApiKeyAsync(ApiKeyClientTestBaseAsync): - @ApiKeyPreparer() - @recorded_by_proxy_async - async def test_valid(self, apikey_endpoint): - client = self.create_async_client(endpoint=apikey_endpoint) - response = await client.valid() - - # please add some check logic here by yourself - # ... - - @ApiKeyPreparer() - @recorded_by_proxy_async - async def test_invalid(self, apikey_endpoint): - client = self.create_async_client(endpoint=apikey_endpoint) - response = await client.invalid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py deleted file mode 100644 index 5af4c923ab1..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.apikey import ApiKeyClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class ApiKeyClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ApiKeyClient) - return self.create_client_from_credential( - ApiKeyClient, - credential=credential, - endpoint=endpoint, - ) - - -ApiKeyPreparer = functools.partial(PowerShellPreparer, "apikey", apikey_endpoint="https://fake_apikey_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py deleted file mode 100644 index 09c26f3fe60..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.apikey.aio import ApiKeyClient -from devtools_testutils import AzureRecordedTestCase - - -class ApiKeyClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ApiKeyClient, is_async=True) - return self.create_client_from_credential( - ApiKeyClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-http-custom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/models/_models.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/models/_models.py index dd2935b5984..c9f6c70366a 100644 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/models/_models.py +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/models/_models.py @@ -16,6 +16,7 @@ class InvalidAuth(_model_base.Model): """InvalidAuth. + All required parameters must be populated in order to send to server. :ivar error: Required. :vartype error: str diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py deleted file mode 100644 index fac3e2bc399..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - custom_subscription_id = os.environ.get("CUSTOM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - custom_tenant_id = os.environ.get("CUSTOM_TENANT_ID", "00000000-0000-0000-0000-000000000000") - custom_client_id = os.environ.get("CUSTOM_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - custom_client_secret = os.environ.get("CUSTOM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=custom_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=custom_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=custom_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=custom_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py deleted file mode 100644 index d87f2c0f154..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import CustomClientTestBase, CustomPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCustom(CustomClientTestBase): - @CustomPreparer() - @recorded_by_proxy - def test_valid(self, custom_endpoint): - client = self.create_client(endpoint=custom_endpoint) - response = client.valid() - - # please add some check logic here by yourself - # ... - - @CustomPreparer() - @recorded_by_proxy - def test_invalid(self, custom_endpoint): - client = self.create_client(endpoint=custom_endpoint) - response = client.invalid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py deleted file mode 100644 index 97407d6ad8b..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import CustomPreparer -from testpreparer_async import CustomClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCustomAsync(CustomClientTestBaseAsync): - @CustomPreparer() - @recorded_by_proxy_async - async def test_valid(self, custom_endpoint): - client = self.create_async_client(endpoint=custom_endpoint) - response = await client.valid() - - # please add some check logic here by yourself - # ... - - @CustomPreparer() - @recorded_by_proxy_async - async def test_invalid(self, custom_endpoint): - client = self.create_async_client(endpoint=custom_endpoint) - response = await client.invalid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py deleted file mode 100644 index 5d6664b2552..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.http.custom import CustomClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class CustomClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(CustomClient) - return self.create_client_from_credential( - CustomClient, - credential=credential, - endpoint=endpoint, - ) - - -CustomPreparer = functools.partial(PowerShellPreparer, "custom", custom_endpoint="https://fake_custom_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py deleted file mode 100644 index 3355fa69b2b..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.http.custom.aio import CustomClient -from devtools_testutils import AzureRecordedTestCase - - -class CustomClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(CustomClient, is_async=True) - return self.create_client_from_credential( - CustomClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-oauth2/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/models/_models.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/models/_models.py index dd2935b5984..c9f6c70366a 100644 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/models/_models.py +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/models/_models.py @@ -16,6 +16,7 @@ class InvalidAuth(_model_base.Model): """InvalidAuth. + All required parameters must be populated in order to send to server. :ivar error: Required. :vartype error: str diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py deleted file mode 100644 index f8a1115d861..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - oauth2_subscription_id = os.environ.get("OAUTH2_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - oauth2_tenant_id = os.environ.get("OAUTH2_TENANT_ID", "00000000-0000-0000-0000-000000000000") - oauth2_client_id = os.environ.get("OAUTH2_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - oauth2_client_secret = os.environ.get("OAUTH2_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=oauth2_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=oauth2_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=oauth2_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=oauth2_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py deleted file mode 100644 index d1c8fee03c3..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OAuth2ClientTestBase, OAuth2Preparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOAuth2(OAuth2ClientTestBase): - @OAuth2Preparer() - @recorded_by_proxy - def test_valid(self, oauth2_endpoint): - client = self.create_client(endpoint=oauth2_endpoint) - response = client.valid() - - # please add some check logic here by yourself - # ... - - @OAuth2Preparer() - @recorded_by_proxy - def test_invalid(self, oauth2_endpoint): - client = self.create_client(endpoint=oauth2_endpoint) - response = client.invalid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py deleted file mode 100644 index f7f88060283..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OAuth2Preparer -from testpreparer_async import OAuth2ClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOAuth2Async(OAuth2ClientTestBaseAsync): - @OAuth2Preparer() - @recorded_by_proxy_async - async def test_valid(self, oauth2_endpoint): - client = self.create_async_client(endpoint=oauth2_endpoint) - response = await client.valid() - - # please add some check logic here by yourself - # ... - - @OAuth2Preparer() - @recorded_by_proxy_async - async def test_invalid(self, oauth2_endpoint): - client = self.create_async_client(endpoint=oauth2_endpoint) - response = await client.invalid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py deleted file mode 100644 index 24fb53e3ec1..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.oauth2 import OAuth2Client -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class OAuth2ClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(OAuth2Client) - return self.create_client_from_credential( - OAuth2Client, - credential=credential, - endpoint=endpoint, - ) - - -OAuth2Preparer = functools.partial(PowerShellPreparer, "oauth2", oauth2_endpoint="https://fake_oauth2_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py deleted file mode 100644 index 6eff123aef9..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.oauth2.aio import OAuth2Client -from devtools_testutils import AzureRecordedTestCase - - -class OAuth2ClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(OAuth2Client, is_async=True) - return self.create_client_from_credential( - OAuth2Client, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/authentication-union/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-union/authentication/union/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-union/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-union/authentication/union/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py deleted file mode 100644 index 5e28d6377c8..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - union_subscription_id = os.environ.get("UNION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - union_tenant_id = os.environ.get("UNION_TENANT_ID", "00000000-0000-0000-0000-000000000000") - union_client_id = os.environ.get("UNION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - union_client_secret = os.environ.get("UNION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py deleted file mode 100644 index fdfea7e808b..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnion(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_valid_key(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.valid_key() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_valid_token(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.valid_token() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py deleted file mode 100644 index 7efe4a658f5..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_valid_key(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.valid_key() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_valid_token(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.valid_token() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py deleted file mode 100644 index 70237769077..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.union import UnionClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class UnionClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(UnionClient) - return self.create_client_from_credential( - UnionClient, - credential=credential, - endpoint=endpoint, - ) - - -UnionPreparer = functools.partial(PowerShellPreparer, "union", union_endpoint="https://fake_union_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py deleted file mode 100644 index 7b184e26233..00000000000 --- a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from authentication.union.aio import UnionClient -from devtools_testutils import AzureRecordedTestCase - - -class UnionClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(UnionClient, is_async=True) - return self.create_client_from_credential( - UnionClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py deleted file mode 100644 index 33880e8e0f8..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - access_subscription_id = os.environ.get("ACCESS_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - access_tenant_id = os.environ.get("ACCESS_TENANT_ID", "00000000-0000-0000-0000-000000000000") - access_client_id = os.environ.get("ACCESS_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - access_client_secret = os.environ.get("ACCESS_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=access_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=access_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=access_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=access_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py deleted file mode 100644 index bf07380e8fa..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AccessClientTestBase, AccessPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAccessPublicOperationOperations(AccessClientTestBase): - @AccessPreparer() - @recorded_by_proxy - def test_public_operation_no_decorator_in_public(self, access_endpoint): - client = self.create_client(endpoint=access_endpoint) - response = client.public_operation.no_decorator_in_public( - name="str", - ) - - # please add some check logic here by yourself - # ... - - @AccessPreparer() - @recorded_by_proxy - def test_public_operation_public_decorator_in_public(self, access_endpoint): - client = self.create_client(endpoint=access_endpoint) - response = client.public_operation.public_decorator_in_public( - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py deleted file mode 100644 index 2396a14d56f..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AccessPreparer -from testpreparer_async import AccessClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAccessPublicOperationOperationsAsync(AccessClientTestBaseAsync): - @AccessPreparer() - @recorded_by_proxy_async - async def test_public_operation_no_decorator_in_public(self, access_endpoint): - client = self.create_async_client(endpoint=access_endpoint) - response = await client.public_operation.no_decorator_in_public( - name="str", - ) - - # please add some check logic here by yourself - # ... - - @AccessPreparer() - @recorded_by_proxy_async - async def test_public_operation_public_decorator_in_public(self, access_endpoint): - client = self.create_async_client(endpoint=access_endpoint) - response = await client.public_operation.public_decorator_in_public( - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py deleted file mode 100644 index df2954375fd..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AccessClientTestBase, AccessPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAccessSharedModelInOperationOperations(AccessClientTestBase): - @AccessPreparer() - @recorded_by_proxy - def test_shared_model_in_operation_public(self, access_endpoint): - client = self.create_client(endpoint=access_endpoint) - response = client.shared_model_in_operation.public( - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py deleted file mode 100644 index a4ce11fd46b..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AccessPreparer -from testpreparer_async import AccessClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAccessSharedModelInOperationOperationsAsync(AccessClientTestBaseAsync): - @AccessPreparer() - @recorded_by_proxy_async - async def test_shared_model_in_operation_public(self, access_endpoint): - client = self.create_async_client(endpoint=access_endpoint) - response = await client.shared_model_in_operation.public( - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py deleted file mode 100644 index 7deee53be35..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.clientgenerator.core.access import AccessClient - - -class AccessClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(AccessClient) - return self.create_client_from_credential( - AccessClient, - credential=credential, - endpoint=endpoint, - ) - - -AccessPreparer = functools.partial(PowerShellPreparer, "access", access_endpoint="https://fake_access_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py deleted file mode 100644 index 7d4e8352e2b..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.clientgenerator.core.access.aio import AccessClient - - -class AccessClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(AccessClient, is_async=True) - return self.create_client_from_credential( - AccessClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-client-generator-core-access/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py deleted file mode 100644 index 77400d8847a..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - flattenproperty_subscription_id = os.environ.get( - "FLATTENPROPERTY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - flattenproperty_tenant_id = os.environ.get("FLATTENPROPERTY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - flattenproperty_client_id = os.environ.get("FLATTENPROPERTY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - flattenproperty_client_secret = os.environ.get( - "FLATTENPROPERTY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=flattenproperty_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=flattenproperty_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=flattenproperty_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=flattenproperty_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py deleted file mode 100644 index 8563ebaa3b4..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import FlattenPropertyClientTestBase, FlattenPropertyPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestFlattenProperty(FlattenPropertyClientTestBase): - @FlattenPropertyPreparer() - @recorded_by_proxy - def test_put_flatten_model(self, flattenproperty_endpoint): - client = self.create_client(endpoint=flattenproperty_endpoint) - response = client.put_flatten_model( - input={"name": "str", "properties": {"age": 0, "description": "str"}}, - ) - - # please add some check logic here by yourself - # ... - - @FlattenPropertyPreparer() - @recorded_by_proxy - def test_put_nested_flatten_model(self, flattenproperty_endpoint): - client = self.create_client(endpoint=flattenproperty_endpoint) - response = client.put_nested_flatten_model( - input={"name": "str", "properties": {"properties": {"age": 0, "description": "str"}, "summary": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py deleted file mode 100644 index 93a1b1b5eff..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import FlattenPropertyPreparer -from testpreparer_async import FlattenPropertyClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestFlattenPropertyAsync(FlattenPropertyClientTestBaseAsync): - @FlattenPropertyPreparer() - @recorded_by_proxy_async - async def test_put_flatten_model(self, flattenproperty_endpoint): - client = self.create_async_client(endpoint=flattenproperty_endpoint) - response = await client.put_flatten_model( - input={"name": "str", "properties": {"age": 0, "description": "str"}}, - ) - - # please add some check logic here by yourself - # ... - - @FlattenPropertyPreparer() - @recorded_by_proxy_async - async def test_put_nested_flatten_model(self, flattenproperty_endpoint): - client = self.create_async_client(endpoint=flattenproperty_endpoint) - response = await client.put_nested_flatten_model( - input={"name": "str", "properties": {"properties": {"age": 0, "description": "str"}, "summary": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py deleted file mode 100644 index 739c5ff2de4..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.clientgenerator.core.flattenproperty import FlattenPropertyClient - - -class FlattenPropertyClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(FlattenPropertyClient) - return self.create_client_from_credential( - FlattenPropertyClient, - credential=credential, - endpoint=endpoint, - ) - - -FlattenPropertyPreparer = functools.partial( - PowerShellPreparer, "flattenproperty", flattenproperty_endpoint="https://fake_flattenproperty_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py deleted file mode 100644 index f66cc7610b9..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.clientgenerator.core.flattenproperty.aio import FlattenPropertyClient - - -class FlattenPropertyClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(FlattenPropertyClient, is_async=True) - return self.create_client_from_credential( - FlattenPropertyClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py deleted file mode 100644 index 6a990b92bd7..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - usage_subscription_id = os.environ.get("USAGE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - usage_tenant_id = os.environ.get("USAGE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - usage_client_id = os.environ.get("USAGE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - usage_client_secret = os.environ.get("USAGE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py deleted file mode 100644 index 1594a7225fd..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py +++ /dev/null @@ -1,44 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UsageClientTestBase, UsagePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUsageModelInOperationOperations(UsageClientTestBase): - @UsagePreparer() - @recorded_by_proxy - def test_model_in_operation_input_to_input_output(self, usage_endpoint): - client = self.create_client(endpoint=usage_endpoint) - response = client.model_in_operation.input_to_input_output( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy - def test_model_in_operation_output_to_input_output(self, usage_endpoint): - client = self.create_client(endpoint=usage_endpoint) - response = client.model_in_operation.output_to_input_output() - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy - def test_model_in_operation_model_in_read_only_property(self, usage_endpoint): - client = self.create_client(endpoint=usage_endpoint) - response = client.model_in_operation.model_in_read_only_property( - body={"result": {"name": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py deleted file mode 100644 index c0b04a06e4d..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UsagePreparer -from testpreparer_async import UsageClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUsageModelInOperationOperationsAsync(UsageClientTestBaseAsync): - @UsagePreparer() - @recorded_by_proxy_async - async def test_model_in_operation_input_to_input_output(self, usage_endpoint): - client = self.create_async_client(endpoint=usage_endpoint) - response = await client.model_in_operation.input_to_input_output( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy_async - async def test_model_in_operation_output_to_input_output(self, usage_endpoint): - client = self.create_async_client(endpoint=usage_endpoint) - response = await client.model_in_operation.output_to_input_output() - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy_async - async def test_model_in_operation_model_in_read_only_property(self, usage_endpoint): - client = self.create_async_client(endpoint=usage_endpoint) - response = await client.model_in_operation.model_in_read_only_property( - body={"result": {"name": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py deleted file mode 100644 index 6d1b1d9481f..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.clientgenerator.core.usage import UsageClient - - -class UsageClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(UsageClient) - return self.create_client_from_credential( - UsageClient, - credential=credential, - endpoint=endpoint, - ) - - -UsagePreparer = functools.partial(PowerShellPreparer, "usage", usage_endpoint="https://fake_usage_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py deleted file mode 100644 index d27978d8baa..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.clientgenerator.core.usage.aio import UsageClient - - -class UsageClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(UsageClient, is_async=True) - return self.create_client_from_credential( - UsageClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py deleted file mode 100644 index 635e5035817..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - basic_subscription_id = os.environ.get("BASIC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - basic_tenant_id = os.environ.get("BASIC_TENANT_ID", "00000000-0000-0000-0000-000000000000") - basic_client_id = os.environ.get("BASIC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - basic_client_secret = os.environ.get("BASIC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py deleted file mode 100644 index 4e1621dc840..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py +++ /dev/null @@ -1,91 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BasicClientTestBase, BasicPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBasic(BasicClientTestBase): - @BasicPreparer() - @recorded_by_proxy - def test_create_or_update(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.create_or_update( - id=0, - resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy - def test_create_or_replace(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.create_or_replace( - id=0, - resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy - def test_get(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.get( - id=0, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy - def test_list(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.list() - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy - def test_delete(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.delete( - id=0, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy - def test_export(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.export( - id=0, - format="str", - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy - def test_export_all_users(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.export_all_users( - format="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py deleted file mode 100644 index c3c7f05a6cf..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py +++ /dev/null @@ -1,92 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BasicPreparer -from testpreparer_async import BasicClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBasicAsync(BasicClientTestBaseAsync): - @BasicPreparer() - @recorded_by_proxy_async - async def test_create_or_update(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.create_or_update( - id=0, - resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy_async - async def test_create_or_replace(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.create_or_replace( - id=0, - resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy_async - async def test_get(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.get( - id=0, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy_async - async def test_list(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = client.list() - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy_async - async def test_delete(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.delete( - id=0, - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy_async - async def test_export(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.export( - id=0, - format="str", - ) - - # please add some check logic here by yourself - # ... - - @BasicPreparer() - @recorded_by_proxy_async - async def test_export_all_users(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.export_all_users( - format="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py deleted file mode 100644 index b662845bf16..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.core.basic import BasicClient - - -class BasicClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(BasicClient) - return self.create_client_from_credential( - BasicClient, - credential=credential, - endpoint=endpoint, - ) - - -BasicPreparer = functools.partial(PowerShellPreparer, "basic", basic_endpoint="https://fake_basic_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py deleted file mode 100644 index 2b63ad3c8f4..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.core.basic.aio import BasicClient - - -class BasicClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(BasicClient, is_async=True) - return self.create_client_from_credential( - BasicClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-basic/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py deleted file mode 100644 index 420b04f8797..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - standard_subscription_id = os.environ.get("STANDARD_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - standard_tenant_id = os.environ.get("STANDARD_TENANT_ID", "00000000-0000-0000-0000-000000000000") - standard_client_id = os.environ.get("STANDARD_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - standard_client_secret = os.environ.get("STANDARD_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=standard_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=standard_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=standard_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=standard_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py deleted file mode 100644 index e697fd1634b..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py +++ /dev/null @@ -1,48 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import StandardClientTestBase, StandardPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStandard(StandardClientTestBase): - @StandardPreparer() - @recorded_by_proxy - def test_begin_create_or_replace(self, standard_endpoint): - client = self.create_client(endpoint=standard_endpoint) - response = client.begin_create_or_replace( - name="str", - resource={"name": "str", "role": "str"}, - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @StandardPreparer() - @recorded_by_proxy - def test_begin_delete(self, standard_endpoint): - client = self.create_client(endpoint=standard_endpoint) - response = client.begin_delete( - name="str", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @StandardPreparer() - @recorded_by_proxy - def test_begin_export(self, standard_endpoint): - client = self.create_client(endpoint=standard_endpoint) - response = client.begin_export( - name="str", - format="str", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py deleted file mode 100644 index 5b8c4cb2479..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py +++ /dev/null @@ -1,55 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import StandardPreparer -from testpreparer_async import StandardClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestStandardAsync(StandardClientTestBaseAsync): - @StandardPreparer() - @recorded_by_proxy_async - async def test_begin_create_or_replace(self, standard_endpoint): - client = self.create_async_client(endpoint=standard_endpoint) - response = await ( - await client.begin_create_or_replace( - name="str", - resource={"name": "str", "role": "str"}, - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @StandardPreparer() - @recorded_by_proxy_async - async def test_begin_delete(self, standard_endpoint): - client = self.create_async_client(endpoint=standard_endpoint) - response = await ( - await client.begin_delete( - name="str", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @StandardPreparer() - @recorded_by_proxy_async - async def test_begin_export(self, standard_endpoint): - client = self.create_async_client(endpoint=standard_endpoint) - response = await ( - await client.begin_export( - name="str", - format="str", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py deleted file mode 100644 index e17f19a3271..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.core.lro.standard import StandardClient - - -class StandardClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(StandardClient) - return self.create_client_from_credential( - StandardClient, - credential=credential, - endpoint=endpoint, - ) - - -StandardPreparer = functools.partial( - PowerShellPreparer, "standard", standard_endpoint="https://fake_standard_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py deleted file mode 100644 index 9b1a3cbea5c..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.core.lro.standard.aio import StandardClient - - -class StandardClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(StandardClient, is_async=True) - return self.create_client_from_credential( - StandardClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-lro-standard/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py deleted file mode 100644 index b11e17f60f7..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - model_subscription_id = os.environ.get("MODEL_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - model_tenant_id = os.environ.get("MODEL_TENANT_ID", "00000000-0000-0000-0000-000000000000") - model_client_id = os.environ.get("MODEL_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - model_client_secret = os.environ.get("MODEL_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=model_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=model_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=model_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=model_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py deleted file mode 100644 index 824eea481e1..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py +++ /dev/null @@ -1,44 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ModelClientTestBase, ModelPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestModelAzureCoreEmbeddingVectorOperations(ModelClientTestBase): - @ModelPreparer() - @recorded_by_proxy - def test_azure_core_embedding_vector_get(self, model_endpoint): - client = self.create_client(endpoint=model_endpoint) - response = client.azure_core_embedding_vector.get() - - # please add some check logic here by yourself - # ... - - @ModelPreparer() - @recorded_by_proxy - def test_azure_core_embedding_vector_put(self, model_endpoint): - client = self.create_client(endpoint=model_endpoint) - response = client.azure_core_embedding_vector.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... - - @ModelPreparer() - @recorded_by_proxy - def test_azure_core_embedding_vector_post(self, model_endpoint): - client = self.create_client(endpoint=model_endpoint) - response = client.azure_core_embedding_vector.post( - body={"embedding": [0]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py deleted file mode 100644 index b7f1463844b..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ModelPreparer -from testpreparer_async import ModelClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestModelAzureCoreEmbeddingVectorOperationsAsync(ModelClientTestBaseAsync): - @ModelPreparer() - @recorded_by_proxy_async - async def test_azure_core_embedding_vector_get(self, model_endpoint): - client = self.create_async_client(endpoint=model_endpoint) - response = await client.azure_core_embedding_vector.get() - - # please add some check logic here by yourself - # ... - - @ModelPreparer() - @recorded_by_proxy_async - async def test_azure_core_embedding_vector_put(self, model_endpoint): - client = self.create_async_client(endpoint=model_endpoint) - response = await client.azure_core_embedding_vector.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... - - @ModelPreparer() - @recorded_by_proxy_async - async def test_azure_core_embedding_vector_post(self, model_endpoint): - client = self.create_async_client(endpoint=model_endpoint) - response = await client.azure_core_embedding_vector.post( - body={"embedding": [0]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py deleted file mode 100644 index e8743f4445d..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.core.model import ModelClient - - -class ModelClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ModelClient) - return self.create_client_from_credential( - ModelClient, - credential=credential, - endpoint=endpoint, - ) - - -ModelPreparer = functools.partial(PowerShellPreparer, "model", model_endpoint="https://fake_model_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py deleted file mode 100644 index 6eafdc7e92f..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.core.model.aio import ModelClient - - -class ModelClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ModelClient, is_async=True) - return self.create_client_from_credential( - ModelClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-model/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py deleted file mode 100644 index 76efbc92590..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - page_subscription_id = os.environ.get("PAGE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - page_tenant_id = os.environ.get("PAGE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - page_client_id = os.environ.get("PAGE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - page_client_secret = os.environ.get("PAGE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=page_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=page_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=page_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=page_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py deleted file mode 100644 index 053081e9ec6..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py +++ /dev/null @@ -1,42 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import PageClientTestBase, PagePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestPage(PageClientTestBase): - @PagePreparer() - @recorded_by_proxy - def test_list_with_page(self, page_endpoint): - client = self.create_client(endpoint=page_endpoint) - response = client.list_with_page() - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @PagePreparer() - @recorded_by_proxy - def test_list_with_parameters(self, page_endpoint): - client = self.create_client(endpoint=page_endpoint) - response = client.list_with_parameters( - body_input={"inputName": "str"}, - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @PagePreparer() - @recorded_by_proxy - def test_list_with_custom_page_model(self, page_endpoint): - client = self.create_client(endpoint=page_endpoint) - response = client.list_with_custom_page_model() - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py deleted file mode 100644 index 3bfc3a98048..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py +++ /dev/null @@ -1,43 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import PagePreparer -from testpreparer_async import PageClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestPageAsync(PageClientTestBaseAsync): - @PagePreparer() - @recorded_by_proxy_async - async def test_list_with_page(self, page_endpoint): - client = self.create_async_client(endpoint=page_endpoint) - response = client.list_with_page() - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @PagePreparer() - @recorded_by_proxy_async - async def test_list_with_parameters(self, page_endpoint): - client = self.create_async_client(endpoint=page_endpoint) - response = client.list_with_parameters( - body_input={"inputName": "str"}, - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @PagePreparer() - @recorded_by_proxy_async - async def test_list_with_custom_page_model(self, page_endpoint): - client = self.create_async_client(endpoint=page_endpoint) - response = client.list_with_custom_page_model() - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py deleted file mode 100644 index e3b9755a70d..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import PageClientTestBase, PagePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestPageTwoModelsAsPageItemOperations(PageClientTestBase): - @PagePreparer() - @recorded_by_proxy - def test_two_models_as_page_item_list_first_item(self, page_endpoint): - client = self.create_client(endpoint=page_endpoint) - response = client.two_models_as_page_item.list_first_item() - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @PagePreparer() - @recorded_by_proxy - def test_two_models_as_page_item_list_second_item(self, page_endpoint): - client = self.create_client(endpoint=page_endpoint) - response = client.two_models_as_page_item.list_second_item() - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py deleted file mode 100644 index 6ce3f441718..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import PagePreparer -from testpreparer_async import PageClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestPageTwoModelsAsPageItemOperationsAsync(PageClientTestBaseAsync): - @PagePreparer() - @recorded_by_proxy_async - async def test_two_models_as_page_item_list_first_item(self, page_endpoint): - client = self.create_async_client(endpoint=page_endpoint) - response = client.two_models_as_page_item.list_first_item() - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @PagePreparer() - @recorded_by_proxy_async - async def test_two_models_as_page_item_list_second_item(self, page_endpoint): - client = self.create_async_client(endpoint=page_endpoint) - response = client.two_models_as_page_item.list_second_item() - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py deleted file mode 100644 index 921bc83a56c..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.core.page import PageClient - - -class PageClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(PageClient) - return self.create_client_from_credential( - PageClient, - credential=credential, - endpoint=endpoint, - ) - - -PagePreparer = functools.partial(PowerShellPreparer, "page", page_endpoint="https://fake_page_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py deleted file mode 100644 index fa9ad45c103..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.core.page.aio import PageClient - - -class PageClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(PageClient, is_async=True) - return self.create_client_from_credential( - PageClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-page/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py deleted file mode 100644 index f7252bc7a1b..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - scalar_subscription_id = os.environ.get("SCALAR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - scalar_tenant_id = os.environ.get("SCALAR_TENANT_ID", "00000000-0000-0000-0000-000000000000") - scalar_client_id = os.environ.get("SCALAR_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - scalar_client_secret = os.environ.get("SCALAR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py deleted file mode 100644 index 234ea6351a1..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py +++ /dev/null @@ -1,67 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarAzureLocationScalarOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_azure_location_scalar_get(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.azure_location_scalar.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_azure_location_scalar_put(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.azure_location_scalar.put( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_azure_location_scalar_post(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.azure_location_scalar.post( - body={"location": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_azure_location_scalar_header(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.azure_location_scalar.header( - region="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_azure_location_scalar_query(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.azure_location_scalar.query( - region="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py deleted file mode 100644 index 38cff3d36b5..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py +++ /dev/null @@ -1,68 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarAzureLocationScalarOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_azure_location_scalar_get(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.azure_location_scalar.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_azure_location_scalar_put(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.azure_location_scalar.put( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_azure_location_scalar_post(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.azure_location_scalar.post( - body={"location": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_azure_location_scalar_header(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.azure_location_scalar.header( - region="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_azure_location_scalar_query(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.azure_location_scalar.query( - region="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py deleted file mode 100644 index cb3c43946e6..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.core.scalar import ScalarClient - - -class ScalarClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ScalarClient) - return self.create_client_from_credential( - ScalarClient, - credential=credential, - endpoint=endpoint, - ) - - -ScalarPreparer = functools.partial(PowerShellPreparer, "scalar", scalar_endpoint="https://fake_scalar_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py deleted file mode 100644 index 5d70e7f1fa2..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.core.scalar.aio import ScalarClient - - -class ScalarClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ScalarClient, is_async=True) - return self.create_client_from_credential( - ScalarClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-scalar/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py deleted file mode 100644 index fd65dd2df68..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - traits_subscription_id = os.environ.get("TRAITS_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - traits_tenant_id = os.environ.get("TRAITS_TENANT_ID", "00000000-0000-0000-0000-000000000000") - traits_client_id = os.environ.get("TRAITS_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - traits_client_secret = os.environ.get("TRAITS_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=traits_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=traits_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=traits_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=traits_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py deleted file mode 100644 index eca09365ca5..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import TraitsClientTestBase, TraitsPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTraits(TraitsClientTestBase): - @TraitsPreparer() - @recorded_by_proxy - def test_smoke_test(self, traits_endpoint): - client = self.create_client(endpoint=traits_endpoint) - response = client.smoke_test( - id=0, - foo="str", - ) - - # please add some check logic here by yourself - # ... - - @TraitsPreparer() - @recorded_by_proxy - def test_repeatable_action(self, traits_endpoint): - client = self.create_client(endpoint=traits_endpoint) - response = client.repeatable_action( - id=0, - body={"userActionValue": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py deleted file mode 100644 index 7188bd6c1e7..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py +++ /dev/null @@ -1,38 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import TraitsPreparer -from testpreparer_async import TraitsClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTraitsAsync(TraitsClientTestBaseAsync): - @TraitsPreparer() - @recorded_by_proxy_async - async def test_smoke_test(self, traits_endpoint): - client = self.create_async_client(endpoint=traits_endpoint) - response = await client.smoke_test( - id=0, - foo="str", - ) - - # please add some check logic here by yourself - # ... - - @TraitsPreparer() - @recorded_by_proxy_async - async def test_repeatable_action(self, traits_endpoint): - client = self.create_async_client(endpoint=traits_endpoint) - response = await client.repeatable_action( - id=0, - body={"userActionValue": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py deleted file mode 100644 index 661fdabcd75..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.core.traits import TraitsClient - - -class TraitsClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(TraitsClient) - return self.create_client_from_credential( - TraitsClient, - credential=credential, - endpoint=endpoint, - ) - - -TraitsPreparer = functools.partial(PowerShellPreparer, "traits", traits_endpoint="https://fake_traits_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py deleted file mode 100644 index ee53fde0dfb..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.core.traits.aio import TraitsClient - - -class TraitsClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(TraitsClient, is_async=True) - return self.create_client_from_credential( - TraitsClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-traits/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py deleted file mode 100644 index 8fc73742343..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from specs.azure.example.basic import AzureExampleClient - -""" -# PREREQUISITES - pip install specs-azure-example-basic -# USAGE - python basic.py -""" - - -def main(): - client = AzureExampleClient() - - response = client.basic_action( - body={ - "arrayProperty": ["item"], - "modelProperty": {"enumProperty": "EnumValue1", "float32Property": 1.5, "int32Property": 1}, - "recordProperty": {"record": "value"}, - "stringProperty": "text", - }, - query_param="query", - header_param="header", - ) - print(response) - - -# x-ms-original-file: 2022-12-01-preview/basic.json -if __name__ == "__main__": - main() diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py deleted file mode 100644 index dfc189b22aa..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - azureexample_subscription_id = os.environ.get( - "AZUREEXAMPLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - azureexample_tenant_id = os.environ.get("AZUREEXAMPLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - azureexample_client_id = os.environ.get("AZUREEXAMPLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - azureexample_client_secret = os.environ.get("AZUREEXAMPLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=azureexample_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=azureexample_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=azureexample_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=azureexample_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py deleted file mode 100644 index 9d66b6994ae..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AzureExampleClientTestBase, AzureExamplePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAzureExample(AzureExampleClientTestBase): - @AzureExamplePreparer() - @recorded_by_proxy - def test_basic_action(self, azureexample_endpoint): - client = self.create_client(endpoint=azureexample_endpoint) - response = client.basic_action( - body={ - "stringProperty": "str", - "arrayProperty": ["str"], - "modelProperty": {"enumProperty": "str", "float32Property": 0.0, "int32Property": 0}, - "recordProperty": {"str": "str"}, - }, - query_param="str", - header_param="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py deleted file mode 100644 index ec5969f68e3..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AzureExamplePreparer -from testpreparer_async import AzureExampleClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAzureExampleAsync(AzureExampleClientTestBaseAsync): - @AzureExamplePreparer() - @recorded_by_proxy_async - async def test_basic_action(self, azureexample_endpoint): - client = self.create_async_client(endpoint=azureexample_endpoint) - response = await client.basic_action( - body={ - "stringProperty": "str", - "arrayProperty": ["str"], - "modelProperty": {"enumProperty": "str", "float32Property": 0.0, "int32Property": 0}, - "recordProperty": {"str": "str"}, - }, - query_param="str", - header_param="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py deleted file mode 100644 index 6039e937d65..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specs.azure.example.basic import AzureExampleClient - - -class AzureExampleClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(AzureExampleClient) - return self.create_client_from_credential( - AzureExampleClient, - credential=credential, - endpoint=endpoint, - ) - - -AzureExamplePreparer = functools.partial( - PowerShellPreparer, "azureexample", azureexample_endpoint="https://fake_azureexample_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py deleted file mode 100644 index 64243921636..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specs.azure.example.basic.aio import AzureExampleClient - - -class AzureExampleClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(AzureExampleClient, is_async=True) - return self.create_client_from_credential( - AzureExampleClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-example-basic/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py deleted file mode 100644 index 49852579a99..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - commonproperties_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - commonproperties_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - commonproperties_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - commonproperties_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=commonproperties_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=commonproperties_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=commonproperties_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=commonproperties_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py deleted file mode 100644 index 92226f4ed8f..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py +++ /dev/null @@ -1,96 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.commonproperties import CommonPropertiesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCommonPropertiesManagedIdentityOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(CommonPropertiesClient) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_managed_identity_get(self, resource_group): - response = self.client.managed_identity.get( - resource_group_name=resource_group.name, - managed_identity_tracked_resource_name="str", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_managed_identity_create_with_system_assigned(self, resource_group): - response = self.client.managed_identity.create_with_system_assigned( - resource_group_name=resource_group.name, - managed_identity_tracked_resource_name="str", - resource={ - "location": "str", - "id": "str", - "identity": { - "type": "str", - "principalId": "str", - "tenantId": "str", - "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, - }, - "name": "str", - "properties": {"provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_managed_identity_update_with_user_assigned_and_system_assigned(self, resource_group): - response = self.client.managed_identity.update_with_user_assigned_and_system_assigned( - resource_group_name=resource_group.name, - managed_identity_tracked_resource_name="str", - properties={ - "location": "str", - "id": "str", - "identity": { - "type": "str", - "principalId": "str", - "tenantId": "str", - "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, - }, - "name": "str", - "properties": {"provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py deleted file mode 100644 index 5b0776d5e57..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py +++ /dev/null @@ -1,97 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.commonproperties.aio import CommonPropertiesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCommonPropertiesManagedIdentityOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(CommonPropertiesClient, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_managed_identity_get(self, resource_group): - response = await self.client.managed_identity.get( - resource_group_name=resource_group.name, - managed_identity_tracked_resource_name="str", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_managed_identity_create_with_system_assigned(self, resource_group): - response = await self.client.managed_identity.create_with_system_assigned( - resource_group_name=resource_group.name, - managed_identity_tracked_resource_name="str", - resource={ - "location": "str", - "id": "str", - "identity": { - "type": "str", - "principalId": "str", - "tenantId": "str", - "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, - }, - "name": "str", - "properties": {"provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_managed_identity_update_with_user_assigned_and_system_assigned(self, resource_group): - response = await self.client.managed_identity.update_with_user_assigned_and_system_assigned( - resource_group_name=resource_group.name, - managed_identity_tracked_resource_name="str", - properties={ - "location": "str", - "id": "str", - "identity": { - "type": "str", - "principalId": "str", - "tenantId": "str", - "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, - }, - "name": "str", - "properties": {"provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-resource-manager-resources/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py deleted file mode 100644 index fe8753d01f3..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - resources_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - resources_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - resources_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - resources_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=resources_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=resources_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=resources_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=resources_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py deleted file mode 100644 index b60fadc33d8..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py +++ /dev/null @@ -1,106 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.resources import ResourcesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResourcesNestedOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(ResourcesClient) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_nested_get(self, resource_group): - response = self.client.nested.get( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_nested_begin_create_or_replace(self, resource_group): - response = self.client.nested.begin_create_or_replace( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - resource={ - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "type": "str", - }, - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_nested_begin_update(self, resource_group): - response = self.client.nested.begin_update( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - properties={ - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "type": "str", - }, - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_nested_begin_delete(self, resource_group): - response = self.client.nested.begin_delete( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_nested_list_by_top_level_tracked_resource(self, resource_group): - response = self.client.nested.list_by_top_level_tracked_resource( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py deleted file mode 100644 index bb86a6ff911..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py +++ /dev/null @@ -1,113 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.resources.aio import ResourcesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResourcesNestedOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(ResourcesClient, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_nested_get(self, resource_group): - response = await self.client.nested.get( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_nested_begin_create_or_replace(self, resource_group): - response = await ( - await self.client.nested.begin_create_or_replace( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - resource={ - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "type": "str", - }, - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_nested_begin_update(self, resource_group): - response = await ( - await self.client.nested.begin_update( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - properties={ - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "type": "str", - }, - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_nested_begin_delete(self, resource_group): - response = await ( - await self.client.nested.begin_delete( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - nexted_proxy_resource_name="str", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_nested_list_by_top_level_tracked_resource(self, resource_group): - response = self.client.nested.list_by_top_level_tracked_resource( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py deleted file mode 100644 index bef8b7bbe35..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py +++ /dev/null @@ -1,91 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.resources import ResourcesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResourcesSingletonOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(ResourcesClient) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_singleton_get_by_resource_group(self, resource_group): - response = self.client.singleton.get_by_resource_group( - resource_group_name=resource_group.name, - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_singleton_begin_create_or_update(self, resource_group): - response = self.client.singleton.begin_create_or_update( - resource_group_name=resource_group.name, - resource={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_singleton_update(self, resource_group): - response = self.client.singleton.update( - resource_group_name=resource_group.name, - properties={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_singleton_list_by_resource_group(self, resource_group): - response = self.client.singleton.list_by_resource_group( - resource_group_name=resource_group.name, - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py deleted file mode 100644 index dc973f54204..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py +++ /dev/null @@ -1,94 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.resources.aio import ResourcesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResourcesSingletonOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(ResourcesClient, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_singleton_get_by_resource_group(self, resource_group): - response = await self.client.singleton.get_by_resource_group( - resource_group_name=resource_group.name, - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_singleton_begin_create_or_update(self, resource_group): - response = await ( - await self.client.singleton.begin_create_or_update( - resource_group_name=resource_group.name, - resource={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_singleton_update(self, resource_group): - response = await self.client.singleton.update( - resource_group_name=resource_group.name, - properties={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_singleton_list_by_resource_group(self, resource_group): - response = self.client.singleton.list_by_resource_group( - resource_group_name=resource_group.name, - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py deleted file mode 100644 index 8c685fd5000..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py +++ /dev/null @@ -1,125 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.resources import ResourcesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResourcesTopLevelOperations(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(ResourcesClient) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_top_level_get(self, resource_group): - response = self.client.top_level.get( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_top_level_begin_create_or_replace(self, resource_group): - response = self.client.top_level.begin_create_or_replace( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - resource={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_top_level_begin_update(self, resource_group): - response = self.client.top_level.begin_update( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - properties={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_top_level_begin_delete(self, resource_group): - response = self.client.top_level.begin_delete( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_top_level_list_by_resource_group(self, resource_group): - response = self.client.top_level.list_by_resource_group( - resource_group_name=resource_group.name, - ) - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_top_level_list_by_subscription(self, resource_group): - response = self.client.top_level.list_by_subscription() - result = [r for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy - def test_top_level_action_sync(self, resource_group): - response = self.client.top_level.action_sync( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - body={"message": "str", "urgent": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py deleted file mode 100644 index c3cec3db79a..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py +++ /dev/null @@ -1,132 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from azure.resourcemanager.resources.aio import ResourcesClient - -from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer -from devtools_testutils.aio import recorded_by_proxy_async - -AZURE_LOCATION = "eastus" - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResourcesTopLevelOperationsAsync(AzureMgmtRecordedTestCase): - def setup_method(self, method): - self.client = self.create_mgmt_client(ResourcesClient, is_async=True) - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_top_level_get(self, resource_group): - response = await self.client.top_level.get( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - ) - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_top_level_begin_create_or_replace(self, resource_group): - response = await ( - await self.client.top_level.begin_create_or_replace( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - resource={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_top_level_begin_update(self, resource_group): - response = await ( - await self.client.top_level.begin_update( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - properties={ - "location": "str", - "id": "str", - "name": "str", - "properties": {"description": "str", "provisioningState": "str"}, - "systemData": { - "createdAt": "2020-02-20 00:00:00", - "createdBy": "str", - "createdByType": "str", - "lastModifiedAt": "2020-02-20 00:00:00", - "lastModifiedBy": "str", - "lastModifiedByType": "str", - }, - "tags": {"str": "str"}, - "type": "str", - }, - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_top_level_begin_delete(self, resource_group): - response = await ( - await self.client.top_level.begin_delete( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_top_level_list_by_resource_group(self, resource_group): - response = self.client.top_level.list_by_resource_group( - resource_group_name=resource_group.name, - ) - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_top_level_list_by_subscription(self, resource_group): - response = self.client.top_level.list_by_subscription() - result = [r async for r in response] - # please add some check logic here by yourself - # ... - - @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) - @recorded_by_proxy_async - async def test_top_level_action_sync(self, resource_group): - response = await self.client.top_level.action_sync( - resource_group_name=resource_group.name, - top_level_tracked_resource_name="str", - body={"message": "str", "urgent": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py deleted file mode 100644 index 81b0a14c303..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - xmsrequestid_subscription_id = os.environ.get( - "XMSREQUESTID_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - xmsrequestid_tenant_id = os.environ.get("XMSREQUESTID_TENANT_ID", "00000000-0000-0000-0000-000000000000") - xmsrequestid_client_id = os.environ.get("XMSREQUESTID_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - xmsrequestid_client_secret = os.environ.get("XMSREQUESTID_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xmsrequestid_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xmsrequestid_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xmsrequestid_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xmsrequestid_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py deleted file mode 100644 index adb49afe641..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmsClientRequestIdClientTestBase, XmsRequestIdPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmsRequestId(XmsClientRequestIdClientTestBase): - @XmsRequestIdPreparer() - @recorded_by_proxy - def test_get(self, xmsrequestid_endpoint): - client = self.create_client(endpoint=xmsrequestid_endpoint) - response = client.get() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py deleted file mode 100644 index 8830328ad72..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmsRequestIdPreparer -from testpreparer_async import XmsClientRequestIdClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmsRequestIdAsync(XmsClientRequestIdClientTestBaseAsync): - @XmsRequestIdPreparer() - @recorded_by_proxy_async - async def test_get(self, xmsrequestid_endpoint): - client = self.create_async_client(endpoint=xmsrequestid_endpoint) - response = await client.get() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py deleted file mode 100644 index 975eab0459d..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from azure.specialheaders.xmsclientrequestid import XmsClientRequestIdClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class XmsClientRequestIdClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(XmsClientRequestIdClient) - return self.create_client_from_credential( - XmsClientRequestIdClient, - credential=credential, - endpoint=endpoint, - ) - - -XmsRequestIdPreparer = functools.partial( - PowerShellPreparer, "xmsrequestid", xmsrequestid_endpoint="https://fake_xmsrequestid_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py deleted file mode 100644 index df7d540a17b..00000000000 --- a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from azure.specialheaders.xmsclientrequestid.aio import XmsClientRequestIdClient -from devtools_testutils import AzureRecordedTestCase - - -class XmsClientRequestIdClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(XmsClientRequestIdClient, is_async=True) - return self.create_client_from_credential( - XmsClientRequestIdClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azurecore-lro-rpc/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py deleted file mode 100644 index 066375380f8..00000000000 --- a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - rpc_subscription_id = os.environ.get("RPC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - rpc_tenant_id = os.environ.get("RPC_TENANT_ID", "00000000-0000-0000-0000-000000000000") - rpc_client_id = os.environ.get("RPC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - rpc_client_secret = os.environ.get("RPC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=rpc_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=rpc_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=rpc_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=rpc_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py deleted file mode 100644 index 1f3b18e28ab..00000000000 --- a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RpcClientTestBase, RpcPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRpc(RpcClientTestBase): - @RpcPreparer() - @recorded_by_proxy - def test_begin_long_running_rpc(self, rpc_endpoint): - client = self.create_client(endpoint=rpc_endpoint) - response = client.begin_long_running_rpc( - body={"prompt": "str"}, - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py deleted file mode 100644 index 9d55e7fa603..00000000000 --- a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py +++ /dev/null @@ -1,27 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RpcPreparer -from testpreparer_async import RpcClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRpcAsync(RpcClientTestBaseAsync): - @RpcPreparer() - @recorded_by_proxy_async - async def test_begin_long_running_rpc(self, rpc_endpoint): - client = self.create_async_client(endpoint=rpc_endpoint) - response = await ( - await client.begin_long_running_rpc( - body={"prompt": "str"}, - ) - ).result() # call '.result()' to poll until service return final result - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py deleted file mode 100644 index b3da2928846..00000000000 --- a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from azurecore.lro.rpc import RpcClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class RpcClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(RpcClient) - return self.create_client_from_credential( - RpcClient, - credential=credential, - endpoint=endpoint, - ) - - -RpcPreparer = functools.partial(PowerShellPreparer, "rpc", rpc_endpoint="https://fake_rpc_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py deleted file mode 100644 index c9b185a7228..00000000000 --- a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from azurecore.lro.rpc.aio import RpcClient -from devtools_testutils import AzureRecordedTestCase - - -class RpcClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(RpcClient, is_async=True) - return self.create_client_from_credential( - RpcClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/client-naming/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-naming/client/naming/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-naming/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-naming/client/naming/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py deleted file mode 100644 index e1548f36b47..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - naming_subscription_id = os.environ.get("NAMING_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - naming_tenant_id = os.environ.get("NAMING_TENANT_ID", "00000000-0000-0000-0000-000000000000") - naming_client_id = os.environ.get("NAMING_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - naming_client_secret = os.environ.get("NAMING_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=naming_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=naming_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=naming_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=naming_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py deleted file mode 100644 index d61a3d6139f..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py +++ /dev/null @@ -1,86 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NamingClientTestBase, NamingPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNaming(NamingClientTestBase): - @NamingPreparer() - @recorded_by_proxy - def test_client_name(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.client_name() - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_parameter(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.parameter( - client_name="str", - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_client(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.client( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_language(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.language( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_compatible_with_encoded_name(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.compatible_with_encoded_name( - body={"wireName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_request(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.request( - client_name="str", - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_response(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.response() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py deleted file mode 100644 index 8e359563300..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py +++ /dev/null @@ -1,87 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NamingPreparer -from testpreparer_async import NamingClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNamingAsync(NamingClientTestBaseAsync): - @NamingPreparer() - @recorded_by_proxy_async - async def test_client_name(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.client_name() - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_parameter(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.parameter( - client_name="str", - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_client(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.client( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_language(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.language( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_compatible_with_encoded_name(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.compatible_with_encoded_name( - body={"wireName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_request(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.request( - client_name="str", - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_response(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.response() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py deleted file mode 100644 index 854de9bc234..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NamingClientTestBase, NamingPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNamingClientModelOperations(NamingClientTestBase): - @NamingPreparer() - @recorded_by_proxy - def test_client_model_client(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.client_model.client( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_client_model_language(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.client_model.language( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py deleted file mode 100644 index ef1c0109021..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NamingPreparer -from testpreparer_async import NamingClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNamingClientModelOperationsAsync(NamingClientTestBaseAsync): - @NamingPreparer() - @recorded_by_proxy_async - async def test_client_model_client(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.client_model.client( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_client_model_language(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.client_model.language( - body={"defaultName": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py deleted file mode 100644 index 45e18f9ddc8..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NamingClientTestBase, NamingPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNamingUnionEnumOperations(NamingClientTestBase): - @NamingPreparer() - @recorded_by_proxy - def test_union_enum_union_enum_name(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.union_enum.union_enum_name( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy - def test_union_enum_union_enum_member_name(self, naming_endpoint): - client = self.create_client(endpoint=naming_endpoint) - response = client.union_enum.union_enum_member_name( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py deleted file mode 100644 index 141e125b601..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py +++ /dev/null @@ -1,38 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NamingPreparer -from testpreparer_async import NamingClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNamingUnionEnumOperationsAsync(NamingClientTestBaseAsync): - @NamingPreparer() - @recorded_by_proxy_async - async def test_union_enum_union_enum_name(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.union_enum.union_enum_name( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @NamingPreparer() - @recorded_by_proxy_async - async def test_union_enum_union_enum_member_name(self, naming_endpoint): - client = self.create_async_client(endpoint=naming_endpoint) - response = await client.union_enum.union_enum_member_name( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py deleted file mode 100644 index 0b6e8178c61..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.naming import NamingClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class NamingClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(NamingClient) - return self.create_client_from_credential( - NamingClient, - credential=credential, - endpoint=endpoint, - ) - - -NamingPreparer = functools.partial(PowerShellPreparer, "naming", naming_endpoint="https://fake_naming_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py deleted file mode 100644 index f306afdbed6..00000000000 --- a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.naming.aio import NamingClient -from devtools_testutils import AzureRecordedTestCase - - -class NamingClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(NamingClient, is_async=True) - return self.create_client_from_credential( - NamingClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-default/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py deleted file mode 100644 index 91256e0344e..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - service_subscription_id = os.environ.get("SERVICE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - service_tenant_id = os.environ.get("SERVICE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - service_client_id = os.environ.get("SERVICE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - service_client_secret = os.environ.get("SERVICE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=service_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=service_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=service_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=service_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py deleted file mode 100644 index 1e7ce2517d5..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ServiceClientTestBase, ServicePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestService(ServiceClientTestBase): - @ServicePreparer() - @recorded_by_proxy - def test_one(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.one() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy - def test_two(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.two() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py deleted file mode 100644 index 2947d4222c5..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ServicePreparer -from testpreparer_async import ServiceClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceAsync(ServiceClientTestBaseAsync): - @ServicePreparer() - @recorded_by_proxy_async - async def test_one(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.one() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy_async - async def test_two(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.two() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py deleted file mode 100644 index ca4600e7c2c..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ServiceClientTestBase, ServicePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceBarOperations(ServiceClientTestBase): - @ServicePreparer() - @recorded_by_proxy - def test_bar_five(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.bar.five() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy - def test_bar_six(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.bar.six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py deleted file mode 100644 index 563b02c2ac8..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ServicePreparer -from testpreparer_async import ServiceClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceBarOperationsAsync(ServiceClientTestBaseAsync): - @ServicePreparer() - @recorded_by_proxy_async - async def test_bar_five(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.bar.five() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy_async - async def test_bar_six(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.bar.six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py deleted file mode 100644 index 25e68492cd5..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ServiceClientTestBase, ServicePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceBazOperations(ServiceClientTestBase): - @ServicePreparer() - @recorded_by_proxy - def test_baz_foo_seven(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.baz.foo.seven() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py deleted file mode 100644 index 6fc9529df7e..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ServicePreparer -from testpreparer_async import ServiceClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceBazOperationsAsync(ServiceClientTestBaseAsync): - @ServicePreparer() - @recorded_by_proxy_async - async def test_baz_foo_seven(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.baz.foo.seven() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py deleted file mode 100644 index 5a62fa3eed0..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ServiceClientTestBase, ServicePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceFooOperations(ServiceClientTestBase): - @ServicePreparer() - @recorded_by_proxy - def test_foo_three(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.foo.three() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy - def test_foo_four(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.foo.four() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py deleted file mode 100644 index 547fc58c746..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ServicePreparer -from testpreparer_async import ServiceClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceFooOperationsAsync(ServiceClientTestBaseAsync): - @ServicePreparer() - @recorded_by_proxy_async - async def test_foo_three(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.foo.three() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy_async - async def test_foo_four(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.foo.four() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py deleted file mode 100644 index 19079ec2a92..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ServiceClientTestBase, ServicePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceQuxOperations(ServiceClientTestBase): - @ServicePreparer() - @recorded_by_proxy - def test_qux_eight(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.qux.eight() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy - def test_qux_bar_nine(self, service_endpoint): - client = self.create_client(endpoint=service_endpoint) - response = client.qux.bar.nine() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py deleted file mode 100644 index 4eb28c55ad0..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ServicePreparer -from testpreparer_async import ServiceClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestServiceQuxOperationsAsync(ServiceClientTestBaseAsync): - @ServicePreparer() - @recorded_by_proxy_async - async def test_qux_eight(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.qux.eight() - - # please add some check logic here by yourself - # ... - - @ServicePreparer() - @recorded_by_proxy_async - async def test_qux_bar_nine(self, service_endpoint): - client = self.create_async_client(endpoint=service_endpoint) - response = await client.qux.bar.nine() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py deleted file mode 100644 index 957dd158a79..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.service import ServiceClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class ServiceClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ServiceClient) - return self.create_client_from_credential( - ServiceClient, - credential=credential, - endpoint=endpoint, - ) - - -ServicePreparer = functools.partial(PowerShellPreparer, "service", service_endpoint="https://fake_service_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py deleted file mode 100644 index 5ff957026f2..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.service.aio import ServiceClient -from devtools_testutils import AzureRecordedTestCase - - -class ServiceClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ServiceClient, is_async=True) - return self.create_client_from_credential( - ServiceClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-multiclient/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py deleted file mode 100644 index 89cc9a005d6..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py +++ /dev/null @@ -1,44 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - a_subscription_id = os.environ.get("A_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - a_tenant_id = os.environ.get("A_TENANT_ID", "00000000-0000-0000-0000-000000000000") - a_client_id = os.environ.get("A_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - a_client_secret = os.environ.get("A_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=a_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=a_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=a_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=a_client_secret, value="00000000-0000-0000-0000-000000000000") - - b_subscription_id = os.environ.get("B_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - b_tenant_id = os.environ.get("B_TENANT_ID", "00000000-0000-0000-0000-000000000000") - b_client_id = os.environ.get("B_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - b_client_secret = os.environ.get("B_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=b_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=b_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=b_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=b_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py deleted file mode 100644 index 758a7aa3087..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py +++ /dev/null @@ -1,40 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import APreparer, ClientAClientTestBase - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestA(ClientAClientTestBase): - @APreparer() - @recorded_by_proxy - def test_renamed_one(self, a_endpoint): - client = self.create_client(endpoint=a_endpoint) - response = client.renamed_one() - - # please add some check logic here by yourself - # ... - - @APreparer() - @recorded_by_proxy - def test_renamed_three(self, a_endpoint): - client = self.create_client(endpoint=a_endpoint) - response = client.renamed_three() - - # please add some check logic here by yourself - # ... - - @APreparer() - @recorded_by_proxy - def test_renamed_five(self, a_endpoint): - client = self.create_client(endpoint=a_endpoint) - response = client.renamed_five() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py deleted file mode 100644 index 350a49c78cd..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py +++ /dev/null @@ -1,41 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import APreparer -from testpreparer_async import ClientAClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAAsync(ClientAClientTestBaseAsync): - @APreparer() - @recorded_by_proxy_async - async def test_renamed_one(self, a_endpoint): - client = self.create_async_client(endpoint=a_endpoint) - response = await client.renamed_one() - - # please add some check logic here by yourself - # ... - - @APreparer() - @recorded_by_proxy_async - async def test_renamed_three(self, a_endpoint): - client = self.create_async_client(endpoint=a_endpoint) - response = await client.renamed_three() - - # please add some check logic here by yourself - # ... - - @APreparer() - @recorded_by_proxy_async - async def test_renamed_five(self, a_endpoint): - client = self.create_async_client(endpoint=a_endpoint) - response = await client.renamed_five() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py deleted file mode 100644 index 0c35b7bfba4..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py +++ /dev/null @@ -1,40 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BPreparer, ClientBClientTestBase - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestB(ClientBClientTestBase): - @BPreparer() - @recorded_by_proxy - def test_renamed_two(self, b_endpoint): - client = self.create_client(endpoint=b_endpoint) - response = client.renamed_two() - - # please add some check logic here by yourself - # ... - - @BPreparer() - @recorded_by_proxy - def test_renamed_four(self, b_endpoint): - client = self.create_client(endpoint=b_endpoint) - response = client.renamed_four() - - # please add some check logic here by yourself - # ... - - @BPreparer() - @recorded_by_proxy - def test_renamed_six(self, b_endpoint): - client = self.create_client(endpoint=b_endpoint) - response = client.renamed_six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py deleted file mode 100644 index 93e20b2c5f5..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py +++ /dev/null @@ -1,41 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BPreparer -from testpreparer_async import ClientBClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBAsync(ClientBClientTestBaseAsync): - @BPreparer() - @recorded_by_proxy_async - async def test_renamed_two(self, b_endpoint): - client = self.create_async_client(endpoint=b_endpoint) - response = await client.renamed_two() - - # please add some check logic here by yourself - # ... - - @BPreparer() - @recorded_by_proxy_async - async def test_renamed_four(self, b_endpoint): - client = self.create_async_client(endpoint=b_endpoint) - response = await client.renamed_four() - - # please add some check logic here by yourself - # ... - - @BPreparer() - @recorded_by_proxy_async - async def test_renamed_six(self, b_endpoint): - client = self.create_async_client(endpoint=b_endpoint) - response = await client.renamed_six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py deleted file mode 100644 index de2a68557e5..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py +++ /dev/null @@ -1,38 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.multiclient import ClientAClient, ClientBClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class ClientAClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ClientAClient) - return self.create_client_from_credential( - ClientAClient, - credential=credential, - endpoint=endpoint, - ) - - -APreparer = functools.partial(PowerShellPreparer, "a", a_endpoint="https://fake_a_endpoint.com") - - -class ClientBClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ClientBClient) - return self.create_client_from_credential( - ClientBClient, - credential=credential, - endpoint=endpoint, - ) - - -BPreparer = functools.partial(PowerShellPreparer, "b", b_endpoint="https://fake_b_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py deleted file mode 100644 index c8d0f416c46..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.multiclient.aio import ClientAClient, ClientBClient -from devtools_testutils import AzureRecordedTestCase - - -class ClientAClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ClientAClient, is_async=True) - return self.create_client_from_credential( - ClientAClient, - credential=credential, - endpoint=endpoint, - ) - - -class ClientBClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ClientBClient, is_async=True) - return self.create_client_from_credential( - ClientBClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-renamedoperation/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py deleted file mode 100644 index 351cda6c0b8..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - renamedoperation_subscription_id = os.environ.get( - "RENAMEDOPERATION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - renamedoperation_tenant_id = os.environ.get("RENAMEDOPERATION_TENANT_ID", "00000000-0000-0000-0000-000000000000") - renamedoperation_client_id = os.environ.get("RENAMEDOPERATION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - renamedoperation_client_secret = os.environ.get( - "RENAMEDOPERATION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=renamedoperation_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=renamedoperation_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=renamedoperation_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=renamedoperation_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py deleted file mode 100644 index e62c8f801d3..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py +++ /dev/null @@ -1,40 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RenamedOperationClientTestBase, RenamedOperationPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedOperation(RenamedOperationClientTestBase): - @RenamedOperationPreparer() - @recorded_by_proxy - def test_renamed_one(self, renamedoperation_endpoint): - client = self.create_client(endpoint=renamedoperation_endpoint) - response = client.renamed_one() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy - def test_renamed_three(self, renamedoperation_endpoint): - client = self.create_client(endpoint=renamedoperation_endpoint) - response = client.renamed_three() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy - def test_renamed_five(self, renamedoperation_endpoint): - client = self.create_client(endpoint=renamedoperation_endpoint) - response = client.renamed_five() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py deleted file mode 100644 index ce2bd7c8e4c..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py +++ /dev/null @@ -1,41 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RenamedOperationPreparer -from testpreparer_async import RenamedOperationClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedOperationAsync(RenamedOperationClientTestBaseAsync): - @RenamedOperationPreparer() - @recorded_by_proxy_async - async def test_renamed_one(self, renamedoperation_endpoint): - client = self.create_async_client(endpoint=renamedoperation_endpoint) - response = await client.renamed_one() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy_async - async def test_renamed_three(self, renamedoperation_endpoint): - client = self.create_async_client(endpoint=renamedoperation_endpoint) - response = await client.renamed_three() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy_async - async def test_renamed_five(self, renamedoperation_endpoint): - client = self.create_async_client(endpoint=renamedoperation_endpoint) - response = await client.renamed_five() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py deleted file mode 100644 index bba0c3a932d..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py +++ /dev/null @@ -1,40 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RenamedOperationClientTestBase, RenamedOperationPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedOperationGroupOperations(RenamedOperationClientTestBase): - @RenamedOperationPreparer() - @recorded_by_proxy - def test_group_renamed_two(self, renamedoperation_endpoint): - client = self.create_client(endpoint=renamedoperation_endpoint) - response = client.group.renamed_two() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy - def test_group_renamed_four(self, renamedoperation_endpoint): - client = self.create_client(endpoint=renamedoperation_endpoint) - response = client.group.renamed_four() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy - def test_group_renamed_six(self, renamedoperation_endpoint): - client = self.create_client(endpoint=renamedoperation_endpoint) - response = client.group.renamed_six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py deleted file mode 100644 index 5ce2eb78017..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py +++ /dev/null @@ -1,41 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RenamedOperationPreparer -from testpreparer_async import RenamedOperationClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedOperationGroupOperationsAsync(RenamedOperationClientTestBaseAsync): - @RenamedOperationPreparer() - @recorded_by_proxy_async - async def test_group_renamed_two(self, renamedoperation_endpoint): - client = self.create_async_client(endpoint=renamedoperation_endpoint) - response = await client.group.renamed_two() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy_async - async def test_group_renamed_four(self, renamedoperation_endpoint): - client = self.create_async_client(endpoint=renamedoperation_endpoint) - response = await client.group.renamed_four() - - # please add some check logic here by yourself - # ... - - @RenamedOperationPreparer() - @recorded_by_proxy_async - async def test_group_renamed_six(self, renamedoperation_endpoint): - client = self.create_async_client(endpoint=renamedoperation_endpoint) - response = await client.group.renamed_six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py deleted file mode 100644 index 5a277b43b52..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.renamedoperation import RenamedOperationClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class RenamedOperationClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(RenamedOperationClient) - return self.create_client_from_credential( - RenamedOperationClient, - credential=credential, - endpoint=endpoint, - ) - - -RenamedOperationPreparer = functools.partial( - PowerShellPreparer, "renamedoperation", renamedoperation_endpoint="https://fake_renamedoperation_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py deleted file mode 100644 index b44022452e6..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.renamedoperation.aio import RenamedOperationClient -from devtools_testutils import AzureRecordedTestCase - - -class RenamedOperationClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(RenamedOperationClient, is_async=True) - return self.create_client_from_credential( - RenamedOperationClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py deleted file mode 100644 index c7671c2b0f6..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - twooperationgroup_subscription_id = os.environ.get( - "TWOOPERATIONGROUP_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - twooperationgroup_tenant_id = os.environ.get("TWOOPERATIONGROUP_TENANT_ID", "00000000-0000-0000-0000-000000000000") - twooperationgroup_client_id = os.environ.get("TWOOPERATIONGROUP_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - twooperationgroup_client_secret = os.environ.get( - "TWOOPERATIONGROUP_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=twooperationgroup_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=twooperationgroup_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=twooperationgroup_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=twooperationgroup_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py deleted file mode 100644 index c3c32ec56f0..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py +++ /dev/null @@ -1,40 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import TwoOperationGroupClientTestBase, TwoOperationGroupPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTwoOperationGroupGroup1Operations(TwoOperationGroupClientTestBase): - @TwoOperationGroupPreparer() - @recorded_by_proxy - def test_group1_one(self, twooperationgroup_endpoint): - client = self.create_client(endpoint=twooperationgroup_endpoint) - response = client.group1.one() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy - def test_group1_three(self, twooperationgroup_endpoint): - client = self.create_client(endpoint=twooperationgroup_endpoint) - response = client.group1.three() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy - def test_group1_four(self, twooperationgroup_endpoint): - client = self.create_client(endpoint=twooperationgroup_endpoint) - response = client.group1.four() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py deleted file mode 100644 index 539bfde4da6..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py +++ /dev/null @@ -1,41 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import TwoOperationGroupPreparer -from testpreparer_async import TwoOperationGroupClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTwoOperationGroupGroup1OperationsAsync(TwoOperationGroupClientTestBaseAsync): - @TwoOperationGroupPreparer() - @recorded_by_proxy_async - async def test_group1_one(self, twooperationgroup_endpoint): - client = self.create_async_client(endpoint=twooperationgroup_endpoint) - response = await client.group1.one() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy_async - async def test_group1_three(self, twooperationgroup_endpoint): - client = self.create_async_client(endpoint=twooperationgroup_endpoint) - response = await client.group1.three() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy_async - async def test_group1_four(self, twooperationgroup_endpoint): - client = self.create_async_client(endpoint=twooperationgroup_endpoint) - response = await client.group1.four() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py deleted file mode 100644 index 0e4194269bb..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py +++ /dev/null @@ -1,40 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import TwoOperationGroupClientTestBase, TwoOperationGroupPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTwoOperationGroupGroup2Operations(TwoOperationGroupClientTestBase): - @TwoOperationGroupPreparer() - @recorded_by_proxy - def test_group2_two(self, twooperationgroup_endpoint): - client = self.create_client(endpoint=twooperationgroup_endpoint) - response = client.group2.two() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy - def test_group2_five(self, twooperationgroup_endpoint): - client = self.create_client(endpoint=twooperationgroup_endpoint) - response = client.group2.five() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy - def test_group2_six(self, twooperationgroup_endpoint): - client = self.create_client(endpoint=twooperationgroup_endpoint) - response = client.group2.six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py deleted file mode 100644 index 2097a90a553..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py +++ /dev/null @@ -1,41 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import TwoOperationGroupPreparer -from testpreparer_async import TwoOperationGroupClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTwoOperationGroupGroup2OperationsAsync(TwoOperationGroupClientTestBaseAsync): - @TwoOperationGroupPreparer() - @recorded_by_proxy_async - async def test_group2_two(self, twooperationgroup_endpoint): - client = self.create_async_client(endpoint=twooperationgroup_endpoint) - response = await client.group2.two() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy_async - async def test_group2_five(self, twooperationgroup_endpoint): - client = self.create_async_client(endpoint=twooperationgroup_endpoint) - response = await client.group2.five() - - # please add some check logic here by yourself - # ... - - @TwoOperationGroupPreparer() - @recorded_by_proxy_async - async def test_group2_six(self, twooperationgroup_endpoint): - client = self.create_async_client(endpoint=twooperationgroup_endpoint) - response = await client.group2.six() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py deleted file mode 100644 index a015a7808f7..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.twooperationgroup import TwoOperationGroupClient -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools - - -class TwoOperationGroupClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(TwoOperationGroupClient) - return self.create_client_from_credential( - TwoOperationGroupClient, - credential=credential, - endpoint=endpoint, - ) - - -TwoOperationGroupPreparer = functools.partial( - PowerShellPreparer, "twooperationgroup", twooperationgroup_endpoint="https://fake_twooperationgroup_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py deleted file mode 100644 index 4f86f927c06..00000000000 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from client.structure.twooperationgroup.aio import TwoOperationGroupClient -from devtools_testutils import AzureRecordedTestCase - - -class TwoOperationGroupClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(TwoOperationGroupClient, is_async=True) - return self.create_client_from_credential( - TwoOperationGroupClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-bytes/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py deleted file mode 100644 index 9238da04a06..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - bytes_subscription_id = os.environ.get("BYTES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - bytes_tenant_id = os.environ.get("BYTES_TENANT_ID", "00000000-0000-0000-0000-000000000000") - bytes_client_id = os.environ.get("BYTES_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - bytes_client_secret = os.environ.get("BYTES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=bytes_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=bytes_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=bytes_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=bytes_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py deleted file mode 100644 index 616fd7ecfa3..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py +++ /dev/null @@ -1,57 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BytesClientTestBase, BytesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesHeaderOperations(BytesClientTestBase): - @BytesPreparer() - @recorded_by_proxy - def test_header_default(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.header.default( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_header_base64(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.header.base64( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_header_base64_url(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.header.base64_url( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_header_base64_url_array(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.header.base64_url_array( - value=[bytes("bytes", encoding="utf-8")], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py deleted file mode 100644 index 8a957f4dbfc..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py +++ /dev/null @@ -1,58 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BytesPreparer -from testpreparer_async import BytesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesHeaderOperationsAsync(BytesClientTestBaseAsync): - @BytesPreparer() - @recorded_by_proxy_async - async def test_header_default(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.header.default( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_header_base64(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.header.base64( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_header_base64_url(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.header.base64_url( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_header_base64_url_array(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.header.base64_url_array( - value=[bytes("bytes", encoding="utf-8")], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py deleted file mode 100644 index a797ba0131d..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py +++ /dev/null @@ -1,57 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BytesClientTestBase, BytesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesPropertyOperations(BytesClientTestBase): - @BytesPreparer() - @recorded_by_proxy - def test_property_default(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.property.default( - body={"value": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_property_base64(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.property.base64( - body={"value": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_property_base64_url(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.property.base64_url( - body={"value": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_property_base64_url_array(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.property.base64_url_array( - body={"value": [bytes("bytes", encoding="utf-8")]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py deleted file mode 100644 index fd1c0f97a12..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py +++ /dev/null @@ -1,58 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BytesPreparer -from testpreparer_async import BytesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesPropertyOperationsAsync(BytesClientTestBaseAsync): - @BytesPreparer() - @recorded_by_proxy_async - async def test_property_default(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.property.default( - body={"value": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_property_base64(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.property.base64( - body={"value": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_property_base64_url(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.property.base64_url( - body={"value": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_property_base64_url_array(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.property.base64_url_array( - body={"value": [bytes("bytes", encoding="utf-8")]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py deleted file mode 100644 index 49f4b3ff3db..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py +++ /dev/null @@ -1,57 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BytesClientTestBase, BytesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesQueryOperations(BytesClientTestBase): - @BytesPreparer() - @recorded_by_proxy - def test_query_default(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.query.default( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_query_base64(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.query.base64( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_query_base64_url(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.query.base64_url( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_query_base64_url_array(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.query.base64_url_array( - value=[bytes("bytes", encoding="utf-8")], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py deleted file mode 100644 index b108afe5b12..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py +++ /dev/null @@ -1,58 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BytesPreparer -from testpreparer_async import BytesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesQueryOperationsAsync(BytesClientTestBaseAsync): - @BytesPreparer() - @recorded_by_proxy_async - async def test_query_default(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.query.default( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_query_base64(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.query.base64( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_query_base64_url(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.query.base64_url( - value=bytes("bytes", encoding="utf-8"), - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_query_base64_url_array(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.query.base64_url_array( - value=[bytes("bytes", encoding="utf-8")], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py deleted file mode 100644 index 1d083b7c7b3..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py +++ /dev/null @@ -1,73 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BytesClientTestBase, BytesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesRequestBodyOperations(BytesClientTestBase): - @BytesPreparer() - @recorded_by_proxy - def test_request_body_default(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.request_body.default( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_request_body_octet_stream(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.request_body.octet_stream( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_request_body_custom_content_type(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.request_body.custom_content_type( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_request_body_base64(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.request_body.base64( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_request_body_base64_url(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.request_body.base64_url( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py deleted file mode 100644 index a02a0e071e2..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py +++ /dev/null @@ -1,74 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BytesPreparer -from testpreparer_async import BytesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesRequestBodyOperationsAsync(BytesClientTestBaseAsync): - @BytesPreparer() - @recorded_by_proxy_async - async def test_request_body_default(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.request_body.default( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_request_body_octet_stream(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.request_body.octet_stream( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_request_body_custom_content_type(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.request_body.custom_content_type( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_request_body_base64(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.request_body.base64( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_request_body_base64_url(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.request_body.base64_url( - value=bytes("bytes", encoding="utf-8"), - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py deleted file mode 100644 index 1bb3aafaef9..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py +++ /dev/null @@ -1,58 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BytesClientTestBase, BytesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesResponseBodyOperations(BytesClientTestBase): - @BytesPreparer() - @recorded_by_proxy - def test_response_body_default(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.response_body.default() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_response_body_octet_stream(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.response_body.octet_stream() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_response_body_custom_content_type(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.response_body.custom_content_type() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_response_body_base64(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.response_body.base64() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy - def test_response_body_base64_url(self, bytes_endpoint): - client = self.create_client(endpoint=bytes_endpoint) - response = client.response_body.base64_url() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py deleted file mode 100644 index 8e702b41e39..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py +++ /dev/null @@ -1,59 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BytesPreparer -from testpreparer_async import BytesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBytesResponseBodyOperationsAsync(BytesClientTestBaseAsync): - @BytesPreparer() - @recorded_by_proxy_async - async def test_response_body_default(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.response_body.default() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_response_body_octet_stream(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.response_body.octet_stream() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_response_body_custom_content_type(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.response_body.custom_content_type() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_response_body_base64(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.response_body.base64() - - # please add some check logic here by yourself - # ... - - @BytesPreparer() - @recorded_by_proxy_async - async def test_response_body_base64_url(self, bytes_endpoint): - client = self.create_async_client(endpoint=bytes_endpoint) - response = await client.response_body.base64_url() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py deleted file mode 100644 index 0885f387943..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -from encode.bytes import BytesClient -import functools - - -class BytesClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(BytesClient) - return self.create_client_from_credential( - BytesClient, - credential=credential, - endpoint=endpoint, - ) - - -BytesPreparer = functools.partial(PowerShellPreparer, "bytes", bytes_endpoint="https://fake_bytes_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py deleted file mode 100644 index 8d0f600ec98..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from encode.bytes.aio import BytesClient - - -class BytesClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(BytesClient, is_async=True) - return self.create_client_from_credential( - BytesClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-datetime/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py deleted file mode 100644 index f3723e069e8..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - datetime_subscription_id = os.environ.get("DATETIME_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - datetime_tenant_id = os.environ.get("DATETIME_TENANT_ID", "00000000-0000-0000-0000-000000000000") - datetime_client_id = os.environ.get("DATETIME_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - datetime_client_secret = os.environ.get("DATETIME_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=datetime_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=datetime_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=datetime_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=datetime_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py deleted file mode 100644 index db1045ed06e..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py +++ /dev/null @@ -1,68 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DatetimeClientTestBase, DatetimePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimeHeaderOperations(DatetimeClientTestBase): - @DatetimePreparer() - @recorded_by_proxy - def test_header_default(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.header.default( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_header_rfc3339(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.header.rfc3339( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_header_rfc7231(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.header.rfc7231( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_header_unix_timestamp(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.header.unix_timestamp( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_header_unix_timestamp_array(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.header.unix_timestamp_array( - value=["2020-02-20 00:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py deleted file mode 100644 index bdf46c102fb..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py +++ /dev/null @@ -1,69 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DatetimePreparer -from testpreparer_async import DatetimeClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimeHeaderOperationsAsync(DatetimeClientTestBaseAsync): - @DatetimePreparer() - @recorded_by_proxy_async - async def test_header_default(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.header.default( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_header_rfc3339(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.header.rfc3339( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_header_rfc7231(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.header.rfc7231( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_header_unix_timestamp(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.header.unix_timestamp( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_header_unix_timestamp_array(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.header.unix_timestamp_array( - value=["2020-02-20 00:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py deleted file mode 100644 index 53447582c24..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py +++ /dev/null @@ -1,68 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DatetimeClientTestBase, DatetimePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimePropertyOperations(DatetimeClientTestBase): - @DatetimePreparer() - @recorded_by_proxy - def test_property_default(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.property.default( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_property_rfc3339(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.property.rfc3339( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_property_rfc7231(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.property.rfc7231( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_property_unix_timestamp(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.property.unix_timestamp( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_property_unix_timestamp_array(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.property.unix_timestamp_array( - body={"value": ["2020-02-20 00:00:00"]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py deleted file mode 100644 index c7062cd1d6f..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py +++ /dev/null @@ -1,69 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DatetimePreparer -from testpreparer_async import DatetimeClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimePropertyOperationsAsync(DatetimeClientTestBaseAsync): - @DatetimePreparer() - @recorded_by_proxy_async - async def test_property_default(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.property.default( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_property_rfc3339(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.property.rfc3339( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_property_rfc7231(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.property.rfc7231( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_property_unix_timestamp(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.property.unix_timestamp( - body={"value": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_property_unix_timestamp_array(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.property.unix_timestamp_array( - body={"value": ["2020-02-20 00:00:00"]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py deleted file mode 100644 index 43419883b18..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py +++ /dev/null @@ -1,68 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DatetimeClientTestBase, DatetimePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimeQueryOperations(DatetimeClientTestBase): - @DatetimePreparer() - @recorded_by_proxy - def test_query_default(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.query.default( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_query_rfc3339(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.query.rfc3339( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_query_rfc7231(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.query.rfc7231( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_query_unix_timestamp(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.query.unix_timestamp( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_query_unix_timestamp_array(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.query.unix_timestamp_array( - value=["2020-02-20 00:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py deleted file mode 100644 index d2ea13edbfd..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py +++ /dev/null @@ -1,69 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DatetimePreparer -from testpreparer_async import DatetimeClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimeQueryOperationsAsync(DatetimeClientTestBaseAsync): - @DatetimePreparer() - @recorded_by_proxy_async - async def test_query_default(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.query.default( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_query_rfc3339(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.query.rfc3339( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_query_rfc7231(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.query.rfc7231( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_query_unix_timestamp(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.query.unix_timestamp( - value="2020-02-20 00:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_query_unix_timestamp_array(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.query.unix_timestamp_array( - value=["2020-02-20 00:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py deleted file mode 100644 index 7facfe78744..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py +++ /dev/null @@ -1,49 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DatetimeClientTestBase, DatetimePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimeResponseHeaderOperations(DatetimeClientTestBase): - @DatetimePreparer() - @recorded_by_proxy - def test_response_header_default(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.response_header.default() - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_response_header_rfc3339(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.response_header.rfc3339() - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_response_header_rfc7231(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.response_header.rfc7231() - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy - def test_response_header_unix_timestamp(self, datetime_endpoint): - client = self.create_client(endpoint=datetime_endpoint) - response = client.response_header.unix_timestamp() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py deleted file mode 100644 index 304b646609e..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py +++ /dev/null @@ -1,50 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DatetimePreparer -from testpreparer_async import DatetimeClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDatetimeResponseHeaderOperationsAsync(DatetimeClientTestBaseAsync): - @DatetimePreparer() - @recorded_by_proxy_async - async def test_response_header_default(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.response_header.default() - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_response_header_rfc3339(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.response_header.rfc3339() - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_response_header_rfc7231(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.response_header.rfc7231() - - # please add some check logic here by yourself - # ... - - @DatetimePreparer() - @recorded_by_proxy_async - async def test_response_header_unix_timestamp(self, datetime_endpoint): - client = self.create_async_client(endpoint=datetime_endpoint) - response = await client.response_header.unix_timestamp() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py deleted file mode 100644 index b05b3815661..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -from encode.datetime import DatetimeClient -import functools - - -class DatetimeClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(DatetimeClient) - return self.create_client_from_credential( - DatetimeClient, - credential=credential, - endpoint=endpoint, - ) - - -DatetimePreparer = functools.partial( - PowerShellPreparer, "datetime", datetime_endpoint="https://fake_datetime_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py deleted file mode 100644 index fa3fcd42461..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from encode.datetime.aio import DatetimeClient - - -class DatetimeClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(DatetimeClient, is_async=True) - return self.create_client_from_credential( - DatetimeClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/encode-duration/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-duration/encode/duration/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-duration/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-duration/encode/duration/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py deleted file mode 100644 index 7c8e921ac05..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - duration_subscription_id = os.environ.get("DURATION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - duration_tenant_id = os.environ.get("DURATION_TENANT_ID", "00000000-0000-0000-0000-000000000000") - duration_client_id = os.environ.get("DURATION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - duration_client_secret = os.environ.get("DURATION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=duration_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=duration_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=duration_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=duration_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py deleted file mode 100644 index f3bf725bed1..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py +++ /dev/null @@ -1,79 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DurationClientTestBase, DurationPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDurationHeaderOperations(DurationClientTestBase): - @DurationPreparer() - @recorded_by_proxy - def test_header_default(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.header.default( - duration="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_header_iso8601(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.header.iso8601( - duration="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_header_iso8601_array(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.header.iso8601_array( - duration=["1 day, 0:00:00"], - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_header_int32_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.header.int32_seconds( - duration=0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_header_float_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.header.float_seconds( - duration=0.0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_header_float64_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.header.float64_seconds( - duration=0.0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py deleted file mode 100644 index 8437b4e599c..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py +++ /dev/null @@ -1,80 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DurationPreparer -from testpreparer_async import DurationClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDurationHeaderOperationsAsync(DurationClientTestBaseAsync): - @DurationPreparer() - @recorded_by_proxy_async - async def test_header_default(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.header.default( - duration="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_header_iso8601(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.header.iso8601( - duration="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_header_iso8601_array(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.header.iso8601_array( - duration=["1 day, 0:00:00"], - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_header_int32_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.header.int32_seconds( - duration=0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_header_float_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.header.float_seconds( - duration=0.0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_header_float64_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.header.float64_seconds( - duration=0.0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py deleted file mode 100644 index 696c38e6897..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py +++ /dev/null @@ -1,79 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DurationClientTestBase, DurationPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDurationPropertyOperations(DurationClientTestBase): - @DurationPreparer() - @recorded_by_proxy - def test_property_default(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.property.default( - body={"value": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_property_iso8601(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.property.iso8601( - body={"value": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_property_int32_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.property.int32_seconds( - body={"value": 0}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_property_float_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.property.float_seconds( - body={"value": 0.0}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_property_float64_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.property.float64_seconds( - body={"value": 0.0}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_property_float_seconds_array(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.property.float_seconds_array( - body={"value": [0.0]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py deleted file mode 100644 index 8b6db79b5d8..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py +++ /dev/null @@ -1,80 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DurationPreparer -from testpreparer_async import DurationClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDurationPropertyOperationsAsync(DurationClientTestBaseAsync): - @DurationPreparer() - @recorded_by_proxy_async - async def test_property_default(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.property.default( - body={"value": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_property_iso8601(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.property.iso8601( - body={"value": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_property_int32_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.property.int32_seconds( - body={"value": 0}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_property_float_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.property.float_seconds( - body={"value": 0.0}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_property_float64_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.property.float64_seconds( - body={"value": 0.0}, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_property_float_seconds_array(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.property.float_seconds_array( - body={"value": [0.0]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py deleted file mode 100644 index f01cfb4019e..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py +++ /dev/null @@ -1,79 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DurationClientTestBase, DurationPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDurationQueryOperations(DurationClientTestBase): - @DurationPreparer() - @recorded_by_proxy - def test_query_default(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.query.default( - input="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_query_iso8601(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.query.iso8601( - input="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_query_int32_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.query.int32_seconds( - input=0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_query_float_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.query.float_seconds( - input=0.0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_query_float64_seconds(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.query.float64_seconds( - input=0.0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy - def test_query_int32_seconds_array(self, duration_endpoint): - client = self.create_client(endpoint=duration_endpoint) - response = client.query.int32_seconds_array( - input=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py deleted file mode 100644 index ce3da7ce8e0..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py +++ /dev/null @@ -1,80 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DurationPreparer -from testpreparer_async import DurationClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDurationQueryOperationsAsync(DurationClientTestBaseAsync): - @DurationPreparer() - @recorded_by_proxy_async - async def test_query_default(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.query.default( - input="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_query_iso8601(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.query.iso8601( - input="1 day, 0:00:00", - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_query_int32_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.query.int32_seconds( - input=0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_query_float_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.query.float_seconds( - input=0.0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_query_float64_seconds(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.query.float64_seconds( - input=0.0, - ) - - # please add some check logic here by yourself - # ... - - @DurationPreparer() - @recorded_by_proxy_async - async def test_query_int32_seconds_array(self, duration_endpoint): - client = self.create_async_client(endpoint=duration_endpoint) - response = await client.query.int32_seconds_array( - input=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py deleted file mode 100644 index 92daf94d4fc..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -from encode.duration import DurationClient -import functools - - -class DurationClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(DurationClient) - return self.create_client_from_credential( - DurationClient, - credential=credential, - endpoint=endpoint, - ) - - -DurationPreparer = functools.partial( - PowerShellPreparer, "duration", duration_endpoint="https://fake_duration_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py deleted file mode 100644 index 9024e6891fc..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from encode.duration.aio import DurationClient - - -class DurationClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(DurationClient, is_async=True) - return self.create_client_from_credential( - DurationClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-numeric/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py deleted file mode 100644 index 4d23d2b7057..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - numeric_subscription_id = os.environ.get("NUMERIC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - numeric_tenant_id = os.environ.get("NUMERIC_TENANT_ID", "00000000-0000-0000-0000-000000000000") - numeric_client_id = os.environ.get("NUMERIC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - numeric_client_secret = os.environ.get("NUMERIC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=numeric_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=numeric_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=numeric_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=numeric_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py deleted file mode 100644 index fffcf2d5f4e..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py +++ /dev/null @@ -1,46 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NumericClientTestBase, NumericPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNumericPropertyOperations(NumericClientTestBase): - @NumericPreparer() - @recorded_by_proxy - def test_property_safeint_as_string(self, numeric_endpoint): - client = self.create_client(endpoint=numeric_endpoint) - response = client.property.safeint_as_string( - value={"value": 0}, - ) - - # please add some check logic here by yourself - # ... - - @NumericPreparer() - @recorded_by_proxy - def test_property_uint32_as_string_optional(self, numeric_endpoint): - client = self.create_client(endpoint=numeric_endpoint) - response = client.property.uint32_as_string_optional( - value={"value": 0}, - ) - - # please add some check logic here by yourself - # ... - - @NumericPreparer() - @recorded_by_proxy - def test_property_uint8_as_string(self, numeric_endpoint): - client = self.create_client(endpoint=numeric_endpoint) - response = client.property.uint8_as_string( - value={"value": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py deleted file mode 100644 index 8de4f2ad73e..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py +++ /dev/null @@ -1,47 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NumericPreparer -from testpreparer_async import NumericClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNumericPropertyOperationsAsync(NumericClientTestBaseAsync): - @NumericPreparer() - @recorded_by_proxy_async - async def test_property_safeint_as_string(self, numeric_endpoint): - client = self.create_async_client(endpoint=numeric_endpoint) - response = await client.property.safeint_as_string( - value={"value": 0}, - ) - - # please add some check logic here by yourself - # ... - - @NumericPreparer() - @recorded_by_proxy_async - async def test_property_uint32_as_string_optional(self, numeric_endpoint): - client = self.create_async_client(endpoint=numeric_endpoint) - response = await client.property.uint32_as_string_optional( - value={"value": 0}, - ) - - # please add some check logic here by yourself - # ... - - @NumericPreparer() - @recorded_by_proxy_async - async def test_property_uint8_as_string(self, numeric_endpoint): - client = self.create_async_client(endpoint=numeric_endpoint) - response = await client.property.uint8_as_string( - value={"value": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py deleted file mode 100644 index f69f6a84bd9..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -from encode.numeric import NumericClient -import functools - - -class NumericClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(NumericClient) - return self.create_client_from_credential( - NumericClient, - credential=credential, - endpoint=endpoint, - ) - - -NumericPreparer = functools.partial(PowerShellPreparer, "numeric", numeric_endpoint="https://fake_numeric_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py deleted file mode 100644 index 1f2469a6268..00000000000 --- a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from encode.numeric.aio import NumericClient - - -class NumericClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(NumericClient, is_async=True) - return self.create_client_from_credential( - NumericClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py deleted file mode 100644 index 3fc4942cf79..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - visibility_subscription_id = os.environ.get("VISIBILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - visibility_tenant_id = os.environ.get("VISIBILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - visibility_client_id = os.environ.get("VISIBILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - visibility_client_secret = os.environ.get("VISIBILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py deleted file mode 100644 index 7ed8cd7db1d..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py +++ /dev/null @@ -1,90 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import VisibilityClientTestBase, VisibilityPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVisibility(VisibilityClientTestBase): - @VisibilityPreparer() - @recorded_by_proxy - def test_get_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.get_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_head_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.head_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_put_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.put_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_patch_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.patch_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_post_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.post_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_delete_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.delete_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_put_read_only_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.put_read_only_model( - input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py deleted file mode 100644 index 9909a3a67eb..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py +++ /dev/null @@ -1,91 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import VisibilityPreparer -from testpreparer_async import VisibilityClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVisibilityAsync(VisibilityClientTestBaseAsync): - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_get_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.get_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_head_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.head_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_put_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.put_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_patch_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.patch_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_post_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.post_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_delete_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.delete_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_put_read_only_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.put_read_only_model( - input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py deleted file mode 100644 index 877033c71e3..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from headasbooleanfalse import VisibilityClient - - -class VisibilityClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(VisibilityClient) - return self.create_client_from_credential( - VisibilityClient, - credential=credential, - endpoint=endpoint, - ) - - -VisibilityPreparer = functools.partial( - PowerShellPreparer, "visibility", visibility_endpoint="https://fake_visibility_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py deleted file mode 100644 index 2d82231904d..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from headasbooleanfalse.aio import VisibilityClient - - -class VisibilityClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(VisibilityClient, is_async=True) - return self.create_client_from_credential( - VisibilityClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/headasbooleanfalse/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py deleted file mode 100644 index 3fc4942cf79..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - visibility_subscription_id = os.environ.get("VISIBILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - visibility_tenant_id = os.environ.get("VISIBILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - visibility_client_id = os.environ.get("VISIBILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - visibility_client_secret = os.environ.get("VISIBILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py deleted file mode 100644 index 7ed8cd7db1d..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py +++ /dev/null @@ -1,90 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import VisibilityClientTestBase, VisibilityPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVisibility(VisibilityClientTestBase): - @VisibilityPreparer() - @recorded_by_proxy - def test_get_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.get_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_head_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.head_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_put_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.put_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_patch_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.patch_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_post_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.post_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_delete_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.delete_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_put_read_only_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.put_read_only_model( - input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py deleted file mode 100644 index 9909a3a67eb..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py +++ /dev/null @@ -1,91 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import VisibilityPreparer -from testpreparer_async import VisibilityClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVisibilityAsync(VisibilityClientTestBaseAsync): - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_get_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.get_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_head_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.head_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_put_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.put_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_patch_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.patch_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_post_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.post_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_delete_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.delete_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_put_read_only_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.put_read_only_model( - input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py deleted file mode 100644 index 370f4302de3..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from headasbooleantrue import VisibilityClient - - -class VisibilityClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(VisibilityClient) - return self.create_client_from_credential( - VisibilityClient, - credential=credential, - endpoint=endpoint, - ) - - -VisibilityPreparer = functools.partial( - PowerShellPreparer, "visibility", visibility_endpoint="https://fake_visibility_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py deleted file mode 100644 index 816ff818734..00000000000 --- a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from headasbooleantrue.aio import VisibilityClient - - -class VisibilityClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(VisibilityClient, is_async=True) - return self.create_client_from_credential( - VisibilityClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/headasbooleantrue/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py deleted file mode 100644 index 635e5035817..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - basic_subscription_id = os.environ.get("BASIC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - basic_tenant_id = os.environ.get("BASIC_TENANT_ID", "00000000-0000-0000-0000-000000000000") - basic_client_id = os.environ.get("BASIC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - basic_client_secret = os.environ.get("BASIC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=basic_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py deleted file mode 100644 index a764b421795..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BasicClientTestBase, BasicPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBasicExplicitBodyOperations(BasicClientTestBase): - @BasicPreparer() - @recorded_by_proxy - def test_explicit_body_simple(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.explicit_body.simple( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py deleted file mode 100644 index bef8dd97225..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BasicPreparer -from testpreparer_async import BasicClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBasicExplicitBodyOperationsAsync(BasicClientTestBaseAsync): - @BasicPreparer() - @recorded_by_proxy_async - async def test_explicit_body_simple(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.explicit_body.simple( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py deleted file mode 100644 index 525c35f6dd0..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BasicClientTestBase, BasicPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBasicImplicitBodyOperations(BasicClientTestBase): - @BasicPreparer() - @recorded_by_proxy - def test_implicit_body_simple(self, basic_endpoint): - client = self.create_client(endpoint=basic_endpoint) - response = client.implicit_body.simple( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py deleted file mode 100644 index 83322a2b281..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BasicPreparer -from testpreparer_async import BasicClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBasicImplicitBodyOperationsAsync(BasicClientTestBaseAsync): - @BasicPreparer() - @recorded_by_proxy_async - async def test_implicit_body_simple(self, basic_endpoint): - client = self.create_async_client(endpoint=basic_endpoint) - response = await client.implicit_body.simple( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py deleted file mode 100644 index 7b6afef93f7..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from parameters.basic import BasicClient - - -class BasicClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(BasicClient) - return self.create_client_from_credential( - BasicClient, - credential=credential, - endpoint=endpoint, - ) - - -BasicPreparer = functools.partial(PowerShellPreparer, "basic", basic_endpoint="https://fake_basic_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py deleted file mode 100644 index 24f41a78773..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from parameters.basic.aio import BasicClient - - -class BasicClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(BasicClient, is_async=True) - return self.create_client_from_credential( - BasicClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-basic/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py deleted file mode 100644 index a125385319e..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - bodyoptionality_subscription_id = os.environ.get( - "BODYOPTIONALITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - bodyoptionality_tenant_id = os.environ.get("BODYOPTIONALITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - bodyoptionality_client_id = os.environ.get("BODYOPTIONALITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - bodyoptionality_client_secret = os.environ.get( - "BODYOPTIONALITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=bodyoptionality_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=bodyoptionality_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=bodyoptionality_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=bodyoptionality_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py deleted file mode 100644 index ef56ba5b5a8..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BodyOptionalityClientTestBase, BodyOptionalityPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBodyOptionality(BodyOptionalityClientTestBase): - @BodyOptionalityPreparer() - @recorded_by_proxy - def test_required_explicit(self, bodyoptionality_endpoint): - client = self.create_client(endpoint=bodyoptionality_endpoint) - response = client.required_explicit( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @BodyOptionalityPreparer() - @recorded_by_proxy - def test_required_implicit(self, bodyoptionality_endpoint): - client = self.create_client(endpoint=bodyoptionality_endpoint) - response = client.required_implicit( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py deleted file mode 100644 index 5024fbdff9d..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BodyOptionalityPreparer -from testpreparer_async import BodyOptionalityClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBodyOptionalityAsync(BodyOptionalityClientTestBaseAsync): - @BodyOptionalityPreparer() - @recorded_by_proxy_async - async def test_required_explicit(self, bodyoptionality_endpoint): - client = self.create_async_client(endpoint=bodyoptionality_endpoint) - response = await client.required_explicit( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @BodyOptionalityPreparer() - @recorded_by_proxy_async - async def test_required_implicit(self, bodyoptionality_endpoint): - client = self.create_async_client(endpoint=bodyoptionality_endpoint) - response = await client.required_implicit( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py deleted file mode 100644 index 5dfe0e95ed4..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py +++ /dev/null @@ -1,31 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import BodyOptionalityClientTestBase, BodyOptionalityPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBodyOptionalityOptionalExplicitOperations(BodyOptionalityClientTestBase): - @BodyOptionalityPreparer() - @recorded_by_proxy - def test_optional_explicit_set(self, bodyoptionality_endpoint): - client = self.create_client(endpoint=bodyoptionality_endpoint) - response = client.optional_explicit.set() - - # please add some check logic here by yourself - # ... - - @BodyOptionalityPreparer() - @recorded_by_proxy - def test_optional_explicit_omit(self, bodyoptionality_endpoint): - client = self.create_client(endpoint=bodyoptionality_endpoint) - response = client.optional_explicit.omit() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py deleted file mode 100644 index aa6ee862b35..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py +++ /dev/null @@ -1,32 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import BodyOptionalityPreparer -from testpreparer_async import BodyOptionalityClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestBodyOptionalityOptionalExplicitOperationsAsync(BodyOptionalityClientTestBaseAsync): - @BodyOptionalityPreparer() - @recorded_by_proxy_async - async def test_optional_explicit_set(self, bodyoptionality_endpoint): - client = self.create_async_client(endpoint=bodyoptionality_endpoint) - response = await client.optional_explicit.set() - - # please add some check logic here by yourself - # ... - - @BodyOptionalityPreparer() - @recorded_by_proxy_async - async def test_optional_explicit_omit(self, bodyoptionality_endpoint): - client = self.create_async_client(endpoint=bodyoptionality_endpoint) - response = await client.optional_explicit.omit() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py deleted file mode 100644 index 0e733a2adbb..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from parameters.bodyoptionality import BodyOptionalityClient - - -class BodyOptionalityClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(BodyOptionalityClient) - return self.create_client_from_credential( - BodyOptionalityClient, - credential=credential, - endpoint=endpoint, - ) - - -BodyOptionalityPreparer = functools.partial( - PowerShellPreparer, "bodyoptionality", bodyoptionality_endpoint="https://fake_bodyoptionality_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py deleted file mode 100644 index 0b22395b14c..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from parameters.bodyoptionality.aio import BodyOptionalityClient - - -class BodyOptionalityClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(BodyOptionalityClient, is_async=True) - return self.create_client_from_credential( - BodyOptionalityClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-body-optionality/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py deleted file mode 100644 index d97918929d3..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - collectionformat_subscription_id = os.environ.get( - "COLLECTIONFORMAT_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - collectionformat_tenant_id = os.environ.get("COLLECTIONFORMAT_TENANT_ID", "00000000-0000-0000-0000-000000000000") - collectionformat_client_id = os.environ.get("COLLECTIONFORMAT_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - collectionformat_client_secret = os.environ.get( - "COLLECTIONFORMAT_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=collectionformat_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=collectionformat_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=collectionformat_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=collectionformat_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py deleted file mode 100644 index 8d22f920b98..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import CollectionFormatClientTestBase, CollectionFormatPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCollectionFormatHeaderOperations(CollectionFormatClientTestBase): - @CollectionFormatPreparer() - @recorded_by_proxy - def test_header_csv(self, collectionformat_endpoint): - client = self.create_client(endpoint=collectionformat_endpoint) - response = client.header.csv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py deleted file mode 100644 index 3b596202fb2..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import CollectionFormatPreparer -from testpreparer_async import CollectionFormatClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCollectionFormatHeaderOperationsAsync(CollectionFormatClientTestBaseAsync): - @CollectionFormatPreparer() - @recorded_by_proxy_async - async def test_header_csv(self, collectionformat_endpoint): - client = self.create_async_client(endpoint=collectionformat_endpoint) - response = await client.header.csv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py deleted file mode 100644 index 12397560740..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py +++ /dev/null @@ -1,68 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import CollectionFormatClientTestBase, CollectionFormatPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCollectionFormatQueryOperations(CollectionFormatClientTestBase): - @CollectionFormatPreparer() - @recorded_by_proxy - def test_query_multi(self, collectionformat_endpoint): - client = self.create_client(endpoint=collectionformat_endpoint) - response = client.query.multi( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy - def test_query_ssv(self, collectionformat_endpoint): - client = self.create_client(endpoint=collectionformat_endpoint) - response = client.query.ssv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy - def test_query_tsv(self, collectionformat_endpoint): - client = self.create_client(endpoint=collectionformat_endpoint) - response = client.query.tsv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy - def test_query_pipes(self, collectionformat_endpoint): - client = self.create_client(endpoint=collectionformat_endpoint) - response = client.query.pipes( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy - def test_query_csv(self, collectionformat_endpoint): - client = self.create_client(endpoint=collectionformat_endpoint) - response = client.query.csv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py deleted file mode 100644 index 4a73943c601..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py +++ /dev/null @@ -1,69 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import CollectionFormatPreparer -from testpreparer_async import CollectionFormatClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestCollectionFormatQueryOperationsAsync(CollectionFormatClientTestBaseAsync): - @CollectionFormatPreparer() - @recorded_by_proxy_async - async def test_query_multi(self, collectionformat_endpoint): - client = self.create_async_client(endpoint=collectionformat_endpoint) - response = await client.query.multi( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy_async - async def test_query_ssv(self, collectionformat_endpoint): - client = self.create_async_client(endpoint=collectionformat_endpoint) - response = await client.query.ssv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy_async - async def test_query_tsv(self, collectionformat_endpoint): - client = self.create_async_client(endpoint=collectionformat_endpoint) - response = await client.query.tsv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy_async - async def test_query_pipes(self, collectionformat_endpoint): - client = self.create_async_client(endpoint=collectionformat_endpoint) - response = await client.query.pipes( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... - - @CollectionFormatPreparer() - @recorded_by_proxy_async - async def test_query_csv(self, collectionformat_endpoint): - client = self.create_async_client(endpoint=collectionformat_endpoint) - response = await client.query.csv( - colors=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py deleted file mode 100644 index 94fd34f3df5..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from parameters.collectionformat import CollectionFormatClient - - -class CollectionFormatClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(CollectionFormatClient) - return self.create_client_from_credential( - CollectionFormatClient, - credential=credential, - endpoint=endpoint, - ) - - -CollectionFormatPreparer = functools.partial( - PowerShellPreparer, "collectionformat", collectionformat_endpoint="https://fake_collectionformat_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py deleted file mode 100644 index fd91ec8b8cb..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from parameters.collectionformat.aio import CollectionFormatClient - - -class CollectionFormatClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(CollectionFormatClient, is_async=True) - return self.create_client_from_credential( - CollectionFormatClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-collection-format/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py deleted file mode 100644 index 67d88cfd75b..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - spread_subscription_id = os.environ.get("SPREAD_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - spread_tenant_id = os.environ.get("SPREAD_TENANT_ID", "00000000-0000-0000-0000-000000000000") - spread_client_id = os.environ.get("SPREAD_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - spread_client_secret = os.environ.get("SPREAD_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=spread_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=spread_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=spread_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=spread_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py deleted file mode 100644 index c1be4f1f508..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py +++ /dev/null @@ -1,83 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SpreadClientTestBase, SpreadPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpreadAliasOperations(SpreadClientTestBase): - @SpreadPreparer() - @recorded_by_proxy - def test_alias_spread_as_request_body(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.alias.spread_as_request_body( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_alias_spread_parameter_with_inner_model(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.alias.spread_parameter_with_inner_model( - id="str", - body={"name": "str"}, - x_ms_test_header="str", - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_alias_spread_as_request_parameter(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.alias.spread_as_request_parameter( - id="str", - body={"name": "str"}, - x_ms_test_header="str", - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_alias_spread_with_multiple_parameters(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.alias.spread_with_multiple_parameters( - id="str", - body={"requiredIntList": [0], "requiredString": "str", "optionalInt": 0, "optionalStringList": ["str"]}, - x_ms_test_header="str", - required_string="str", - required_int_list=[0], - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_alias_spread_parameter_with_inner_alias(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.alias.spread_parameter_with_inner_alias( - id="str", - body={"age": 0, "name": "str"}, - x_ms_test_header="str", - name="str", - age=0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py deleted file mode 100644 index e54868a73d8..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py +++ /dev/null @@ -1,84 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SpreadPreparer -from testpreparer_async import SpreadClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpreadAliasOperationsAsync(SpreadClientTestBaseAsync): - @SpreadPreparer() - @recorded_by_proxy_async - async def test_alias_spread_as_request_body(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.alias.spread_as_request_body( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_alias_spread_parameter_with_inner_model(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.alias.spread_parameter_with_inner_model( - id="str", - body={"name": "str"}, - x_ms_test_header="str", - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_alias_spread_as_request_parameter(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.alias.spread_as_request_parameter( - id="str", - body={"name": "str"}, - x_ms_test_header="str", - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_alias_spread_with_multiple_parameters(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.alias.spread_with_multiple_parameters( - id="str", - body={"requiredIntList": [0], "requiredString": "str", "optionalInt": 0, "optionalStringList": ["str"]}, - x_ms_test_header="str", - required_string="str", - required_int_list=[0], - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_alias_spread_parameter_with_inner_alias(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.alias.spread_parameter_with_inner_alias( - id="str", - body={"age": 0, "name": "str"}, - x_ms_test_header="str", - name="str", - age=0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py deleted file mode 100644 index e223de65762..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py +++ /dev/null @@ -1,75 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SpreadClientTestBase, SpreadPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpreadModelOperations(SpreadClientTestBase): - @SpreadPreparer() - @recorded_by_proxy - def test_model_spread_as_request_body(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.model.spread_as_request_body( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_model_spread_composite_request_only_with_body(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.model.spread_composite_request_only_with_body( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_model_spread_composite_request_without_body(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.model.spread_composite_request_without_body( - name="str", - test_header="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_model_spread_composite_request(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.model.spread_composite_request( - name="str", - body={"name": "str"}, - test_header="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy - def test_model_spread_composite_request_mix(self, spread_endpoint): - client = self.create_client(endpoint=spread_endpoint) - response = client.model.spread_composite_request_mix( - name="str", - body={"prop": "str"}, - test_header="str", - prop="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py deleted file mode 100644 index d8919fbfd48..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py +++ /dev/null @@ -1,76 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SpreadPreparer -from testpreparer_async import SpreadClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpreadModelOperationsAsync(SpreadClientTestBaseAsync): - @SpreadPreparer() - @recorded_by_proxy_async - async def test_model_spread_as_request_body(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.model.spread_as_request_body( - body={"name": "str"}, - name="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_model_spread_composite_request_only_with_body(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.model.spread_composite_request_only_with_body( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_model_spread_composite_request_without_body(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.model.spread_composite_request_without_body( - name="str", - test_header="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_model_spread_composite_request(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.model.spread_composite_request( - name="str", - body={"name": "str"}, - test_header="str", - ) - - # please add some check logic here by yourself - # ... - - @SpreadPreparer() - @recorded_by_proxy_async - async def test_model_spread_composite_request_mix(self, spread_endpoint): - client = self.create_async_client(endpoint=spread_endpoint) - response = await client.model.spread_composite_request_mix( - name="str", - body={"prop": "str"}, - test_header="str", - prop="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py deleted file mode 100644 index 2901c04003f..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from parameters.spread import SpreadClient - - -class SpreadClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(SpreadClient) - return self.create_client_from_credential( - SpreadClient, - credential=credential, - endpoint=endpoint, - ) - - -SpreadPreparer = functools.partial(PowerShellPreparer, "spread", spread_endpoint="https://fake_spread_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py deleted file mode 100644 index 1416ad2c18b..00000000000 --- a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from parameters.spread.aio import SpreadClient - - -class SpreadClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(SpreadClient, is_async=True) - return self.create_client_from_credential( - SpreadClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-spread/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py deleted file mode 100644 index acf7adc84bf..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py +++ /dev/null @@ -1,43 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - contentnegotiation_subscription_id = os.environ.get( - "CONTENTNEGOTIATION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - contentnegotiation_tenant_id = os.environ.get( - "CONTENTNEGOTIATION_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - contentnegotiation_client_id = os.environ.get( - "CONTENTNEGOTIATION_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - contentnegotiation_client_secret = os.environ.get( - "CONTENTNEGOTIATION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=contentnegotiation_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=contentnegotiation_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=contentnegotiation_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=contentnegotiation_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py deleted file mode 100644 index da9558d9c13..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ContentNegotiationClientTestBase, ContentNegotiationPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestContentNegotiationDifferentBodyOperations(ContentNegotiationClientTestBase): - @ContentNegotiationPreparer() - @recorded_by_proxy - def test_different_body_get_avatar_as_png(self, contentnegotiation_endpoint): - client = self.create_client(endpoint=contentnegotiation_endpoint) - response = client.different_body.get_avatar_as_png( - accept="image/png", - ) - - # please add some check logic here by yourself - # ... - - @ContentNegotiationPreparer() - @recorded_by_proxy - def test_different_body_get_avatar_as_json(self, contentnegotiation_endpoint): - client = self.create_client(endpoint=contentnegotiation_endpoint) - response = client.different_body.get_avatar_as_json( - accept="application/json", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py deleted file mode 100644 index 62d47e75f10..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ContentNegotiationPreparer -from testpreparer_async import ContentNegotiationClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestContentNegotiationDifferentBodyOperationsAsync(ContentNegotiationClientTestBaseAsync): - @ContentNegotiationPreparer() - @recorded_by_proxy_async - async def test_different_body_get_avatar_as_png(self, contentnegotiation_endpoint): - client = self.create_async_client(endpoint=contentnegotiation_endpoint) - response = await client.different_body.get_avatar_as_png( - accept="image/png", - ) - - # please add some check logic here by yourself - # ... - - @ContentNegotiationPreparer() - @recorded_by_proxy_async - async def test_different_body_get_avatar_as_json(self, contentnegotiation_endpoint): - client = self.create_async_client(endpoint=contentnegotiation_endpoint) - response = await client.different_body.get_avatar_as_json( - accept="application/json", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py deleted file mode 100644 index 84186b5709d..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ContentNegotiationClientTestBase, ContentNegotiationPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestContentNegotiationSameBodyOperations(ContentNegotiationClientTestBase): - @ContentNegotiationPreparer() - @recorded_by_proxy - def test_same_body_get_avatar_as_png(self, contentnegotiation_endpoint): - client = self.create_client(endpoint=contentnegotiation_endpoint) - response = client.same_body.get_avatar_as_png( - accept="image/png", - ) - - # please add some check logic here by yourself - # ... - - @ContentNegotiationPreparer() - @recorded_by_proxy - def test_same_body_get_avatar_as_jpeg(self, contentnegotiation_endpoint): - client = self.create_client(endpoint=contentnegotiation_endpoint) - response = client.same_body.get_avatar_as_jpeg( - accept="image/jpeg", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py deleted file mode 100644 index 2ece9daf85f..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ContentNegotiationPreparer -from testpreparer_async import ContentNegotiationClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestContentNegotiationSameBodyOperationsAsync(ContentNegotiationClientTestBaseAsync): - @ContentNegotiationPreparer() - @recorded_by_proxy_async - async def test_same_body_get_avatar_as_png(self, contentnegotiation_endpoint): - client = self.create_async_client(endpoint=contentnegotiation_endpoint) - response = await client.same_body.get_avatar_as_png( - accept="image/png", - ) - - # please add some check logic here by yourself - # ... - - @ContentNegotiationPreparer() - @recorded_by_proxy_async - async def test_same_body_get_avatar_as_jpeg(self, contentnegotiation_endpoint): - client = self.create_async_client(endpoint=contentnegotiation_endpoint) - response = await client.same_body.get_avatar_as_jpeg( - accept="image/jpeg", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py deleted file mode 100644 index da4060d62c8..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from payload.contentnegotiation import ContentNegotiationClient - - -class ContentNegotiationClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ContentNegotiationClient) - return self.create_client_from_credential( - ContentNegotiationClient, - credential=credential, - endpoint=endpoint, - ) - - -ContentNegotiationPreparer = functools.partial( - PowerShellPreparer, "contentnegotiation", contentnegotiation_endpoint="https://fake_contentnegotiation_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py deleted file mode 100644 index 7133bd2e095..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from payload.contentnegotiation.aio import ContentNegotiationClient - - -class ContentNegotiationClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ContentNegotiationClient, is_async=True) - return self.create_client_from_credential( - ContentNegotiationClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-content-negotiation/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py deleted file mode 100644 index 687cbc1cd18..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - jsonmergepatch_subscription_id = os.environ.get( - "JSONMERGEPATCH_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - jsonmergepatch_tenant_id = os.environ.get("JSONMERGEPATCH_TENANT_ID", "00000000-0000-0000-0000-000000000000") - jsonmergepatch_client_id = os.environ.get("JSONMERGEPATCH_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - jsonmergepatch_client_secret = os.environ.get( - "JSONMERGEPATCH_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=jsonmergepatch_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=jsonmergepatch_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=jsonmergepatch_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=jsonmergepatch_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py deleted file mode 100644 index 5719033ace2..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py +++ /dev/null @@ -1,61 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import JsonMergePatchClientTestBase, JsonMergePatchPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestJsonMergePatch(JsonMergePatchClientTestBase): - @JsonMergePatchPreparer() - @recorded_by_proxy - def test_create_resource(self, jsonmergepatch_endpoint): - client = self.create_client(endpoint=jsonmergepatch_endpoint) - response = client.create_resource( - body={ - "name": "str", - "array": [{"description": "str", "name": "str"}], - "description": "str", - "floatValue": 0.0, - "innerModel": {"description": "str", "name": "str"}, - "intArray": [0], - "intValue": 0, - "map": {"str": {"description": "str", "name": "str"}}, - }, - ) - - # please add some check logic here by yourself - # ... - - @JsonMergePatchPreparer() - @recorded_by_proxy - def test_update_resource(self, jsonmergepatch_endpoint): - client = self.create_client(endpoint=jsonmergepatch_endpoint) - response = client.update_resource( - body={ - "array": [{"description": "str", "name": "str"}], - "description": "str", - "floatValue": 0.0, - "innerModel": {"description": "str", "name": "str"}, - "intArray": [0], - "intValue": 0, - "map": {"str": {"description": "str", "name": "str"}}, - }, - ) - - # please add some check logic here by yourself - # ... - - @JsonMergePatchPreparer() - @recorded_by_proxy - def test_update_optional_resource(self, jsonmergepatch_endpoint): - client = self.create_client(endpoint=jsonmergepatch_endpoint) - response = client.update_optional_resource() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py deleted file mode 100644 index bfc931ddf4b..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py +++ /dev/null @@ -1,62 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import JsonMergePatchPreparer -from testpreparer_async import JsonMergePatchClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestJsonMergePatchAsync(JsonMergePatchClientTestBaseAsync): - @JsonMergePatchPreparer() - @recorded_by_proxy_async - async def test_create_resource(self, jsonmergepatch_endpoint): - client = self.create_async_client(endpoint=jsonmergepatch_endpoint) - response = await client.create_resource( - body={ - "name": "str", - "array": [{"description": "str", "name": "str"}], - "description": "str", - "floatValue": 0.0, - "innerModel": {"description": "str", "name": "str"}, - "intArray": [0], - "intValue": 0, - "map": {"str": {"description": "str", "name": "str"}}, - }, - ) - - # please add some check logic here by yourself - # ... - - @JsonMergePatchPreparer() - @recorded_by_proxy_async - async def test_update_resource(self, jsonmergepatch_endpoint): - client = self.create_async_client(endpoint=jsonmergepatch_endpoint) - response = await client.update_resource( - body={ - "array": [{"description": "str", "name": "str"}], - "description": "str", - "floatValue": 0.0, - "innerModel": {"description": "str", "name": "str"}, - "intArray": [0], - "intValue": 0, - "map": {"str": {"description": "str", "name": "str"}}, - }, - ) - - # please add some check logic here by yourself - # ... - - @JsonMergePatchPreparer() - @recorded_by_proxy_async - async def test_update_optional_resource(self, jsonmergepatch_endpoint): - client = self.create_async_client(endpoint=jsonmergepatch_endpoint) - response = await client.update_optional_resource() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py deleted file mode 100644 index 4be2411f2f8..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from payload.jsonmergepatch import JsonMergePatchClient - - -class JsonMergePatchClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(JsonMergePatchClient) - return self.create_client_from_credential( - JsonMergePatchClient, - credential=credential, - endpoint=endpoint, - ) - - -JsonMergePatchPreparer = functools.partial( - PowerShellPreparer, "jsonmergepatch", jsonmergepatch_endpoint="https://fake_jsonmergepatch_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py deleted file mode 100644 index fcc202c7605..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from payload.jsonmergepatch.aio import JsonMergePatchClient - - -class JsonMergePatchClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(JsonMergePatchClient, is_async=True) - return self.create_client_from_credential( - JsonMergePatchClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-json-merge-patch/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py deleted file mode 100644 index 9c6c4e96dc8..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - mediatype_subscription_id = os.environ.get("MEDIATYPE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - mediatype_tenant_id = os.environ.get("MEDIATYPE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - mediatype_client_id = os.environ.get("MEDIATYPE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - mediatype_client_secret = os.environ.get("MEDIATYPE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=mediatype_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=mediatype_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=mediatype_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=mediatype_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py deleted file mode 100644 index f3f77dafaf9..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py +++ /dev/null @@ -1,55 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import MediaTypeClientTestBase, MediaTypePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMediaTypeStringBodyOperations(MediaTypeClientTestBase): - @MediaTypePreparer() - @recorded_by_proxy - def test_string_body_send_as_text(self, mediatype_endpoint): - client = self.create_client(endpoint=mediatype_endpoint) - response = client.string_body.send_as_text( - text="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @MediaTypePreparer() - @recorded_by_proxy - def test_string_body_get_as_text(self, mediatype_endpoint): - client = self.create_client(endpoint=mediatype_endpoint) - response = client.string_body.get_as_text() - - # please add some check logic here by yourself - # ... - - @MediaTypePreparer() - @recorded_by_proxy - def test_string_body_send_as_json(self, mediatype_endpoint): - client = self.create_client(endpoint=mediatype_endpoint) - response = client.string_body.send_as_json( - text="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @MediaTypePreparer() - @recorded_by_proxy - def test_string_body_get_as_json(self, mediatype_endpoint): - client = self.create_client(endpoint=mediatype_endpoint) - response = client.string_body.get_as_json() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py deleted file mode 100644 index 12547676b6f..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py +++ /dev/null @@ -1,56 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import MediaTypePreparer -from testpreparer_async import MediaTypeClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMediaTypeStringBodyOperationsAsync(MediaTypeClientTestBaseAsync): - @MediaTypePreparer() - @recorded_by_proxy_async - async def test_string_body_send_as_text(self, mediatype_endpoint): - client = self.create_async_client(endpoint=mediatype_endpoint) - response = await client.string_body.send_as_text( - text="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @MediaTypePreparer() - @recorded_by_proxy_async - async def test_string_body_get_as_text(self, mediatype_endpoint): - client = self.create_async_client(endpoint=mediatype_endpoint) - response = await client.string_body.get_as_text() - - # please add some check logic here by yourself - # ... - - @MediaTypePreparer() - @recorded_by_proxy_async - async def test_string_body_send_as_json(self, mediatype_endpoint): - client = self.create_async_client(endpoint=mediatype_endpoint) - response = await client.string_body.send_as_json( - text="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @MediaTypePreparer() - @recorded_by_proxy_async - async def test_string_body_get_as_json(self, mediatype_endpoint): - client = self.create_async_client(endpoint=mediatype_endpoint) - response = await client.string_body.get_as_json() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py deleted file mode 100644 index f32364a5010..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from payload.mediatype import MediaTypeClient - - -class MediaTypeClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(MediaTypeClient) - return self.create_client_from_credential( - MediaTypeClient, - credential=credential, - endpoint=endpoint, - ) - - -MediaTypePreparer = functools.partial( - PowerShellPreparer, "mediatype", mediatype_endpoint="https://fake_mediatype_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py deleted file mode 100644 index 34e76d5eeaa..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from payload.mediatype.aio import MediaTypeClient - - -class MediaTypeClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(MediaTypeClient, is_async=True) - return self.create_client_from_credential( - MediaTypeClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-media-type/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py deleted file mode 100644 index 7b1948a5cb1..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - multipart_subscription_id = os.environ.get("MULTIPART_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - multipart_tenant_id = os.environ.get("MULTIPART_TENANT_ID", "00000000-0000-0000-0000-000000000000") - multipart_client_id = os.environ.get("MULTIPART_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - multipart_client_secret = os.environ.get("MULTIPART_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multipart_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multipart_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multipart_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multipart_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py deleted file mode 100644 index c0787c2ae85..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py +++ /dev/null @@ -1,152 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import MultiPartClientTestBase, MultiPartPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMultiPartFormDataOperations(MultiPartClientTestBase): - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_basic(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.basic( - body={"id": "str", "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_file_array_and_basic(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.file_array_and_basic( - body={"address": {"city": "str"}, "id": "str", "pictures": ["filetype"], "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_json_part(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.json_part( - body={"address": {"city": "str"}, "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_binary_array_parts(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.binary_array_parts( - body={"id": "str", "pictures": ["filetype"]}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_multi_binary_parts(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.multi_binary_parts( - body={"profileImage": "filetype", "picture": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_check_file_name_and_content_type(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.check_file_name_and_content_type( - body={"id": "str", "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_anonymous_model(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.anonymous_model( - body={"profileImage": "filetype"}, - profile_image="filetype", - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_http_parts_json_array_and_file_array(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.http_parts.json_array_and_file_array( - body={ - "address": {"city": "str"}, - "id": "str", - "pictures": ["filetype"], - "previousAddresses": [{"city": "str"}], - "profileImage": "filetype", - }, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_http_parts_content_type_image_jpeg_content_type(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.http_parts.content_type.image_jpeg_content_type( - body={"profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_http_parts_content_type_required_content_type(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.http_parts.content_type.required_content_type( - body={"profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_http_parts_content_type_optional_content_type(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.http_parts.content_type.optional_content_type( - body={"profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy - def test_form_data_http_parts_non_string_float(self, multipart_endpoint): - client = self.create_client(endpoint=multipart_endpoint) - response = client.form_data.http_parts.non_string.float( - body={"temperature": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py deleted file mode 100644 index e030fa8600a..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py +++ /dev/null @@ -1,153 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import MultiPartPreparer -from testpreparer_async import MultiPartClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMultiPartFormDataOperationsAsync(MultiPartClientTestBaseAsync): - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_basic(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.basic( - body={"id": "str", "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_file_array_and_basic(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.file_array_and_basic( - body={"address": {"city": "str"}, "id": "str", "pictures": ["filetype"], "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_json_part(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.json_part( - body={"address": {"city": "str"}, "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_binary_array_parts(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.binary_array_parts( - body={"id": "str", "pictures": ["filetype"]}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_multi_binary_parts(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.multi_binary_parts( - body={"profileImage": "filetype", "picture": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_check_file_name_and_content_type(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.check_file_name_and_content_type( - body={"id": "str", "profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_anonymous_model(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.anonymous_model( - body={"profileImage": "filetype"}, - profile_image="filetype", - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_http_parts_json_array_and_file_array(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.http_parts.json_array_and_file_array( - body={ - "address": {"city": "str"}, - "id": "str", - "pictures": ["filetype"], - "previousAddresses": [{"city": "str"}], - "profileImage": "filetype", - }, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_http_parts_content_type_image_jpeg_content_type(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.http_parts.content_type.image_jpeg_content_type( - body={"profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_http_parts_content_type_required_content_type(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.http_parts.content_type.required_content_type( - body={"profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_http_parts_content_type_optional_content_type(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.http_parts.content_type.optional_content_type( - body={"profileImage": "filetype"}, - ) - - # please add some check logic here by yourself - # ... - - @MultiPartPreparer() - @recorded_by_proxy_async - async def test_form_data_http_parts_non_string_float(self, multipart_endpoint): - client = self.create_async_client(endpoint=multipart_endpoint) - response = await client.form_data.http_parts.non_string.float( - body={"temperature": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py deleted file mode 100644 index ad58b84f2c2..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from payload.multipart import MultiPartClient - - -class MultiPartClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(MultiPartClient) - return self.create_client_from_credential( - MultiPartClient, - credential=credential, - endpoint=endpoint, - ) - - -MultiPartPreparer = functools.partial( - PowerShellPreparer, "multipart", multipart_endpoint="https://fake_multipart_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py deleted file mode 100644 index 98b451ec0c0..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from payload.multipart.aio import MultiPartClient - - -class MultiPartClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(MultiPartClient, is_async=True) - return self.create_client_from_credential( - MultiPartClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-multipart/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py deleted file mode 100644 index 164dcd89702..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - pageable_subscription_id = os.environ.get("PAGEABLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - pageable_tenant_id = os.environ.get("PAGEABLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - pageable_client_id = os.environ.get("PAGEABLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - pageable_client_secret = os.environ.get("PAGEABLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=pageable_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=pageable_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=pageable_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=pageable_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py deleted file mode 100644 index e66879ed799..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import PageableClientTestBase, PageablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestPageable(PageableClientTestBase): - @PageablePreparer() - @recorded_by_proxy - def test_list(self, pageable_endpoint): - client = self.create_client(endpoint=pageable_endpoint) - response = client.list() - result = [r for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py deleted file mode 100644 index 44faaa807bd..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import PageablePreparer -from testpreparer_async import PageableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestPageableAsync(PageableClientTestBaseAsync): - @PageablePreparer() - @recorded_by_proxy_async - async def test_list(self, pageable_endpoint): - client = self.create_async_client(endpoint=pageable_endpoint) - response = client.list() - result = [r async for r in response] - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py deleted file mode 100644 index 9f5844c2124..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from payload.pageable import PageableClient - - -class PageableClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(PageableClient) - return self.create_client_from_credential( - PageableClient, - credential=credential, - endpoint=endpoint, - ) - - -PageablePreparer = functools.partial( - PowerShellPreparer, "pageable", pageable_endpoint="https://fake_pageable_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py deleted file mode 100644 index 21379ef4880..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from payload.pageable.aio import PageableClient - - -class PageableClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(PageableClient, is_async=True) - return self.create_client_from_credential( - PageableClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-pageable/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py deleted file mode 100644 index b7edb687c98..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - xml_subscription_id = os.environ.get("XML_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - xml_tenant_id = os.environ.get("XML_TENANT_ID", "00000000-0000-0000-0000-000000000000") - xml_client_id = os.environ.get("XML_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - xml_client_secret = os.environ.get("XML_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xml_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xml_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xml_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=xml_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py deleted file mode 100644 index 513a9286363..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithArrayOfModelValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_array_of_model_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_array_of_model_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_array_of_model_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_array_of_model_value.put( - input={"items": [{"age": 0, "name": "str"}]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py deleted file mode 100644 index 7f8831671cf..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithArrayOfModelValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_array_of_model_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_array_of_model_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_array_of_model_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_array_of_model_value.put( - input={"items": [{"age": 0, "name": "str"}]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py deleted file mode 100644 index 75477c169a0..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithAttributesValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_attributes_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_attributes_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_attributes_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_attributes_value.put( - input={"enabled": bool, "id1": 0, "id2": "str"}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py deleted file mode 100644 index 38e116b8bf0..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithAttributesValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_attributes_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_attributes_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_attributes_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_attributes_value.put( - input={"enabled": bool, "id1": 0, "id2": "str"}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py deleted file mode 100644 index 16928d7c7ab..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithDictionaryValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_dictionary_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_dictionary_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_dictionary_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_dictionary_value.put( - input={"metadata": {"str": "str"}}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py deleted file mode 100644 index 9bf17851386..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithDictionaryValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_dictionary_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_dictionary_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_dictionary_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_dictionary_value.put( - input={"metadata": {"str": "str"}}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py deleted file mode 100644 index ad415cb3ba3..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithEmptyArrayValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_empty_array_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_empty_array_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_empty_array_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_empty_array_value.put( - input={"items": [{"age": 0, "name": "str"}]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py deleted file mode 100644 index 15c4211942d..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithEmptyArrayValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_empty_array_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_empty_array_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_empty_array_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_empty_array_value.put( - input={"items": [{"age": 0, "name": "str"}]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py deleted file mode 100644 index d5af0600ff8..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithEncodedNamesValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_encoded_names_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_encoded_names_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_encoded_names_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_encoded_names_value.put( - input={"colors": ["str"], "modelData": {"age": 0, "name": "str"}}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py deleted file mode 100644 index 26b306b521e..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithEncodedNamesValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_encoded_names_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_encoded_names_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_encoded_names_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_encoded_names_value.put( - input={"colors": ["str"], "modelData": {"age": 0, "name": "str"}}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py deleted file mode 100644 index 3390b3458d8..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithOptionalFieldValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_optional_field_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_optional_field_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_optional_field_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_optional_field_value.put( - input={"item": "str", "value": 0}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py deleted file mode 100644 index 4ce92bfc3e5..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithOptionalFieldValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_optional_field_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_optional_field_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_optional_field_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_optional_field_value.put( - input={"item": "str", "value": 0}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py deleted file mode 100644 index 1616afab05b..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithRenamedArraysValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_renamed_arrays_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_renamed_arrays_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_renamed_arrays_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_renamed_arrays_value.put( - input={"colors": ["str"], "counts": [0]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py deleted file mode 100644 index 85c8109461f..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithRenamedArraysValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_renamed_arrays_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_renamed_arrays_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_renamed_arrays_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_renamed_arrays_value.put( - input={"colors": ["str"], "counts": [0]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py deleted file mode 100644 index 55fa6ec8b3a..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithRenamedFieldsValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_renamed_fields_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_renamed_fields_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_renamed_fields_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_renamed_fields_value.put( - input={"inputData": {"age": 0, "name": "str"}, "outputData": {"age": 0, "name": "str"}}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py deleted file mode 100644 index 7ab55fa7257..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithRenamedFieldsValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_renamed_fields_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_renamed_fields_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_renamed_fields_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_renamed_fields_value.put( - input={"inputData": {"age": 0, "name": "str"}, "outputData": {"age": 0, "name": "str"}}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py deleted file mode 100644 index 097a57230ca..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithSimpleArraysValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_simple_arrays_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_simple_arrays_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_simple_arrays_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_simple_arrays_value.put( - input={"colors": ["str"], "counts": [0]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py deleted file mode 100644 index 4d4d4a54266..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithSimpleArraysValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_simple_arrays_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_simple_arrays_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_simple_arrays_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_simple_arrays_value.put( - input={"colors": ["str"], "counts": [0]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py deleted file mode 100644 index c19e8ce8aba..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithTextValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_text_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_text_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_text_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_text_value.put( - input={"content": "str", "language": "str"}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py deleted file mode 100644 index 65eab4fe6b5..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithTextValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_text_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_text_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_text_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_text_value.put( - input={"content": "str", "language": "str"}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py deleted file mode 100644 index 06dfa2d6fa9..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithUnwrappedArrayValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_model_with_unwrapped_array_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_unwrapped_array_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_model_with_unwrapped_array_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.model_with_unwrapped_array_value.put( - input={"colors": ["str"], "counts": [0]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py deleted file mode 100644 index 7057cddd2c0..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlModelWithUnwrappedArrayValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_unwrapped_array_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_unwrapped_array_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_model_with_unwrapped_array_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.model_with_unwrapped_array_value.put( - input={"colors": ["str"], "counts": [0]}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py deleted file mode 100644 index a8376712a5d..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import XmlClientTestBase, XmlPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlSimpleModelValueOperations(XmlClientTestBase): - @XmlPreparer() - @recorded_by_proxy - def test_simple_model_value_get(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.simple_model_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy - def test_simple_model_value_put(self, xml_endpoint): - client = self.create_client(endpoint=xml_endpoint) - response = client.simple_model_value.put( - input={"age": 0, "name": "str"}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py deleted file mode 100644 index 0c42bd7d059..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import XmlPreparer -from testpreparer_async import XmlClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestXmlSimpleModelValueOperationsAsync(XmlClientTestBaseAsync): - @XmlPreparer() - @recorded_by_proxy_async - async def test_simple_model_value_get(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.simple_model_value.get() - - # please add some check logic here by yourself - # ... - - @XmlPreparer() - @recorded_by_proxy_async - async def test_simple_model_value_put(self, xml_endpoint): - client = self.create_async_client(endpoint=xml_endpoint) - response = await client.simple_model_value.put( - input={"age": 0, "name": "str"}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py deleted file mode 100644 index 7470b7afb70..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from payload.xml import XmlClient - - -class XmlClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(XmlClient) - return self.create_client_from_credential( - XmlClient, - credential=credential, - endpoint=endpoint, - ) - - -XmlPreparer = functools.partial(PowerShellPreparer, "xml", xml_endpoint="https://fake_xml_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py deleted file mode 100644 index d993ba8306c..00000000000 --- a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from payload.xml.aio import XmlClient - - -class XmlClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(XmlClient, is_async=True) - return self.create_client_from_credential( - XmlClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/payload-xml/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-xml/payload/xml/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-xml/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-xml/payload/xml/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py deleted file mode 100644 index a2a87317d0a..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py +++ /dev/null @@ -1,47 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - resiliencyservicedriven_subscription_id = os.environ.get( - "RESILIENCYSERVICEDRIVEN_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - resiliencyservicedriven_tenant_id = os.environ.get( - "RESILIENCYSERVICEDRIVEN_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - resiliencyservicedriven_client_id = os.environ.get( - "RESILIENCYSERVICEDRIVEN_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - resiliencyservicedriven_client_secret = os.environ.get( - "RESILIENCYSERVICEDRIVEN_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer( - regex=resiliencyservicedriven_subscription_id, value="00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=resiliencyservicedriven_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=resiliencyservicedriven_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer( - regex=resiliencyservicedriven_client_secret, value="00000000-0000-0000-0000-000000000000" - ) - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py deleted file mode 100644 index 39eb11630ef..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py +++ /dev/null @@ -1,42 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ResiliencyServiceDrivenClientTestBase, ResiliencyServiceDrivenPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResiliencyServiceDriven(ResiliencyServiceDrivenClientTestBase): - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy - def test_from_none(self, resiliencyservicedriven_endpoint): - client = self.create_client(endpoint=resiliencyservicedriven_endpoint) - response = client.from_none() - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy - def test_from_one_required(self, resiliencyservicedriven_endpoint): - client = self.create_client(endpoint=resiliencyservicedriven_endpoint) - response = client.from_one_required( - parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy - def test_from_one_optional(self, resiliencyservicedriven_endpoint): - client = self.create_client(endpoint=resiliencyservicedriven_endpoint) - response = client.from_one_optional() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py deleted file mode 100644 index dd7d11a9487..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py +++ /dev/null @@ -1,43 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ResiliencyServiceDrivenPreparer -from testpreparer_async import ResiliencyServiceDrivenClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResiliencyServiceDrivenAsync(ResiliencyServiceDrivenClientTestBaseAsync): - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy_async - async def test_from_none(self, resiliencyservicedriven_endpoint): - client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) - response = await client.from_none() - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy_async - async def test_from_one_required(self, resiliencyservicedriven_endpoint): - client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) - response = await client.from_one_required( - parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy_async - async def test_from_one_optional(self, resiliencyservicedriven_endpoint): - client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) - response = await client.from_one_optional() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py deleted file mode 100644 index d657bd047dc..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from resiliency.srv.driven1 import ResiliencyServiceDrivenClient - - -class ResiliencyServiceDrivenClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ResiliencyServiceDrivenClient) - return self.create_client_from_credential( - ResiliencyServiceDrivenClient, - credential=credential, - endpoint=endpoint, - ) - - -ResiliencyServiceDrivenPreparer = functools.partial( - PowerShellPreparer, - "resiliencyservicedriven", - resiliencyservicedriven_endpoint="https://fake_resiliencyservicedriven_endpoint.com", -) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py deleted file mode 100644 index bb886d1ab0a..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from resiliency.srv.driven1.aio import ResiliencyServiceDrivenClient - - -class ResiliencyServiceDrivenClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ResiliencyServiceDrivenClient, is_async=True) - return self.create_client_from_credential( - ResiliencyServiceDrivenClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/resiliency-srv-driven1/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py deleted file mode 100644 index a2a87317d0a..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py +++ /dev/null @@ -1,47 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - resiliencyservicedriven_subscription_id = os.environ.get( - "RESILIENCYSERVICEDRIVEN_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - resiliencyservicedriven_tenant_id = os.environ.get( - "RESILIENCYSERVICEDRIVEN_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - resiliencyservicedriven_client_id = os.environ.get( - "RESILIENCYSERVICEDRIVEN_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - resiliencyservicedriven_client_secret = os.environ.get( - "RESILIENCYSERVICEDRIVEN_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer( - regex=resiliencyservicedriven_subscription_id, value="00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=resiliencyservicedriven_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=resiliencyservicedriven_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer( - regex=resiliencyservicedriven_client_secret, value="00000000-0000-0000-0000-000000000000" - ) - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py deleted file mode 100644 index a827df42fc3..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py +++ /dev/null @@ -1,51 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ResiliencyServiceDrivenClientTestBase, ResiliencyServiceDrivenPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResiliencyServiceDriven(ResiliencyServiceDrivenClientTestBase): - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy - def test_add_operation(self, resiliencyservicedriven_endpoint): - client = self.create_client(endpoint=resiliencyservicedriven_endpoint) - response = client.add_operation() - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy - def test_from_none(self, resiliencyservicedriven_endpoint): - client = self.create_client(endpoint=resiliencyservicedriven_endpoint) - response = client.from_none() - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy - def test_from_one_required(self, resiliencyservicedriven_endpoint): - client = self.create_client(endpoint=resiliencyservicedriven_endpoint) - response = client.from_one_required( - parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy - def test_from_one_optional(self, resiliencyservicedriven_endpoint): - client = self.create_client(endpoint=resiliencyservicedriven_endpoint) - response = client.from_one_optional() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py deleted file mode 100644 index 05d9bcc89f6..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py +++ /dev/null @@ -1,52 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ResiliencyServiceDrivenPreparer -from testpreparer_async import ResiliencyServiceDrivenClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestResiliencyServiceDrivenAsync(ResiliencyServiceDrivenClientTestBaseAsync): - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy_async - async def test_add_operation(self, resiliencyservicedriven_endpoint): - client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) - response = await client.add_operation() - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy_async - async def test_from_none(self, resiliencyservicedriven_endpoint): - client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) - response = await client.from_none() - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy_async - async def test_from_one_required(self, resiliencyservicedriven_endpoint): - client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) - response = await client.from_one_required( - parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @ResiliencyServiceDrivenPreparer() - @recorded_by_proxy_async - async def test_from_one_optional(self, resiliencyservicedriven_endpoint): - client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) - response = await client.from_one_optional() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py deleted file mode 100644 index 8ce298e478a..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from resiliency.srv.driven2 import ResiliencyServiceDrivenClient - - -class ResiliencyServiceDrivenClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ResiliencyServiceDrivenClient) - return self.create_client_from_credential( - ResiliencyServiceDrivenClient, - credential=credential, - endpoint=endpoint, - ) - - -ResiliencyServiceDrivenPreparer = functools.partial( - PowerShellPreparer, - "resiliencyservicedriven", - resiliencyservicedriven_endpoint="https://fake_resiliencyservicedriven_endpoint.com", -) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py deleted file mode 100644 index e94f4e5ea90..00000000000 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from resiliency.srv.driven2.aio import ResiliencyServiceDrivenClient - - -class ResiliencyServiceDrivenClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ResiliencyServiceDrivenClient, is_async=True) - return self.create_client_from_credential( - ResiliencyServiceDrivenClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/resiliency-srv-driven2/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py deleted file mode 100644 index 0416b7b5099..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - routes_subscription_id = os.environ.get("ROUTES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - routes_tenant_id = os.environ.get("ROUTES_TENANT_ID", "00000000-0000-0000-0000-000000000000") - routes_client_id = os.environ.get("ROUTES_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - routes_client_secret = os.environ.get("ROUTES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=routes_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=routes_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=routes_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=routes_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py deleted file mode 100644 index 76ea321422e..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RoutesClientTestBase, RoutesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutes(RoutesClientTestBase): - @RoutesPreparer() - @recorded_by_proxy - def test_fixed(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.fixed() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py deleted file mode 100644 index 705ab5d1955..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RoutesPreparer -from testpreparer_async import RoutesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutesAsync(RoutesClientTestBaseAsync): - @RoutesPreparer() - @recorded_by_proxy_async - async def test_fixed(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.fixed() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py deleted file mode 100644 index 4c2c9a7be14..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RoutesClientTestBase, RoutesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutesInInterfaceOperations(RoutesClientTestBase): - @RoutesPreparer() - @recorded_by_proxy - def test_in_interface_fixed(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.in_interface.fixed() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py deleted file mode 100644 index 1fac7cb10fe..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RoutesPreparer -from testpreparer_async import RoutesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutesInInterfaceOperationsAsync(RoutesClientTestBaseAsync): - @RoutesPreparer() - @recorded_by_proxy_async - async def test_in_interface_fixed(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.in_interface.fixed() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py deleted file mode 100644 index 20cf78b6765..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py +++ /dev/null @@ -1,332 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RoutesClientTestBase, RoutesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutesPathParametersOperations(RoutesClientTestBase): - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_template_only(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.template_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_explicit(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.explicit( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_annotation_only(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.annotation_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_reserved_expansion_template(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.reserved_expansion.template( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_reserved_expansion_annotation(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.reserved_expansion.annotation( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_simple_expansion_standard_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.simple_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_simple_expansion_standard_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.simple_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_simple_expansion_standard_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.simple_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_simple_expansion_explode_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.simple_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_simple_expansion_explode_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.simple_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_simple_expansion_explode_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.simple_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_path_expansion_standard_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.path_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_path_expansion_standard_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.path_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_path_expansion_standard_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.path_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_path_expansion_explode_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.path_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_path_expansion_explode_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.path_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_path_expansion_explode_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.path_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_label_expansion_standard_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.label_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_label_expansion_standard_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.label_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_label_expansion_standard_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.label_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_label_expansion_explode_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.label_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_label_expansion_explode_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.label_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_label_expansion_explode_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.label_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_matrix_expansion_standard_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.matrix_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_matrix_expansion_standard_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.matrix_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_matrix_expansion_standard_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.matrix_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_matrix_expansion_explode_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.matrix_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_matrix_expansion_explode_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.matrix_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_path_parameters_matrix_expansion_explode_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.path_parameters.matrix_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py deleted file mode 100644 index c73a227f87e..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py +++ /dev/null @@ -1,333 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RoutesPreparer -from testpreparer_async import RoutesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutesPathParametersOperationsAsync(RoutesClientTestBaseAsync): - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_template_only(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.template_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_explicit(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.explicit( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_annotation_only(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.annotation_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_reserved_expansion_template(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.reserved_expansion.template( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_reserved_expansion_annotation(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.reserved_expansion.annotation( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_simple_expansion_standard_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.simple_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_simple_expansion_standard_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.simple_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_simple_expansion_standard_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.simple_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_simple_expansion_explode_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.simple_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_simple_expansion_explode_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.simple_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_simple_expansion_explode_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.simple_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_path_expansion_standard_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.path_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_path_expansion_standard_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.path_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_path_expansion_standard_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.path_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_path_expansion_explode_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.path_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_path_expansion_explode_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.path_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_path_expansion_explode_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.path_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_label_expansion_standard_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.label_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_label_expansion_standard_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.label_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_label_expansion_standard_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.label_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_label_expansion_explode_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.label_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_label_expansion_explode_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.label_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_label_expansion_explode_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.label_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_matrix_expansion_standard_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.matrix_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_matrix_expansion_standard_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.matrix_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_matrix_expansion_standard_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.matrix_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_matrix_expansion_explode_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.matrix_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_matrix_expansion_explode_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.matrix_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_path_parameters_matrix_expansion_explode_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.path_parameters.matrix_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py deleted file mode 100644 index 566ff148af5..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py +++ /dev/null @@ -1,178 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RoutesClientTestBase, RoutesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutesQueryParametersOperations(RoutesClientTestBase): - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_template_only(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.template_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_explicit(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.explicit( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_annotation_only(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.annotation_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_expansion_standard_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_expansion_standard_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_expansion_standard_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_expansion_explode_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_expansion_explode_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_expansion_explode_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_continuation_standard_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_continuation.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_continuation_standard_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_continuation.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_continuation_standard_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_continuation.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_continuation_explode_primitive(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_continuation.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_continuation_explode_array(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_continuation.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy - def test_query_parameters_query_continuation_explode_record(self, routes_endpoint): - client = self.create_client(endpoint=routes_endpoint) - response = client.query_parameters.query_continuation.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py deleted file mode 100644 index c85cfc7ebd7..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py +++ /dev/null @@ -1,179 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RoutesPreparer -from testpreparer_async import RoutesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRoutesQueryParametersOperationsAsync(RoutesClientTestBaseAsync): - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_template_only(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.template_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_explicit(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.explicit( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_annotation_only(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.annotation_only( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_expansion_standard_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_expansion.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_expansion_standard_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_expansion.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_expansion_standard_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_expansion.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_expansion_explode_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_expansion.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_expansion_explode_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_expansion.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_expansion_explode_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_expansion.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_continuation_standard_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_continuation.standard.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_continuation_standard_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_continuation.standard.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_continuation_standard_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_continuation.standard.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_continuation_explode_primitive(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_continuation.explode.primitive( - param="str", - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_continuation_explode_array(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_continuation.explode.array( - param=["str"], - ) - - # please add some check logic here by yourself - # ... - - @RoutesPreparer() - @recorded_by_proxy_async - async def test_query_parameters_query_continuation_explode_record(self, routes_endpoint): - client = self.create_async_client(endpoint=routes_endpoint) - response = await client.query_parameters.query_continuation.explode.record( - param={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py deleted file mode 100644 index baf88af09ab..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from routes import RoutesClient - - -class RoutesClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(RoutesClient) - return self.create_client_from_credential( - RoutesClient, - credential=credential, - endpoint=endpoint, - ) - - -RoutesPreparer = functools.partial(PowerShellPreparer, "routes", routes_endpoint="https://fake_routes_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py deleted file mode 100644 index 3d66391fd7d..00000000000 --- a/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from routes.aio import RoutesClient - - -class RoutesClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(RoutesClient, is_async=True) - return self.create_client_from_credential( - RoutesClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/routes/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/routes/routes/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/routes/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/routes/routes/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py deleted file mode 100644 index 9e4e68059bf..00000000000 --- a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - json_subscription_id = os.environ.get("JSON_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - json_tenant_id = os.environ.get("JSON_TENANT_ID", "00000000-0000-0000-0000-000000000000") - json_client_id = os.environ.get("JSON_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - json_client_secret = os.environ.get("JSON_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=json_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=json_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=json_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=json_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py deleted file mode 100644 index ffa148b4644..00000000000 --- a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import JsonClientTestBase, JsonPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestJsonPropertyOperations(JsonClientTestBase): - @JsonPreparer() - @recorded_by_proxy - def test_property_send(self, json_endpoint): - client = self.create_client(endpoint=json_endpoint) - response = client.property.send( - body={"wireName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @JsonPreparer() - @recorded_by_proxy - def test_property_get(self, json_endpoint): - client = self.create_client(endpoint=json_endpoint) - response = client.property.get() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py deleted file mode 100644 index 44de344c6f5..00000000000 --- a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import JsonPreparer -from testpreparer_async import JsonClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestJsonPropertyOperationsAsync(JsonClientTestBaseAsync): - @JsonPreparer() - @recorded_by_proxy_async - async def test_property_send(self, json_endpoint): - client = self.create_async_client(endpoint=json_endpoint) - response = await client.property.send( - body={"wireName": bool}, - ) - - # please add some check logic here by yourself - # ... - - @JsonPreparer() - @recorded_by_proxy_async - async def test_property_get(self, json_endpoint): - client = self.create_async_client(endpoint=json_endpoint) - response = await client.property.get() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py deleted file mode 100644 index c451c83f0f7..00000000000 --- a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from serialization.encodedname.json import JsonClient - - -class JsonClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(JsonClient) - return self.create_client_from_credential( - JsonClient, - credential=credential, - endpoint=endpoint, - ) - - -JsonPreparer = functools.partial(PowerShellPreparer, "json", json_endpoint="https://fake_json_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py deleted file mode 100644 index ede52b11d00..00000000000 --- a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from serialization.encodedname.json.aio import JsonClient - - -class JsonClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(JsonClient, is_async=True) - return self.create_client_from_credential( - JsonClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/serialization-encoded-name-json/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py deleted file mode 100644 index 0ab89c65aa8..00000000000 --- a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - notdefined_subscription_id = os.environ.get("NOTDEFINED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - notdefined_tenant_id = os.environ.get("NOTDEFINED_TENANT_ID", "00000000-0000-0000-0000-000000000000") - notdefined_client_id = os.environ.get("NOTDEFINED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - notdefined_client_secret = os.environ.get("NOTDEFINED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notdefined_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notdefined_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notdefined_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notdefined_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py deleted file mode 100644 index d8ac72c4999..00000000000 --- a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NotDefinedClientTestBase, NotDefinedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNotDefined(NotDefinedClientTestBase): - @NotDefinedPreparer() - @recorded_by_proxy - def test_valid(self, notdefined_endpoint): - client = self.create_client(endpoint=notdefined_endpoint) - response = client.valid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py deleted file mode 100644 index debae5f767c..00000000000 --- a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NotDefinedPreparer -from testpreparer_async import NotDefinedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNotDefinedAsync(NotDefinedClientTestBaseAsync): - @NotDefinedPreparer() - @recorded_by_proxy_async - async def test_valid(self, notdefined_endpoint): - client = self.create_async_client(endpoint=notdefined_endpoint) - response = await client.valid() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py deleted file mode 100644 index d376459f692..00000000000 --- a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from server.endpoint.notdefined import NotDefinedClient - - -class NotDefinedClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(NotDefinedClient) - return self.create_client_from_credential( - NotDefinedClient, - credential=credential, - endpoint=endpoint, - ) - - -NotDefinedPreparer = functools.partial( - PowerShellPreparer, "notdefined", notdefined_endpoint="https://fake_notdefined_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py deleted file mode 100644 index 18259ec1835..00000000000 --- a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from server.endpoint.notdefined.aio import NotDefinedClient - - -class NotDefinedClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(NotDefinedClient, is_async=True) - return self.create_client_from_credential( - NotDefinedClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-endpoint-not-defined/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py deleted file mode 100644 index ecb5f89d885..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - multiple_subscription_id = os.environ.get("MULTIPLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - multiple_tenant_id = os.environ.get("MULTIPLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - multiple_client_id = os.environ.get("MULTIPLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - multiple_client_secret = os.environ.get("MULTIPLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multiple_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multiple_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multiple_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=multiple_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py deleted file mode 100644 index ff7fdbce871..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import MultipleClientTestBase, MultiplePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMultiple(MultipleClientTestBase): - @MultiplePreparer() - @recorded_by_proxy - def test_no_operation_params(self, multiple_endpoint): - client = self.create_client(endpoint=multiple_endpoint) - response = client.no_operation_params() - - # please add some check logic here by yourself - # ... - - @MultiplePreparer() - @recorded_by_proxy - def test_with_operation_path_param(self, multiple_endpoint): - client = self.create_client(endpoint=multiple_endpoint) - response = client.with_operation_path_param( - keyword="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py deleted file mode 100644 index af09dbc3f87..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import MultiplePreparer -from testpreparer_async import MultipleClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMultipleAsync(MultipleClientTestBaseAsync): - @MultiplePreparer() - @recorded_by_proxy_async - async def test_no_operation_params(self, multiple_endpoint): - client = self.create_async_client(endpoint=multiple_endpoint) - response = await client.no_operation_params() - - # please add some check logic here by yourself - # ... - - @MultiplePreparer() - @recorded_by_proxy_async - async def test_with_operation_path_param(self, multiple_endpoint): - client = self.create_async_client(endpoint=multiple_endpoint) - response = await client.with_operation_path_param( - keyword="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py deleted file mode 100644 index 058cc89c40c..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from server.path.multiple import MultipleClient - - -class MultipleClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(MultipleClient) - return self.create_client_from_credential( - MultipleClient, - credential=credential, - endpoint=endpoint, - ) - - -MultiplePreparer = functools.partial( - PowerShellPreparer, "multiple", multiple_endpoint="https://fake_multiple_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py deleted file mode 100644 index 3b5a7b2c131..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from server.path.multiple.aio import MultipleClient - - -class MultipleClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(MultipleClient, is_async=True) - return self.create_client_from_credential( - MultipleClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-path-multiple/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py deleted file mode 100644 index 2b689de1153..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - single_subscription_id = os.environ.get("SINGLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - single_tenant_id = os.environ.get("SINGLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - single_client_id = os.environ.get("SINGLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - single_client_secret = os.environ.get("SINGLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=single_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=single_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=single_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=single_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py deleted file mode 100644 index e6c45a6e182..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SingleClientTestBase, SinglePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSingle(SingleClientTestBase): - @SinglePreparer() - @recorded_by_proxy - def test_my_op(self, single_endpoint): - client = self.create_client(endpoint=single_endpoint) - response = client.my_op() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py deleted file mode 100644 index b841788ad37..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SinglePreparer -from testpreparer_async import SingleClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSingleAsync(SingleClientTestBaseAsync): - @SinglePreparer() - @recorded_by_proxy_async - async def test_my_op(self, single_endpoint): - client = self.create_async_client(endpoint=single_endpoint) - response = await client.my_op() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py deleted file mode 100644 index 37982e18354..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from server.path.single import SingleClient - - -class SingleClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(SingleClient) - return self.create_client_from_credential( - SingleClient, - credential=credential, - endpoint=endpoint, - ) - - -SinglePreparer = functools.partial(PowerShellPreparer, "single", single_endpoint="https://fake_single_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py deleted file mode 100644 index 131dc39a8ce..00000000000 --- a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from server.path.single.aio import SingleClient - - -class SingleClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(SingleClient, is_async=True) - return self.create_client_from_credential( - SingleClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/server-path-single/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-path-single/server/path/single/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-path-single/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-path-single/server/path/single/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py deleted file mode 100644 index 113408d780d..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - notversioned_subscription_id = os.environ.get( - "NOTVERSIONED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - notversioned_tenant_id = os.environ.get("NOTVERSIONED_TENANT_ID", "00000000-0000-0000-0000-000000000000") - notversioned_client_id = os.environ.get("NOTVERSIONED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - notversioned_client_secret = os.environ.get("NOTVERSIONED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notversioned_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notversioned_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notversioned_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notversioned_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py deleted file mode 100644 index 0168620cb05..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py +++ /dev/null @@ -1,44 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NotVersionedClientTestBase, NotVersionedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNotVersioned(NotVersionedClientTestBase): - @NotVersionedPreparer() - @recorded_by_proxy - def test_without_api_version(self, notversioned_endpoint): - client = self.create_client(endpoint=notversioned_endpoint) - response = client.without_api_version() - - # please add some check logic here by yourself - # ... - - @NotVersionedPreparer() - @recorded_by_proxy - def test_with_query_api_version(self, notversioned_endpoint): - client = self.create_client(endpoint=notversioned_endpoint) - response = client.with_query_api_version( - api_version="str", - ) - - # please add some check logic here by yourself - # ... - - @NotVersionedPreparer() - @recorded_by_proxy - def test_with_path_api_version(self, notversioned_endpoint): - client = self.create_client(endpoint=notversioned_endpoint) - response = client.with_path_api_version( - api_version="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py deleted file mode 100644 index 10167440b12..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NotVersionedPreparer -from testpreparer_async import NotVersionedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNotVersionedAsync(NotVersionedClientTestBaseAsync): - @NotVersionedPreparer() - @recorded_by_proxy_async - async def test_without_api_version(self, notversioned_endpoint): - client = self.create_async_client(endpoint=notversioned_endpoint) - response = await client.without_api_version() - - # please add some check logic here by yourself - # ... - - @NotVersionedPreparer() - @recorded_by_proxy_async - async def test_with_query_api_version(self, notversioned_endpoint): - client = self.create_async_client(endpoint=notversioned_endpoint) - response = await client.with_query_api_version( - api_version="str", - ) - - # please add some check logic here by yourself - # ... - - @NotVersionedPreparer() - @recorded_by_proxy_async - async def test_with_path_api_version(self, notversioned_endpoint): - client = self.create_async_client(endpoint=notversioned_endpoint) - response = await client.with_path_api_version( - api_version="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py deleted file mode 100644 index d60bf7abed6..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from server.versions.notversioned import NotVersionedClient - - -class NotVersionedClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(NotVersionedClient) - return self.create_client_from_credential( - NotVersionedClient, - credential=credential, - endpoint=endpoint, - ) - - -NotVersionedPreparer = functools.partial( - PowerShellPreparer, "notversioned", notversioned_endpoint="https://fake_notversioned_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py deleted file mode 100644 index 4d083176898..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from server.versions.notversioned.aio import NotVersionedClient - - -class NotVersionedClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(NotVersionedClient, is_async=True) - return self.create_client_from_credential( - NotVersionedClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-versions-not-versioned/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py deleted file mode 100644 index b21f87d9301..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - versioned_subscription_id = os.environ.get("VERSIONED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - versioned_tenant_id = os.environ.get("VERSIONED_TENANT_ID", "00000000-0000-0000-0000-000000000000") - versioned_client_id = os.environ.get("VERSIONED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - versioned_client_secret = os.environ.get("VERSIONED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=versioned_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=versioned_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=versioned_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=versioned_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py deleted file mode 100644 index c3377064708..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py +++ /dev/null @@ -1,49 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import VersionedClientTestBase, VersionedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVersioned(VersionedClientTestBase): - @VersionedPreparer() - @recorded_by_proxy - def test_without_api_version(self, versioned_endpoint): - client = self.create_client(endpoint=versioned_endpoint) - response = client.without_api_version() - - # please add some check logic here by yourself - # ... - - @VersionedPreparer() - @recorded_by_proxy - def test_with_query_api_version(self, versioned_endpoint): - client = self.create_client(endpoint=versioned_endpoint) - response = client.with_query_api_version() - - # please add some check logic here by yourself - # ... - - @VersionedPreparer() - @recorded_by_proxy - def test_with_path_api_version(self, versioned_endpoint): - client = self.create_client(endpoint=versioned_endpoint) - response = client.with_path_api_version() - - # please add some check logic here by yourself - # ... - - @VersionedPreparer() - @recorded_by_proxy - def test_with_query_old_api_version(self, versioned_endpoint): - client = self.create_client(endpoint=versioned_endpoint) - response = client.with_query_old_api_version() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py deleted file mode 100644 index b5a2f83d581..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py +++ /dev/null @@ -1,50 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import VersionedPreparer -from testpreparer_async import VersionedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVersionedAsync(VersionedClientTestBaseAsync): - @VersionedPreparer() - @recorded_by_proxy_async - async def test_without_api_version(self, versioned_endpoint): - client = self.create_async_client(endpoint=versioned_endpoint) - response = await client.without_api_version() - - # please add some check logic here by yourself - # ... - - @VersionedPreparer() - @recorded_by_proxy_async - async def test_with_query_api_version(self, versioned_endpoint): - client = self.create_async_client(endpoint=versioned_endpoint) - response = await client.with_query_api_version() - - # please add some check logic here by yourself - # ... - - @VersionedPreparer() - @recorded_by_proxy_async - async def test_with_path_api_version(self, versioned_endpoint): - client = self.create_async_client(endpoint=versioned_endpoint) - response = await client.with_path_api_version() - - # please add some check logic here by yourself - # ... - - @VersionedPreparer() - @recorded_by_proxy_async - async def test_with_query_old_api_version(self, versioned_endpoint): - client = self.create_async_client(endpoint=versioned_endpoint) - response = await client.with_query_old_api_version() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py deleted file mode 100644 index 8d7f9d4965f..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from server.versions.versioned import VersionedClient - - -class VersionedClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(VersionedClient) - return self.create_client_from_credential( - VersionedClient, - credential=credential, - endpoint=endpoint, - ) - - -VersionedPreparer = functools.partial( - PowerShellPreparer, "versioned", versioned_endpoint="https://fake_versioned_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py deleted file mode 100644 index b0610e33752..00000000000 --- a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from server.versions.versioned.aio import VersionedClient - - -class VersionedClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(VersionedClient, is_async=True) - return self.create_client_from_credential( - VersionedClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-versions-versioned/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py deleted file mode 100644 index 732c2c5da0d..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py +++ /dev/null @@ -1,43 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - conditionalrequest_subscription_id = os.environ.get( - "CONDITIONALREQUEST_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - conditionalrequest_tenant_id = os.environ.get( - "CONDITIONALREQUEST_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - conditionalrequest_client_id = os.environ.get( - "CONDITIONALREQUEST_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - conditionalrequest_client_secret = os.environ.get( - "CONDITIONALREQUEST_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=conditionalrequest_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=conditionalrequest_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=conditionalrequest_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=conditionalrequest_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py deleted file mode 100644 index 3850898ec82..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py +++ /dev/null @@ -1,49 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ConditionalRequestClientTestBase, ConditionalRequestPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestConditionalRequest(ConditionalRequestClientTestBase): - @ConditionalRequestPreparer() - @recorded_by_proxy - def test_post_if_match(self, conditionalrequest_endpoint): - client = self.create_client(endpoint=conditionalrequest_endpoint) - response = client.post_if_match() - - # please add some check logic here by yourself - # ... - - @ConditionalRequestPreparer() - @recorded_by_proxy - def test_post_if_none_match(self, conditionalrequest_endpoint): - client = self.create_client(endpoint=conditionalrequest_endpoint) - response = client.post_if_none_match() - - # please add some check logic here by yourself - # ... - - @ConditionalRequestPreparer() - @recorded_by_proxy - def test_head_if_modified_since(self, conditionalrequest_endpoint): - client = self.create_client(endpoint=conditionalrequest_endpoint) - response = client.head_if_modified_since() - - # please add some check logic here by yourself - # ... - - @ConditionalRequestPreparer() - @recorded_by_proxy - def test_post_if_unmodified_since(self, conditionalrequest_endpoint): - client = self.create_client(endpoint=conditionalrequest_endpoint) - response = client.post_if_unmodified_since() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py deleted file mode 100644 index 714becb95a0..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py +++ /dev/null @@ -1,50 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ConditionalRequestPreparer -from testpreparer_async import ConditionalRequestClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestConditionalRequestAsync(ConditionalRequestClientTestBaseAsync): - @ConditionalRequestPreparer() - @recorded_by_proxy_async - async def test_post_if_match(self, conditionalrequest_endpoint): - client = self.create_async_client(endpoint=conditionalrequest_endpoint) - response = await client.post_if_match() - - # please add some check logic here by yourself - # ... - - @ConditionalRequestPreparer() - @recorded_by_proxy_async - async def test_post_if_none_match(self, conditionalrequest_endpoint): - client = self.create_async_client(endpoint=conditionalrequest_endpoint) - response = await client.post_if_none_match() - - # please add some check logic here by yourself - # ... - - @ConditionalRequestPreparer() - @recorded_by_proxy_async - async def test_head_if_modified_since(self, conditionalrequest_endpoint): - client = self.create_async_client(endpoint=conditionalrequest_endpoint) - response = await client.head_if_modified_since() - - # please add some check logic here by yourself - # ... - - @ConditionalRequestPreparer() - @recorded_by_proxy_async - async def test_post_if_unmodified_since(self, conditionalrequest_endpoint): - client = self.create_async_client(endpoint=conditionalrequest_endpoint) - response = await client.post_if_unmodified_since() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py deleted file mode 100644 index 147f64006bd..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specialheaders.conditionalrequest import ConditionalRequestClient - - -class ConditionalRequestClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ConditionalRequestClient) - return self.create_client_from_credential( - ConditionalRequestClient, - credential=credential, - endpoint=endpoint, - ) - - -ConditionalRequestPreparer = functools.partial( - PowerShellPreparer, "conditionalrequest", conditionalrequest_endpoint="https://fake_conditionalrequest_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py deleted file mode 100644 index 9d23fc53553..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specialheaders.conditionalrequest.aio import ConditionalRequestClient - - -class ConditionalRequestClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ConditionalRequestClient, is_async=True) - return self.create_client_from_credential( - ConditionalRequestClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/special-headers-conditional-request/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py deleted file mode 100644 index 420e0fefcb5..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - repeatability_subscription_id = os.environ.get( - "REPEATABILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - repeatability_tenant_id = os.environ.get("REPEATABILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - repeatability_client_id = os.environ.get("REPEATABILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - repeatability_client_secret = os.environ.get("REPEATABILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=repeatability_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=repeatability_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=repeatability_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=repeatability_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py deleted file mode 100644 index f90fb946847..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py +++ /dev/null @@ -1,22 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RepeatabilityClientTestBase, RepeatabilityPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRepeatability(RepeatabilityClientTestBase): - @RepeatabilityPreparer() - @recorded_by_proxy - def test_immediate_success(self, repeatability_endpoint): - client = self.create_client(endpoint=repeatability_endpoint) - response = client.immediate_success() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py deleted file mode 100644 index 96a827ca7de..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py +++ /dev/null @@ -1,23 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RepeatabilityPreparer -from testpreparer_async import RepeatabilityClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRepeatabilityAsync(RepeatabilityClientTestBaseAsync): - @RepeatabilityPreparer() - @recorded_by_proxy_async - async def test_immediate_success(self, repeatability_endpoint): - client = self.create_async_client(endpoint=repeatability_endpoint) - response = await client.immediate_success() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py deleted file mode 100644 index 20f2aa53457..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specialheaders.repeatability import RepeatabilityClient - - -class RepeatabilityClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(RepeatabilityClient) - return self.create_client_from_credential( - RepeatabilityClient, - credential=credential, - endpoint=endpoint, - ) - - -RepeatabilityPreparer = functools.partial( - PowerShellPreparer, "repeatability", repeatability_endpoint="https://fake_repeatability_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py deleted file mode 100644 index 1aee05e0694..00000000000 --- a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specialheaders.repeatability.aio import RepeatabilityClient - - -class RepeatabilityClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(RepeatabilityClient, is_async=True) - return self.create_client_from_credential( - RepeatabilityClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/special-headers-repeatability/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py deleted file mode 100644 index 299af9f3cf3..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - specialwords_subscription_id = os.environ.get( - "SPECIALWORDS_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - specialwords_tenant_id = os.environ.get("SPECIALWORDS_TENANT_ID", "00000000-0000-0000-0000-000000000000") - specialwords_client_id = os.environ.get("SPECIALWORDS_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - specialwords_client_secret = os.environ.get("SPECIALWORDS_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=specialwords_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=specialwords_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=specialwords_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=specialwords_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py deleted file mode 100644 index b0d93f73c52..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsModelPropertiesOperations(SpecialWordsClientTestBase): - @SpecialWordsPreparer() - @recorded_by_proxy - def test_model_properties_same_as_model(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.model_properties.same_as_model( - body={"SameAsModel": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py deleted file mode 100644 index fe0b9e2887b..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SpecialWordsPreparer -from testpreparer_async import SpecialWordsClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsModelPropertiesOperationsAsync(SpecialWordsClientTestBaseAsync): - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_model_properties_same_as_model(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.model_properties.same_as_model( - body={"SameAsModel": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py deleted file mode 100644 index 1e4b1cce356..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py +++ /dev/null @@ -1,376 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsModelsOperations(SpecialWordsClientTestBase): - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_and(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_and( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_as(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_as( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_assert(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_assert( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_async(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_async( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_await(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_await( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_break(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_break( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_class(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_class( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_constructor(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_constructor( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_continue(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_continue( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_def(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_def( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_del(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_del( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_elif(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_elif( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_else(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_else( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_except(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_except( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_exec(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_exec( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_finally(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_finally( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_for(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_for( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_from(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_from( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_global(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_global( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_if(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_if( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_import(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_import( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_in(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_in( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_is(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_is( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_lambda(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_lambda( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_not(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_not( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_or(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_or( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_pass(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_pass( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_raise(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_raise( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_return(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_return( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_try(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_try( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_while(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_while( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_with(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_with( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_models_with_yield(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.models.with_yield( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py deleted file mode 100644 index 36c089dc3a4..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py +++ /dev/null @@ -1,377 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SpecialWordsPreparer -from testpreparer_async import SpecialWordsClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsModelsOperationsAsync(SpecialWordsClientTestBaseAsync): - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_and(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_and( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_as(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_as( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_assert(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_assert( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_async(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_async( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_await(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_await( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_break(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_break( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_class(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_class( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_constructor(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_constructor( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_continue(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_continue( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_def(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_def( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_del(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_del( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_elif(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_elif( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_else(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_else( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_except(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_except( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_exec(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_exec( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_finally(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_finally( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_for(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_for( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_from(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_from( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_global(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_global( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_if(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_if( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_import(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_import( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_in(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_in( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_is(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_is( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_lambda(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_lambda( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_not(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_not( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_or(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_or( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_pass(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_pass( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_raise(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_raise( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_return(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_return( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_try(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_try( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_while(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_while( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_with(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_with( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_models_with_yield(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.models.with_yield( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py deleted file mode 100644 index e73807af358..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py +++ /dev/null @@ -1,310 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsOperations(SpecialWordsClientTestBase): - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_and_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.and_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_as_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.as_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_assert_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.assert_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_async_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.async_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_await_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.await_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_break_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.break_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_class_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.class_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_constructor(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.constructor() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_continue_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.continue_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_def_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.def_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_del_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.del_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_elif_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.elif_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_else_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.else_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_except_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.except_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_exec_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.exec_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_finally_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.finally_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_for_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.for_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_from_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.from_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_global_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.global_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_if_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.if_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_import_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.import_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_in_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.in_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_is_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.is_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_lambda_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.lambda_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_not_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.not_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_or_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.or_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_pass_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.pass_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_raise_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.raise_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_return_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.return_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_try_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.try_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_while_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.while_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_with_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.with_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_operations_yield_method(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.operations.yield_method() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py deleted file mode 100644 index 1404d0b1ef8..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py +++ /dev/null @@ -1,311 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SpecialWordsPreparer -from testpreparer_async import SpecialWordsClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsOperationsAsync(SpecialWordsClientTestBaseAsync): - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_and_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.and_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_as_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.as_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_assert_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.assert_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_async_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.async_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_await_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.await_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_break_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.break_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_class_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.class_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_constructor(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.constructor() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_continue_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.continue_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_def_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.def_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_del_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.del_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_elif_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.elif_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_else_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.else_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_except_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.except_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_exec_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.exec_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_finally_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.finally_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_for_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.for_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_from_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.from_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_global_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.global_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_if_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.if_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_import_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.import_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_in_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.in_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_is_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.is_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_lambda_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.lambda_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_not_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.not_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_or_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.or_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_pass_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.pass_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_raise_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.raise_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_return_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.return_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_try_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.try_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_while_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.while_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_with_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.with_method() - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_operations_yield_method(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.operations.yield_method() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py deleted file mode 100644 index 3096a60a28e..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py +++ /dev/null @@ -1,387 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsParametersOperations(SpecialWordsClientTestBase): - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_and(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_and( - and_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_as(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_as( - as_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_assert(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_assert( - assert_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_async(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_async( - async_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_await(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_await( - await_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_break(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_break( - break_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_class(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_class( - class_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_constructor(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_constructor( - constructor="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_continue(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_continue( - continue_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_def(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_def( - def_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_del(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_del( - del_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_elif(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_elif( - elif_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_else(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_else( - else_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_except(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_except( - except_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_exec(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_exec( - exec_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_finally(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_finally( - finally_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_for(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_for( - for_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_from(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_from( - from_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_global(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_global( - global_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_if(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_if( - if_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_import(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_import( - import_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_in(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_in( - in_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_is(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_is( - is_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_lambda(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_lambda( - lambda_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_not(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_not( - not_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_or(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_or( - or_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_pass(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_pass( - pass_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_raise(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_raise( - raise_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_return(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_return( - return_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_try(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_try( - try_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_while(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_while( - while_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_with(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_with( - with_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_yield(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_yield( - yield_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy - def test_parameters_with_cancellation_token(self, specialwords_endpoint): - client = self.create_client(endpoint=specialwords_endpoint) - response = client.parameters.with_cancellation_token( - cancellation_token="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py deleted file mode 100644 index 94ca98849b1..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py +++ /dev/null @@ -1,388 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SpecialWordsPreparer -from testpreparer_async import SpecialWordsClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSpecialWordsParametersOperationsAsync(SpecialWordsClientTestBaseAsync): - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_and(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_and( - and_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_as(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_as( - as_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_assert(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_assert( - assert_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_async(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_async( - async_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_await(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_await( - await_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_break(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_break( - break_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_class(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_class( - class_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_constructor(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_constructor( - constructor="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_continue(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_continue( - continue_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_def(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_def( - def_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_del(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_del( - del_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_elif(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_elif( - elif_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_else(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_else( - else_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_except(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_except( - except_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_exec(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_exec( - exec_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_finally(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_finally( - finally_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_for(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_for( - for_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_from(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_from( - from_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_global(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_global( - global_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_if(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_if( - if_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_import(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_import( - import_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_in(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_in( - in_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_is(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_is( - is_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_lambda(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_lambda( - lambda_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_not(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_not( - not_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_or(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_or( - or_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_pass(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_pass( - pass_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_raise(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_raise( - raise_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_return(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_return( - return_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_try(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_try( - try_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_while(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_while( - while_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_with(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_with( - with_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_yield(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_yield( - yield_parameter="str", - ) - - # please add some check logic here by yourself - # ... - - @SpecialWordsPreparer() - @recorded_by_proxy_async - async def test_parameters_with_cancellation_token(self, specialwords_endpoint): - client = self.create_async_client(endpoint=specialwords_endpoint) - response = await client.parameters.with_cancellation_token( - cancellation_token="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py deleted file mode 100644 index 09070149dec..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from specialwords import SpecialWordsClient - - -class SpecialWordsClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(SpecialWordsClient) - return self.create_client_from_credential( - SpecialWordsClient, - credential=credential, - endpoint=endpoint, - ) - - -SpecialWordsPreparer = functools.partial( - PowerShellPreparer, "specialwords", specialwords_endpoint="https://fake_specialwords_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py deleted file mode 100644 index 877ce52b4e9..00000000000 --- a/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from specialwords.aio import SpecialWordsClient - - -class SpecialWordsClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(SpecialWordsClient, is_async=True) - return self.create_client_from_credential( - SpecialWordsClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/special-words/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/special-words/specialwords/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/special-words/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/special-words/specialwords/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py deleted file mode 100644 index 76591609ab0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - array_subscription_id = os.environ.get("ARRAY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - array_tenant_id = os.environ.get("ARRAY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - array_client_id = os.environ.get("ARRAY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - array_client_secret = os.environ.get("ARRAY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=array_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=array_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=array_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=array_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py deleted file mode 100644 index a92d8a1de47..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayBooleanValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_boolean_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.boolean_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_boolean_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.boolean_value.put( - body=[bool], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py deleted file mode 100644 index dbb00a2f7d6..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayBooleanValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_boolean_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.boolean_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_boolean_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.boolean_value.put( - body=[bool], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py deleted file mode 100644 index 51c85df3067..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayDatetimeValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_datetime_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.datetime_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_datetime_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.datetime_value.put( - body=["2020-02-20 00:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py deleted file mode 100644 index c5143d890c3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayDatetimeValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_datetime_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.datetime_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_datetime_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.datetime_value.put( - body=["2020-02-20 00:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py deleted file mode 100644 index e5f5d6f5398..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayDurationValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_duration_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.duration_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_duration_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.duration_value.put( - body=["1 day, 0:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py deleted file mode 100644 index d6dca6b97bb..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayDurationValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_duration_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.duration_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_duration_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.duration_value.put( - body=["1 day, 0:00:00"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py deleted file mode 100644 index 7667d59ab18..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayFloat32ValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_float32_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.float32_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_float32_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.float32_value.put( - body=[0.0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py deleted file mode 100644 index 32d2941545f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayFloat32ValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_float32_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.float32_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_float32_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.float32_value.put( - body=[0.0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py deleted file mode 100644 index 5d3d8e7b084..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayInt32ValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_int32_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.int32_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_int32_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.int32_value.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py deleted file mode 100644 index 945de66811c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayInt32ValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_int32_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.int32_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_int32_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.int32_value.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py deleted file mode 100644 index f7e0c27b1c4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayInt64ValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_int64_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.int64_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_int64_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.int64_value.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py deleted file mode 100644 index 0a65a3409bc..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayInt64ValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_int64_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.int64_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_int64_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.int64_value.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py deleted file mode 100644 index 902a16cc99b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayModelValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_model_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.model_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_model_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.model_value.put( - body=[{"property": "str", "children": [...]}], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py deleted file mode 100644 index 3f3fd3c8302..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayModelValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_model_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.model_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_model_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.model_value.put( - body=[{"property": "str", "children": [...]}], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py deleted file mode 100644 index 424edfcac74..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableBooleanValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_boolean_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_boolean_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_boolean_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_boolean_value.put( - body=[bool], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py deleted file mode 100644 index 00f98afecd6..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableBooleanValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_boolean_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_boolean_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_boolean_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_boolean_value.put( - body=[bool], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py deleted file mode 100644 index 22e6eb6f081..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableFloatValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_float_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_float_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_float_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_float_value.put( - body=[0.0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py deleted file mode 100644 index 51eb6323a14..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableFloatValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_float_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_float_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_float_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_float_value.put( - body=[0.0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py deleted file mode 100644 index b15da9d989f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableInt32ValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_int32_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_int32_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_int32_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_int32_value.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py deleted file mode 100644 index 99ab79992ad..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableInt32ValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_int32_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_int32_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_int32_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_int32_value.put( - body=[0], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py deleted file mode 100644 index 5af02b802ea..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableModelValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_model_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_model_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_model_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_model_value.put( - body=[{"property": "str", "children": [...]}], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py deleted file mode 100644 index add8c12e990..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableModelValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_model_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_model_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_model_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_model_value.put( - body=[{"property": "str", "children": [...]}], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py deleted file mode 100644 index 6f822baef0c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableStringValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_string_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_string_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_nullable_string_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.nullable_string_value.put( - body=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py deleted file mode 100644 index 872997efda8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayNullableStringValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_string_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_string_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_nullable_string_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.nullable_string_value.put( - body=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py deleted file mode 100644 index 88918304f26..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayStringValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_string_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.string_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_string_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.string_value.put( - body=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py deleted file mode 100644 index fd99dcd54fc..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayStringValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_string_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.string_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_string_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.string_value.put( - body=["str"], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py deleted file mode 100644 index 2a018ba6fa0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ArrayClientTestBase, ArrayPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayUnknownValueOperations(ArrayClientTestBase): - @ArrayPreparer() - @recorded_by_proxy - def test_unknown_value_get(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.unknown_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy - def test_unknown_value_put(self, array_endpoint): - client = self.create_client(endpoint=array_endpoint) - response = client.unknown_value.put( - body=[{}], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py deleted file mode 100644 index ffbc2d2c7df..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ArrayPreparer -from testpreparer_async import ArrayClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestArrayUnknownValueOperationsAsync(ArrayClientTestBaseAsync): - @ArrayPreparer() - @recorded_by_proxy_async - async def test_unknown_value_get(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.unknown_value.get() - - # please add some check logic here by yourself - # ... - - @ArrayPreparer() - @recorded_by_proxy_async - async def test_unknown_value_put(self, array_endpoint): - client = self.create_async_client(endpoint=array_endpoint) - response = await client.unknown_value.put( - body=[{}], - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py deleted file mode 100644 index 7e657f1d6f7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.array import ArrayClient - - -class ArrayClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ArrayClient) - return self.create_client_from_credential( - ArrayClient, - credential=credential, - endpoint=endpoint, - ) - - -ArrayPreparer = functools.partial(PowerShellPreparer, "array", array_endpoint="https://fake_array_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py deleted file mode 100644 index b0e2f641713..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.array.aio import ArrayClient - - -class ArrayClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ArrayClient, is_async=True) - return self.create_client_from_credential( - ArrayClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-array/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-array/typetest/array/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-array/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-array/typetest/array/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py deleted file mode 100644 index 694a0f94554..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - dictionary_subscription_id = os.environ.get("DICTIONARY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - dictionary_tenant_id = os.environ.get("DICTIONARY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - dictionary_client_id = os.environ.get("DICTIONARY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - dictionary_client_secret = os.environ.get("DICTIONARY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=dictionary_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=dictionary_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=dictionary_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=dictionary_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py deleted file mode 100644 index 066f228c89a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryBooleanValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_boolean_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.boolean_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_boolean_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.boolean_value.put( - body={"str": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py deleted file mode 100644 index ea0403b733d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryBooleanValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_boolean_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.boolean_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_boolean_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.boolean_value.put( - body={"str": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py deleted file mode 100644 index 30fb62304ce..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryDatetimeValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_datetime_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.datetime_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_datetime_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.datetime_value.put( - body={"str": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py deleted file mode 100644 index 3e83eefc827..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryDatetimeValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_datetime_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.datetime_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_datetime_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.datetime_value.put( - body={"str": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py deleted file mode 100644 index b0b3e702188..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryDurationValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_duration_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.duration_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_duration_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.duration_value.put( - body={"str": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py deleted file mode 100644 index 0d5394008dd..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryDurationValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_duration_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.duration_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_duration_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.duration_value.put( - body={"str": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py deleted file mode 100644 index 859239e7348..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryFloat32ValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_float32_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.float32_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_float32_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.float32_value.put( - body={"str": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py deleted file mode 100644 index a52a03e857f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryFloat32ValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_float32_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.float32_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_float32_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.float32_value.put( - body={"str": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py deleted file mode 100644 index a2199a45778..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryInt32ValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_int32_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.int32_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_int32_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.int32_value.put( - body={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py deleted file mode 100644 index 77d4bb4577d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryInt32ValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_int32_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.int32_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_int32_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.int32_value.put( - body={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py deleted file mode 100644 index 6f7322982d7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryInt64ValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_int64_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.int64_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_int64_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.int64_value.put( - body={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py deleted file mode 100644 index 12b6e15ef77..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryInt64ValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_int64_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.int64_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_int64_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.int64_value.put( - body={"str": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py deleted file mode 100644 index 040afe8e20b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryModelValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_model_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.model_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_model_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.model_value.put( - body={"str": {"property": "str", "children": {"str": ...}}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py deleted file mode 100644 index 54fbc74db5b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryModelValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_model_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.model_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_model_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.model_value.put( - body={"str": {"property": "str", "children": {"str": ...}}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py deleted file mode 100644 index 3f17f0f6982..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryNullableFloatValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_nullable_float_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.nullable_float_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_nullable_float_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.nullable_float_value.put( - body={"str": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py deleted file mode 100644 index 1e8f5f07b35..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryNullableFloatValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_nullable_float_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.nullable_float_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_nullable_float_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.nullable_float_value.put( - body={"str": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py deleted file mode 100644 index fe00987385c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryRecursiveModelValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_recursive_model_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.recursive_model_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_recursive_model_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.recursive_model_value.put( - body={"str": {"property": "str", "children": {"str": ...}}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py deleted file mode 100644 index 63ed9c3ca9f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryRecursiveModelValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_recursive_model_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.recursive_model_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_recursive_model_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.recursive_model_value.put( - body={"str": {"property": "str", "children": {"str": ...}}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py deleted file mode 100644 index 6309558db05..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryStringValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_string_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.string_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_string_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.string_value.put( - body={"str": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py deleted file mode 100644 index 91fdb59db9f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryStringValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_string_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.string_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_string_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.string_value.put( - body={"str": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py deleted file mode 100644 index e84f0564e91..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import DictionaryClientTestBase, DictionaryPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryUnknownValueOperations(DictionaryClientTestBase): - @DictionaryPreparer() - @recorded_by_proxy - def test_unknown_value_get(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.unknown_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy - def test_unknown_value_put(self, dictionary_endpoint): - client = self.create_client(endpoint=dictionary_endpoint) - response = client.unknown_value.put( - body={"str": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py deleted file mode 100644 index 6bd7a2bf28a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import DictionaryPreparer -from testpreparer_async import DictionaryClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestDictionaryUnknownValueOperationsAsync(DictionaryClientTestBaseAsync): - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_unknown_value_get(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.unknown_value.get() - - # please add some check logic here by yourself - # ... - - @DictionaryPreparer() - @recorded_by_proxy_async - async def test_unknown_value_put(self, dictionary_endpoint): - client = self.create_async_client(endpoint=dictionary_endpoint) - response = await client.unknown_value.put( - body={"str": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py deleted file mode 100644 index 8f9f9085cc3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.dictionary import DictionaryClient - - -class DictionaryClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(DictionaryClient) - return self.create_client_from_credential( - DictionaryClient, - credential=credential, - endpoint=endpoint, - ) - - -DictionaryPreparer = functools.partial( - PowerShellPreparer, "dictionary", dictionary_endpoint="https://fake_dictionary_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py deleted file mode 100644 index d6149b7d469..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.dictionary.aio import DictionaryClient - - -class DictionaryClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(DictionaryClient, is_async=True) - return self.create_client_from_credential( - DictionaryClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-dictionary/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py deleted file mode 100644 index ef77f38e7b8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - extensible_subscription_id = os.environ.get("EXTENSIBLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - extensible_tenant_id = os.environ.get("EXTENSIBLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - extensible_client_id = os.environ.get("EXTENSIBLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - extensible_client_secret = os.environ.get("EXTENSIBLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=extensible_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=extensible_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=extensible_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=extensible_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py deleted file mode 100644 index be910b162cb..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py +++ /dev/null @@ -1,55 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ExtensibleClientTestBase, ExtensiblePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestExtensibleStringOperations(ExtensibleClientTestBase): - @ExtensiblePreparer() - @recorded_by_proxy - def test_string_get_known_value(self, extensible_endpoint): - client = self.create_client(endpoint=extensible_endpoint) - response = client.string.get_known_value() - - # please add some check logic here by yourself - # ... - - @ExtensiblePreparer() - @recorded_by_proxy - def test_string_get_unknown_value(self, extensible_endpoint): - client = self.create_client(endpoint=extensible_endpoint) - response = client.string.get_unknown_value() - - # please add some check logic here by yourself - # ... - - @ExtensiblePreparer() - @recorded_by_proxy - def test_string_put_known_value(self, extensible_endpoint): - client = self.create_client(endpoint=extensible_endpoint) - response = client.string.put_known_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ExtensiblePreparer() - @recorded_by_proxy - def test_string_put_unknown_value(self, extensible_endpoint): - client = self.create_client(endpoint=extensible_endpoint) - response = client.string.put_unknown_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py deleted file mode 100644 index 3e12585bf55..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py +++ /dev/null @@ -1,56 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ExtensiblePreparer -from testpreparer_async import ExtensibleClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestExtensibleStringOperationsAsync(ExtensibleClientTestBaseAsync): - @ExtensiblePreparer() - @recorded_by_proxy_async - async def test_string_get_known_value(self, extensible_endpoint): - client = self.create_async_client(endpoint=extensible_endpoint) - response = await client.string.get_known_value() - - # please add some check logic here by yourself - # ... - - @ExtensiblePreparer() - @recorded_by_proxy_async - async def test_string_get_unknown_value(self, extensible_endpoint): - client = self.create_async_client(endpoint=extensible_endpoint) - response = await client.string.get_unknown_value() - - # please add some check logic here by yourself - # ... - - @ExtensiblePreparer() - @recorded_by_proxy_async - async def test_string_put_known_value(self, extensible_endpoint): - client = self.create_async_client(endpoint=extensible_endpoint) - response = await client.string.put_known_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ExtensiblePreparer() - @recorded_by_proxy_async - async def test_string_put_unknown_value(self, extensible_endpoint): - client = self.create_async_client(endpoint=extensible_endpoint) - response = await client.string.put_unknown_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py deleted file mode 100644 index 67cc900aa4c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.enum.extensible import ExtensibleClient - - -class ExtensibleClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ExtensibleClient) - return self.create_client_from_credential( - ExtensibleClient, - credential=credential, - endpoint=endpoint, - ) - - -ExtensiblePreparer = functools.partial( - PowerShellPreparer, "extensible", extensible_endpoint="https://fake_extensible_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py deleted file mode 100644 index 2ffefb6d6b1..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.enum.extensible.aio import ExtensibleClient - - -class ExtensibleClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ExtensibleClient, is_async=True) - return self.create_client_from_credential( - ExtensibleClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-enum-extensible/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py deleted file mode 100644 index 4dd457b05f4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - fixed_subscription_id = os.environ.get("FIXED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - fixed_tenant_id = os.environ.get("FIXED_TENANT_ID", "00000000-0000-0000-0000-000000000000") - fixed_client_id = os.environ.get("FIXED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - fixed_client_secret = os.environ.get("FIXED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=fixed_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=fixed_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=fixed_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=fixed_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py deleted file mode 100644 index 01f766ccb9c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py +++ /dev/null @@ -1,46 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import FixedClientTestBase, FixedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestFixedStringOperations(FixedClientTestBase): - @FixedPreparer() - @recorded_by_proxy - def test_string_get_known_value(self, fixed_endpoint): - client = self.create_client(endpoint=fixed_endpoint) - response = client.string.get_known_value() - - # please add some check logic here by yourself - # ... - - @FixedPreparer() - @recorded_by_proxy - def test_string_put_known_value(self, fixed_endpoint): - client = self.create_client(endpoint=fixed_endpoint) - response = client.string.put_known_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @FixedPreparer() - @recorded_by_proxy - def test_string_put_unknown_value(self, fixed_endpoint): - client = self.create_client(endpoint=fixed_endpoint) - response = client.string.put_unknown_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py deleted file mode 100644 index fd4a6d20c4c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py +++ /dev/null @@ -1,47 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import FixedPreparer -from testpreparer_async import FixedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestFixedStringOperationsAsync(FixedClientTestBaseAsync): - @FixedPreparer() - @recorded_by_proxy_async - async def test_string_get_known_value(self, fixed_endpoint): - client = self.create_async_client(endpoint=fixed_endpoint) - response = await client.string.get_known_value() - - # please add some check logic here by yourself - # ... - - @FixedPreparer() - @recorded_by_proxy_async - async def test_string_put_known_value(self, fixed_endpoint): - client = self.create_async_client(endpoint=fixed_endpoint) - response = await client.string.put_known_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @FixedPreparer() - @recorded_by_proxy_async - async def test_string_put_unknown_value(self, fixed_endpoint): - client = self.create_async_client(endpoint=fixed_endpoint) - response = await client.string.put_unknown_value( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py deleted file mode 100644 index 5a5bb459bc4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.enum.fixed import FixedClient - - -class FixedClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(FixedClient) - return self.create_client_from_credential( - FixedClient, - credential=credential, - endpoint=endpoint, - ) - - -FixedPreparer = functools.partial(PowerShellPreparer, "fixed", fixed_endpoint="https://fake_fixed_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py deleted file mode 100644 index ec4217913e8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.enum.fixed.aio import FixedClient - - -class FixedClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(FixedClient, is_async=True) - return self.create_client_from_credential( - FixedClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-enum-fixed/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py deleted file mode 100644 index 2a895bea492..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - empty_subscription_id = os.environ.get("EMPTY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - empty_tenant_id = os.environ.get("EMPTY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - empty_client_id = os.environ.get("EMPTY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - empty_client_secret = os.environ.get("EMPTY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=empty_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=empty_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=empty_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=empty_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py deleted file mode 100644 index cdadb9d894a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py +++ /dev/null @@ -1,44 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import EmptyClientTestBase, EmptyPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestEmpty(EmptyClientTestBase): - @EmptyPreparer() - @recorded_by_proxy - def test_put_empty(self, empty_endpoint): - client = self.create_client(endpoint=empty_endpoint) - response = client.put_empty( - input={}, - ) - - # please add some check logic here by yourself - # ... - - @EmptyPreparer() - @recorded_by_proxy - def test_get_empty(self, empty_endpoint): - client = self.create_client(endpoint=empty_endpoint) - response = client.get_empty() - - # please add some check logic here by yourself - # ... - - @EmptyPreparer() - @recorded_by_proxy - def test_post_round_trip_empty(self, empty_endpoint): - client = self.create_client(endpoint=empty_endpoint) - response = client.post_round_trip_empty( - body={}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py deleted file mode 100644 index ed450cbb2e3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import EmptyPreparer -from testpreparer_async import EmptyClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestEmptyAsync(EmptyClientTestBaseAsync): - @EmptyPreparer() - @recorded_by_proxy_async - async def test_put_empty(self, empty_endpoint): - client = self.create_async_client(endpoint=empty_endpoint) - response = await client.put_empty( - input={}, - ) - - # please add some check logic here by yourself - # ... - - @EmptyPreparer() - @recorded_by_proxy_async - async def test_get_empty(self, empty_endpoint): - client = self.create_async_client(endpoint=empty_endpoint) - response = await client.get_empty() - - # please add some check logic here by yourself - # ... - - @EmptyPreparer() - @recorded_by_proxy_async - async def test_post_round_trip_empty(self, empty_endpoint): - client = self.create_async_client(endpoint=empty_endpoint) - response = await client.post_round_trip_empty( - body={}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py deleted file mode 100644 index 697114f9f4b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.empty import EmptyClient - - -class EmptyClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(EmptyClient) - return self.create_client_from_credential( - EmptyClient, - credential=credential, - endpoint=endpoint, - ) - - -EmptyPreparer = functools.partial(PowerShellPreparer, "empty", empty_endpoint="https://fake_empty_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py deleted file mode 100644 index de9919c1869..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.empty.aio import EmptyClient - - -class EmptyClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(EmptyClient, is_async=True) - return self.create_client_from_credential( - EmptyClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-empty/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py deleted file mode 100644 index 8f844ad92c7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - enumdiscriminator_subscription_id = os.environ.get( - "ENUMDISCRIMINATOR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - enumdiscriminator_tenant_id = os.environ.get("ENUMDISCRIMINATOR_TENANT_ID", "00000000-0000-0000-0000-000000000000") - enumdiscriminator_client_id = os.environ.get("ENUMDISCRIMINATOR_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - enumdiscriminator_client_secret = os.environ.get( - "ENUMDISCRIMINATOR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=enumdiscriminator_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=enumdiscriminator_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=enumdiscriminator_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=enumdiscriminator_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py deleted file mode 100644 index efc5b841192..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py +++ /dev/null @@ -1,89 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import EnumDiscriminatorClientTestBase, EnumDiscriminatorPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestEnumDiscriminator(EnumDiscriminatorClientTestBase): - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_get_extensible_model(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.get_extensible_model() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_put_extensible_model(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.put_extensible_model( - input={"kind": "golden", "weight": 0}, - ) - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_get_extensible_model_missing_discriminator(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.get_extensible_model_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_get_extensible_model_wrong_discriminator(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.get_extensible_model_wrong_discriminator() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_get_fixed_model(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.get_fixed_model() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_put_fixed_model(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.put_fixed_model( - input={"kind": "cobra", "length": 0}, - ) - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_get_fixed_model_missing_discriminator(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.get_fixed_model_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy - def test_get_fixed_model_wrong_discriminator(self, enumdiscriminator_endpoint): - client = self.create_client(endpoint=enumdiscriminator_endpoint) - response = client.get_fixed_model_wrong_discriminator() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py deleted file mode 100644 index d4c0c170943..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py +++ /dev/null @@ -1,90 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import EnumDiscriminatorPreparer -from testpreparer_async import EnumDiscriminatorClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestEnumDiscriminatorAsync(EnumDiscriminatorClientTestBaseAsync): - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_extensible_model(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.get_extensible_model() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_put_extensible_model(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.put_extensible_model( - input={"kind": "golden", "weight": 0}, - ) - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_extensible_model_missing_discriminator(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.get_extensible_model_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_extensible_model_wrong_discriminator(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.get_extensible_model_wrong_discriminator() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_fixed_model(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.get_fixed_model() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_put_fixed_model(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.put_fixed_model( - input={"kind": "cobra", "length": 0}, - ) - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_fixed_model_missing_discriminator(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.get_fixed_model_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @EnumDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_fixed_model_wrong_discriminator(self, enumdiscriminator_endpoint): - client = self.create_async_client(endpoint=enumdiscriminator_endpoint) - response = await client.get_fixed_model_wrong_discriminator() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py deleted file mode 100644 index 99473c8b5b6..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.enumdiscriminator import EnumDiscriminatorClient - - -class EnumDiscriminatorClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(EnumDiscriminatorClient) - return self.create_client_from_credential( - EnumDiscriminatorClient, - credential=credential, - endpoint=endpoint, - ) - - -EnumDiscriminatorPreparer = functools.partial( - PowerShellPreparer, "enumdiscriminator", enumdiscriminator_endpoint="https://fake_enumdiscriminator_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py deleted file mode 100644 index 479d42f79ee..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.enumdiscriminator.aio import EnumDiscriminatorClient - - -class EnumDiscriminatorClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(EnumDiscriminatorClient, is_async=True) - return self.create_client_from_credential( - EnumDiscriminatorClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py deleted file mode 100644 index 8ba1f6f463b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py +++ /dev/null @@ -1,43 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - nesteddiscriminator_subscription_id = os.environ.get( - "NESTEDDISCRIMINATOR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - nesteddiscriminator_tenant_id = os.environ.get( - "NESTEDDISCRIMINATOR_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - nesteddiscriminator_client_id = os.environ.get( - "NESTEDDISCRIMINATOR_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - nesteddiscriminator_client_secret = os.environ.get( - "NESTEDDISCRIMINATOR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=nesteddiscriminator_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=nesteddiscriminator_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=nesteddiscriminator_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=nesteddiscriminator_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py deleted file mode 100644 index 76206b37b92..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py +++ /dev/null @@ -1,71 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NestedDiscriminatorClientTestBase, NestedDiscriminatorPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNestedDiscriminator(NestedDiscriminatorClientTestBase): - @NestedDiscriminatorPreparer() - @recorded_by_proxy - def test_get_model(self, nesteddiscriminator_endpoint): - client = self.create_client(endpoint=nesteddiscriminator_endpoint) - response = client.get_model() - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy - def test_put_model(self, nesteddiscriminator_endpoint): - client = self.create_client(endpoint=nesteddiscriminator_endpoint) - response = client.put_model( - input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, - ) - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy - def test_get_recursive_model(self, nesteddiscriminator_endpoint): - client = self.create_client(endpoint=nesteddiscriminator_endpoint) - response = client.get_recursive_model() - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy - def test_put_recursive_model(self, nesteddiscriminator_endpoint): - client = self.create_client(endpoint=nesteddiscriminator_endpoint) - response = client.put_recursive_model( - input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, - ) - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy - def test_get_missing_discriminator(self, nesteddiscriminator_endpoint): - client = self.create_client(endpoint=nesteddiscriminator_endpoint) - response = client.get_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy - def test_get_wrong_discriminator(self, nesteddiscriminator_endpoint): - client = self.create_client(endpoint=nesteddiscriminator_endpoint) - response = client.get_wrong_discriminator() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py deleted file mode 100644 index 08270488ab8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py +++ /dev/null @@ -1,72 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NestedDiscriminatorPreparer -from testpreparer_async import NestedDiscriminatorClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNestedDiscriminatorAsync(NestedDiscriminatorClientTestBaseAsync): - @NestedDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_model(self, nesteddiscriminator_endpoint): - client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) - response = await client.get_model() - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_put_model(self, nesteddiscriminator_endpoint): - client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) - response = await client.put_model( - input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, - ) - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_recursive_model(self, nesteddiscriminator_endpoint): - client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) - response = await client.get_recursive_model() - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_put_recursive_model(self, nesteddiscriminator_endpoint): - client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) - response = await client.put_recursive_model( - input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, - ) - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_missing_discriminator(self, nesteddiscriminator_endpoint): - client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) - response = await client.get_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @NestedDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_wrong_discriminator(self, nesteddiscriminator_endpoint): - client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) - response = await client.get_wrong_discriminator() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py deleted file mode 100644 index 11eb32cb7e3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.nesteddiscriminator import NestedDiscriminatorClient - - -class NestedDiscriminatorClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(NestedDiscriminatorClient) - return self.create_client_from_credential( - NestedDiscriminatorClient, - credential=credential, - endpoint=endpoint, - ) - - -NestedDiscriminatorPreparer = functools.partial( - PowerShellPreparer, - "nesteddiscriminator", - nesteddiscriminator_endpoint="https://fake_nesteddiscriminator_endpoint.com", -) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py deleted file mode 100644 index 7913db8a24b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.nesteddiscriminator.aio import NestedDiscriminatorClient - - -class NestedDiscriminatorClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(NestedDiscriminatorClient, is_async=True) - return self.create_client_from_credential( - NestedDiscriminatorClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py deleted file mode 100644 index f6b00219293..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - notdiscriminated_subscription_id = os.environ.get( - "NOTDISCRIMINATED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - notdiscriminated_tenant_id = os.environ.get("NOTDISCRIMINATED_TENANT_ID", "00000000-0000-0000-0000-000000000000") - notdiscriminated_client_id = os.environ.get("NOTDISCRIMINATED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - notdiscriminated_client_secret = os.environ.get( - "NOTDISCRIMINATED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=notdiscriminated_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notdiscriminated_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notdiscriminated_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=notdiscriminated_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py deleted file mode 100644 index 8cae28e36be..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py +++ /dev/null @@ -1,44 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NotDiscriminatedClientTestBase, NotDiscriminatedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNotDiscriminated(NotDiscriminatedClientTestBase): - @NotDiscriminatedPreparer() - @recorded_by_proxy - def test_post_valid(self, notdiscriminated_endpoint): - client = self.create_client(endpoint=notdiscriminated_endpoint) - response = client.post_valid( - input={"age": 0, "name": "str", "smart": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NotDiscriminatedPreparer() - @recorded_by_proxy - def test_get_valid(self, notdiscriminated_endpoint): - client = self.create_client(endpoint=notdiscriminated_endpoint) - response = client.get_valid() - - # please add some check logic here by yourself - # ... - - @NotDiscriminatedPreparer() - @recorded_by_proxy - def test_put_valid(self, notdiscriminated_endpoint): - client = self.create_client(endpoint=notdiscriminated_endpoint) - response = client.put_valid( - input={"age": 0, "name": "str", "smart": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py deleted file mode 100644 index a9df6bb895b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NotDiscriminatedPreparer -from testpreparer_async import NotDiscriminatedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNotDiscriminatedAsync(NotDiscriminatedClientTestBaseAsync): - @NotDiscriminatedPreparer() - @recorded_by_proxy_async - async def test_post_valid(self, notdiscriminated_endpoint): - client = self.create_async_client(endpoint=notdiscriminated_endpoint) - response = await client.post_valid( - input={"age": 0, "name": "str", "smart": bool}, - ) - - # please add some check logic here by yourself - # ... - - @NotDiscriminatedPreparer() - @recorded_by_proxy_async - async def test_get_valid(self, notdiscriminated_endpoint): - client = self.create_async_client(endpoint=notdiscriminated_endpoint) - response = await client.get_valid() - - # please add some check logic here by yourself - # ... - - @NotDiscriminatedPreparer() - @recorded_by_proxy_async - async def test_put_valid(self, notdiscriminated_endpoint): - client = self.create_async_client(endpoint=notdiscriminated_endpoint) - response = await client.put_valid( - input={"age": 0, "name": "str", "smart": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py deleted file mode 100644 index 82ddc70adbe..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.notdiscriminated import NotDiscriminatedClient - - -class NotDiscriminatedClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(NotDiscriminatedClient) - return self.create_client_from_credential( - NotDiscriminatedClient, - credential=credential, - endpoint=endpoint, - ) - - -NotDiscriminatedPreparer = functools.partial( - PowerShellPreparer, "notdiscriminated", notdiscriminated_endpoint="https://fake_notdiscriminated_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py deleted file mode 100644 index ecfe643396f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.notdiscriminated.aio import NotDiscriminatedClient - - -class NotDiscriminatedClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(NotDiscriminatedClient, is_async=True) - return self.create_client_from_credential( - NotDiscriminatedClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py deleted file mode 100644 index 338abb9869a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - recursive_subscription_id = os.environ.get("RECURSIVE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - recursive_tenant_id = os.environ.get("RECURSIVE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - recursive_client_id = os.environ.get("RECURSIVE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - recursive_client_secret = os.environ.get("RECURSIVE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=recursive_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=recursive_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=recursive_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=recursive_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py deleted file mode 100644 index 11b06922531..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RecursiveClientTestBase, RecursivePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRecursive(RecursiveClientTestBase): - @RecursivePreparer() - @recorded_by_proxy - def test_put(self, recursive_endpoint): - client = self.create_client(endpoint=recursive_endpoint) - response = client.put( - input={"level": 0, "extension": [...]}, - ) - - # please add some check logic here by yourself - # ... - - @RecursivePreparer() - @recorded_by_proxy - def test_get(self, recursive_endpoint): - client = self.create_client(endpoint=recursive_endpoint) - response = client.get() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py deleted file mode 100644 index a09a896c425..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RecursivePreparer -from testpreparer_async import RecursiveClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRecursiveAsync(RecursiveClientTestBaseAsync): - @RecursivePreparer() - @recorded_by_proxy_async - async def test_put(self, recursive_endpoint): - client = self.create_async_client(endpoint=recursive_endpoint) - response = await client.put( - input={"level": 0, "extension": [...]}, - ) - - # please add some check logic here by yourself - # ... - - @RecursivePreparer() - @recorded_by_proxy_async - async def test_get(self, recursive_endpoint): - client = self.create_async_client(endpoint=recursive_endpoint) - response = await client.get() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py deleted file mode 100644 index 33dbd65de15..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.recursive import RecursiveClient - - -class RecursiveClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(RecursiveClient) - return self.create_client_from_credential( - RecursiveClient, - credential=credential, - endpoint=endpoint, - ) - - -RecursivePreparer = functools.partial( - PowerShellPreparer, "recursive", recursive_endpoint="https://fake_recursive_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py deleted file mode 100644 index e57630973f5..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.recursive.aio import RecursiveClient - - -class RecursiveClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(RecursiveClient, is_async=True) - return self.create_client_from_credential( - RecursiveClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-recursive/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py deleted file mode 100644 index 633ac1e52f0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py +++ /dev/null @@ -1,43 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - singlediscriminator_subscription_id = os.environ.get( - "SINGLEDISCRIMINATOR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - singlediscriminator_tenant_id = os.environ.get( - "SINGLEDISCRIMINATOR_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - singlediscriminator_client_id = os.environ.get( - "SINGLEDISCRIMINATOR_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - singlediscriminator_client_secret = os.environ.get( - "SINGLEDISCRIMINATOR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=singlediscriminator_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=singlediscriminator_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=singlediscriminator_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=singlediscriminator_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py deleted file mode 100644 index efc2bac6ce8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py +++ /dev/null @@ -1,80 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import SingleDiscriminatorClientTestBase, SingleDiscriminatorPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSingleDiscriminator(SingleDiscriminatorClientTestBase): - @SingleDiscriminatorPreparer() - @recorded_by_proxy - def test_get_model(self, singlediscriminator_endpoint): - client = self.create_client(endpoint=singlediscriminator_endpoint) - response = client.get_model() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy - def test_put_model(self, singlediscriminator_endpoint): - client = self.create_client(endpoint=singlediscriminator_endpoint) - response = client.put_model( - input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, - ) - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy - def test_get_recursive_model(self, singlediscriminator_endpoint): - client = self.create_client(endpoint=singlediscriminator_endpoint) - response = client.get_recursive_model() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy - def test_put_recursive_model(self, singlediscriminator_endpoint): - client = self.create_client(endpoint=singlediscriminator_endpoint) - response = client.put_recursive_model( - input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, - ) - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy - def test_get_missing_discriminator(self, singlediscriminator_endpoint): - client = self.create_client(endpoint=singlediscriminator_endpoint) - response = client.get_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy - def test_get_wrong_discriminator(self, singlediscriminator_endpoint): - client = self.create_client(endpoint=singlediscriminator_endpoint) - response = client.get_wrong_discriminator() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy - def test_get_legacy_model(self, singlediscriminator_endpoint): - client = self.create_client(endpoint=singlediscriminator_endpoint) - response = client.get_legacy_model() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py deleted file mode 100644 index 96fc035088f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py +++ /dev/null @@ -1,81 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import SingleDiscriminatorPreparer -from testpreparer_async import SingleDiscriminatorClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestSingleDiscriminatorAsync(SingleDiscriminatorClientTestBaseAsync): - @SingleDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_model(self, singlediscriminator_endpoint): - client = self.create_async_client(endpoint=singlediscriminator_endpoint) - response = await client.get_model() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_put_model(self, singlediscriminator_endpoint): - client = self.create_async_client(endpoint=singlediscriminator_endpoint) - response = await client.put_model( - input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, - ) - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_recursive_model(self, singlediscriminator_endpoint): - client = self.create_async_client(endpoint=singlediscriminator_endpoint) - response = await client.get_recursive_model() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_put_recursive_model(self, singlediscriminator_endpoint): - client = self.create_async_client(endpoint=singlediscriminator_endpoint) - response = await client.put_recursive_model( - input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, - ) - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_missing_discriminator(self, singlediscriminator_endpoint): - client = self.create_async_client(endpoint=singlediscriminator_endpoint) - response = await client.get_missing_discriminator() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_wrong_discriminator(self, singlediscriminator_endpoint): - client = self.create_async_client(endpoint=singlediscriminator_endpoint) - response = await client.get_wrong_discriminator() - - # please add some check logic here by yourself - # ... - - @SingleDiscriminatorPreparer() - @recorded_by_proxy_async - async def test_get_legacy_model(self, singlediscriminator_endpoint): - client = self.create_async_client(endpoint=singlediscriminator_endpoint) - response = await client.get_legacy_model() - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py deleted file mode 100644 index 97b81d355e1..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.singlediscriminator import SingleDiscriminatorClient - - -class SingleDiscriminatorClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(SingleDiscriminatorClient) - return self.create_client_from_credential( - SingleDiscriminatorClient, - credential=credential, - endpoint=endpoint, - ) - - -SingleDiscriminatorPreparer = functools.partial( - PowerShellPreparer, - "singlediscriminator", - singlediscriminator_endpoint="https://fake_singlediscriminator_endpoint.com", -) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py deleted file mode 100644 index 71110d7df51..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.singlediscriminator.aio import SingleDiscriminatorClient - - -class SingleDiscriminatorClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(SingleDiscriminatorClient, is_async=True) - return self.create_client_from_credential( - SingleDiscriminatorClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py deleted file mode 100644 index 6a990b92bd7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - usage_subscription_id = os.environ.get("USAGE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - usage_tenant_id = os.environ.get("USAGE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - usage_client_id = os.environ.get("USAGE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - usage_client_secret = os.environ.get("USAGE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=usage_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py deleted file mode 100644 index 7514cae181d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py +++ /dev/null @@ -1,44 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UsageClientTestBase, UsagePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUsage(UsageClientTestBase): - @UsagePreparer() - @recorded_by_proxy - def test_input(self, usage_endpoint): - client = self.create_client(endpoint=usage_endpoint) - response = client.input( - input={"requiredProp": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy - def test_output(self, usage_endpoint): - client = self.create_client(endpoint=usage_endpoint) - response = client.output() - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy - def test_input_and_output(self, usage_endpoint): - client = self.create_client(endpoint=usage_endpoint) - response = client.input_and_output( - body={"requiredProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py deleted file mode 100644 index c2a66f7642b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UsagePreparer -from testpreparer_async import UsageClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUsageAsync(UsageClientTestBaseAsync): - @UsagePreparer() - @recorded_by_proxy_async - async def test_input(self, usage_endpoint): - client = self.create_async_client(endpoint=usage_endpoint) - response = await client.input( - input={"requiredProp": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy_async - async def test_output(self, usage_endpoint): - client = self.create_async_client(endpoint=usage_endpoint) - response = await client.output() - - # please add some check logic here by yourself - # ... - - @UsagePreparer() - @recorded_by_proxy_async - async def test_input_and_output(self, usage_endpoint): - client = self.create_async_client(endpoint=usage_endpoint) - response = await client.input_and_output( - body={"requiredProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py deleted file mode 100644 index 1c43d7f3f9e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.usage import UsageClient - - -class UsageClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(UsageClient) - return self.create_client_from_credential( - UsageClient, - credential=credential, - endpoint=endpoint, - ) - - -UsagePreparer = functools.partial(PowerShellPreparer, "usage", usage_endpoint="https://fake_usage_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py deleted file mode 100644 index feabf6b2b2e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.usage.aio import UsageClient - - -class UsageClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(UsageClient, is_async=True) - return self.create_client_from_credential( - UsageClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-usage/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py deleted file mode 100644 index 3fc4942cf79..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - visibility_subscription_id = os.environ.get("VISIBILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - visibility_tenant_id = os.environ.get("VISIBILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") - visibility_client_id = os.environ.get("VISIBILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - visibility_client_secret = os.environ.get("VISIBILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=visibility_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py deleted file mode 100644 index 7ed8cd7db1d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py +++ /dev/null @@ -1,90 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import VisibilityClientTestBase, VisibilityPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVisibility(VisibilityClientTestBase): - @VisibilityPreparer() - @recorded_by_proxy - def test_get_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.get_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_head_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.head_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_put_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.put_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_patch_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.patch_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_post_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.post_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_delete_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.delete_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy - def test_put_read_only_model(self, visibility_endpoint): - client = self.create_client(endpoint=visibility_endpoint) - response = client.put_read_only_model( - input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py deleted file mode 100644 index 9909a3a67eb..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py +++ /dev/null @@ -1,91 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import VisibilityPreparer -from testpreparer_async import VisibilityClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestVisibilityAsync(VisibilityClientTestBaseAsync): - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_get_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.get_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_head_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.head_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_put_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.put_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_patch_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.patch_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_post_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.post_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_delete_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.delete_model( - input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, - ) - - # please add some check logic here by yourself - # ... - - @VisibilityPreparer() - @recorded_by_proxy_async - async def test_put_read_only_model(self, visibility_endpoint): - client = self.create_async_client(endpoint=visibility_endpoint) - response = await client.put_read_only_model( - input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py deleted file mode 100644 index 5b050ddfa6b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.model.visibility import VisibilityClient - - -class VisibilityClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(VisibilityClient) - return self.create_client_from_credential( - VisibilityClient, - credential=credential, - endpoint=endpoint, - ) - - -VisibilityPreparer = functools.partial( - PowerShellPreparer, "visibility", visibility_endpoint="https://fake_visibility_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py deleted file mode 100644 index d029cadcb66..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.model.visibility.aio import VisibilityClient - - -class VisibilityClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(VisibilityClient, is_async=True) - return self.create_client_from_credential( - VisibilityClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-visibility/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py deleted file mode 100644 index 1468d06c958..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - additionalproperties_subscription_id = os.environ.get( - "ADDITIONALPROPERTIES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - additionalproperties_tenant_id = os.environ.get( - "ADDITIONALPROPERTIES_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - additionalproperties_client_id = os.environ.get( - "ADDITIONALPROPERTIES_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - additionalproperties_client_secret = os.environ.get( - "ADDITIONALPROPERTIES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer( - regex=additionalproperties_subscription_id, value="00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=additionalproperties_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=additionalproperties_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=additionalproperties_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py deleted file mode 100644 index bce95a06639..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadFloatOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_float_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_float_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_float.put( - body={"derivedProp": 0.0, "name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py deleted file mode 100644 index efbdeeb9588..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_float_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_float_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_float.put( - body={"derivedProp": 0.0, "name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py deleted file mode 100644 index 048d80dbb9e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadModelArrayOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_model_array_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_model_array_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_model_array.put( - body={"derivedProp": [{"state": "str"}], "knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py deleted file mode 100644 index dad7cd099de..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_model_array_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_model_array_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_model_array.put( - body={"derivedProp": [{"state": "str"}], "knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py deleted file mode 100644 index 9a3ab0104f4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadModelOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_model_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_model_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_model.put( - body={"derivedProp": {"state": "str"}, "knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py deleted file mode 100644 index eb9aed8f98e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_model_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_model_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_model.put( - body={"derivedProp": {"state": "str"}, "knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py deleted file mode 100644 index c2d8c804b7a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadStringOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_string_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_different_spread_string_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_different_spread_string.put( - body={"derivedProp": "str", "id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py deleted file mode 100644 index 117ae0ce9d0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsDifferentSpreadStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_string_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_different_spread_string_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_different_spread_string.put( - body={"derivedProp": "str", "id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py deleted file mode 100644 index 9f7581b3d8c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsFloatOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_float_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_float_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_float.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py deleted file mode 100644 index 737ed0bd682..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_float_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_float_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_float.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py deleted file mode 100644 index 6e539bb68c5..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsModelArrayOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_model_array_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_model_array_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_model_array.put( - body={"knownProp": [{"state": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py deleted file mode 100644 index 1c0cb2da785..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_model_array_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_model_array_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_model_array.put( - body={"knownProp": [{"state": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py deleted file mode 100644 index 71f8be6216c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsModelOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_model_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_model_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_model.put( - body={"knownProp": {"state": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py deleted file mode 100644 index 1ed0b99ca3f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_model_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_model_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_model.put( - body={"knownProp": {"state": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py deleted file mode 100644 index ba56033e4af..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsStringOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_string_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_string_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_string.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py deleted file mode 100644 index 58e4e63d8ea..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_string_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_string_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_string.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py deleted file mode 100644 index 3a65da7a619..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsUnknownDerivedOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_unknown_derived_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_unknown_derived.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_unknown_derived_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_unknown_derived.put( - body={"index": 0, "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py deleted file mode 100644 index ce20df60dc4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsUnknownDerivedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_unknown_derived_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_unknown_derived.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_unknown_derived_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_unknown_derived.put( - body={"index": 0, "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py deleted file mode 100644 index c0a56d2ff1a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsUnknownDiscriminatedOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_unknown_discriminated_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_unknown_discriminated.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_unknown_discriminated_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_unknown_discriminated.put( - body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py deleted file mode 100644 index ad6020d0192..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsUnknownDiscriminatedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_unknown_discriminated_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_unknown_discriminated.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_unknown_discriminated_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_unknown_discriminated.put( - body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py deleted file mode 100644 index 31c2bed53f2..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsUnknownOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_unknown_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_unknown.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_extends_unknown_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.extends_unknown.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py deleted file mode 100644 index 83136a4fe7e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesExtendsUnknownOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_unknown_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_unknown.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_extends_unknown_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.extends_unknown.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py deleted file mode 100644 index 354263e1857..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsFloatOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_float_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_float_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_float.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py deleted file mode 100644 index da7825acb80..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_float_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_float_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_float.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py deleted file mode 100644 index bdb8bd90138..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsModelArrayOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_model_array_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_model_array_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_model_array.put( - body={"knownProp": [{"state": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py deleted file mode 100644 index 31593031a17..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_model_array_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_model_array_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_model_array.put( - body={"knownProp": [{"state": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py deleted file mode 100644 index 5ccd7fddce3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsModelOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_model_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_model_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_model.put( - body={"knownProp": {"state": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py deleted file mode 100644 index 1606f5f21cc..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_model_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_model_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_model.put( - body={"knownProp": {"state": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py deleted file mode 100644 index 7fbd6658191..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsStringOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_string_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_string_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_string.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py deleted file mode 100644 index 60854f96eb7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_string_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_string_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_string.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py deleted file mode 100644 index b3798ae7950..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsUnknownDerivedOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_unknown_derived_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_unknown_derived.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_unknown_derived_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_unknown_derived.put( - body={"index": 0, "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py deleted file mode 100644 index ff24780610d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsUnknownDerivedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_unknown_derived_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_unknown_derived.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_unknown_derived_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_unknown_derived.put( - body={"index": 0, "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py deleted file mode 100644 index 6721020fb1a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsUnknownDiscriminatedOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_unknown_discriminated_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_unknown_discriminated.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_unknown_discriminated_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_unknown_discriminated.put( - body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py deleted file mode 100644 index baabdf3a52e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsUnknownDiscriminatedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_unknown_discriminated_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_unknown_discriminated.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_unknown_discriminated_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_unknown_discriminated.put( - body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py deleted file mode 100644 index 8886d068ecd..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsUnknownOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_unknown_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_unknown.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_is_unknown_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.is_unknown.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py deleted file mode 100644 index a382ceef639..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesIsUnknownOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_unknown_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_unknown.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_is_unknown_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.is_unknown.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py deleted file mode 100644 index 05639d3daf0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesMultipleSpreadOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_multiple_spread_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.multiple_spread.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_multiple_spread_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.multiple_spread.put( - body={"flag": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py deleted file mode 100644 index d252e42b2e6..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesMultipleSpreadOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_multiple_spread_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.multiple_spread.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_multiple_spread_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.multiple_spread.put( - body={"flag": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py deleted file mode 100644 index c49babb04e8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentFloatOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_float_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_float_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_float.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py deleted file mode 100644 index a809a2f2a04..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_float_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_float_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_float.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py deleted file mode 100644 index 72f42e0aa6d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentModelArrayOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_model_array_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_model_array_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_model_array.put( - body={"knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py deleted file mode 100644 index 87aa47bc9e0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_model_array_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_model_array_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_model_array.put( - body={"knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py deleted file mode 100644 index f98dbe3ac01..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentModelOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_model_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_model_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_model.put( - body={"knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py deleted file mode 100644 index bf9d2bed264..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_model_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_model_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_model.put( - body={"knownProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py deleted file mode 100644 index 6dc582546a5..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentStringOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_string_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_different_string_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_different_string.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py deleted file mode 100644 index 65018748418..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadDifferentStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_string_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_different_string_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_different_string.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py deleted file mode 100644 index 6061341f6b0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadFloatOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_float_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_float_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_float.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py deleted file mode 100644 index 429930f2719..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_float_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_float.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_float_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_float.put( - body={"id": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py deleted file mode 100644 index ba829536ceb..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadModelArrayOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_model_array_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_model_array_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_model_array.put( - body={"knownProp": [{"state": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py deleted file mode 100644 index 2647755f4dd..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_model_array_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_model_array.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_model_array_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_model_array.put( - body={"knownProp": [{"state": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py deleted file mode 100644 index 9fe7fb6e2cc..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadModelOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_model_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_model_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_model.put( - body={"knownProp": {"state": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py deleted file mode 100644 index d1987ed0a6d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_model_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_model.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_model_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_model.put( - body={"knownProp": {"state": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py deleted file mode 100644 index d783fba2a7e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordDiscriminatedUnionOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_discriminated_union_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_discriminated_union.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_discriminated_union_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_discriminated_union.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py deleted file mode 100644 index f9afa893275..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordDiscriminatedUnionOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_discriminated_union_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_discriminated_union.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_discriminated_union_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_discriminated_union.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py deleted file mode 100644 index 1e2055dfb52..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion2Operations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_non_discriminated_union2_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_non_discriminated_union2.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_non_discriminated_union2_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_non_discriminated_union2.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py deleted file mode 100644 index b4fc9526888..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion2OperationsAsync( - AdditionalPropertiesClientTestBaseAsync -): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_non_discriminated_union2_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_non_discriminated_union2.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_non_discriminated_union2_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_non_discriminated_union2.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py deleted file mode 100644 index d80b30b4424..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion3Operations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_non_discriminated_union3_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_non_discriminated_union3.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_non_discriminated_union3_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_non_discriminated_union3.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py deleted file mode 100644 index 9b6f4cf260e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion3OperationsAsync( - AdditionalPropertiesClientTestBaseAsync -): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_non_discriminated_union3_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_non_discriminated_union3.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_non_discriminated_union3_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_non_discriminated_union3.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py deleted file mode 100644 index fc4e613ba15..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnionOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_non_discriminated_union_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_non_discriminated_union.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_non_discriminated_union_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_non_discriminated_union.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py deleted file mode 100644 index af876dc59cc..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnionOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_non_discriminated_union_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_non_discriminated_union.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_non_discriminated_union_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_non_discriminated_union.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py deleted file mode 100644 index a78a589899d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordUnionOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_union_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_union.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_record_union_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_record_union.put( - body={"flag": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py deleted file mode 100644 index 6738840018d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadRecordUnionOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_union_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_union.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_record_union_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_record_union.put( - body={"flag": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py deleted file mode 100644 index b01e760174d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadStringOperations(AdditionalPropertiesClientTestBase): - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_string_get(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy - def test_spread_string_put(self, additionalproperties_endpoint): - client = self.create_client(endpoint=additionalproperties_endpoint) - response = client.spread_string.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py deleted file mode 100644 index 6647bf7b020..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AdditionalPropertiesPreparer -from testpreparer_async import AdditionalPropertiesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdditionalPropertiesSpreadStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_string_get(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_string.get() - - # please add some check logic here by yourself - # ... - - @AdditionalPropertiesPreparer() - @recorded_by_proxy_async - async def test_spread_string_put(self, additionalproperties_endpoint): - client = self.create_async_client(endpoint=additionalproperties_endpoint) - response = await client.spread_string.put( - body={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py deleted file mode 100644 index 84f49b4c073..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.property.additionalproperties import AdditionalPropertiesClient - - -class AdditionalPropertiesClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(AdditionalPropertiesClient) - return self.create_client_from_credential( - AdditionalPropertiesClient, - credential=credential, - endpoint=endpoint, - ) - - -AdditionalPropertiesPreparer = functools.partial( - PowerShellPreparer, - "additionalproperties", - additionalproperties_endpoint="https://fake_additionalproperties_endpoint.com", -) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py deleted file mode 100644 index a512953aeed..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.property.additionalproperties.aio import AdditionalPropertiesClient - - -class AdditionalPropertiesClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(AdditionalPropertiesClient, is_async=True) - return self.create_client_from_credential( - AdditionalPropertiesClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py deleted file mode 100644 index 3bb2820941c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - nullable_subscription_id = os.environ.get("NULLABLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - nullable_tenant_id = os.environ.get("NULLABLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") - nullable_client_id = os.environ.get("NULLABLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - nullable_client_secret = os.environ.get("NULLABLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=nullable_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=nullable_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=nullable_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=nullable_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py deleted file mode 100644 index b6cb74e9109..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NullableClientTestBase, NullablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableBytesOperations(NullableClientTestBase): - @NullablePreparer() - @recorded_by_proxy - def test_bytes_get_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.bytes.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_bytes_get_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.bytes.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_bytes_patch_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.bytes.patch_non_null( - body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_bytes_patch_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.bytes.patch_null( - body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py deleted file mode 100644 index 792c936c76e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NullablePreparer -from testpreparer_async import NullableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableBytesOperationsAsync(NullableClientTestBaseAsync): - @NullablePreparer() - @recorded_by_proxy_async - async def test_bytes_get_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.bytes.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_bytes_get_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.bytes.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_bytes_patch_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.bytes.patch_non_null( - body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_bytes_patch_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.bytes.patch_null( - body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py deleted file mode 100644 index ab22c0a94d7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NullableClientTestBase, NullablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableCollectionsByteOperations(NullableClientTestBase): - @NullablePreparer() - @recorded_by_proxy - def test_collections_byte_get_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_byte.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_byte_get_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_byte.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_byte_patch_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_byte.patch_non_null( - body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_byte_patch_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_byte.patch_null( - body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py deleted file mode 100644 index b970b004499..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NullablePreparer -from testpreparer_async import NullableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableCollectionsByteOperationsAsync(NullableClientTestBaseAsync): - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_byte_get_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_byte.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_byte_get_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_byte.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_byte_patch_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_byte.patch_non_null( - body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_byte_patch_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_byte.patch_null( - body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py deleted file mode 100644 index 36ead8cceca..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NullableClientTestBase, NullablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableCollectionsModelOperations(NullableClientTestBase): - @NullablePreparer() - @recorded_by_proxy - def test_collections_model_get_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_model.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_model_get_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_model.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_model_patch_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_model.patch_non_null( - body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_model_patch_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_model.patch_null( - body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py deleted file mode 100644 index dd0a92fa140..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NullablePreparer -from testpreparer_async import NullableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableCollectionsModelOperationsAsync(NullableClientTestBaseAsync): - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_model_get_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_model.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_model_get_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_model.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_model_patch_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_model.patch_non_null( - body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_model_patch_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_model.patch_null( - body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py deleted file mode 100644 index ec73d86cce2..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NullableClientTestBase, NullablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableCollectionsStringOperations(NullableClientTestBase): - @NullablePreparer() - @recorded_by_proxy - def test_collections_string_get_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_string.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_string_get_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_string.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_string_patch_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_string.patch_non_null( - body={"nullableProperty": ["str"], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_collections_string_patch_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.collections_string.patch_null( - body={"nullableProperty": ["str"], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py deleted file mode 100644 index f8e87b572b6..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NullablePreparer -from testpreparer_async import NullableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableCollectionsStringOperationsAsync(NullableClientTestBaseAsync): - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_string_get_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_string.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_string_get_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_string.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_string_patch_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_string.patch_non_null( - body={"nullableProperty": ["str"], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_collections_string_patch_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.collections_string.patch_null( - body={"nullableProperty": ["str"], "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py deleted file mode 100644 index afdf8b9796a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NullableClientTestBase, NullablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableDatetimeOperations(NullableClientTestBase): - @NullablePreparer() - @recorded_by_proxy - def test_datetime_get_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.datetime.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_datetime_get_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.datetime.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_datetime_patch_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.datetime.patch_non_null( - body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_datetime_patch_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.datetime.patch_null( - body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py deleted file mode 100644 index 0e3f62d7660..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NullablePreparer -from testpreparer_async import NullableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableDatetimeOperationsAsync(NullableClientTestBaseAsync): - @NullablePreparer() - @recorded_by_proxy_async - async def test_datetime_get_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.datetime.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_datetime_get_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.datetime.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_datetime_patch_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.datetime.patch_non_null( - body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_datetime_patch_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.datetime.patch_null( - body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py deleted file mode 100644 index 00c29045174..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NullableClientTestBase, NullablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableDurationOperations(NullableClientTestBase): - @NullablePreparer() - @recorded_by_proxy - def test_duration_get_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.duration.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_duration_get_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.duration.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_duration_patch_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.duration.patch_non_null( - body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_duration_patch_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.duration.patch_null( - body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py deleted file mode 100644 index e2fc03a01e0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NullablePreparer -from testpreparer_async import NullableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableDurationOperationsAsync(NullableClientTestBaseAsync): - @NullablePreparer() - @recorded_by_proxy_async - async def test_duration_get_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.duration.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_duration_get_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.duration.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_duration_patch_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.duration.patch_non_null( - body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_duration_patch_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.duration.patch_null( - body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py deleted file mode 100644 index 9dcc24d95a3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import NullableClientTestBase, NullablePreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableStringOperations(NullableClientTestBase): - @NullablePreparer() - @recorded_by_proxy - def test_string_get_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.string.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_string_get_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.string.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_string_patch_non_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.string.patch_non_null( - body={"nullableProperty": "str", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy - def test_string_patch_null(self, nullable_endpoint): - client = self.create_client(endpoint=nullable_endpoint) - response = client.string.patch_null( - body={"nullableProperty": "str", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py deleted file mode 100644 index dbf4e588182..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import NullablePreparer -from testpreparer_async import NullableClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestNullableStringOperationsAsync(NullableClientTestBaseAsync): - @NullablePreparer() - @recorded_by_proxy_async - async def test_string_get_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.string.get_non_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_string_get_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.string.get_null() - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_string_patch_non_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.string.patch_non_null( - body={"nullableProperty": "str", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @NullablePreparer() - @recorded_by_proxy_async - async def test_string_patch_null(self, nullable_endpoint): - client = self.create_async_client(endpoint=nullable_endpoint) - response = await client.string.patch_null( - body={"nullableProperty": "str", "requiredProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py deleted file mode 100644 index 1a8af063f5a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.property.nullable import NullableClient - - -class NullableClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(NullableClient) - return self.create_client_from_credential( - NullableClient, - credential=credential, - endpoint=endpoint, - ) - - -NullablePreparer = functools.partial( - PowerShellPreparer, "nullable", nullable_endpoint="https://fake_nullable_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py deleted file mode 100644 index 243f7d80b11..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.property.nullable.aio import NullableClient - - -class NullableClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(NullableClient, is_async=True) - return self.create_client_from_credential( - NullableClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-nullable/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py deleted file mode 100644 index 0361d7a1e81..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - optional_subscription_id = os.environ.get("OPTIONAL_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - optional_tenant_id = os.environ.get("OPTIONAL_TENANT_ID", "00000000-0000-0000-0000-000000000000") - optional_client_id = os.environ.get("OPTIONAL_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - optional_client_secret = os.environ.get("OPTIONAL_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=optional_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=optional_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=optional_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=optional_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py deleted file mode 100644 index 54ed36a824b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalBooleanLiteralOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_boolean_literal_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.boolean_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_boolean_literal_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.boolean_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_boolean_literal_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.boolean_literal.put_all( - body={"property": True}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_boolean_literal_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.boolean_literal.put_default( - body={"property": True}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py deleted file mode 100644 index 30609287dc3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalBooleanLiteralOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_boolean_literal_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.boolean_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_boolean_literal_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.boolean_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_boolean_literal_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.boolean_literal.put_all( - body={"property": True}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_boolean_literal_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.boolean_literal.put_default( - body={"property": True}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py deleted file mode 100644 index 305d2612ee3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalBytesOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_bytes_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.bytes.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_bytes_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.bytes.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_bytes_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.bytes.put_all( - body={"property": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_bytes_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.bytes.put_default( - body={"property": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py deleted file mode 100644 index ce7a116254f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalBytesOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_bytes_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.bytes.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_bytes_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.bytes.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_bytes_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.bytes.put_all( - body={"property": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_bytes_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.bytes.put_default( - body={"property": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py deleted file mode 100644 index 1f6dcf106f3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalCollectionsByteOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_collections_byte_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_byte.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_collections_byte_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_byte.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_collections_byte_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_byte.put_all( - body={"property": [bytes("bytes", encoding="utf-8")]}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_collections_byte_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_byte.put_default( - body={"property": [bytes("bytes", encoding="utf-8")]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py deleted file mode 100644 index 896f44ea6ed..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalCollectionsByteOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_byte_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_byte.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_byte_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_byte.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_byte_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_byte.put_all( - body={"property": [bytes("bytes", encoding="utf-8")]}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_byte_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_byte.put_default( - body={"property": [bytes("bytes", encoding="utf-8")]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py deleted file mode 100644 index 18cc93f8a5a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalCollectionsModelOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_collections_model_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_model.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_collections_model_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_model.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_collections_model_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_model.put_all( - body={"property": [{"property": "str"}]}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_collections_model_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.collections_model.put_default( - body={"property": [{"property": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py deleted file mode 100644 index 06e36e8e9f8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalCollectionsModelOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_model_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_model.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_model_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_model.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_model_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_model.put_all( - body={"property": [{"property": "str"}]}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_collections_model_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.collections_model.put_default( - body={"property": [{"property": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py deleted file mode 100644 index bc19d57fef3..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalDatetimeOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_datetime_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.datetime.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_datetime_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.datetime.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_datetime_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.datetime.put_all( - body={"property": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_datetime_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.datetime.put_default( - body={"property": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py deleted file mode 100644 index 280a0014040..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalDatetimeOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_datetime_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.datetime.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_datetime_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.datetime.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_datetime_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.datetime.put_all( - body={"property": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_datetime_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.datetime.put_default( - body={"property": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py deleted file mode 100644 index e920893260a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalDurationOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_duration_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.duration.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_duration_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.duration.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_duration_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.duration.put_all( - body={"property": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_duration_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.duration.put_default( - body={"property": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py deleted file mode 100644 index c31f449ec75..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalDurationOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_duration_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.duration.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_duration_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.duration.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_duration_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.duration.put_all( - body={"property": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_duration_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.duration.put_default( - body={"property": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py deleted file mode 100644 index f546fda9550..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalFloatLiteralOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_float_literal_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.float_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_float_literal_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.float_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_float_literal_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.float_literal.put_all( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_float_literal_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.float_literal.put_default( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py deleted file mode 100644 index f10606012a1..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalFloatLiteralOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_float_literal_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.float_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_float_literal_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.float_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_float_literal_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.float_literal.put_all( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_float_literal_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.float_literal.put_default( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py deleted file mode 100644 index 044da751628..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalIntLiteralOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_int_literal_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.int_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_int_literal_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.int_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_int_literal_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.int_literal.put_all( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_int_literal_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.int_literal.put_default( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py deleted file mode 100644 index a84ba860d26..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalIntLiteralOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_int_literal_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.int_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_int_literal_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.int_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_int_literal_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.int_literal.put_all( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_int_literal_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.int_literal.put_default( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py deleted file mode 100644 index 8d7331a5528..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalPlainDateOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_plain_date_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_date.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_plain_date_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_date.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_plain_date_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_date.put_all( - body={"property": "2020-02-20"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_plain_date_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_date.put_default( - body={"property": "2020-02-20"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py deleted file mode 100644 index ecea48e2582..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalPlainDateOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_date_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_date.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_date_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_date.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_date_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_date.put_all( - body={"property": "2020-02-20"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_date_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_date.put_default( - body={"property": "2020-02-20"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py deleted file mode 100644 index 223b55f35ea..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalPlainTimeOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_plain_time_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_time.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_plain_time_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_time.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_plain_time_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_time.put_all( - body={"property": "12:30:00"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_plain_time_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.plain_time.put_default( - body={"property": "12:30:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py deleted file mode 100644 index 062b2c1a48d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalPlainTimeOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_time_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_time.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_time_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_time.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_time_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_time.put_all( - body={"property": "12:30:00"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_plain_time_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.plain_time.put_default( - body={"property": "12:30:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py deleted file mode 100644 index 38b641dad1a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalRequiredAndOptionalOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_required_and_optional_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.required_and_optional.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_required_and_optional_get_required_only(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.required_and_optional.get_required_only() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_required_and_optional_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.required_and_optional.put_all( - body={"requiredProperty": 0, "optionalProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_required_and_optional_put_required_only(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.required_and_optional.put_required_only( - body={"requiredProperty": 0, "optionalProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py deleted file mode 100644 index 0d694036bda..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalRequiredAndOptionalOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_required_and_optional_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.required_and_optional.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_required_and_optional_get_required_only(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.required_and_optional.get_required_only() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_required_and_optional_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.required_and_optional.put_all( - body={"requiredProperty": 0, "optionalProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_required_and_optional_put_required_only(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.required_and_optional.put_required_only( - body={"requiredProperty": 0, "optionalProperty": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py deleted file mode 100644 index 3c75efd4b0a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalStringLiteralOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_string_literal_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_string_literal_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_string_literal_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string_literal.put_all( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_string_literal_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string_literal.put_default( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py deleted file mode 100644 index 0f48cbf56a8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalStringLiteralOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_literal_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_literal_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_literal_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string_literal.put_all( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_literal_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string_literal.put_default( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py deleted file mode 100644 index 1dcc560ba06..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalStringOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_string_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_string_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_string_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string.put_all( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_string_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.string.put_default( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py deleted file mode 100644 index d757488c9b5..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalStringOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string.put_all( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_string_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.string.put_default( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py deleted file mode 100644 index 775c2e338e9..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalUnionFloatLiteralOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_union_float_literal_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_float_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_float_literal_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_float_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_float_literal_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_float_literal.put_all( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_float_literal_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_float_literal.put_default( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py deleted file mode 100644 index 1c7555ba53e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalUnionFloatLiteralOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_float_literal_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_float_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_float_literal_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_float_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_float_literal_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_float_literal.put_all( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_float_literal_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_float_literal.put_default( - body={"property": 1.25}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py deleted file mode 100644 index a301c507aea..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalUnionIntLiteralOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_union_int_literal_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_int_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_int_literal_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_int_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_int_literal_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_int_literal.put_all( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_int_literal_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_int_literal.put_default( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py deleted file mode 100644 index 552dbfca317..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalUnionIntLiteralOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_int_literal_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_int_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_int_literal_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_int_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_int_literal_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_int_literal.put_all( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_int_literal_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_int_literal.put_default( - body={"property": 1}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py deleted file mode 100644 index 4cde960c6d4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py +++ /dev/null @@ -1,53 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import OptionalClientTestBase, OptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalUnionStringLiteralOperations(OptionalClientTestBase): - @OptionalPreparer() - @recorded_by_proxy - def test_union_string_literal_get_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_string_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_string_literal_get_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_string_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_string_literal_put_all(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_string_literal.put_all( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy - def test_union_string_literal_put_default(self, optional_endpoint): - client = self.create_client(endpoint=optional_endpoint) - response = client.union_string_literal.put_default( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py deleted file mode 100644 index 785ec15fdc2..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py +++ /dev/null @@ -1,54 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import OptionalPreparer -from testpreparer_async import OptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestOptionalUnionStringLiteralOperationsAsync(OptionalClientTestBaseAsync): - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_string_literal_get_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_string_literal.get_all() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_string_literal_get_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_string_literal.get_default() - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_string_literal_put_all(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_string_literal.put_all( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... - - @OptionalPreparer() - @recorded_by_proxy_async - async def test_union_string_literal_put_default(self, optional_endpoint): - client = self.create_async_client(endpoint=optional_endpoint) - response = await client.union_string_literal.put_default( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py deleted file mode 100644 index 3fea0db046e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.property.optional import OptionalClient - - -class OptionalClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(OptionalClient) - return self.create_client_from_credential( - OptionalClient, - credential=credential, - endpoint=endpoint, - ) - - -OptionalPreparer = functools.partial( - PowerShellPreparer, "optional", optional_endpoint="https://fake_optional_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py deleted file mode 100644 index 617123e80fe..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.property.optional.aio import OptionalClient - - -class OptionalClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(OptionalClient, is_async=True) - return self.create_client_from_credential( - OptionalClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-optional/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py deleted file mode 100644 index c2d5129d87a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - valuetypes_subscription_id = os.environ.get("VALUETYPES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - valuetypes_tenant_id = os.environ.get("VALUETYPES_TENANT_ID", "00000000-0000-0000-0000-000000000000") - valuetypes_client_id = os.environ.get("VALUETYPES_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - valuetypes_client_secret = os.environ.get("VALUETYPES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=valuetypes_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=valuetypes_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=valuetypes_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=valuetypes_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py deleted file mode 100644 index c3c3d54b963..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesBooleanLiteralOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_boolean_literal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.boolean_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_boolean_literal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.boolean_literal.put( - body={"property": True}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py deleted file mode 100644 index e8c8189aae4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesBooleanLiteralOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_boolean_literal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.boolean_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_boolean_literal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.boolean_literal.put( - body={"property": True}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py deleted file mode 100644 index 51309ac9860..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesBooleanOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_boolean_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.boolean.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_boolean_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.boolean.put( - body={"property": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py deleted file mode 100644 index 069b7a93c2b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesBooleanOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_boolean_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.boolean.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_boolean_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.boolean.put( - body={"property": bool}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py deleted file mode 100644 index 3f08982c2f8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesBytesOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_bytes_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.bytes.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_bytes_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.bytes.put( - body={"property": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py deleted file mode 100644 index dd9dcffc483..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesBytesOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_bytes_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.bytes.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_bytes_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.bytes.put( - body={"property": bytes("bytes", encoding="utf-8")}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py deleted file mode 100644 index df06ce23f99..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesCollectionsIntOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_collections_int_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.collections_int.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_collections_int_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.collections_int.put( - body={"property": [0]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py deleted file mode 100644 index 51b31ffdd77..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesCollectionsIntOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_collections_int_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.collections_int.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_collections_int_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.collections_int.put( - body={"property": [0]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py deleted file mode 100644 index 8fcab968ccd..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesCollectionsModelOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_collections_model_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.collections_model.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_collections_model_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.collections_model.put( - body={"property": [{"property": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py deleted file mode 100644 index 991461dc3dd..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesCollectionsModelOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_collections_model_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.collections_model.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_collections_model_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.collections_model.put( - body={"property": [{"property": "str"}]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py deleted file mode 100644 index 1ab22215eab..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesCollectionsStringOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_collections_string_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.collections_string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_collections_string_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.collections_string.put( - body={"property": ["str"]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py deleted file mode 100644 index cb8e324ceb2..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesCollectionsStringOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_collections_string_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.collections_string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_collections_string_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.collections_string.put( - body={"property": ["str"]}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py deleted file mode 100644 index a0eaa45c66a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDatetimeOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_datetime_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.datetime.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_datetime_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.datetime.put( - body={"property": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py deleted file mode 100644 index 888caa316a4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDatetimeOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_datetime_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.datetime.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_datetime_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.datetime.put( - body={"property": "2020-02-20 00:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py deleted file mode 100644 index 00676e9f6a1..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDecimal128Operations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_decimal128_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.decimal128.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_decimal128_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.decimal128.put( - body={"property": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py deleted file mode 100644 index 605c463e8df..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDecimal128OperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_decimal128_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.decimal128.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_decimal128_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.decimal128.put( - body={"property": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py deleted file mode 100644 index f495eea2814..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDecimalOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_decimal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.decimal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_decimal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.decimal.put( - body={"property": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py deleted file mode 100644 index 7153118e60d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDecimalOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_decimal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.decimal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_decimal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.decimal.put( - body={"property": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py deleted file mode 100644 index 9e09f1d71f4..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDictionaryStringOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_dictionary_string_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.dictionary_string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_dictionary_string_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.dictionary_string.put( - body={"property": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py deleted file mode 100644 index f24b8b13007..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDictionaryStringOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_dictionary_string_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.dictionary_string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_dictionary_string_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.dictionary_string.put( - body={"property": {"str": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py deleted file mode 100644 index 135aa3cb772..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDurationOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_duration_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.duration.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_duration_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.duration.put( - body={"property": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py deleted file mode 100644 index cac800420e5..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesDurationOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_duration_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.duration.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_duration_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.duration.put( - body={"property": "1 day, 0:00:00"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py deleted file mode 100644 index cc503ad2394..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesEnumOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_enum_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.enum.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_enum_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.enum.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py deleted file mode 100644 index 778528c75cf..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesEnumOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_enum_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.enum.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_enum_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.enum.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py deleted file mode 100644 index a095e152b17..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesExtensibleEnumOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_extensible_enum_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.extensible_enum.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_extensible_enum_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.extensible_enum.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py deleted file mode 100644 index 362775ee153..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesExtensibleEnumOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_extensible_enum_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.extensible_enum.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_extensible_enum_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.extensible_enum.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py deleted file mode 100644 index df265bbaafc..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesFloatLiteralOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_float_literal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.float_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_float_literal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.float_literal.put( - body={"property": 43.125}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py deleted file mode 100644 index 9bffccf770d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesFloatLiteralOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_float_literal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.float_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_float_literal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.float_literal.put( - body={"property": 43.125}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py deleted file mode 100644 index b0fe9df2e81..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesFloatOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_float_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.float.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_float_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.float.put( - body={"property": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py deleted file mode 100644 index 18f8789a51f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesFloatOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_float_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.float.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_float_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.float.put( - body={"property": 0.0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py deleted file mode 100644 index 061f04a21a5..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesIntLiteralOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_int_literal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.int_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_int_literal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.int_literal.put( - body={"property": 42}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py deleted file mode 100644 index 46d4783f71f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesIntLiteralOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_int_literal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.int_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_int_literal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.int_literal.put( - body={"property": 42}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py deleted file mode 100644 index a59b36e06e1..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesIntOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_int_operations_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.int_operations.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_int_operations_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.int_operations.put( - body={"property": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py deleted file mode 100644 index c694b12be4d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesIntOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_int_operations_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.int_operations.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_int_operations_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.int_operations.put( - body={"property": 0}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py deleted file mode 100644 index a65204abb54..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesModelOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_model_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.model.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_model_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.model.put( - body={"property": {"property": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py deleted file mode 100644 index aa51c3983e5..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesModelOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_model_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.model.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_model_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.model.put( - body={"property": {"property": "str"}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py deleted file mode 100644 index 9271124511a..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesNeverOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_never_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.never.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_never_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.never.put( - body={}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py deleted file mode 100644 index 3a5917baf1b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesNeverOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_never_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.never.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_never_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.never.put( - body={}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py deleted file mode 100644 index 906cdcc0ec9..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesStringLiteralOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_string_literal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.string_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_string_literal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.string_literal.put( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py deleted file mode 100644 index 40d3144d7f1..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesStringLiteralOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_string_literal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.string_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_string_literal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.string_literal.put( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py deleted file mode 100644 index 60cec79a37b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesStringOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_string_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_string_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.string.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py deleted file mode 100644 index 4da3d00b3ff..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesStringOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_string_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_string_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.string.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py deleted file mode 100644 index 8f3ada270c8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionEnumValueOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_enum_value_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_enum_value.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_enum_value_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_enum_value.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py deleted file mode 100644 index 356ae6028a7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionEnumValueOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_enum_value_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_enum_value.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_enum_value_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_enum_value.put( - body={"property": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py deleted file mode 100644 index 014f408a041..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionFloatLiteralOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_float_literal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_float_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_float_literal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_float_literal.put( - body={"property": 43.125}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py deleted file mode 100644 index a9690ec570c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionFloatLiteralOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_float_literal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_float_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_float_literal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_float_literal.put( - body={"property": 43.125}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py deleted file mode 100644 index 5388367dd02..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionIntLiteralOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_int_literal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_int_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_int_literal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_int_literal.put( - body={"property": 42}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py deleted file mode 100644 index 589cebfb201..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionIntLiteralOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_int_literal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_int_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_int_literal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_int_literal.put( - body={"property": 42}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py deleted file mode 100644 index 89f1206661e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionStringLiteralOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_string_literal_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_string_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_union_string_literal_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.union_string_literal.put( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py deleted file mode 100644 index b1d0eeb1cea..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnionStringLiteralOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_string_literal_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_string_literal.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_union_string_literal_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.union_string_literal.put( - body={"property": "hello"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py deleted file mode 100644 index 328bdaea272..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownArrayOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_array_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_array.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_array_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_array.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py deleted file mode 100644 index 7fcf40b5904..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownArrayOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_array_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_array.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_array_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_array.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py deleted file mode 100644 index f1f1b2d139f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownDictOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_dict_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_dict.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_dict_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_dict.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py deleted file mode 100644 index d43b1a5e594..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownDictOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_dict_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_dict.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_dict_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_dict.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py deleted file mode 100644 index 1fd3a8f292b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownIntOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_int_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_int.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_int_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_int.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py deleted file mode 100644 index 688cf49172d..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownIntOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_int_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_int.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_int_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_int.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py deleted file mode 100644 index de51f421a4f..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py +++ /dev/null @@ -1,33 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownStringOperations(ValueTypesClientTestBase): - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_string_get(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy - def test_unknown_string_put(self, valuetypes_endpoint): - client = self.create_client(endpoint=valuetypes_endpoint) - response = client.unknown_string.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py deleted file mode 100644 index 4c60198bcef..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ValueTypesPreparer -from testpreparer_async import ValueTypesClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestValueTypesUnknownStringOperationsAsync(ValueTypesClientTestBaseAsync): - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_string_get(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_string.get() - - # please add some check logic here by yourself - # ... - - @ValueTypesPreparer() - @recorded_by_proxy_async - async def test_unknown_string_put(self, valuetypes_endpoint): - client = self.create_async_client(endpoint=valuetypes_endpoint) - response = await client.unknown_string.put( - body={"property": {}}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py deleted file mode 100644 index 8c8476a2489..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.property.valuetypes import ValueTypesClient - - -class ValueTypesClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ValueTypesClient) - return self.create_client_from_credential( - ValueTypesClient, - credential=credential, - endpoint=endpoint, - ) - - -ValueTypesPreparer = functools.partial( - PowerShellPreparer, "valuetypes", valuetypes_endpoint="https://fake_valuetypes_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py deleted file mode 100644 index 08c15283880..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.property.valuetypes.aio import ValueTypesClient - - -class ValueTypesClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ValueTypesClient, is_async=True) - return self.create_client_from_credential( - ValueTypesClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-valuetypes/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py deleted file mode 100644 index f7252bc7a1b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - scalar_subscription_id = os.environ.get("SCALAR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - scalar_tenant_id = os.environ.get("SCALAR_TENANT_ID", "00000000-0000-0000-0000-000000000000") - scalar_client_id = os.environ.get("SCALAR_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - scalar_client_secret = os.environ.get("SCALAR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=scalar_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py deleted file mode 100644 index 8ed568688c1..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarBooleanOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_boolean_get(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.boolean.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_boolean_put(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.boolean.put( - body=bool, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py deleted file mode 100644 index beb9b7ee726..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarBooleanOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_boolean_get(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.boolean.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_boolean_put(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.boolean.put( - body=bool, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py deleted file mode 100644 index a66b1a98438..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimal128TypeOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_decimal128_type_response_body(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal128_type.response_body() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_decimal128_type_request_body(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal128_type.request_body( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_decimal128_type_request_parameter(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal128_type.request_parameter( - value=0.0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py deleted file mode 100644 index 8d30937e1da..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py +++ /dev/null @@ -1,46 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimal128TypeOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal128_type_response_body(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal128_type.response_body() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal128_type_request_body(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal128_type.request_body( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal128_type_request_parameter(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal128_type.request_parameter( - value=0.0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py deleted file mode 100644 index 575d594f1da..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimal128VerifyOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_decimal128_verify_prepare_verify(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal128_verify.prepare_verify() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_decimal128_verify_verify(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal128_verify.verify( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py deleted file mode 100644 index c8a29772379..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimal128VerifyOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal128_verify_prepare_verify(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal128_verify.prepare_verify() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal128_verify_verify(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal128_verify.verify( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py deleted file mode 100644 index 744251a8c93..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimalTypeOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_decimal_type_response_body(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal_type.response_body() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_decimal_type_request_body(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal_type.request_body( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_decimal_type_request_parameter(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal_type.request_parameter( - value=0.0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py deleted file mode 100644 index 8ceb3f201ac..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py +++ /dev/null @@ -1,46 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimalTypeOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal_type_response_body(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal_type.response_body() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal_type_request_body(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal_type.request_body( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal_type_request_parameter(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal_type.request_parameter( - value=0.0, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py deleted file mode 100644 index 4b4b954a380..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimalVerifyOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_decimal_verify_prepare_verify(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal_verify.prepare_verify() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_decimal_verify_verify(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.decimal_verify.verify( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py deleted file mode 100644 index 0d649c9cdec..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarDecimalVerifyOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal_verify_prepare_verify(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal_verify.prepare_verify() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_decimal_verify_verify(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.decimal_verify.verify( - body=0.0, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py deleted file mode 100644 index 951f291fca6..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarStringOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_string_get(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.string.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_string_put(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.string.put( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py deleted file mode 100644 index 03824836948..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarStringOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_string_get(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.string.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_string_put(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.string.put( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py deleted file mode 100644 index 9af0d808bc9..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ScalarClientTestBase, ScalarPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarUnknownOperations(ScalarClientTestBase): - @ScalarPreparer() - @recorded_by_proxy - def test_unknown_get(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.unknown.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy - def test_unknown_put(self, scalar_endpoint): - client = self.create_client(endpoint=scalar_endpoint) - response = client.unknown.put( - body={}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py deleted file mode 100644 index fd3aa53ac1b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ScalarPreparer -from testpreparer_async import ScalarClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestScalarUnknownOperationsAsync(ScalarClientTestBaseAsync): - @ScalarPreparer() - @recorded_by_proxy_async - async def test_unknown_get(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.unknown.get() - - # please add some check logic here by yourself - # ... - - @ScalarPreparer() - @recorded_by_proxy_async - async def test_unknown_put(self, scalar_endpoint): - client = self.create_async_client(endpoint=scalar_endpoint) - response = await client.unknown.put( - body={}, - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py deleted file mode 100644 index 3ba91005381..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.scalar import ScalarClient - - -class ScalarClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ScalarClient) - return self.create_client_from_credential( - ScalarClient, - credential=credential, - endpoint=endpoint, - ) - - -ScalarPreparer = functools.partial(PowerShellPreparer, "scalar", scalar_endpoint="https://fake_scalar_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py deleted file mode 100644 index f3e076a1b41..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.scalar.aio import ScalarClient - - -class ScalarClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ScalarClient, is_async=True) - return self.create_client_from_credential( - ScalarClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-scalar/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py deleted file mode 100644 index 5e28d6377c8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - union_subscription_id = os.environ.get("UNION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - union_tenant_id = os.environ.get("UNION_TENANT_ID", "00000000-0000-0000-0000-000000000000") - union_client_id = os.environ.get("UNION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - union_client_secret = os.environ.get("UNION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=union_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py deleted file mode 100644 index e35f531e7af..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionEnumsOnlyOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_enums_only_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.enums_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_enums_only_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.enums_only.send( - body={"prop": {"lr": "left", "ud": "up"}}, - prop={"lr": "left", "ud": "up"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py deleted file mode 100644 index 7927950425b..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionEnumsOnlyOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_enums_only_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.enums_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_enums_only_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.enums_only.send( - body={"prop": {"lr": "left", "ud": "up"}}, - prop={"lr": "left", "ud": "up"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py deleted file mode 100644 index 1f6b8c84fc7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionFloatsOnlyOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_floats_only_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.floats_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_floats_only_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.floats_only.send( - body={"prop": 1.1}, - prop=1.1, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py deleted file mode 100644 index 95c162de097..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionFloatsOnlyOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_floats_only_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.floats_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_floats_only_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.floats_only.send( - body={"prop": 1.1}, - prop=1.1, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py deleted file mode 100644 index 8eb6efc6d13..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionIntsOnlyOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_ints_only_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.ints_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_ints_only_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.ints_only.send( - body={"prop": 1}, - prop=1, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py deleted file mode 100644 index 876727cdc0c..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionIntsOnlyOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_ints_only_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.ints_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_ints_only_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.ints_only.send( - body={"prop": 1}, - prop=1, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py deleted file mode 100644 index 2b46b3270d9..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionMixedLiteralsOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_mixed_literals_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.mixed_literals.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_mixed_literals_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.mixed_literals.send( - body={"prop": {"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}}, - prop={"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py deleted file mode 100644 index 356f7496bf0..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionMixedLiteralsOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_mixed_literals_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.mixed_literals.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_mixed_literals_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.mixed_literals.send( - body={"prop": {"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}}, - prop={"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py deleted file mode 100644 index 325317368c6..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py +++ /dev/null @@ -1,48 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionMixedTypesOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_mixed_types_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.mixed_types.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_mixed_types_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.mixed_types.send( - body={ - "prop": { - "array": [{"name": "str"}], - "boolean": {"name": "str"}, - "int": {"name": "str"}, - "literal": {"name": "str"}, - "model": {"name": "str"}, - } - }, - prop={ - "array": [{"name": "str"}], - "boolean": {"name": "str"}, - "int": {"name": "str"}, - "literal": {"name": "str"}, - "model": {"name": "str"}, - }, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py deleted file mode 100644 index b146ea74056..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py +++ /dev/null @@ -1,49 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionMixedTypesOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_mixed_types_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.mixed_types.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_mixed_types_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.mixed_types.send( - body={ - "prop": { - "array": [{"name": "str"}], - "boolean": {"name": "str"}, - "int": {"name": "str"}, - "literal": {"name": "str"}, - "model": {"name": "str"}, - } - }, - prop={ - "array": [{"name": "str"}], - "boolean": {"name": "str"}, - "int": {"name": "str"}, - "literal": {"name": "str"}, - "model": {"name": "str"}, - }, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py deleted file mode 100644 index e932af10a22..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionModelsOnlyOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_models_only_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.models_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_models_only_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.models_only.send( - body={"prop": {"name": "str"}}, - prop={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py deleted file mode 100644 index 1ee10f9b68e..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionModelsOnlyOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_models_only_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.models_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_models_only_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.models_only.send( - body={"prop": {"name": "str"}}, - prop={"name": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py deleted file mode 100644 index 4e68cad1acd..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringAndArrayOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_string_and_array_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.string_and_array.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_string_and_array_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.string_and_array.send( - body={"prop": {"array": "str", "string": "str"}}, - prop={"array": "str", "string": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py deleted file mode 100644 index 7692fd0bd05..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringAndArrayOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_string_and_array_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.string_and_array.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_string_and_array_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.string_and_array.send( - body={"prop": {"array": "str", "string": "str"}}, - prop={"array": "str", "string": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py deleted file mode 100644 index b05ecc2f0c7..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringExtensibleNamedOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_string_extensible_named_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.string_extensible_named.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_string_extensible_named_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.string_extensible_named.send( - body={"prop": "str"}, - prop="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py deleted file mode 100644 index 63d9457c2a8..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringExtensibleNamedOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_string_extensible_named_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.string_extensible_named.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_string_extensible_named_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.string_extensible_named.send( - body={"prop": "str"}, - prop="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py deleted file mode 100644 index 4514fdc2676..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringExtensibleOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_string_extensible_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.string_extensible.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_string_extensible_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.string_extensible.send( - body={"prop": "b"}, - prop="b", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py deleted file mode 100644 index 7150caa0edd..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringExtensibleOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_string_extensible_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.string_extensible.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_string_extensible_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.string_extensible.send( - body={"prop": "b"}, - prop="b", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py deleted file mode 100644 index 2181e8a0ced..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py +++ /dev/null @@ -1,34 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import UnionClientTestBase, UnionPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringsOnlyOperations(UnionClientTestBase): - @UnionPreparer() - @recorded_by_proxy - def test_strings_only_get(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.strings_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy - def test_strings_only_send(self, union_endpoint): - client = self.create_client(endpoint=union_endpoint) - response = client.strings_only.send( - body={"prop": "a"}, - prop="a", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py deleted file mode 100644 index 89d23916804..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import UnionPreparer -from testpreparer_async import UnionClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestUnionStringsOnlyOperationsAsync(UnionClientTestBaseAsync): - @UnionPreparer() - @recorded_by_proxy_async - async def test_strings_only_get(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.strings_only.get() - - # please add some check logic here by yourself - # ... - - @UnionPreparer() - @recorded_by_proxy_async - async def test_strings_only_send(self, union_endpoint): - client = self.create_async_client(endpoint=union_endpoint) - response = await client.strings_only.send( - body={"prop": "a"}, - prop="a", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py deleted file mode 100644 index 0d66b2cf3fa..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from typetest.union import UnionClient - - -class UnionClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(UnionClient) - return self.create_client_from_credential( - UnionClient, - credential=credential, - endpoint=endpoint, - ) - - -UnionPreparer = functools.partial(PowerShellPreparer, "union", union_endpoint="https://fake_union_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py deleted file mode 100644 index d61fa77c079..00000000000 --- a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from typetest.union.aio import UnionClient - - -class UnionClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(UnionClient, is_async=True) - return self.create_client_from_credential( - UnionClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/typetest-union/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-union/typetest/union/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-union/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-union/typetest/union/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py deleted file mode 100644 index 37ef9defab6..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - added_subscription_id = os.environ.get("ADDED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - added_tenant_id = os.environ.get("ADDED_TENANT_ID", "00000000-0000-0000-0000-000000000000") - added_client_id = os.environ.get("ADDED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - added_client_secret = os.environ.get("ADDED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=added_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=added_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=added_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=added_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py deleted file mode 100644 index 21f3c1c245f..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AddedClientTestBase, AddedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAdded(AddedClientTestBase): - @AddedPreparer() - @recorded_by_proxy - def test_v1(self, added_endpoint): - client = self.create_client(endpoint=added_endpoint) - response = client.v1( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - header_v2="str", - ) - - # please add some check logic here by yourself - # ... - - @AddedPreparer() - @recorded_by_proxy - def test_v2(self, added_endpoint): - client = self.create_client(endpoint=added_endpoint) - response = client.v2( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py deleted file mode 100644 index c0aa575e088..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AddedPreparer -from testpreparer_async import AddedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAddedAsync(AddedClientTestBaseAsync): - @AddedPreparer() - @recorded_by_proxy_async - async def test_v1(self, added_endpoint): - client = self.create_async_client(endpoint=added_endpoint) - response = await client.v1( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - header_v2="str", - ) - - # please add some check logic here by yourself - # ... - - @AddedPreparer() - @recorded_by_proxy_async - async def test_v2(self, added_endpoint): - client = self.create_async_client(endpoint=added_endpoint) - response = await client.v2( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py deleted file mode 100644 index d29deadeedd..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import AddedClientTestBase, AddedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAddedInterfaceV2Operations(AddedClientTestBase): - @AddedPreparer() - @recorded_by_proxy - def test_interface_v2_v2_in_interface(self, added_endpoint): - client = self.create_client(endpoint=added_endpoint) - response = client.interface_v2.v2_in_interface( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py deleted file mode 100644 index 653c57ae97f..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import AddedPreparer -from testpreparer_async import AddedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestAddedInterfaceV2OperationsAsync(AddedClientTestBaseAsync): - @AddedPreparer() - @recorded_by_proxy_async - async def test_interface_v2_v2_in_interface(self, added_endpoint): - client = self.create_async_client(endpoint=added_endpoint) - response = await client.interface_v2.v2_in_interface( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py deleted file mode 100644 index da5e39b6fe6..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from versioning.added import AddedClient - - -class AddedClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(AddedClient) - return self.create_client_from_credential( - AddedClient, - credential=credential, - endpoint=endpoint, - ) - - -AddedPreparer = functools.partial(PowerShellPreparer, "added", added_endpoint="https://fake_added_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py deleted file mode 100644 index fb7f4883db3..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from versioning.added.aio import AddedClient - - -class AddedClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(AddedClient, is_async=True) - return self.create_client_from_credential( - AddedClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/versioning-added/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-added/versioning/added/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-added/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-added/versioning/added/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py deleted file mode 100644 index 02d95966ca8..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - madeoptional_subscription_id = os.environ.get( - "MADEOPTIONAL_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - madeoptional_tenant_id = os.environ.get("MADEOPTIONAL_TENANT_ID", "00000000-0000-0000-0000-000000000000") - madeoptional_client_id = os.environ.get("MADEOPTIONAL_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - madeoptional_client_secret = os.environ.get("MADEOPTIONAL_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=madeoptional_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=madeoptional_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=madeoptional_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=madeoptional_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py deleted file mode 100644 index 13cd7b9ea12..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import MadeOptionalClientTestBase, MadeOptionalPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMadeOptional(MadeOptionalClientTestBase): - @MadeOptionalPreparer() - @recorded_by_proxy - def test_test(self, madeoptional_endpoint): - client = self.create_client(endpoint=madeoptional_endpoint) - response = client.test( - body={"prop": "str", "changedProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py deleted file mode 100644 index 6a6baf3922c..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import MadeOptionalPreparer -from testpreparer_async import MadeOptionalClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestMadeOptionalAsync(MadeOptionalClientTestBaseAsync): - @MadeOptionalPreparer() - @recorded_by_proxy_async - async def test_test(self, madeoptional_endpoint): - client = self.create_async_client(endpoint=madeoptional_endpoint) - response = await client.test( - body={"prop": "str", "changedProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py deleted file mode 100644 index e2d307ecbe9..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from versioning.madeoptional import MadeOptionalClient - - -class MadeOptionalClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(MadeOptionalClient) - return self.create_client_from_credential( - MadeOptionalClient, - credential=credential, - endpoint=endpoint, - ) - - -MadeOptionalPreparer = functools.partial( - PowerShellPreparer, "madeoptional", madeoptional_endpoint="https://fake_madeoptional_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py deleted file mode 100644 index 394e086e4ac..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from versioning.madeoptional.aio import MadeOptionalClient - - -class MadeOptionalClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(MadeOptionalClient, is_async=True) - return self.create_client_from_credential( - MadeOptionalClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-madeoptional/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py deleted file mode 100644 index dd62b27f5ea..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - removed_subscription_id = os.environ.get("REMOVED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - removed_tenant_id = os.environ.get("REMOVED_TENANT_ID", "00000000-0000-0000-0000-000000000000") - removed_client_id = os.environ.get("REMOVED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - removed_client_secret = os.environ.get("REMOVED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=removed_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=removed_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=removed_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=removed_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py deleted file mode 100644 index d3f57f2b16a..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RemovedClientTestBase, RemovedPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRemoved(RemovedClientTestBase): - @RemovedPreparer() - @recorded_by_proxy - def test_v2(self, removed_endpoint): - client = self.create_client(endpoint=removed_endpoint) - response = client.v2( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @RemovedPreparer() - @recorded_by_proxy - def test_model_v3(self, removed_endpoint): - client = self.create_client(endpoint=removed_endpoint) - response = client.model_v3( - body={"enumProp": "str", "id": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py deleted file mode 100644 index 0a5c7bee85a..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RemovedPreparer -from testpreparer_async import RemovedClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRemovedAsync(RemovedClientTestBaseAsync): - @RemovedPreparer() - @recorded_by_proxy_async - async def test_v2(self, removed_endpoint): - client = self.create_async_client(endpoint=removed_endpoint) - response = await client.v2( - body={"enumProp": "str", "prop": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... - - @RemovedPreparer() - @recorded_by_proxy_async - async def test_model_v3(self, removed_endpoint): - client = self.create_async_client(endpoint=removed_endpoint) - response = await client.model_v3( - body={"enumProp": "str", "id": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py deleted file mode 100644 index 60f068de6e5..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from versioning.removed import RemovedClient - - -class RemovedClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(RemovedClient) - return self.create_client_from_credential( - RemovedClient, - credential=credential, - endpoint=endpoint, - ) - - -RemovedPreparer = functools.partial(PowerShellPreparer, "removed", removed_endpoint="https://fake_removed_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py deleted file mode 100644 index fb44cbf1a9c..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from versioning.removed.aio import RemovedClient - - -class RemovedClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(RemovedClient, is_async=True) - return self.create_client_from_credential( - RemovedClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-removed/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py deleted file mode 100644 index 4a15c18fbd5..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py +++ /dev/null @@ -1,35 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - renamedfrom_subscription_id = os.environ.get("RENAMEDFROM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") - renamedfrom_tenant_id = os.environ.get("RENAMEDFROM_TENANT_ID", "00000000-0000-0000-0000-000000000000") - renamedfrom_client_id = os.environ.get("RENAMEDFROM_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - renamedfrom_client_secret = os.environ.get("RENAMEDFROM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=renamedfrom_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=renamedfrom_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=renamedfrom_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=renamedfrom_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py deleted file mode 100644 index 724043b1b94..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RenamedFromClientTestBase, RenamedFromPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedFrom(RenamedFromClientTestBase): - @RenamedFromPreparer() - @recorded_by_proxy - def test_new_op(self, renamedfrom_endpoint): - client = self.create_client(endpoint=renamedfrom_endpoint) - response = client.new_op( - body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, - new_query="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py deleted file mode 100644 index aac2319e12b..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RenamedFromPreparer -from testpreparer_async import RenamedFromClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedFromAsync(RenamedFromClientTestBaseAsync): - @RenamedFromPreparer() - @recorded_by_proxy_async - async def test_new_op(self, renamedfrom_endpoint): - client = self.create_async_client(endpoint=renamedfrom_endpoint) - response = await client.new_op( - body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, - new_query="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py deleted file mode 100644 index d3aa2db48ef..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py +++ /dev/null @@ -1,24 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import RenamedFromClientTestBase, RenamedFromPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedFromNewInterfaceOperations(RenamedFromClientTestBase): - @RenamedFromPreparer() - @recorded_by_proxy - def test_new_interface_new_op_in_new_interface(self, renamedfrom_endpoint): - client = self.create_client(endpoint=renamedfrom_endpoint) - response = client.new_interface.new_op_in_new_interface( - body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py deleted file mode 100644 index ded1a984fd9..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import RenamedFromPreparer -from testpreparer_async import RenamedFromClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestRenamedFromNewInterfaceOperationsAsync(RenamedFromClientTestBaseAsync): - @RenamedFromPreparer() - @recorded_by_proxy_async - async def test_new_interface_new_op_in_new_interface(self, renamedfrom_endpoint): - client = self.create_async_client(endpoint=renamedfrom_endpoint) - response = await client.new_interface.new_op_in_new_interface( - body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py deleted file mode 100644 index af5459a16cb..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from versioning.renamedfrom import RenamedFromClient - - -class RenamedFromClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(RenamedFromClient) - return self.create_client_from_credential( - RenamedFromClient, - credential=credential, - endpoint=endpoint, - ) - - -RenamedFromPreparer = functools.partial( - PowerShellPreparer, "renamedfrom", renamedfrom_endpoint="https://fake_renamedfrom_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py deleted file mode 100644 index 2a3a04d4cd9..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from versioning.renamedfrom.aio import RenamedFromClient - - -class RenamedFromClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(RenamedFromClient, is_async=True) - return self.create_client_from_credential( - RenamedFromClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-renamedfrom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py deleted file mode 100644 index 2d8b087a478..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py +++ /dev/null @@ -1,45 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - returntypechangedfrom_subscription_id = os.environ.get( - "RETURNTYPECHANGEDFROM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - returntypechangedfrom_tenant_id = os.environ.get( - "RETURNTYPECHANGEDFROM_TENANT_ID", "00000000-0000-0000-0000-000000000000" - ) - returntypechangedfrom_client_id = os.environ.get( - "RETURNTYPECHANGEDFROM_CLIENT_ID", "00000000-0000-0000-0000-000000000000" - ) - returntypechangedfrom_client_secret = os.environ.get( - "RETURNTYPECHANGEDFROM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer( - regex=returntypechangedfrom_subscription_id, value="00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=returntypechangedfrom_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=returntypechangedfrom_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=returntypechangedfrom_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py deleted file mode 100644 index 27b6ee78d2b..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import ReturnTypeChangedFromClientTestBase, ReturnTypeChangedFromPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestReturnTypeChangedFrom(ReturnTypeChangedFromClientTestBase): - @ReturnTypeChangedFromPreparer() - @recorded_by_proxy - def test_test(self, returntypechangedfrom_endpoint): - client = self.create_client(endpoint=returntypechangedfrom_endpoint) - response = client.test( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py deleted file mode 100644 index b75ea6fa0ab..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import ReturnTypeChangedFromPreparer -from testpreparer_async import ReturnTypeChangedFromClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestReturnTypeChangedFromAsync(ReturnTypeChangedFromClientTestBaseAsync): - @ReturnTypeChangedFromPreparer() - @recorded_by_proxy_async - async def test_test(self, returntypechangedfrom_endpoint): - client = self.create_async_client(endpoint=returntypechangedfrom_endpoint) - response = await client.test( - body="str", - content_type="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py deleted file mode 100644 index 3db19f8ddbe..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from versioning.returntypechangedfrom import ReturnTypeChangedFromClient - - -class ReturnTypeChangedFromClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(ReturnTypeChangedFromClient) - return self.create_client_from_credential( - ReturnTypeChangedFromClient, - credential=credential, - endpoint=endpoint, - ) - - -ReturnTypeChangedFromPreparer = functools.partial( - PowerShellPreparer, - "returntypechangedfrom", - returntypechangedfrom_endpoint="https://fake_returntypechangedfrom_endpoint.com", -) diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py deleted file mode 100644 index 9b673fad3f1..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from versioning.returntypechangedfrom.aio import ReturnTypeChangedFromClient - - -class ReturnTypeChangedFromClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(ReturnTypeChangedFromClient, is_async=True) - return self.create_client_from_credential( - ReturnTypeChangedFromClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py deleted file mode 100644 index 9fcd84c05df..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py +++ /dev/null @@ -1,39 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import os -import pytest -from dotenv import load_dotenv -from devtools_testutils import ( - test_proxy, - add_general_regex_sanitizer, - add_body_key_sanitizer, - add_header_regex_sanitizer, -) - -load_dotenv() - - -# For security, please avoid record sensitive identity information in recordings -@pytest.fixture(scope="session", autouse=True) -def add_sanitizers(test_proxy): - typechangedfrom_subscription_id = os.environ.get( - "TYPECHANGEDFROM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" - ) - typechangedfrom_tenant_id = os.environ.get("TYPECHANGEDFROM_TENANT_ID", "00000000-0000-0000-0000-000000000000") - typechangedfrom_client_id = os.environ.get("TYPECHANGEDFROM_CLIENT_ID", "00000000-0000-0000-0000-000000000000") - typechangedfrom_client_secret = os.environ.get( - "TYPECHANGEDFROM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" - ) - add_general_regex_sanitizer(regex=typechangedfrom_subscription_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=typechangedfrom_tenant_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=typechangedfrom_client_id, value="00000000-0000-0000-0000-000000000000") - add_general_regex_sanitizer(regex=typechangedfrom_client_secret, value="00000000-0000-0000-0000-000000000000") - - add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") - add_header_regex_sanitizer(key="Cookie", value="cookie;") - add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py deleted file mode 100644 index f923927e7d8..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py +++ /dev/null @@ -1,25 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils import recorded_by_proxy -from testpreparer import TypeChangedFromClientTestBase, TypeChangedFromPreparer - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTypeChangedFrom(TypeChangedFromClientTestBase): - @TypeChangedFromPreparer() - @recorded_by_proxy - def test_test(self, typechangedfrom_endpoint): - client = self.create_client(endpoint=typechangedfrom_endpoint) - response = client.test( - body={"changedProp": "str", "prop": "str"}, - param="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py deleted file mode 100644 index cc7fe639551..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -import pytest -from devtools_testutils.aio import recorded_by_proxy_async -from testpreparer import TypeChangedFromPreparer -from testpreparer_async import TypeChangedFromClientTestBaseAsync - - -@pytest.mark.skip("you may need to update the auto-generated test case before run it") -class TestTypeChangedFromAsync(TypeChangedFromClientTestBaseAsync): - @TypeChangedFromPreparer() - @recorded_by_proxy_async - async def test_test(self, typechangedfrom_endpoint): - client = self.create_async_client(endpoint=typechangedfrom_endpoint) - response = await client.test( - body={"changedProp": "str", "prop": "str"}, - param="str", - ) - - # please add some check logic here by yourself - # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py deleted file mode 100644 index 69d22259ad2..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py +++ /dev/null @@ -1,26 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer -import functools -from versioning.typechangedfrom import TypeChangedFromClient - - -class TypeChangedFromClientTestBase(AzureRecordedTestCase): - - def create_client(self, endpoint): - credential = self.get_credential(TypeChangedFromClient) - return self.create_client_from_credential( - TypeChangedFromClient, - credential=credential, - endpoint=endpoint, - ) - - -TypeChangedFromPreparer = functools.partial( - PowerShellPreparer, "typechangedfrom", typechangedfrom_endpoint="https://fake_typechangedfrom_endpoint.com" -) diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py deleted file mode 100644 index cf09aeeeca5..00000000000 --- a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py +++ /dev/null @@ -1,20 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) Python Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- -from devtools_testutils import AzureRecordedTestCase -from versioning.typechangedfrom.aio import TypeChangedFromClient - - -class TypeChangedFromClientTestBaseAsync(AzureRecordedTestCase): - - def create_async_client(self, endpoint): - credential = self.get_credential(TypeChangedFromClient, is_async=True) - return self.create_client_from_credential( - TypeChangedFromClient, - credential=credential, - endpoint=endpoint, - ) diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-typechangedfrom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/models/_models.py b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/models/_models.py index 7fd18199e95..89f206a7df2 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/models/_models.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/models/_models.py @@ -16,6 +16,7 @@ class InvalidAuth(_model_base.Model): """InvalidAuth. + All required parameters must be populated in order to send to server. :ivar error: Required. :vartype error: str diff --git a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/models/_models.py b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/models/_models.py index 7fd18199e95..89f206a7df2 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/models/_models.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/models/_models.py @@ -16,6 +16,7 @@ class InvalidAuth(_model_base.Model): """InvalidAuth. + All required parameters must be populated in order to send to server. :ivar error: Required. :vartype error: str diff --git a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/models/_models.py b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/models/_models.py index 7fd18199e95..89f206a7df2 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/models/_models.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/models/_models.py @@ -16,6 +16,7 @@ class InvalidAuth(_model_base.Model): """InvalidAuth. + All required parameters must be populated in order to send to server. :ivar error: Required. :vartype error: str From c44571926c4e3c4a2f63a3fd94438121ab69c1b0 Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 08:48:41 +0000 Subject: [PATCH 3/8] Regenerate for autorest.python (2024-12-06 08:48:41) --- .../multiapi/generated/azure/__init__.py | 1 + .../generated/azure/multiapi/__init__.py | 1 + .../azure/multiapi/sample/v1/_metadata.json | 4 +- .../multiapi/sample/v1/models/_models_py3.py | 5 +- .../azure/multiapi/sample/v3/_metadata.json | 4 +- .../multiapi/sample/v3/models/_models_py3.py | 5 +- .../generated_samples/duration_get_null.py | 27 - .../duration_put_positive_duration.py | 30 - .../generated_samples/paths_get_empty.py | 28 - .../custompollerpager/models/_models_py3.py | 5 +- .../custombaseurlpaging/models/_models_py3.py | 5 +- .../paging_get_pages_partial_url.py | 30 - .../paging_get_pages_partial_url_operation.py | 30 - .../generated_samples/http_success_head200.py | 37 - .../generated_samples/http_success_head204.py | 37 - .../generated_samples/http_success_head404.py | 37 - .../head_exception_head200.py | 37 - .../head_exception_head204.py | 37 - .../generated_samples/http_success_head200.py | 36 - .../generated_samples/http_success_head204.py | 36 - .../generated_samples/http_success_head404.py | 36 - .../Lro/lro/models/_models_py3.py | 5 +- .../batch/paging/models/_models_py3.py | 5 +- .../Paging/paging/models/_models_py3.py | 5 +- .../storage/models/_models_py3.py | 5 +- .../generated_samples/duration_get_null.py | 27 - .../duration_put_positive_duration.py | 30 - .../generated_samples/paths_get_empty.py | 28 - .../paging_get_pages_partial_url.py | 30 - .../paging_get_pages_partial_url_operation.py | 30 - .../head_exception_head200.py | 37 - .../head_exception_head204.py | 37 - .../generated_samples/http_success_head200.py | 37 - .../generated_samples/http_success_head204.py | 37 - .../generated_samples/http_success_head404.py | 37 - .../models/_models.py | 5 +- .../Multiapi/multiapi/v0/__init__.py | 5 +- .../Multiapi/multiapi/v0/_configuration.py | 43 +- .../multiapi/v0/_multiapi_service_client.py | 37 +- .../Multiapi/multiapi/v0/aio/__init__.py | 5 +- .../multiapi/v0/aio/_configuration.py | 43 +- .../v0/aio/_multiapi_service_client.py | 33 +- .../multiapi/v0/aio/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 46 +- .../Multiapi/multiapi/v0/models/__init__.py | 9 +- .../multiapi/v0/models/_models_py3.py | 29 +- .../multiapi/v0/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 66 +- .../Multiapi/multiapi/v1/__init__.py | 5 +- .../Multiapi/multiapi/v1/_configuration.py | 43 +- .../Multiapi/multiapi/v1/_metadata.json | 4 +- .../multiapi/v1/_multiapi_service_client.py | 37 +- .../Multiapi/multiapi/v1/_vendor.py | 7 +- .../Multiapi/multiapi/v1/aio/__init__.py | 5 +- .../multiapi/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 33 +- .../Multiapi/multiapi/v1/aio/_vendor.py | 7 +- .../multiapi/v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 ++++--- .../_operation_group_one_operations.py | 46 +- .../Multiapi/multiapi/v1/models/__init__.py | 11 +- .../multiapi/v1/models/_models_py3.py | 38 +- .../multiapi/v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++++---- .../_operation_group_one_operations.py | 66 +- .../Multiapi/multiapi/v2/__init__.py | 5 +- .../Multiapi/multiapi/v2/_configuration.py | 43 +- .../multiapi/v2/_multiapi_service_client.py | 39 +- .../Multiapi/multiapi/v2/_vendor.py | 7 +- .../Multiapi/multiapi/v2/aio/__init__.py | 5 +- .../multiapi/v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 35 +- .../Multiapi/multiapi/v2/aio/_vendor.py | 7 +- .../multiapi/v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 ++- .../_operation_group_two_operations.py | 47 +- .../Multiapi/multiapi/v2/models/__init__.py | 7 +- .../multiapi/v2/models/_models_py3.py | 18 +- .../multiapi/v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../Multiapi/multiapi/v3/__init__.py | 5 +- .../Multiapi/multiapi/v3/_configuration.py | 43 +- .../Multiapi/multiapi/v3/_metadata.json | 4 +- .../multiapi/v3/_multiapi_service_client.py | 39 +- .../Multiapi/multiapi/v3/_vendor.py | 7 +- .../Multiapi/multiapi/v3/aio/__init__.py | 5 +- .../multiapi/v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 35 +- .../Multiapi/multiapi/v3/aio/_vendor.py | 7 +- .../multiapi/v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 ++- .../_operation_group_one_operations.py | 115 +-- .../_operation_group_two_operations.py | 92 ++- .../Multiapi/multiapi/v3/models/__init__.py | 13 +- .../multiapi/v3/models/_models_py3.py | 39 +- .../models/_multiapi_service_client_enums.py | 3 +- .../multiapi/v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../v1/__init__.py | 5 +- .../v1/_configuration.py | 37 +- .../v1/_metadata.json | 4 +- .../v1/_multiapi_service_client.py | 37 +- .../v1/_vendor.py | 7 +- .../v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 37 +- .../v1/aio/_multiapi_service_client.py | 35 +- .../v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 ++++--- .../_operation_group_one_operations.py | 46 +- .../v1/models/__init__.py | 11 +- .../v1/models/_models_py3.py | 38 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++++---- .../_operation_group_one_operations.py | 66 +- .../v2/__init__.py | 5 +- .../v2/_configuration.py | 37 +- .../v2/_multiapi_service_client.py | 39 +- .../v2/_vendor.py | 7 +- .../v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 37 +- .../v2/aio/_multiapi_service_client.py | 37 +- .../v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 ++- .../_operation_group_two_operations.py | 47 +- .../v2/models/__init__.py | 7 +- .../v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../v3/__init__.py | 5 +- .../v3/_configuration.py | 37 +- .../v3/_metadata.json | 4 +- .../v3/_multiapi_service_client.py | 39 +- .../v3/_vendor.py | 7 +- .../v3/aio/__init__.py | 5 +- .../v3/aio/_configuration.py | 37 +- .../v3/aio/_multiapi_service_client.py | 37 +- .../v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 ++- .../_operation_group_one_operations.py | 115 +-- .../_operation_group_two_operations.py | 92 ++- .../v3/models/__init__.py | 13 +- .../v3/models/_models_py3.py | 39 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapicustombaseurl/v1/__init__.py | 5 +- .../v1/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 42 +- .../multiapicustombaseurl/v1/_vendor.py | 7 +- .../multiapicustombaseurl/v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 40 +- .../multiapicustombaseurl/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 44 +- .../v1/models/__init__.py | 5 +- .../v1/models/_models_py3.py | 9 +- .../v1/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 70 +- .../multiapicustombaseurl/v2/__init__.py | 5 +- .../v2/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 42 +- .../multiapicustombaseurl/v2/_vendor.py | 7 +- .../multiapicustombaseurl/v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 40 +- .../multiapicustombaseurl/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 44 +- .../v2/models/__init__.py | 5 +- .../v2/models/_models_py3.py | 9 +- .../v2/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 70 +- .../multiapidataplane/v1/__init__.py | 5 +- .../multiapidataplane/v1/_configuration.py | 43 +- .../multiapidataplane/v1/_metadata.json | 4 +- .../v1/_multiapi_service_client.py | 36 +- .../multiapidataplane/v1/_vendor.py | 7 +- .../multiapidataplane/v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 32 +- .../multiapidataplane/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 ++++--- .../_operation_group_one_operations.py | 46 +- .../multiapidataplane/v1/models/__init__.py | 11 +- .../v1/models/_models_py3.py | 38 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++++---- .../_operation_group_one_operations.py | 66 +- .../multiapidataplane/v2/__init__.py | 5 +- .../multiapidataplane/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 38 +- .../multiapidataplane/v2/_vendor.py | 7 +- .../multiapidataplane/v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 34 +- .../multiapidataplane/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 ++- .../_operation_group_two_operations.py | 47 +- .../multiapidataplane/v2/models/__init__.py | 7 +- .../v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../multiapidataplane/v3/__init__.py | 5 +- .../multiapidataplane/v3/_configuration.py | 43 +- .../multiapidataplane/v3/_metadata.json | 4 +- .../v3/_multiapi_service_client.py | 38 +- .../multiapidataplane/v3/_vendor.py | 7 +- .../multiapidataplane/v3/aio/__init__.py | 5 +- .../v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 34 +- .../multiapidataplane/v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 ++- .../_operation_group_one_operations.py | 115 +-- .../_operation_group_two_operations.py | 92 ++- .../multiapidataplane/v3/models/__init__.py | 13 +- .../v3/models/_models_py3.py | 39 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapikeywordonly/v1/__init__.py | 5 +- .../multiapikeywordonly/v1/_configuration.py | 43 +- .../multiapikeywordonly/v1/_metadata.json | 4 +- .../v1/_multiapi_service_client.py | 36 +- .../multiapikeywordonly/v1/_vendor.py | 7 +- .../multiapikeywordonly/v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 32 +- .../multiapikeywordonly/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 275 ++++--- .../_operation_group_one_operations.py | 46 +- .../multiapikeywordonly/v1/models/__init__.py | 11 +- .../v1/models/_models_py3.py | 38 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 371 ++++++---- .../_operation_group_one_operations.py | 66 +- .../multiapikeywordonly/v2/__init__.py | 5 +- .../multiapikeywordonly/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 38 +- .../multiapikeywordonly/v2/_vendor.py | 7 +- .../multiapikeywordonly/v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 34 +- .../multiapikeywordonly/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 79 +- .../_operation_group_one_operations.py | 95 ++- .../_operation_group_two_operations.py | 48 +- .../multiapikeywordonly/v2/models/__init__.py | 7 +- .../v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 139 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 70 +- .../multiapikeywordonly/v3/__init__.py | 5 +- .../multiapikeywordonly/v3/_configuration.py | 43 +- .../multiapikeywordonly/v3/_metadata.json | 4 +- .../v3/_multiapi_service_client.py | 38 +- .../multiapikeywordonly/v3/_vendor.py | 7 +- .../multiapikeywordonly/v3/aio/__init__.py | 5 +- .../v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 34 +- .../multiapikeywordonly/v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 ++- .../_operation_group_one_operations.py | 115 +-- .../_operation_group_two_operations.py | 92 ++- .../multiapikeywordonly/v3/models/__init__.py | 13 +- .../v3/models/_models_py3.py | 39 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapinoasync/v1/__init__.py | 5 +- .../multiapinoasync/v1/_configuration.py | 43 +- .../multiapinoasync/v1/_metadata.json | 4 +- .../v1/_multiapi_service_client.py | 37 +- .../multiapinoasync/v1/_vendor.py | 7 +- .../multiapinoasync/v1/models/__init__.py | 11 +- .../multiapinoasync/v1/models/_models_py3.py | 38 +- .../multiapinoasync/v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++++---- .../_operation_group_one_operations.py | 66 +- .../multiapinoasync/v2/__init__.py | 5 +- .../multiapinoasync/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 39 +- .../multiapinoasync/v2/_vendor.py | 7 +- .../multiapinoasync/v2/models/__init__.py | 7 +- .../multiapinoasync/v2/models/_models_py3.py | 18 +- .../multiapinoasync/v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../multiapinoasync/v3/__init__.py | 5 +- .../multiapinoasync/v3/_configuration.py | 43 +- .../multiapinoasync/v3/_metadata.json | 4 +- .../v3/_multiapi_service_client.py | 39 +- .../multiapinoasync/v3/_vendor.py | 7 +- .../multiapinoasync/v3/models/__init__.py | 13 +- .../multiapinoasync/v3/models/_models_py3.py | 39 +- .../models/_multiapi_service_client_enums.py | 3 +- .../multiapinoasync/v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapisecurity/v0/__init__.py | 5 +- .../multiapisecurity/v0/_configuration.py | 43 +- .../v0/_multiapi_service_client.py | 36 +- .../multiapisecurity/v0/aio/__init__.py | 5 +- .../multiapisecurity/v0/aio/_configuration.py | 43 +- .../v0/aio/_multiapi_service_client.py | 32 +- .../v0/aio/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 46 +- .../multiapisecurity/v0/models/__init__.py | 9 +- .../multiapisecurity/v0/models/_models_py3.py | 29 +- .../v0/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 66 +- .../multiapisecurity/v1/__init__.py | 5 +- .../multiapisecurity/v1/_configuration.py | 43 +- .../multiapisecurity/v1/_metadata.json | 4 +- .../v1/_multiapi_service_client.py | 36 +- .../multiapisecurity/v1/_vendor.py | 7 +- .../multiapisecurity/v1/aio/__init__.py | 5 +- .../multiapisecurity/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 32 +- .../multiapisecurity/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 ++++--- .../_operation_group_one_operations.py | 46 +- .../multiapisecurity/v1/models/__init__.py | 11 +- .../multiapisecurity/v1/models/_models_py3.py | 38 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++++---- .../_operation_group_one_operations.py | 66 +- .../multiapiwithsubmodule/__init__.py | 1 + .../submodule/v1/__init__.py | 5 +- .../submodule/v1/_configuration.py | 43 +- .../submodule/v1/_metadata.json | 4 +- .../submodule/v1/_multiapi_service_client.py | 37 +- .../submodule/v1/_vendor.py | 7 +- .../submodule/v1/aio/__init__.py | 5 +- .../submodule/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 33 +- .../submodule/v1/aio/_vendor.py | 7 +- .../submodule/v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 ++++--- .../_operation_group_one_operations.py | 46 +- .../submodule/v1/models/__init__.py | 11 +- .../submodule/v1/models/_models_py3.py | 38 +- .../submodule/v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++++---- .../_operation_group_one_operations.py | 66 +- .../submodule/v2/__init__.py | 5 +- .../submodule/v2/_configuration.py | 43 +- .../submodule/v2/_multiapi_service_client.py | 39 +- .../submodule/v2/_vendor.py | 7 +- .../submodule/v2/aio/__init__.py | 5 +- .../submodule/v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 35 +- .../submodule/v2/aio/_vendor.py | 7 +- .../submodule/v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 ++- .../_operation_group_two_operations.py | 47 +- .../submodule/v2/models/__init__.py | 7 +- .../submodule/v2/models/_models_py3.py | 18 +- .../submodule/v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../submodule/v3/__init__.py | 5 +- .../submodule/v3/_configuration.py | 43 +- .../submodule/v3/_metadata.json | 4 +- .../submodule/v3/_multiapi_service_client.py | 39 +- .../submodule/v3/_vendor.py | 7 +- .../submodule/v3/aio/__init__.py | 5 +- .../submodule/v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 35 +- .../submodule/v3/aio/_vendor.py | 7 +- .../submodule/v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 ++- .../_operation_group_one_operations.py | 115 +-- .../_operation_group_two_operations.py | 92 ++- .../submodule/v3/models/__init__.py | 13 +- .../submodule/v3/models/_models_py3.py | 39 +- .../models/_multiapi_service_client_enums.py | 3 +- .../submodule/v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapicombiner/__init__.py | 5 +- .../multiapicombiner/_configuration.py | 43 +- .../multiapicombiner/_vendor.py | 7 +- .../multiapicombiner/aio/__init__.py | 5 +- .../multiapicombiner/aio/_configuration.py | 43 +- .../multiapicombiner/aio/_vendor.py | 7 +- .../aio/operations/_operations.py | 462 ++++++------ .../multiapicombiner/models/_enums.py | 3 +- .../multiapicombiner/models/_models.py | 59 +- .../operations/_operations.py | 689 +++++++++++------- .../multiapicombiner/v1/__init__.py | 5 +- .../multiapicombiner/v1/_configuration.py | 43 +- .../multiapicombiner/v1/_metadata.json | 4 +- .../v1/_multiapi_service_client.py | 37 +- .../multiapicombiner/v1/_vendor.py | 7 +- .../multiapicombiner/v1/aio/__init__.py | 5 +- .../multiapicombiner/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 33 +- .../multiapicombiner/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../v1/aio/operations/_operations.py | 308 ++++---- .../multiapicombiner/v1/models/__init__.py | 11 +- .../multiapicombiner/v1/models/_models_py3.py | 38 +- .../v1/operations/__init__.py | 7 +- .../v1/operations/_operations.py | 421 ++++++----- .../multiapicombiner/v2/__init__.py | 5 +- .../multiapicombiner/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 39 +- .../multiapicombiner/v2/_vendor.py | 7 +- .../multiapicombiner/v2/aio/__init__.py | 5 +- .../multiapicombiner/v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 35 +- .../multiapicombiner/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../v2/aio/operations/_operations.py | 202 +++-- .../multiapicombiner/v2/models/__init__.py | 7 +- .../multiapicombiner/v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../v2/operations/_operations.py | 318 +++++--- .../multiapicombiner/v3/__init__.py | 5 +- .../multiapicombiner/v3/_configuration.py | 43 +- .../multiapicombiner/v3/_metadata.json | 4 +- .../v3/_multiapi_service_client.py | 39 +- .../multiapicombiner/v3/_vendor.py | 7 +- .../multiapicombiner/v3/aio/__init__.py | 5 +- .../multiapicombiner/v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 35 +- .../multiapicombiner/v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../v3/aio/operations/_operations.py | 269 ++++--- .../multiapicombiner/v3/models/__init__.py | 13 +- .../multiapicombiner/v3/models/_models_py3.py | 39 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../v3/operations/_operations.py | 390 ++++++---- .../bodycomplex/models/_models_py3.py | 5 +- .../models/_models_py3.py | 5 +- .../Constants/constants/models/_models_py3.py | 5 +- .../errorwithsecrets/models/_models_py3.py | 5 +- .../models/_models_py3.py | 5 +- .../modelflattening/models/_models_py3.py | 5 +- .../NoOperations/nooperations/__init__.py | 5 +- .../nooperations/_configuration.py | 38 + .../_no_operations_service_client.py | 82 +++ .../requiredoptional/models/_models_py3.py | 5 +- .../validation/models/_models_py3.py | 5 +- .../Xml/xmlservice/models/_models_py3.py | 5 +- .../aio/operations/_pet_operations.py | 6 +- .../operations/_pet_operations.py | 6 +- .../aio/operations/_operations.py | 2 +- .../aio/operations/_operations.py | 2 +- .../aio/_operations/_operations.py | 3 +- .../nooperationsversiontolerant/__init__.py | 5 +- .../nooperationsversiontolerant/_client.py | 80 ++ .../_configuration.py | 38 + .../aio/operations/_operations.py | 3 +- .../models/_models.py | 5 +- 490 files changed, 12588 insertions(+), 10298 deletions(-) create mode 100644 packages/autorest.python/samples/specification/multiapi/generated/azure/__init__.py create mode 100644 packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/__init__.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py delete mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py delete mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py create mode 100644 packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_configuration.py create mode 100644 packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_no_operations_service_client.py create mode 100644 packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_client.py create mode 100644 packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_configuration.py diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/__init__.py b/packages/autorest.python/samples/specification/multiapi/generated/azure/__init__.py new file mode 100644 index 00000000000..d55ccad1f57 --- /dev/null +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/__init__.py @@ -0,0 +1 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/__init__.py b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/__init__.py new file mode 100644 index 00000000000..d55ccad1f57 --- /dev/null +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/__init__.py @@ -0,0 +1 @@ +__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/_metadata.json b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/_metadata.json index f07290e2f59..95e2f0c1049 100644 --- a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/_metadata.json +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py index 3748e0032bc..83aa365da8d 100644 --- a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/_metadata.json b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/_metadata.json index ce0ac43abfe..bd5edd71d2a 100644 --- a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/_metadata.json +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py index 0f284e05cf1..2f695839c35 100644 --- a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py deleted file mode 100644 index 045bab5d418..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py +++ /dev/null @@ -1,27 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from bodyduration import AutoRestDurationTestService - -""" -# PREREQUISITES - pip install autorestdurationtestservice -# USAGE - python duration_get_null.py -""" - - -def main(): - client = AutoRestDurationTestService() - - response = client.duration.get_null() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py deleted file mode 100644 index 69e50626bbd..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py +++ /dev/null @@ -1,30 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -import isodate - -from bodyduration import AutoRestDurationTestService - -""" -# PREREQUISITES - pip install autorestdurationtestservice -# USAGE - python duration_put_positive_duration.py -""" - - -def main(): - client = AutoRestDurationTestService() - - client.duration.put_positive_duration( - duration_body=isodate.parse_duration("P123DT22H14M12.011S"), - ) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py deleted file mode 100644 index c8c33fed4b6..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from custombaseurl import AutoRestParameterizedHostTestClient - -""" -# PREREQUISITES - pip install autorestparameterizedhosttestclient -# USAGE - python paths_get_empty.py -""" - - -def main(): - client = AutoRestParameterizedHostTestClient() - - client.paths.get_empty( - account_name="testaccount", - ) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py index e661461d066..399fd76b242 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING, Union +from typing import Any, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class BodyParam(_serialization.Model): """BodyParam. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py index fc88a8d66ac..177edc08ae3 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py deleted file mode 100644 index 318995c0864..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py +++ /dev/null @@ -1,30 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from custombaseurlpaging import AutoRestParameterizedHostTestPagingClient - -""" -# PREREQUISITES - pip install autorestparameterizedhosttestpagingclient -# USAGE - python paging_get_pages_partial_url.py -""" - - -def main(): - client = AutoRestParameterizedHostTestPagingClient() - - response = client.paging.get_pages_partial_url( - account_name="testaccount", - ) - for item in response: - print(item) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py deleted file mode 100644 index 37cd45add32..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py +++ /dev/null @@ -1,30 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from custombaseurlpaging import AutoRestParameterizedHostTestPagingClient - -""" -# PREREQUISITES - pip install autorestparameterizedhosttestpagingclient -# USAGE - python paging_get_pages_partial_url_operation.py -""" - - -def main(): - client = AutoRestParameterizedHostTestPagingClient() - - response = client.paging.get_pages_partial_url_operation( - account_name="testaccount", - ) - for item in response: - print(item) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py deleted file mode 100644 index be86f08131e..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from head import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadtestservice -# USAGE - python http_success_head200.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadTestService( - credential=DefaultAzureCredential(), - ) - - response = client.http_success.head200() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py deleted file mode 100644 index 804edc5597d..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from head import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadtestservice -# USAGE - python http_success_head204.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadTestService( - credential=DefaultAzureCredential(), - ) - - response = client.http_success.head204() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py deleted file mode 100644 index 4b50f67a8d9..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from head import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadtestservice -# USAGE - python http_success_head404.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadTestService( - credential=DefaultAzureCredential(), - ) - - response = client.http_success.head404() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py deleted file mode 100644 index 4285729aa20..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from headexceptions import AutoRestHeadExceptionTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadexceptiontestservice -# USAGE - python head_exception_head200.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadExceptionTestService( - credential=DefaultAzureCredential(), - ) - - response = client.head_exception.head200() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py deleted file mode 100644 index 6ea218aa8e1..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from headexceptions import AutoRestHeadExceptionTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadexceptiontestservice -# USAGE - python head_exception_head204.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadExceptionTestService( - credential=DefaultAzureCredential(), - ) - - response = client.head_exception.head204() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py deleted file mode 100644 index d0928b7aa33..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -import os - -from azure.core.credentials import AzureKeyCredential - -from headwithazurekeycredentialpolicy import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install autorestheadtestservice -# USAGE - python http_success_head200.py - - Before run the sample, please set environment variables AZURE_KEY with real value - which can access your service -""" - - -def main(): - client = AutoRestHeadTestService( - credential=AzureKeyCredential(key=os.getenv("AZURE_KEY")), - ) - - response = client.http_success.head200() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py deleted file mode 100644 index 8dad096cb16..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -import os - -from azure.core.credentials import AzureKeyCredential - -from headwithazurekeycredentialpolicy import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install autorestheadtestservice -# USAGE - python http_success_head204.py - - Before run the sample, please set environment variables AZURE_KEY with real value - which can access your service -""" - - -def main(): - client = AutoRestHeadTestService( - credential=AzureKeyCredential(key=os.getenv("AZURE_KEY")), - ) - - response = client.http_success.head204() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py deleted file mode 100644 index b6c3e61f065..00000000000 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -import os - -from azure.core.credentials import AzureKeyCredential - -from headwithazurekeycredentialpolicy import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install autorestheadtestservice -# USAGE - python http_success_head404.py - - Before run the sample, please set environment variables AZURE_KEY with real value - which can access your service -""" - - -def main(): - client = AutoRestHeadTestService( - credential=AzureKeyCredential(key=os.getenv("AZURE_KEY")), - ) - - response = client.http_success.head404() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py index 68b6288aa8a..5abc01aa699 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Dict, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class OperationResult(_serialization.Model): """OperationResult. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py index e59e87a4ac9..995c0f52b3c 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING, Union +from typing import Any, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class BodyParam(_serialization.Model): """BodyParam. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py index 54d8bfc2674..0e95655c3bc 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING, Union +from typing import Any, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class BodyParam(_serialization.Model): """BodyParam. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py index 406b6861daa..2e15afca307 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py @@ -7,13 +7,10 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Bar(_serialization.Model): """The URIs that are used to perform a retrieval of a public blob, queue or table object. diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py deleted file mode 100644 index 9197f6d6e6e..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py +++ /dev/null @@ -1,27 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from bodydurationversiontolerant import AutoRestDurationTestService - -""" -# PREREQUISITES - pip install autorestdurationtestservice -# USAGE - python duration_get_null.py -""" - - -def main(): - client = AutoRestDurationTestService() - - response = client.duration.get_null() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py deleted file mode 100644 index e3d92a29334..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py +++ /dev/null @@ -1,30 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -import isodate - -from bodydurationversiontolerant import AutoRestDurationTestService - -""" -# PREREQUISITES - pip install autorestdurationtestservice -# USAGE - python duration_put_positive_duration.py -""" - - -def main(): - client = AutoRestDurationTestService() - - client.duration.put_positive_duration( - duration_body=isodate.parse_duration("P123DT22H14M12.011S"), - ) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py deleted file mode 100644 index cd2f88a83ff..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py +++ /dev/null @@ -1,28 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from custombaseurlversiontolerant import AutoRestParameterizedHostTestClient - -""" -# PREREQUISITES - pip install autorestparameterizedhosttestclient -# USAGE - python paths_get_empty.py -""" - - -def main(): - client = AutoRestParameterizedHostTestClient() - - client.paths.get_empty( - account_name="testaccount", - ) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py deleted file mode 100644 index b9c7cfcfc2b..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py +++ /dev/null @@ -1,30 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from custombaseurlpagingversiontolerant import AutoRestParameterizedHostTestPagingClient - -""" -# PREREQUISITES - pip install autorestparameterizedhosttestpagingclient -# USAGE - python paging_get_pages_partial_url.py -""" - - -def main(): - client = AutoRestParameterizedHostTestPagingClient() - - response = client.paging.get_pages_partial_url( - account_name="testaccount", - ) - for item in response: - print(item) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py deleted file mode 100644 index 9c315c6b84e..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py +++ /dev/null @@ -1,30 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from custombaseurlpagingversiontolerant import AutoRestParameterizedHostTestPagingClient - -""" -# PREREQUISITES - pip install autorestparameterizedhosttestpagingclient -# USAGE - python paging_get_pages_partial_url_operation.py -""" - - -def main(): - client = AutoRestParameterizedHostTestPagingClient() - - response = client.paging.get_pages_partial_url_operation( - account_name="testaccount", - ) - for item in response: - print(item) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py deleted file mode 100644 index 95892b1b19a..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from headexceptionsversiontolerant import AutoRestHeadExceptionTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadexceptiontestservice -# USAGE - python head_exception_head200.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadExceptionTestService( - credential=DefaultAzureCredential(), - ) - - response = client.head_exception.head200() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py deleted file mode 100644 index 90ceb384219..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from headexceptionsversiontolerant import AutoRestHeadExceptionTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadexceptiontestservice -# USAGE - python head_exception_head204.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadExceptionTestService( - credential=DefaultAzureCredential(), - ) - - response = client.head_exception.head204() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py deleted file mode 100644 index e036e7e5142..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from headversiontolerant import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadtestservice -# USAGE - python http_success_head200.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadTestService( - credential=DefaultAzureCredential(), - ) - - response = client.http_success.head200() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py deleted file mode 100644 index 8637361f364..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from headversiontolerant import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadtestservice -# USAGE - python http_success_head204.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadTestService( - credential=DefaultAzureCredential(), - ) - - response = client.http_success.head204() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py deleted file mode 100644 index ed7221131d0..00000000000 --- a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py +++ /dev/null @@ -1,37 +0,0 @@ -# coding=utf-8 -# -------------------------------------------------------------------------- -# Copyright (c) Microsoft Corporation. All rights reserved. -# Licensed under the MIT License. See License.txt in the project root for license information. -# Code generated by Microsoft (R) AutoRest Code Generator. -# Changes may cause incorrect behavior and will be lost if the code is regenerated. -# -------------------------------------------------------------------------- - -from azure.identity import DefaultAzureCredential - -from headversiontolerant import AutoRestHeadTestService - -""" -# PREREQUISITES - pip install azure-identity - pip install autorestheadtestservice -# USAGE - python http_success_head404.py - - Before run the sample, please set the values of the client ID, tenant ID and client secret - of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, - AZURE_CLIENT_SECRET. For more info about how to get the value, please see: - https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal -""" - - -def main(): - client = AutoRestHeadTestService( - credential=DefaultAzureCredential(), - ) - - response = client.http_success.head404() - print(response) - - -if __name__ == "__main__": - main() diff --git a/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py b/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py index 2779069dfd1..100da986cd1 100644 --- a/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py +++ b/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING, Union +from typing import Any, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Input(_serialization.Model): """Input. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py index 5d72190ec6e..ff9fe140cda 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "0.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py index 9955ea175a5..02297b40119 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient: """Service client for multiapi client testing. @@ -38,37 +37,33 @@ class MultiapiServiceClient: :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "0.0.0" + self._client, self._config, self._serialize, self._deserialize, '0.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py index cc3420f9112..611e61dc65b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "0.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py index 99f361877eb..f8e04726218 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient: """Service client for multiapi client testing. @@ -39,39 +38,31 @@ class MultiapiServiceClient: """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "0.0.0" + self._client, self._config, self._serialize, self._deserialize, '0.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py index f8443c0f5fe..9631441cd6a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "OperationGroupOneOperations", + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py index e04784a2b44..ca42dc5232a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,8 +45,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -62,19 +60,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -83,8 +81,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -95,4 +95,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py index 513632adef7..0b2869020ef 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py @@ -21,11 +21,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", + 'Error', + 'PagingResult', + 'Product', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py index 89176e0e663..15d7bb640ad 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py index f8443c0f5fe..9631441cd6a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "OperationGroupOneOperations", + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py index 3213f256983..a4756b1a8ca 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,33 +22,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "0.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "0.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,8 +75,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -84,19 +90,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -105,8 +111,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -117,4 +125,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py index 4d91e0a380e..1c9a5b0ff2d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_metadata.json index 0fde66cae75..83970d1cfef 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py index 3dc74cddca7..fc1ba252cf5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,37 +39,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py index 093bce4e8d9..9e3856bf5cc 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py index 7cc3d40eb7d..0c4d5171992 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,39 +40,31 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py index ade2701e6d6..96dfced3f55 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,16 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -31,23 +22,19 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_lro_and_paging_request, - build_test_lro_request, - build_test_one_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -55,7 +42,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -67,21 +59,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -93,7 +83,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -104,24 +96,27 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -130,7 +125,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -143,10 +138,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -163,13 +160,19 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload async def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -186,7 +189,11 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -201,9 +208,12 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -218,45 +228,59 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + async def _test_lro_and_paging_initial( self, @@ -265,17 +289,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -292,10 +315,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -311,10 +336,12 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -338,16 +365,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -355,7 +380,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -368,25 +393,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -394,7 +416,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -404,20 +428,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -425,27 +453,38 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace_async - async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: + async def test_different_calls( + self, + greeting_in_english: str, + **kwargs: Any + ) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -455,21 +494,19 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -480,7 +517,9 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -491,4 +530,6 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py index 7b5ed061b53..f1aa6183dd8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,8 +45,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -62,19 +60,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -83,8 +81,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -95,4 +95,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py index 996681e3e2c..ed6f229313a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py index 576f88f1b4e..3848a9c8fdf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -37,95 +28,133 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_test_lro_and_paging_request( - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: +def build_test_different_calls_request( + *, + greeting_in_english: str, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -134,7 +163,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, id: int, message: Optional[str] = None, **kwargs: Any + self, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -147,21 +179,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -172,8 +202,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -184,24 +216,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -210,7 +245,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -223,10 +258,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -243,13 +280,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -265,7 +308,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -279,9 +326,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -295,45 +345,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -342,17 +406,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -369,10 +432,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -388,10 +453,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -415,16 +482,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -432,7 +497,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -445,33 +510,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -481,20 +545,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -502,28 +570,37 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, **kwargs: Any + self, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -534,21 +611,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -558,8 +633,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -570,4 +647,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py index 92c05501686..d1aff3a2365 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,33 +22,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,8 +75,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -84,19 +90,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -105,8 +111,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -117,4 +125,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py index 8a6a6365ac6..0c09e0d6074 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py index 267df04dacc..73b1e3da5ac 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py index 10f716d8dd6..c961d19c35b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py index 3a081ccc216..7c58e8f62b4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,42 +40,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py index fb9447b4402..1dc47f74947 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,12 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -42,7 +36,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -54,21 +53,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -80,7 +77,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -90,16 +89,24 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def test_different_calls( - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -112,21 +119,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -138,7 +143,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -149,4 +156,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py index 78943a9d4be..4d3b3f9ceb6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -54,9 +46,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -72,7 +71,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -86,9 +89,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -100,19 +106,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -121,7 +126,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -136,8 +141,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -147,15 +154,23 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async - async def test_three(self, **kwargs: Any) -> None: + async def test_three( + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -163,19 +178,19 @@ async def test_three(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -184,8 +199,10 @@ async def test_three(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -196,4 +213,6 @@ async def test_three(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py index dd3455c83c3..055d7801caa 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,8 +45,15 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: + async def test_four( + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -64,19 +63,19 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -86,8 +85,10 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -98,4 +99,6 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py index ed8e322c54e..bdef0772a67 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py @@ -20,10 +20,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelTwo", + 'Error', + 'ModelTwo', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py index 46990252065..14a84ef76da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -36,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -49,7 +53,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -58,7 +62,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py index b094ad6f29a..5c2816d8007 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,60 +23,81 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( - *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +105,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -103,21 +122,19 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -128,8 +145,10 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -139,16 +158,24 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -161,21 +188,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -186,8 +211,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -198,4 +225,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py index 3cd1ed59cf0..91fb46a415a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_three_request(**kwargs: Any) -> HttpRequest: +def build_test_three_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -116,7 +131,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -130,9 +149,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -144,19 +166,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -165,7 +186,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -180,8 +201,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -191,15 +214,23 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -207,19 +238,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -228,8 +259,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -240,4 +273,6 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py index 8194728a647..d58c381f306 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,34 +22,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: +def build_test_four_request( + *, + parameter_one: bool, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -76,8 +78,15 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_four( # pylint: disable=inconsistent-return-statements + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -87,19 +96,19 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -109,8 +118,10 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -121,4 +132,6 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py index 633f146ec60..f7b05316ad2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_metadata.json index 5f505a520c3..359498e32bc 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py index 119f8588d79..24f2baa1933 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py index 7a4058edefe..b5996cca8ef 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py index e41a3533676..adaac305718 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,42 +40,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py index 97c4d332d0c..29467d39007 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,14 +10,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -26,21 +19,19 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_paging_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -48,7 +39,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -58,19 +52,17 @@ def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -80,25 +72,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -106,7 +95,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -116,7 +107,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @distributed_trace_async async def test_different_calls( @@ -139,21 +134,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -166,7 +159,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -177,4 +172,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py index 117bb536374..88c037aeaa6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py @@ -11,14 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -27,20 +20,16 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operation_group_one_operations import ( - build_test_operation_group_paging_request, - build_test_two_request, -) +from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -60,8 +49,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -71,19 +66,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.M _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -93,33 +86,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -129,7 +121,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @overload async def test_two( @@ -153,7 +149,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -167,9 +167,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -181,19 +184,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -202,7 +204,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -217,8 +219,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -228,9 +232,14 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py index c33382ea2c6..80e0bbb37d5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -54,9 +46,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -72,7 +71,11 @@ async def test_four( @overload async def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -87,8 +90,13 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async - async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: + async def test_four( + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -99,19 +107,18 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -119,7 +126,7 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -135,8 +142,10 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -147,10 +156,15 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace_async - async def test_five(self, **kwargs: Any) -> None: + async def test_five( + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -158,19 +172,19 @@ async def test_five(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -179,8 +193,10 @@ async def test_five(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -191,4 +207,6 @@ async def test_five(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py index 63672cad01d..a21d2c11b05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py @@ -26,13 +26,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelThree", - "PagingResult", - "SourcePath", - "ContentType", + 'Error', + 'ModelThree', + 'PagingResult', + 'SourcePath', + 'ContentType', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py index 49a4432418f..a8292170171 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -51,7 +52,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -59,7 +65,6 @@ def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> super().__init__(**kwargs) self.optional_property = optional_property - class PagingResult(_serialization.Model): """PagingResult. @@ -75,7 +80,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -87,7 +96,6 @@ def __init__( self.values = values self.next_link = next_link - class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -96,14 +104,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py index 7179ffb9c14..853a409c981 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py @@ -11,7 +11,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py index d4e34c679e3..5a4b0724f9b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py @@ -9,14 +9,7 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,25 +25,32 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request(**kwargs: Any) -> HttpRequest: +def build_test_paging_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( @@ -63,27 +63,34 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +98,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -101,19 +111,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -123,33 +131,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -159,7 +166,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -182,21 +193,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -208,8 +217,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -220,4 +231,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py index 4edc0e9ea1b..b715f292c50 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,50 +25,64 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_test_operation_group_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class OperationGroupOneOperations: + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -95,8 +102,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -106,19 +119,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -128,33 +139,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -164,7 +174,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @overload def test_two( @@ -188,7 +202,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -202,9 +220,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -216,19 +237,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -237,7 +257,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -252,8 +272,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -263,9 +285,14 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py index eeb9c5e3872..22f547ac52a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(**kwargs: Any) -> HttpRequest: +def build_test_four_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_five_request(**kwargs: Any) -> HttpRequest: +def build_test_five_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -116,7 +131,11 @@ def test_four( @overload def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,9 +150,12 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -145,19 +167,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -165,7 +186,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -181,8 +202,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -193,10 +216,15 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -204,19 +232,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -225,8 +253,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -237,4 +267,6 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py index a8be991d4c3..712154c9ff9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py @@ -14,8 +14,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,27 +27,35 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: AzureKeyCredential, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) + kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_metadata.json index 95778b846ac..e0f1d908799 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py index 324fbdc58e0..e85f4f96d9a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py @@ -21,7 +21,6 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,37 +38,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: AzureKeyCredential, + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py index c0bfc9b848e..b1341bff761 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py @@ -14,8 +14,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,27 +27,35 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: AzureKeyCredential, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) + kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py index 7afa7077e11..4036c86c1af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py @@ -21,7 +21,6 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,38 +38,32 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: AzureKeyCredential, + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py index 97a2ca1cb9f..471b1ff46fb 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,16 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -31,23 +22,19 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_lro_and_paging_request, - build_test_lro_request, - build_test_one_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -55,7 +42,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -67,21 +59,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -93,7 +83,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -104,24 +96,27 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -130,7 +125,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -143,10 +138,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -163,13 +160,19 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload async def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -186,7 +189,11 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -201,9 +208,12 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -218,45 +228,59 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + async def _test_lro_and_paging_initial( self, @@ -265,17 +289,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -292,10 +315,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -311,10 +336,12 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -340,16 +367,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -357,7 +382,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -370,25 +395,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -396,7 +418,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -406,20 +430,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -427,27 +455,38 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace_async - async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: + async def test_different_calls( + self, + greeting_in_english: str, + **kwargs: Any + ) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -457,21 +496,19 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -482,7 +519,9 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -493,4 +532,6 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py index ce64ec4c546..acfb0727b49 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,8 +45,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -62,19 +60,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -83,8 +81,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -95,4 +95,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py index 93d38043e62..ffbf90c6f18 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py index ec3d941ed8d..b6bcc863bcf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -37,95 +28,133 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_test_lro_and_paging_request( - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: +def build_test_different_calls_request( + *, + greeting_in_english: str, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -134,7 +163,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, id: int, message: Optional[str] = None, **kwargs: Any + self, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -147,21 +179,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -172,8 +202,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -184,24 +216,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -210,7 +245,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -223,10 +258,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -243,13 +280,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -265,7 +308,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -279,9 +326,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -295,45 +345,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -342,17 +406,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -369,10 +432,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -388,10 +453,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -416,16 +483,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -433,7 +498,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -446,33 +511,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -482,20 +546,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -503,28 +571,37 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, **kwargs: Any + self, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -535,21 +612,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -559,8 +634,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -571,4 +648,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py index f6fee2b20d0..999c0681731 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,33 +22,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,8 +75,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -84,19 +90,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -105,8 +111,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -117,4 +125,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py index 0912525614d..fa0888e64c9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py @@ -14,8 +14,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,27 +27,35 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: AzureKeyCredential, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) + kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py index ab171b0b766..172f5da5b90 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py @@ -21,7 +21,6 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: AzureKeyCredential, + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py index c6dedcb071d..45bee4fee54 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py @@ -14,8 +14,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,27 +27,35 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: AzureKeyCredential, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) + kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py index 4d425a8b9d6..26f62b9439c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py @@ -21,7 +21,6 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,41 +39,35 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: AzureKeyCredential, + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py index 3bb6b8555bc..9c81d309ff7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,12 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -42,7 +36,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -54,21 +53,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -80,7 +77,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -90,16 +89,24 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def test_different_calls( - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -112,21 +119,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -138,7 +143,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -149,4 +156,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py index 015e383a8fa..a490264d924 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -54,9 +46,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -72,7 +71,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -86,9 +89,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -100,19 +106,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -121,7 +126,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -136,8 +141,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -147,15 +154,23 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async - async def test_three(self, **kwargs: Any) -> None: + async def test_three( + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -163,19 +178,19 @@ async def test_three(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -184,8 +199,10 @@ async def test_three(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -196,4 +213,6 @@ async def test_three(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py index a4da2d85b0a..27e67f73930 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,8 +45,15 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: + async def test_four( + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -64,19 +63,19 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -86,8 +85,10 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -98,4 +99,6 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py index ed8e322c54e..bdef0772a67 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py @@ -20,10 +20,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelTwo", + 'Error', + 'ModelTwo', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py index 46990252065..14a84ef76da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -36,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -49,7 +53,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -58,7 +62,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py index be1a9b9e82e..f42959dd244 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,60 +23,81 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( - *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +105,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -103,21 +122,19 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -128,8 +145,10 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -139,16 +158,24 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -161,21 +188,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -186,8 +211,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -198,4 +225,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py index 7f8bf5f5700..503485608ba 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_three_request(**kwargs: Any) -> HttpRequest: +def build_test_three_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -116,7 +131,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -130,9 +149,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -144,19 +166,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -165,7 +186,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -180,8 +201,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -191,15 +214,23 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -207,19 +238,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -228,8 +259,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -240,4 +273,6 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py index ecae6014d5d..39b3c2c79f9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,34 +22,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: +def build_test_four_request( + *, + parameter_one: bool, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -76,8 +78,15 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_four( # pylint: disable=inconsistent-return-statements + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -87,19 +96,19 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -109,8 +118,10 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -121,4 +132,6 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py index eca4a1b0a56..0ad67c0eb34 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py @@ -14,8 +14,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,27 +27,35 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: AzureKeyCredential, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) + kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_metadata.json index 7a34f154b40..436ac1a965e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py index e27caf2ecf8..596b5c66218 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py @@ -21,7 +21,6 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: AzureKeyCredential, + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py index 1f5ce3c0e76..8cc199d2b6a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py @@ -14,8 +14,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,27 +27,35 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: AzureKeyCredential, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) + kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py index fd43b96f213..9b52d66d726 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py @@ -21,7 +21,6 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,41 +39,35 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: AzureKeyCredential, + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py index c1765973ab2..6a987f06fcf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,14 +10,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -26,21 +19,19 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_paging_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -48,7 +39,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -59,19 +53,17 @@ def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -81,25 +73,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -107,7 +96,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -117,7 +108,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @distributed_trace_async async def test_different_calls( @@ -140,21 +135,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -167,7 +160,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -178,4 +173,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py index b1af64630fb..ae6fa327c9d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py @@ -11,14 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -27,20 +20,16 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operation_group_one_operations import ( - build_test_operation_group_paging_request, - build_test_two_request, -) +from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -60,8 +49,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -72,19 +67,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.M _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -94,33 +87,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -130,7 +122,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @overload async def test_two( @@ -154,7 +150,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -168,9 +168,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -182,19 +185,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -203,7 +205,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -218,8 +220,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -229,9 +233,14 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py index f447c159587..840a3b9d2fb 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -54,9 +46,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -72,7 +71,11 @@ async def test_four( @overload async def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -87,8 +90,13 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async - async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: + async def test_four( + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -99,19 +107,18 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -119,7 +126,7 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -135,8 +142,10 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -147,10 +156,15 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace_async - async def test_five(self, **kwargs: Any) -> None: + async def test_five( + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -158,19 +172,19 @@ async def test_five(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -179,8 +193,10 @@ async def test_five(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -191,4 +207,6 @@ async def test_five(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py index 63672cad01d..a21d2c11b05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py @@ -26,13 +26,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelThree", - "PagingResult", - "SourcePath", - "ContentType", + 'Error', + 'ModelThree', + 'PagingResult', + 'SourcePath', + 'ContentType', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py index 79a8e667ab8..2fa2f8f28e0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -51,7 +52,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -59,7 +65,6 @@ def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> super().__init__(**kwargs) self.optional_property = optional_property - class PagingResult(_serialization.Model): """PagingResult. @@ -75,7 +80,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -87,7 +96,6 @@ def __init__( self.values = values self.next_link = next_link - class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -96,14 +104,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py index 7179ffb9c14..853a409c981 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py @@ -11,7 +11,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py index 7225a31c9db..f47bf6113ba 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py @@ -9,14 +9,7 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,25 +25,32 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request(**kwargs: Any) -> HttpRequest: +def build_test_paging_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( @@ -63,27 +63,34 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +98,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -101,19 +111,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -123,33 +131,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -159,7 +166,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -182,21 +193,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -208,8 +217,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -220,4 +231,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py index a0802150ed0..b7ac24273fd 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,50 +25,64 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_test_operation_group_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class OperationGroupOneOperations: + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -95,8 +102,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -106,19 +119,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -128,33 +139,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -164,7 +174,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @overload def test_two( @@ -188,7 +202,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -202,9 +220,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -216,19 +237,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -237,7 +257,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -252,8 +272,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -263,9 +285,14 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py index e37e6579d60..c07001a7d97 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(**kwargs: Any) -> HttpRequest: +def build_test_four_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_five_request(**kwargs: Any) -> HttpRequest: +def build_test_five_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -116,7 +131,11 @@ def test_four( @overload def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,9 +150,12 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -145,19 +167,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -165,7 +186,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -181,8 +202,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -193,10 +216,15 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -204,19 +232,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -225,8 +253,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -237,4 +267,6 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py index ccc133e71e1..7652bc9a164 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClient", + 'MultiapiCustomBaseUrlServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py index e88c1c6d7af..7f0f64c8eb7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -31,8 +30,13 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-ma :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -42,24 +46,26 @@ def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py index e961cb6fe23..86f432fa78d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -35,36 +34,31 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: - _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v1" - self._config = MultiapiCustomBaseUrlServiceClientConfiguration( - credential=credential, endpoint=endpoint, **kwargs - ) - _policies = kwargs.pop("policies", None) + def __init__( + self, + credential: "TokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v1' + self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest @@ -84,7 +78,7 @@ def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py index 856dac1195a..e1066c871a7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py index ccc133e71e1..7652bc9a164 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClient", + 'MultiapiCustomBaseUrlServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py index 9915f2c2e2e..2fdd7034dba 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -31,8 +30,13 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-ma :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -42,24 +46,26 @@ def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py index 37dbb09b777..179bc8893cf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -35,37 +34,30 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: - _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v1" - self._config = MultiapiCustomBaseUrlServiceClientConfiguration( - credential=credential, endpoint=endpoint, **kwargs - ) - _policies = kwargs.pop("policies", None) + def __init__( + self, + credential: "AsyncTokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v1' + self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=_endpoint, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. @@ -86,7 +78,7 @@ def _send_request( request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py index 7ad36d04654..1ab725bccca 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py index 40541c05116..172a9058373 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClientOperationsMixin", + 'MultiapiCustomBaseUrlServiceClientOperationsMixin', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py index dc3cc527caa..18d2c97abdf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -43,7 +35,11 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test(self, id: int, **kwargs: Any) -> None: + async def test( + self, + id: int, + **kwargs: Any + ) -> None: """Should be a mixin operation. Put in 1 for the required parameter and have the correct api version of 1.0.0 to pass. @@ -54,19 +50,19 @@ async def test(self, id: int, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_request( id=id, api_version=api_version, @@ -74,13 +70,15 @@ async def test(self, id: int, **kwargs: Any) -> None: params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -91,4 +89,6 @@ async def test(self, id: int, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py index 187235acc3a..f02be92e17f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py @@ -19,9 +19,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", + 'Error', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py index 4ccf63e55b5..b58173f2878 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py index 40541c05116..172a9058373 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClientOperationsMixin", + 'MultiapiCustomBaseUrlServiceClientOperationsMixin', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py index 0a5f910a9f0..ee05f3883c3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,34 +22,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_request(*, id: int, **kwargs: Any) -> HttpRequest: +def build_test_request( + *, + id: int, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/test") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['id'] = _SERIALIZER.query("id", id, 'int') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -66,7 +68,11 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test( # pylint: disable=inconsistent-return-statements + self, + id: int, + **kwargs: Any + ) -> None: """Should be a mixin operation. Put in 1 for the required parameter and have the correct api version of 1.0.0 to pass. @@ -77,19 +83,19 @@ def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent- :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_request( id=id, api_version=api_version, @@ -97,13 +103,15 @@ def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent- params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -114,4 +122,6 @@ def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent- raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py index ccc133e71e1..7652bc9a164 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClient", + 'MultiapiCustomBaseUrlServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py index 5f2904887f8..9019197a1ba 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -31,8 +30,13 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-ma :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "TokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -42,24 +46,26 @@ def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py index ed15d88585e..7aac04a3271 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -35,36 +34,31 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: - _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v2" - self._config = MultiapiCustomBaseUrlServiceClientConfiguration( - credential=credential, endpoint=endpoint, **kwargs - ) - _policies = kwargs.pop("policies", None) + def __init__( + self, + credential: "TokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v2' + self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest @@ -84,7 +78,7 @@ def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py index 856dac1195a..e1066c871a7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py index ccc133e71e1..7652bc9a164 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClient", + 'MultiapiCustomBaseUrlServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py index 0bdd1c9cfd0..686cb16bcfd 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -31,8 +30,13 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-ma :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -42,24 +46,26 @@ def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py index 98cc9e0b5e0..ab660ded68a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -35,37 +34,30 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: - _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v2" - self._config = MultiapiCustomBaseUrlServiceClientConfiguration( - credential=credential, endpoint=endpoint, **kwargs - ) - _policies = kwargs.pop("policies", None) + def __init__( + self, + credential: "AsyncTokenCredential", + endpoint: str, + **kwargs: Any + ) -> None: + _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v2' + self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=_endpoint, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. @@ -86,7 +78,7 @@ def _send_request( request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py index 7ad36d04654..1ab725bccca 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py index 40541c05116..172a9058373 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClientOperationsMixin", + 'MultiapiCustomBaseUrlServiceClientOperationsMixin', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py index 6bbcee4be50..95c6edf3c1c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -43,7 +35,11 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test(self, id: int, **kwargs: Any) -> None: + async def test( + self, + id: int, + **kwargs: Any + ) -> None: """Should be a mixin operation. Put in 2 for the required parameter and have the correct api version of 2.0.0 to pass. @@ -54,19 +50,19 @@ async def test(self, id: int, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_request( id=id, api_version=api_version, @@ -74,13 +70,15 @@ async def test(self, id: int, **kwargs: Any) -> None: params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -91,4 +89,6 @@ async def test(self, id: int, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py index 187235acc3a..f02be92e17f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py @@ -19,9 +19,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", + 'Error', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py index 4ccf63e55b5..b58173f2878 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py index 40541c05116..172a9058373 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiCustomBaseUrlServiceClientOperationsMixin", + 'MultiapiCustomBaseUrlServiceClientOperationsMixin', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py index a7610c97bcb..e03323f3a05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,34 +22,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_request(*, id: int, **kwargs: Any) -> HttpRequest: +def build_test_request( + *, + id: int, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/test") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['id'] = _SERIALIZER.query("id", id, 'int') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -66,7 +68,11 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test( # pylint: disable=inconsistent-return-statements + self, + id: int, + **kwargs: Any + ) -> None: """Should be a mixin operation. Put in 2 for the required parameter and have the correct api version of 2.0.0 to pass. @@ -77,19 +83,19 @@ def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent- :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_request( id=id, api_version=api_version, @@ -97,13 +103,15 @@ def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent- params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -114,4 +122,6 @@ def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent- raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py index 09e03ece143..19124f3ce57 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_metadata.json index 6afc2663b28..3c61b609e87 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py index 02e0b796d19..ec55857b82e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +38,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py index 3d6432fa758..bde90131f5d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py index c5457654c34..ea5937d3ed3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,38 +39,31 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py index 1592e7e3079..9cf4d485236 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,16 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.polling.async_base_polling import AsyncLROBasePolling @@ -30,23 +21,19 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_lro_and_paging_request, - build_test_lro_request, - build_test_one_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -54,7 +41,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -66,21 +58,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -92,7 +82,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -103,24 +95,27 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -129,7 +124,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -142,10 +137,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -162,13 +159,19 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload async def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -185,7 +188,11 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -200,9 +207,12 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -217,45 +227,59 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + async def _test_lro_and_paging_initial( self, @@ -264,17 +288,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -291,10 +314,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -310,10 +335,12 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -338,16 +365,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -355,7 +380,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -368,25 +393,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -394,7 +416,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -404,20 +428,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -425,27 +453,38 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace_async - async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: + async def test_different_calls( + self, + greeting_in_english: str, + **kwargs: Any + ) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -455,21 +494,19 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -480,7 +517,9 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -491,4 +530,6 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py index 35141ff86c5..c822e39b56f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -28,11 +21,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -52,8 +44,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -61,19 +59,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -82,8 +80,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -94,4 +94,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py index 918a39761c2..7c46380faec 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py index 7deb4528d01..6a017091706 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -36,95 +27,133 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_test_lro_and_paging_request( - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: +def build_test_different_calls_request( + *, + greeting_in_english: str, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -133,7 +162,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, id: int, message: Optional[str] = None, **kwargs: Any + self, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -146,21 +178,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -171,8 +201,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -183,24 +215,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -209,7 +244,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -222,10 +257,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -242,13 +279,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -264,7 +307,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -278,9 +325,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -294,45 +344,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -341,17 +405,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -368,10 +431,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -387,10 +452,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -414,16 +481,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -431,7 +496,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -444,33 +509,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -480,20 +544,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -501,28 +569,37 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, **kwargs: Any + self, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -533,21 +610,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -557,8 +632,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -569,4 +646,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py index 160c63eedd4..6114030a753 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -28,33 +21,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,8 +74,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -83,19 +89,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -104,8 +110,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -116,4 +124,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py index 5e60bea3540..2fd8b4ff762 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py index 2f9a67a9e8c..80b50b79d59 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,39 +38,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py index 8d71e3f2f72..d45199c2a69 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py index 4809ee86266..2ccfa08b81a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,41 +39,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py index 4b446c5f02c..bc6ed439f09 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,12 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -41,7 +35,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -53,21 +52,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -79,7 +76,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -89,16 +88,24 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def test_different_calls( - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -111,21 +118,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -137,7 +142,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -148,4 +155,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py index b6a71a2e585..6e6e6e03d66 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,9 +45,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -71,7 +70,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -85,9 +88,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -99,19 +105,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -120,7 +125,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -135,8 +140,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -146,15 +153,23 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async - async def test_three(self, **kwargs: Any) -> None: + async def test_three( + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -162,19 +177,19 @@ async def test_three(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -183,8 +198,10 @@ async def test_three(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -195,4 +212,6 @@ async def test_three(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py index ded5facbd78..3e751d5baab 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -28,11 +21,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -52,8 +44,15 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: + async def test_four( + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -63,19 +62,19 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -85,8 +84,10 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -97,4 +98,6 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py index ed8e322c54e..bdef0772a67 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py @@ -20,10 +20,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelTwo", + 'Error', + 'ModelTwo', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py index 46990252065..14a84ef76da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -36,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -49,7 +53,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -58,7 +62,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py index 66ca2e3753e..60b659d5d6b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,60 +22,81 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( - *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -90,7 +104,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -102,21 +121,19 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -127,8 +144,10 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -138,16 +157,24 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -160,21 +187,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -185,8 +210,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -197,4 +224,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py index dfed4c01227..69dafaeb3e6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,55 +22,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_three_request(**kwargs: Any) -> HttpRequest: +def build_test_three_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -97,9 +105,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -115,7 +130,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -129,9 +148,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -143,19 +165,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -164,7 +185,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -179,8 +200,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -190,15 +213,23 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -206,19 +237,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -227,8 +258,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -239,4 +272,6 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py index 7488db14ba2..449683618ce 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -28,34 +21,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: +def build_test_four_request( + *, + parameter_one: bool, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,8 +77,15 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_four( # pylint: disable=inconsistent-return-statements + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -86,19 +95,19 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -108,8 +117,10 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -120,4 +131,6 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py index 9826b12b94f..8c1ed8b2ddf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_metadata.json index 70a84e3a7b8..3e0f12a2ab7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py index 7b6fc625cf4..c57b7afeb82 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,39 +38,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py index 0b30d2372cd..827d3959bf9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py index 56e1a56df02..d879d925712 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,41 +39,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py index 9e5c6f71af5..fbbb099e1c0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,14 +10,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -25,21 +18,19 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_paging_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -47,7 +38,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -57,19 +51,17 @@ def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -79,25 +71,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -105,7 +94,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -115,7 +106,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @distributed_trace_async async def test_different_calls( @@ -138,21 +133,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -165,7 +158,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -176,4 +171,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py index a6ead9ba8c6..deadb8ab34d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py @@ -11,14 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -26,20 +19,16 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._operation_group_one_operations import ( - build_test_operation_group_paging_request, - build_test_two_request, -) +from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -59,8 +48,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -70,19 +65,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.M _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -92,33 +85,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -128,7 +120,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @overload async def test_two( @@ -152,7 +148,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -166,9 +166,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -180,19 +183,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -201,7 +203,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -216,8 +218,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -227,9 +231,14 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py index 19e03f22b23..165d04b7fbc 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,9 +45,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -71,7 +70,11 @@ async def test_four( @overload async def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -86,8 +89,13 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async - async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: + async def test_four( + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -98,19 +106,18 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -118,7 +125,7 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -134,8 +141,10 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -146,10 +155,15 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace_async - async def test_five(self, **kwargs: Any) -> None: + async def test_five( + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -157,19 +171,19 @@ async def test_five(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -178,8 +192,10 @@ async def test_five(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -190,4 +206,6 @@ async def test_five(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py index 63672cad01d..a21d2c11b05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py @@ -26,13 +26,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelThree", - "PagingResult", - "SourcePath", - "ContentType", + 'Error', + 'ModelThree', + 'PagingResult', + 'SourcePath', + 'ContentType', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py index 0b7fe77b89e..0935a52d2d0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -51,7 +52,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -59,7 +65,6 @@ def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> super().__init__(**kwargs) self.optional_property = optional_property - class PagingResult(_serialization.Model): """PagingResult. @@ -75,7 +80,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -87,7 +96,6 @@ def __init__( self.values = values self.next_link = next_link - class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -96,14 +104,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py index 7179ffb9c14..853a409c981 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py @@ -11,7 +11,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py index 74b0a8bfab5..0e01b28839e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py @@ -9,14 +9,7 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -31,25 +24,32 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request(**kwargs: Any) -> HttpRequest: +def build_test_paging_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( @@ -62,27 +62,34 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -90,7 +97,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -100,19 +110,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -122,33 +130,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -158,7 +165,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -181,21 +192,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -207,8 +216,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -219,4 +230,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py index 96c3dd80870..3483038be34 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -31,50 +24,64 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_test_operation_group_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class OperationGroupOneOperations: + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -94,8 +101,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -105,19 +118,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -127,33 +138,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -163,7 +173,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @overload def test_two( @@ -187,7 +201,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -201,9 +219,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -215,19 +236,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -236,7 +256,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -251,8 +271,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -262,9 +284,14 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py index 7cb38df4f2c..74799e52c56 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,55 +22,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(**kwargs: Any) -> HttpRequest: +def build_test_four_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_five_request(**kwargs: Any) -> HttpRequest: +def build_test_five_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -97,9 +105,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -115,7 +130,11 @@ def test_four( @overload def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -130,9 +149,12 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -144,19 +166,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -164,7 +185,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -180,8 +201,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -192,10 +215,15 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -203,19 +231,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -224,8 +252,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -236,4 +266,6 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py index 6810552fa19..c33bf8ae0ad 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_metadata.json index cb47fa50483..3edaf131320 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py index 4c91d87e39f..1680db68ebe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +38,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py index e704da2b33e..01d21e467bc 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py index 3d1a97d3380..4eb5c4b4501 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,38 +39,31 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py index 1a4887a3ccf..8f2974137d2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,16 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.polling.async_base_polling import AsyncLROBasePolling @@ -30,23 +21,19 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_lro_and_paging_request, - build_test_lro_request, - build_test_one_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -54,7 +41,13 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> None: + async def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """TestOne should be in an FirstVersionOperationsMixin. :keyword id: An int parameter. Required. @@ -66,21 +59,19 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -92,7 +83,9 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -103,24 +96,27 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -129,7 +125,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -142,10 +138,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -162,13 +160,19 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload async def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -185,7 +189,11 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -200,9 +208,12 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -217,45 +228,59 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + async def _test_lro_and_paging_initial( self, @@ -265,17 +290,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -292,10 +316,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -311,10 +337,12 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -340,16 +368,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -357,7 +383,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -370,25 +396,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -396,7 +419,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -406,20 +431,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -427,27 +456,39 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace_async - async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) -> None: + async def test_different_calls( + self, + *, + greeting_in_english: str, + **kwargs: Any + ) -> None: """Has added parameters across the API versions. :keyword greeting_in_english: pass in 'hello' to pass test. Required. @@ -457,21 +498,19 @@ async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -482,7 +521,9 @@ async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -493,4 +534,6 @@ async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py index 41b52864a73..f7bf3c18dab 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -28,11 +21,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -52,8 +44,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -61,19 +59,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -82,8 +80,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -94,4 +94,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py index 57b29f30605..61d47a63696 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py index b8fe62eda38..588668551b4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -36,95 +27,133 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_test_lro_and_paging_request( - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: +def build_test_different_calls_request( + *, + greeting_in_english: str, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -133,7 +162,11 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, *, id: int, message: Optional[str] = None, **kwargs: Any + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -146,21 +179,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -171,8 +202,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -183,24 +216,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -209,7 +245,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -222,10 +258,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -242,13 +280,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -264,7 +308,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -278,9 +326,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -294,45 +345,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -342,17 +407,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -369,10 +433,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -388,10 +454,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -416,16 +484,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -433,7 +499,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -446,33 +512,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -482,20 +547,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -503,28 +572,38 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, *, greeting_in_english: str, **kwargs: Any + self, + *, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -535,21 +614,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -559,8 +636,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -571,4 +650,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py index 1db6fa5ca6b..cef549b274d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -28,33 +21,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,8 +74,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -83,19 +89,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -104,8 +110,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -116,4 +124,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py index 74eee4f7f7d..6c56eb09de7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py index 30bdc5c6c06..9c30b7b7883 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,39 +38,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py index 2faf9fb852f..340e36ee459 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py index 0dc415a7c7e..e8488e1306d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,41 +39,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py index b6feefbb231..400c5bfdd1c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,12 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -41,7 +35,13 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + async def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -53,21 +53,19 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -79,7 +77,9 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -89,16 +89,25 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def test_different_calls( - self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -111,21 +120,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -137,7 +144,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -148,4 +157,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py index f6f4410b013..138759d999d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,9 +45,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -71,7 +70,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -85,9 +88,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -99,19 +105,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -120,7 +125,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -135,8 +140,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -146,15 +153,23 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async - async def test_three(self, **kwargs: Any) -> None: + async def test_three( + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -162,19 +177,19 @@ async def test_three(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -183,8 +198,10 @@ async def test_three(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -195,4 +212,6 @@ async def test_three(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py index 58c8d1ca210..cb666b70c10 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -28,11 +21,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -52,8 +44,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: + async def test_four( + self, + *, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :keyword parameter_one: A boolean parameter. Required. @@ -63,19 +63,19 @@ async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -85,8 +85,10 @@ async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -97,4 +99,6 @@ async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py index ed8e322c54e..bdef0772a67 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py @@ -20,10 +20,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelTwo", + 'Error', + 'ModelTwo', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py index 46990252065..14a84ef76da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -36,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -49,7 +53,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -58,7 +62,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py index a5233c33e63..b12b08dc834 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,60 +22,81 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( - *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -90,7 +104,13 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -102,21 +122,19 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -127,8 +145,10 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -138,16 +158,25 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -160,21 +189,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -185,8 +212,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -197,4 +226,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py index 9c4025eee8f..86f64e9544f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,55 +22,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_three_request(**kwargs: Any) -> HttpRequest: +def build_test_three_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -97,9 +105,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -115,7 +130,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -129,9 +148,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -143,19 +165,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -164,7 +185,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -179,8 +200,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -190,15 +213,23 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -206,19 +237,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -227,8 +258,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -239,4 +272,6 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py index 9aa1d1c87ce..a635aa4185c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -28,34 +21,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: +def build_test_four_request( + *, + parameter_one: bool, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,9 +77,15 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, *, parameter_one: bool, **kwargs: Any + self, + *, + parameter_one: bool, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -88,19 +96,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -110,8 +118,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -122,4 +132,6 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py index 525afec9c0b..fb0f812979a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_metadata.json index 89a87306b02..93cada4e403 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py index 1e86a7b095e..02a60977c75 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,39 +38,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py index fa0ebc27201..97bbef80a2b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,32 +28,38 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", []) - kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', []) + kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py index 519ff60a57e..63f6cce4446 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,41 +39,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py index e75413f0e1e..37956f27972 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,14 +10,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -25,21 +18,19 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_paging_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -47,7 +38,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -57,19 +51,17 @@ def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -79,25 +71,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -105,7 +94,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -115,7 +106,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @distributed_trace_async async def test_different_calls( @@ -139,21 +134,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -166,7 +159,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -177,4 +172,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py index a8632ecebad..6dcfb2235fe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py @@ -11,14 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -26,20 +19,16 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._operation_group_one_operations import ( - build_test_operation_group_paging_request, - build_test_two_request, -) +from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -59,8 +48,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -70,19 +65,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.M _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -92,33 +85,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -128,7 +120,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @overload async def test_two( @@ -152,7 +148,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -166,9 +166,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -180,19 +183,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -201,7 +203,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -216,8 +218,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -227,9 +231,14 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py index d159601c24a..e4f271fee50 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,9 +45,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -71,7 +70,11 @@ async def test_four( @overload async def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -86,8 +89,13 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async - async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: + async def test_four( + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -98,19 +106,18 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -118,7 +125,7 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -134,8 +141,10 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -146,10 +155,15 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace_async - async def test_five(self, **kwargs: Any) -> None: + async def test_five( + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -157,19 +171,19 @@ async def test_five(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -178,8 +192,10 @@ async def test_five(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -190,4 +206,6 @@ async def test_five(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py index 63672cad01d..a21d2c11b05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py @@ -26,13 +26,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelThree", - "PagingResult", - "SourcePath", - "ContentType", + 'Error', + 'ModelThree', + 'PagingResult', + 'SourcePath', + 'ContentType', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py index 558a0c7c0ca..32d705523ce 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -51,7 +52,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -59,7 +65,6 @@ def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> super().__init__(**kwargs) self.optional_property = optional_property - class PagingResult(_serialization.Model): """PagingResult. @@ -75,7 +80,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -87,7 +96,6 @@ def __init__( self.values = values self.next_link = next_link - class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -96,14 +104,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py index 7179ffb9c14..853a409c981 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py @@ -11,7 +11,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py index 6f2a65b7f58..90018679649 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py @@ -9,14 +9,7 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -31,25 +24,32 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request(**kwargs: Any) -> HttpRequest: +def build_test_paging_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( @@ -62,27 +62,34 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -90,7 +97,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -100,19 +110,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -122,33 +130,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -158,7 +165,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -182,21 +193,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -208,8 +217,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -220,4 +231,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py index f45c29d6e24..62ef65c8a3f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -31,50 +24,64 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_test_operation_group_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class OperationGroupOneOperations: + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -94,8 +101,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -105,19 +118,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -127,33 +138,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -163,7 +173,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @overload def test_two( @@ -187,7 +201,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -201,9 +219,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -215,19 +236,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -236,7 +256,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -251,8 +271,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -262,9 +284,14 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py index 65c837d0fe7..0baf6a1a677 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,55 +22,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(**kwargs: Any) -> HttpRequest: +def build_test_four_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_five_request(**kwargs: Any) -> HttpRequest: +def build_test_five_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -97,9 +105,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -115,7 +130,11 @@ def test_four( @overload def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -130,9 +149,12 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -144,19 +166,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -164,7 +185,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -180,8 +201,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -192,10 +215,15 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -203,19 +231,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -224,8 +252,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -236,4 +266,6 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py index c4224d94b4c..e500e64f9d9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapinoasync/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapinoasync/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_metadata.json index 23c84d47768..23fedc11828 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py index 6c7eda4e245..0e354639f45 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,37 +39,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py index 3379281e03e..24c341cd35b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py index f9a3405e005..8a730caa052 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -37,95 +28,133 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_test_lro_and_paging_request( - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: +def build_test_different_calls_request( + *, + greeting_in_english: str, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -134,7 +163,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, id: int, message: Optional[str] = None, **kwargs: Any + self, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -147,21 +179,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -172,8 +202,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -184,24 +216,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -210,7 +245,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -223,10 +258,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -243,13 +280,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -265,7 +308,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -279,9 +326,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -295,45 +345,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -342,17 +406,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -369,10 +432,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -388,10 +453,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -415,16 +482,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -432,7 +497,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -445,33 +510,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -481,20 +545,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -502,28 +570,37 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, **kwargs: Any + self, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -534,21 +611,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -558,8 +633,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -570,4 +647,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py index fa640fca422..dcc682277c3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,33 +22,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,8 +75,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -84,19 +90,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -105,8 +111,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -117,4 +125,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py index 91e7712fba7..dc493e12998 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapinoasync/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapinoasync/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py index b42aa5f432b..374aa10b974 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py index ed8e322c54e..bdef0772a67 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py @@ -20,10 +20,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelTwo", + 'Error', + 'ModelTwo', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py index 46990252065..14a84ef76da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -36,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -49,7 +53,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -58,7 +62,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py index 9cd9fc89b1d..f733b815f65 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,60 +23,81 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( - *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +105,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -103,21 +122,19 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -128,8 +145,10 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -139,16 +158,24 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -161,21 +188,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -186,8 +211,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -198,4 +225,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py index cd290cb1235..b9bf5a48d4d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_three_request(**kwargs: Any) -> HttpRequest: +def build_test_three_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -116,7 +131,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -130,9 +149,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -144,19 +166,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -165,7 +186,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -180,8 +201,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -191,15 +214,23 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -207,19 +238,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -228,8 +259,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -240,4 +273,6 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py index 240741ec538..d65087d810f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,34 +22,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: +def build_test_four_request( + *, + parameter_one: bool, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -76,8 +78,15 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_four( # pylint: disable=inconsistent-return-statements + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -87,19 +96,19 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -109,8 +118,10 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -121,4 +132,6 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py index 74845e1428b..feb2223ffe9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapinoasync/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapinoasync/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_metadata.json index ca1ede8bb18..4c4a1de399c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py index 089af5a6d19..5043f2ec322 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py index 63672cad01d..a21d2c11b05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py @@ -26,13 +26,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelThree", - "PagingResult", - "SourcePath", - "ContentType", + 'Error', + 'ModelThree', + 'PagingResult', + 'SourcePath', + 'ContentType', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py index 04f6ab18536..284706c8d1a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -51,7 +52,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -59,7 +65,6 @@ def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> super().__init__(**kwargs) self.optional_property = optional_property - class PagingResult(_serialization.Model): """PagingResult. @@ -75,7 +80,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -87,7 +96,6 @@ def __init__( self.values = values self.next_link = next_link - class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -96,14 +104,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py index 7179ffb9c14..853a409c981 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py @@ -11,7 +11,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py index b23e72f4dea..ae49a51f0e9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py @@ -9,14 +9,7 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,25 +25,32 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request(**kwargs: Any) -> HttpRequest: +def build_test_paging_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( @@ -63,27 +63,34 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +98,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -101,19 +111,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -123,33 +131,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -159,7 +166,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -182,21 +193,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -208,8 +217,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -220,4 +231,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py index d230ae6ed31..3b6f2f326a1 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,50 +25,64 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_test_operation_group_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class OperationGroupOneOperations: + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -95,8 +102,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -106,19 +119,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -128,33 +139,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -164,7 +174,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @overload def test_two( @@ -188,7 +202,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -202,9 +220,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -216,19 +237,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -237,7 +257,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -252,8 +272,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -263,9 +285,14 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py index 238f6b6713a..7d5f71cbbe8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(**kwargs: Any) -> HttpRequest: +def build_test_four_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_five_request(**kwargs: Any) -> HttpRequest: +def build_test_five_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -116,7 +131,11 @@ def test_four( @overload def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,9 +150,12 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -145,19 +167,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -165,7 +186,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -181,8 +202,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -193,10 +216,15 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -204,19 +232,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -225,8 +253,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -237,4 +267,6 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py index 9851b5dfa42..113c6839f8b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,30 +28,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "0.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py index fda5b5367df..4cb81c54e62 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient: """Service client for multiapi client testing. @@ -37,36 +36,33 @@ class MultiapiServiceClient: :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "0.0.0" + self._client, self._config, self._serialize, self._deserialize, '0.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py index 74a93dfdd47..c6876a549cf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,30 +28,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "0.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py index 0010438f296..76477c40daf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient: """Service client for multiapi client testing. @@ -38,38 +37,31 @@ class MultiapiServiceClient: """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "0.0.0" + self._client, self._config, self._serialize, self._deserialize, '0.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py index f8443c0f5fe..9631441cd6a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "OperationGroupOneOperations", + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py index e5a310e7549..da36d153d4c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -28,11 +21,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -52,8 +44,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -61,19 +59,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -82,8 +80,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -94,4 +94,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py index 513632adef7..0b2869020ef 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py @@ -21,11 +21,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", + 'Error', + 'PagingResult', + 'Product', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py index 9797a5a8c77..133294256cc 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py index f8443c0f5fe..9631441cd6a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py @@ -17,9 +17,8 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "OperationGroupOneOperations", + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py index 41c1c3b31d9..ed85ed89e58 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -28,33 +21,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "0.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "0.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,8 +74,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -83,19 +89,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -104,8 +110,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -116,4 +124,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py index ab8adb477f6..091c7f939c4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,30 +28,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_metadata.json index 2afa2ae0f40..0e6aee39314 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py index e26d21c5737..9a3cdde8c50 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +38,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py index cc1b4d0032a..e7cd00bea2d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py @@ -15,8 +15,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,30 +28,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py index 7e9382d9988..4529fbd189a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py @@ -22,7 +22,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,38 +39,31 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py index 3dc7f3d0999..5cebd8b5abd 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,16 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.polling.async_base_polling import AsyncLROBasePolling @@ -30,23 +21,19 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_lro_and_paging_request, - build_test_lro_request, - build_test_one_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -54,7 +41,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -66,21 +58,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -92,7 +82,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -103,24 +95,27 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -129,7 +124,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -142,10 +137,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -162,13 +159,19 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload async def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -185,7 +188,11 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -200,9 +207,12 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -217,45 +227,59 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + async def _test_lro_and_paging_initial( self, @@ -264,17 +288,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -291,10 +314,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -310,10 +335,12 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -338,16 +365,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -355,7 +380,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -368,25 +393,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -394,7 +416,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -404,20 +428,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -425,27 +453,38 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace_async - async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: + async def test_different_calls( + self, + greeting_in_english: str, + **kwargs: Any + ) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -455,21 +494,19 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -480,7 +517,9 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -491,4 +530,6 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py index 1a3584ea4a0..aea40996aea 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -28,11 +21,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -52,8 +44,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -61,19 +59,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -82,8 +80,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -94,4 +94,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py index c9ddcd7fa74..f689de5518f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py index 4efe481b2f2..0813100f380 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -36,95 +27,133 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_test_lro_and_paging_request( - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: +def build_test_different_calls_request( + *, + greeting_in_english: str, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -133,7 +162,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, id: int, message: Optional[str] = None, **kwargs: Any + self, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -146,21 +178,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -171,8 +201,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -183,24 +215,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -209,7 +244,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -222,10 +257,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -242,13 +279,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -264,7 +307,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -278,9 +325,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -294,45 +344,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -341,17 +405,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -368,10 +431,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -387,10 +452,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -414,16 +481,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -431,7 +496,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -444,33 +509,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -480,20 +544,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -501,28 +569,37 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, **kwargs: Any + self, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -533,21 +610,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -557,8 +632,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -569,4 +646,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py index 883e25b35e0..e31c7f58bac 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -28,33 +21,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,8 +74,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -83,19 +89,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -104,8 +110,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -116,4 +124,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py index e69de29bb2d..5960c353a89 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py @@ -0,0 +1 @@ +__path__ = __import__('pkgutil').extend_path(__path__, __name__) # type: ignore \ No newline at end of file diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py index 3dd2e667247..8aed01eef43 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_metadata.json index 1d15d3d6390..9c6ece66781 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py index d804d78d30d..64207a9ba7a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,37 +40,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py index d9524f4ef39..fac9f3e204e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py index 4bc4f474464..c3749bdf9c3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -42,39 +41,31 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py index 7f47fddb36c..39ddb101c6a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,16 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -31,23 +22,19 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_lro_and_paging_request, - build_test_lro_request, - build_test_one_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -55,7 +42,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -67,21 +59,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -93,7 +83,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -104,24 +96,27 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -130,7 +125,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -143,10 +138,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -163,13 +160,19 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload async def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -186,7 +189,11 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -201,9 +208,12 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -218,45 +228,59 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + async def _test_lro_and_paging_initial( self, @@ -265,17 +289,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -292,10 +315,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -311,10 +336,12 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -340,16 +367,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -357,7 +382,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -370,25 +395,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -396,7 +418,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -406,20 +430,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -427,27 +455,38 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace_async - async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: + async def test_different_calls( + self, + greeting_in_english: str, + **kwargs: Any + ) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -457,21 +496,19 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -482,7 +519,9 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -493,4 +532,6 @@ async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py index f76a78464d8..28e6b513ae6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,8 +45,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -62,19 +60,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -83,8 +81,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -95,4 +95,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py index cbc5fbd265b..10facdeb65c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py index dc6be7ed447..b928092d59f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py index 3a3cd22d345..a422dcf113e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -37,95 +28,133 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_test_lro_and_paging_request( - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: +def build_test_different_calls_request( + *, + greeting_in_english: str, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -134,7 +163,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, id: int, message: Optional[str] = None, **kwargs: Any + self, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -147,21 +179,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -172,8 +202,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -184,24 +216,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -210,7 +245,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -223,10 +258,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -243,13 +280,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -265,7 +308,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -279,9 +326,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -295,45 +345,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -342,17 +406,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -369,10 +432,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -388,10 +453,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -416,16 +483,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -433,7 +498,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -446,33 +511,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -482,20 +546,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -503,28 +571,37 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, **kwargs: Any + self, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -535,21 +612,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -559,8 +634,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -571,4 +648,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py index f9fd0e6235a..01e61d3cfb0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,33 +22,40 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,8 +75,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -84,19 +90,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -105,8 +111,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -117,4 +125,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py index ccf9343999a..9725fcb3b82 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py index d4e7e77885b..1d59320acea 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -42,40 +41,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py index 3ef47ecd0f2..58f33c99bff 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py index dba17194fa3..2055ea2d2f8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -43,42 +42,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py index 3dcd40687ba..41e0dbe6cda 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,12 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -42,7 +36,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + async def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -54,21 +53,19 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -80,7 +77,9 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -90,16 +89,24 @@ async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def test_different_calls( - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -112,21 +119,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -138,7 +143,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -149,4 +156,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py index 07f2b00ced9..763490fe36b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -54,9 +46,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -72,7 +71,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -86,9 +89,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -100,19 +106,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -121,7 +126,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -136,8 +141,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -147,15 +154,23 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async - async def test_three(self, **kwargs: Any) -> None: + async def test_three( + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -163,19 +178,19 @@ async def test_three(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -184,8 +199,10 @@ async def test_three(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -196,4 +213,6 @@ async def test_three(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py index 9f90431c490..a4e5b03665a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -29,11 +22,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -53,8 +45,15 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: + async def test_four( + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -64,19 +63,19 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -86,8 +85,10 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -98,4 +99,6 @@ async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py index ed8e322c54e..bdef0772a67 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py @@ -20,10 +20,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelTwo", + 'Error', + 'ModelTwo', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py index 46990252065..14a84ef76da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -36,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -49,7 +53,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -58,7 +62,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py index f4643b30912..babbe906b47 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,60 +23,81 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( - *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +105,12 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -103,21 +122,19 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_test_one_request( id=id, message=message, @@ -128,8 +145,10 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -139,16 +158,24 @@ def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _mo error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -161,21 +188,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -186,8 +211,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -198,4 +225,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py index 7348de6757f..2ecb72bfc04 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_three_request(**kwargs: Any) -> HttpRequest: +def build_test_three_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -116,7 +131,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -130,9 +149,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -144,19 +166,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -165,7 +186,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -180,8 +201,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -191,15 +214,23 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -207,19 +238,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -228,8 +259,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -240,4 +273,6 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py index 63bc7ccc0d0..02afdc02e66 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py @@ -8,14 +8,7 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -29,34 +22,43 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: +def build_test_four_request( + *, + parameter_one: bool, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -76,8 +78,15 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_four( # pylint: disable=inconsistent-return-statements + self, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -87,19 +96,19 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -109,8 +118,10 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -121,4 +132,6 @@ def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disa raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py index f39362fec82..8f0d64c37df 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_metadata.json index d3520011b2b..bf467aa3b60 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py index b507e04b23a..9b29d6aa0e6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -42,40 +41,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py index 6da45f95f65..303ce374204 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py index abaf7900ef2..cf20d39419c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -43,42 +42,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py index b1d0c672ba7..8bd8701687c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,14 +10,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -26,21 +19,19 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._multiapi_service_client_operations import ( - build_test_different_calls_request, - build_test_paging_request, -) +from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -48,7 +39,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -59,19 +53,17 @@ def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -81,25 +73,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -107,7 +96,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -117,7 +108,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @distributed_trace_async async def test_different_calls( @@ -140,21 +135,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -167,7 +160,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -178,4 +173,6 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py index 87e66f8c331..b9c451da297 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py @@ -11,14 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -27,20 +20,16 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operation_group_one_operations import ( - build_test_operation_group_paging_request, - build_test_two_request, -) +from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -60,8 +49,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -72,19 +67,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.M _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -94,33 +87,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -130,7 +122,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @overload async def test_two( @@ -154,7 +150,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -168,9 +168,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -182,19 +185,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -203,7 +205,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -218,8 +220,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -229,9 +233,14 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py index 05151725dd8..ccc32916176 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -30,11 +23,10 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -54,9 +46,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -72,7 +71,11 @@ async def test_four( @overload async def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -87,8 +90,13 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async - async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: + async def test_four( + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -99,19 +107,18 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -119,7 +126,7 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -135,8 +142,10 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -147,10 +156,15 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace_async - async def test_five(self, **kwargs: Any) -> None: + async def test_five( + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -158,19 +172,19 @@ async def test_five(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -179,8 +193,10 @@ async def test_five(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -191,4 +207,6 @@ async def test_five(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py index 63672cad01d..a21d2c11b05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py @@ -26,13 +26,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelThree", - "PagingResult", - "SourcePath", - "ContentType", + 'Error', + 'ModelThree', + 'PagingResult', + 'SourcePath', + 'ContentType', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py index a7d63282c19..5768703fec4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -51,7 +52,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -59,7 +65,6 @@ def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> super().__init__(**kwargs) self.optional_property = optional_property - class PagingResult(_serialization.Model): """PagingResult. @@ -75,7 +80,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -87,7 +96,6 @@ def __init__( self.values = values self.next_link = next_link - class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -96,14 +104,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py index 7179ffb9c14..853a409c981 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py @@ -11,7 +11,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py index 63452e61cf6..be08986800c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py index 95b84481ec3..376c4d761fd 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py @@ -9,14 +9,7 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,25 +25,32 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request(**kwargs: Any) -> HttpRequest: +def build_test_paging_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) def build_test_different_calls_request( @@ -63,27 +63,34 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -91,7 +98,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -101,19 +111,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -123,33 +131,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -159,7 +166,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -182,21 +193,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -208,8 +217,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -220,4 +231,6 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py index fb261b27c7a..058189a157d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -32,50 +25,64 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_test_operation_group_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) -def build_test_two_request(**kwargs: Any) -> HttpRequest: +def build_test_two_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class OperationGroupOneOperations: + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -95,8 +102,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -106,19 +119,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -128,33 +139,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -164,7 +174,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @overload def test_two( @@ -188,7 +202,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -202,9 +220,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -216,19 +237,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -237,7 +257,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -252,8 +272,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -263,9 +285,14 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py index efedc6f0ca9..f835e9c2e6f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -30,55 +23,70 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request(**kwargs: Any) -> HttpRequest: +def build_test_four_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_test_five_request(**kwargs: Any) -> HttpRequest: +def build_test_five_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) - -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -98,9 +106,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -116,7 +131,11 @@ def test_four( @overload def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,9 +150,12 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -145,19 +167,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -165,7 +186,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -181,8 +202,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -193,10 +216,15 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -204,19 +232,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -225,8 +253,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -237,4 +267,6 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py index 5213f6b72da..54a3a3b8fda 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py index f26bfbe150b..01d4dc77f92 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py index 5213f6b72da..54a3a3b8fda 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py index 924dbdb2511..cfe7c788bc7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py index c3992e19425..598f872188b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py @@ -11,14 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -27,28 +20,19 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operations import ( - build_multiapi_service_test_different_calls_request, - build_multiapi_service_test_paging_request, - build_operation_group_one_test_operation_group_paging_request, - build_operation_group_one_test_two_request, - build_operation_group_two_test_five_request, - build_operation_group_two_test_four_request, -) +from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_paging_request, build_operation_group_one_test_operation_group_paging_request, build_operation_group_one_test_two_request, build_operation_group_two_test_five_request, build_operation_group_two_test_four_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, Optional, TypeVar, Union, cast, overload -from azure.core.exceptions import StreamClosedError -from azure.core.exceptions import StreamConsumedError +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload @@ -85,7 +69,9 @@ def __init__(self, *args, **kwargs): ) @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -97,19 +83,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -118,7 +103,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -133,8 +118,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -144,10 +131,13 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -156,7 +146,10 @@ async def test_two( method_valid_on=['2.0.0'], ) @distributed_trace_async - async def test_three(self, **kwargs: Any) -> None: + async def test_three( + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -164,19 +157,19 @@ async def test_three(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -185,8 +178,10 @@ async def test_three(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -197,14 +192,17 @@ async def test_three(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -214,19 +212,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.M _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -236,33 +232,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -272,7 +267,10 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) class OperationGroupTwoOperations: @@ -302,7 +300,11 @@ def __init__(self, *args, **kwargs): } ) @distributed_trace_async - async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: + async def test_four( + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -313,19 +315,18 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -333,7 +334,7 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -349,8 +350,10 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -361,14 +364,17 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace_async - async def test_five(self, **kwargs: Any) -> None: + async def test_five( + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -376,19 +382,19 @@ async def test_five(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -397,8 +403,10 @@ async def test_five(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -409,7 +417,7 @@ async def test_five(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): @@ -424,17 +432,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -451,10 +458,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -470,27 +479,28 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -499,7 +509,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -512,10 +522,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -532,7 +544,7 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -542,7 +554,9 @@ async def _test_lro_initial( ) @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -557,45 +571,57 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -626,16 +652,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -643,7 +667,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -656,25 +680,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -682,7 +703,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -692,20 +715,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -713,24 +740,29 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -761,21 +793,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -788,7 +818,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -799,14 +831,20 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['1.0.0', '2.0.0'], ) @distributed_trace_async - async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + async def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -818,21 +856,19 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -844,7 +880,9 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -854,10 +892,13 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -866,7 +907,10 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An method_valid_on=['3.0.0'], ) @distributed_trace - def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -876,19 +920,17 @@ def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -898,25 +940,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -924,7 +963,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -934,6 +975,9 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py index 8f82806ad3c..454cdc8be88 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py @@ -13,7 +13,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py index e572bc9fc4c..7022dd2e9d2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): @@ -30,7 +26,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -56,7 +58,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -79,7 +85,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -102,7 +113,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int @@ -126,7 +143,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -135,7 +152,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. @@ -158,7 +179,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -174,14 +200,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py index f0a399d0445..f0771525fbe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -33,7 +26,7 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() @@ -42,66 +35,86 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload -from azure.core.exceptions import StreamClosedError -from azure.core.exceptions import StreamConsumedError +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.polling import LROPoller, NoPolling, PollingMethod from azure.mgmt.core.polling.arm_polling import ARMPolling from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload from .._validation import api_version_validation -def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_one_test_two_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_operation_group_one_test_three_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_one_test_three_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_operation_group_one_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) class OperationGroupOneOperations: """ @@ -130,7 +143,9 @@ def __init__(self, *args, **kwargs): ) @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -142,19 +157,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -163,7 +177,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -178,8 +192,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -189,10 +205,13 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -201,7 +220,10 @@ def test_two( method_valid_on=['2.0.0'], ) @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -209,19 +231,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -230,8 +252,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -242,14 +266,17 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -259,19 +286,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -281,33 +306,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -317,47 +341,66 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + return ItemPaged( + get_next, extract_data + ) -def build_operation_group_two_test_four_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long + +def build_operation_group_two_test_four_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_operation_group_two_test_five_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_two_test_five_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) class OperationGroupTwoOperations: """ @@ -387,7 +430,9 @@ def __init__(self, *args, **kwargs): ) @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -399,19 +444,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -419,7 +463,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -435,8 +479,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -447,14 +493,17 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -462,19 +511,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -483,8 +532,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -495,108 +546,148 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore def build_multiapi_service_test_lro_and_paging_request( # pylint: disable=name-too-long - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) -def build_multiapi_service_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_multiapi_service_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_multiapi_service_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_multiapi_service_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_multiapi_service_test_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_multiapi_service_test_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: @@ -610,17 +701,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -637,10 +727,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -656,27 +748,28 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -685,7 +778,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -698,10 +791,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -718,7 +813,7 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -728,7 +823,9 @@ def _test_lro_initial( ) @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -742,45 +839,57 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -810,16 +919,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -827,7 +934,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -840,33 +947,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -876,20 +982,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -897,24 +1007,29 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -930,7 +1045,7 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -945,21 +1060,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -971,8 +1084,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -983,14 +1098,20 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['1.0.0', '2.0.0'], ) @distributed_trace - def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -1002,21 +1123,19 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -1027,8 +1146,10 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -1038,10 +1159,13 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -1050,7 +1174,10 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> method_valid_on=['3.0.0'], ) @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -1060,19 +1187,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -1082,33 +1207,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -1118,6 +1242,9 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py index 79e32d87d94..91bd75b1569 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_metadata.json index ff69f821a3d..9211daf25c4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_metadata.json @@ -95,8 +95,8 @@ "operation_group_one": "OperationGroupOneOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"IO\", \"Iterator\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"LROPoller\"], \"azure.core.paging\": [\"ItemPaged\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"stdlib\": {\"typing\": [\"AsyncIterator\", \"IO\", \"Optional\", \"Union\"], \"io\": [\"IOBase\"]}, \"sdkcore\": {\"azure.core.polling\": [\"AsyncLROPoller\"], \"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py index 24b18be3e61..0aabb0fc6b3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,37 +39,33 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py index f706e232a1e..f63da44aaf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "1.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py index b402ddec33b..4db51ca16f9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,39 +40,31 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "1.0.0" + self._client, self._config, self._serialize, self._deserialize, '1.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py index 01939b64a84..8eb92c30ca1 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py index e915b1c289f..e658f770445 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py @@ -11,16 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -31,24 +22,19 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._operations import ( - build_multiapi_service_test_different_calls_request, - build_multiapi_service_test_lro_and_paging_request, - build_multiapi_service_test_lro_request, - build_multiapi_service_test_one_request, - build_operation_group_one_test_two_request, -) +from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_lro_and_paging_request, build_multiapi_service_test_lro_request, build_multiapi_service_test_one_request, build_operation_group_one_test_two_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -56,7 +42,13 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> None: + async def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """TestOne should be in an FirstVersionOperationsMixin. :keyword id: An int parameter. Required. @@ -68,21 +60,19 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -94,7 +84,9 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -105,24 +97,27 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + async def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -131,7 +126,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -144,10 +139,12 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -164,13 +161,19 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload async def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -187,7 +190,11 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -202,9 +209,12 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -219,45 +229,59 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + async def _test_lro_and_paging_initial( self, @@ -267,17 +291,16 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -294,10 +317,12 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -313,10 +338,12 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -342,16 +369,14 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -359,7 +384,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -372,25 +397,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -398,7 +420,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -408,20 +432,24 @@ async def get_next(next_link=None): return pipeline_response - polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + await raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -429,27 +457,39 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged(internal_get_next, extract_data) + return AsyncItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: - polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace_async - async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) -> None: + async def test_different_calls( + self, + *, + greeting_in_english: str, + **kwargs: Any + ) -> None: """Has added parameters across the API versions. :keyword greeting_in_english: pass in 'hello' to pass test. Required. @@ -459,21 +499,19 @@ async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -484,7 +522,9 @@ async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -495,10 +535,10 @@ async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -518,8 +558,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_two(self, **kwargs: Any) -> None: + async def test_two( + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -527,19 +573,19 @@ async def test_two(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_one_test_two_request( api_version=api_version, headers=_headers, @@ -548,8 +594,10 @@ async def test_two(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -560,4 +608,6 @@ async def test_two(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py index e389a34d387..34986ba29ae 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py @@ -22,12 +22,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "PagingResult", - "Product", - "TestLroAndPagingOptions", + 'Error', + 'PagingResult', + 'Product', + 'TestLroAndPagingOptions', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py index d1cf00f88d6..d47335f09b7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class PagingResult(_serialization.Model): """PagingResult. @@ -55,7 +56,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.Product"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -67,7 +72,6 @@ def __init__( self.values = values self.next_link = next_link - class Product(_serialization.Model): """Product. @@ -79,7 +83,12 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin + def __init__( + self, + *, + id: Optional[int] = None, # pylint: disable=redefined-builtin + **kwargs: Any + ) -> None: """ :keyword id: :paramtype id: int @@ -87,7 +96,6 @@ def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylin super().__init__(**kwargs) self.id = id - class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -103,7 +111,13 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: + def __init__( + self, + *, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any + ) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py index 01939b64a84..8eb92c30ca1 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py @@ -18,10 +18,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py index a7e0aeec963..e4dd39c9826 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py @@ -10,16 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - StreamClosedError, - StreamConsumedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -37,116 +28,160 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_multiapi_service_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_multiapi_service_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_multiapi_service_test_lro_request(**kwargs: Any) -> HttpRequest: +def build_multiapi_service_test_lro_request( + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + headers=_headers, + **kwargs + ) def build_multiapi_service_test_lro_and_paging_request( # pylint: disable=name-too-long - *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any + *, + client_request_id: Optional[str] = None, + maxresults: Optional[int] = None, + timeout: int = 30, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") + _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') if maxresults is not None: - _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") + _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') if timeout is not None: - _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + headers=_headers, + **kwargs + ) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long - *, greeting_in_english: str, **kwargs: Any + *, + greeting_in_english: str, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_one_test_two_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -155,7 +190,11 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, *, id: int, message: Optional[str] = None, **kwargs: Any + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -168,21 +207,19 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -193,8 +230,10 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -205,24 +244,27 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + def _test_lro_initial( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -231,7 +273,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, "Product") + _json = self._serialize.body(product, 'Product') else: _json = None @@ -244,10 +286,12 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -264,13 +308,19 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @overload def begin_test_lro( - self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[_models.Product] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -286,7 +336,11 @@ def begin_test_lro( @overload def begin_test_lro( - self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + product: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -300,9 +354,12 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def begin_test_lro( - self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any + self, + product: Optional[Union[_models.Product, IO[bytes]]] = None, + **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -316,45 +373,59 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.Product] = kwargs.pop("cls", None) - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.Product] = kwargs.pop( + 'cls', None + ) + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize("Product", pipeline_response.http_response) + deserialized = self._deserialize( + 'Product', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized + if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + def _test_lro_and_paging_initial( self, @@ -364,17 +435,16 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) + cls: ClsType[Iterator[bytes]] = kwargs.pop( + 'cls', None + ) _maxresults = None _timeout = None @@ -391,10 +461,12 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop("decompress", True) + _decompress = kwargs.pop('decompress', True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -410,10 +482,12 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def begin_test_lro_and_paging( self, @@ -438,16 +512,14 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -455,7 +527,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -468,33 +540,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response.http_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response.http_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -504,20 +575,24 @@ def get_next(next_link=None): return pipeline_response - polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) - lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) - cont_token: Optional[str] = kwargs.pop("continuation_token", None) + + polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) + lro_delay = kwargs.pop( + 'polling_interval', + self._config.polling_interval + ) + cont_token: Optional[str] = kwargs.pop('continuation_token', None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x, y, z: x, + cls=lambda x,y,z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop("error_map", None) + raw_result.http_response.read() # type: ignore + kwargs.pop('error_map', None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -525,28 +600,38 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged(internal_get_next, extract_data) + return ItemPaged( + internal_get_next, extract_data + ) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) - elif polling is False: - polling_method = cast(PollingMethod, NoPolling()) - else: - polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling( + lro_delay, + + + **kwargs + )) + elif polling is False: polling_method = cast(PollingMethod, NoPolling()) + else: polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output, + deserialization_callback=get_long_running_output ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, *, greeting_in_english: str, **kwargs: Any + self, + *, + greeting_in_english: str, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -557,21 +642,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -581,8 +664,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -593,10 +678,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -616,8 +701,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_two( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -625,19 +716,19 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_one_test_two_request( api_version=api_version, headers=_headers, @@ -646,8 +737,10 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -658,4 +751,6 @@ def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retur raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py index f9444070b03..abaaaa33886 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py index 5cef307ba5e..5deb24da5c6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py index 2156d35f402..926aea303a9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "2.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py index 258353eff7a..d4cfa27e1e5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,42 +40,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "2.0.0" + self._client, self._config, self._serialize, self._deserialize, '2.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py index 33560f2557a..6aaa304b8a7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py index 88972f2a779..6ad79f6b9d2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -24,24 +17,19 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operations import ( - build_multiapi_service_test_different_calls_request, - build_multiapi_service_test_one_request, - build_operation_group_one_test_three_request, - build_operation_group_one_test_two_request, - build_operation_group_two_test_four_request, -) +from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_one_request, build_operation_group_one_test_three_request, build_operation_group_one_test_two_request, build_operation_group_two_test_four_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -49,7 +37,13 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + async def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -61,21 +55,19 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -87,7 +79,9 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -97,16 +91,25 @@ async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: An error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async async def test_different_calls( - self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -119,21 +122,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -145,7 +146,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -156,10 +159,10 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -179,9 +182,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -197,7 +207,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -211,9 +225,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -225,19 +242,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -246,7 +262,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -261,8 +277,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -272,15 +290,23 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace_async - async def test_three(self, **kwargs: Any) -> None: + async def test_three( + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -288,19 +314,19 @@ async def test_three(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -309,8 +335,10 @@ async def test_three(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -321,10 +349,10 @@ async def test_three(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -344,8 +372,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace_async - async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: + async def test_four( + self, + *, + parameter_one: bool, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :keyword parameter_one: A boolean parameter. Required. @@ -355,19 +391,19 @@ async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_two_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -377,8 +413,10 @@ async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -389,4 +427,6 @@ async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py index ed8e322c54e..bdef0772a67 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py @@ -20,10 +20,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelTwo", + 'Error', + 'ModelTwo', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py index 46990252065..14a84ef76da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py @@ -10,7 +10,6 @@ from ... import _serialization - class Error(_serialization.Model): """Error. @@ -25,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -36,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -49,7 +53,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - "id": {"required": True}, + 'id': {'required': True}, } _attribute_map = { @@ -58,7 +62,11 @@ class ModelTwo(_serialization.Model): } def __init__( - self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin + self, + *, + id: int, # pylint: disable=redefined-builtin + message: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py index 33560f2557a..6aaa304b8a7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py index b15213994b5..afc59e1e43f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py @@ -9,14 +9,7 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -31,123 +24,168 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_multiapi_service_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: +def build_multiapi_service_test_one_request( + *, + id: int, + message: Optional[str] = None, + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params["id"] = _SERIALIZER.query("id", id, "int") + _params['id'] = _SERIALIZER.query("id", id, 'int') if message is not None: - _params["message"] = _SERIALIZER.query("message", message, "str") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['message'] = _SERIALIZER.query("message", message, 'str') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long - *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_one_test_two_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_operation_group_one_test_three_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_one_test_three_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_operation_group_two_test_four_request( # pylint: disable=name-too-long - *, parameter_one: bool, **kwargs: Any + *, + parameter_one: bool, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -155,7 +193,13 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: + def test_one( + self, + *, + id: int, + message: Optional[str] = None, + **kwargs: Any + ) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -167,21 +211,19 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None ) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -192,8 +234,10 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -203,16 +247,25 @@ def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any + self, + *, + greeting_in_english: str, + greeting_in_chinese: Optional[str] = None, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -225,21 +278,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -250,8 +301,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -262,10 +315,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -285,9 +338,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_two( - self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[_models.ModelTwo] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -303,7 +363,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -317,9 +381,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -331,19 +398,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -352,7 +418,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelTwo") + _json = self._serialize.body(parameter_one, 'ModelTwo') else: _json = None @@ -367,8 +433,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -378,15 +446,23 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelTwo', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore + + @distributed_trace - def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_three( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -394,19 +470,19 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -415,8 +491,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -427,10 +505,10 @@ def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -450,9 +528,15 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, *, parameter_one: bool, **kwargs: Any + self, + *, + parameter_one: bool, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -463,19 +547,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_two_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -485,8 +569,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -497,4 +583,6 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py index f26bfbe150b..01d4dc77f92 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "TokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_metadata.json b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_metadata.json index abe34802c8b..b580ceaa54c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_metadata.json +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_metadata.json @@ -96,8 +96,8 @@ "operation_group_two": "OperationGroupTwoOperations" }, "operation_mixins": { - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"local\": {\".\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"local\": {\"..\": [[\"models\", \"_models\"]]}, \"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}}}", + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Iterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.paging\": [\"ItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\".\": [[\"models\", \"_models\"]]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"AsyncIterable\"]}}, \"regular\": {\"sdkcore\": {\"azure.core.async_paging\": [\"AsyncItemPaged\"]}, \"stdlib\": {\"typing\": [\"Optional\"]}, \"local\": {\"..\": [[\"models\", \"_models\"]]}}}", "sync_mixin_typing_definitions": "", "async_mixin_typing_definitions": "", "operations": { diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py index ad63699a83c..7650ad6847b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,40 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: + def __init__( + self, + credential: "TokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any + ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - ARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) - def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + + def _send_request( + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any + ) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py index 70517e0b7af..614290bca5a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py @@ -15,11 +15,12 @@ from .._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py index 30c0d2f93d3..8f5084e3ef5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py @@ -20,10 +20,9 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClient", + 'MultiapiServiceClient', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py index 924dbdb2511..cfe7c788bc7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py @@ -16,8 +16,7 @@ VERSION = "unknown" - -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,30 +29,36 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-a :paramtype api_version: str """ - def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: - api_version: str = kwargs.pop("api_version", "3.0.0") + def __init__( + self, + credential: "AsyncTokenCredential", + **kwargs: Any + ) -> None: + api_version: str = kwargs.pop('api_version', "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) - kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) + self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) + kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - def _configure(self, **kwargs: Any) -> None: - self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get("authentication_policy") + + def _configure( + self, + **kwargs: Any + ) -> None: + self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get('authentication_policy') if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( - self.credential, *self.credential_scopes, **kwargs - ) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py index f840ea392b0..d7f9a304bc0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py @@ -23,7 +23,6 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential - class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,42 +40,34 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any + self, + credential: "AsyncTokenCredential", + base_url: str = "http://localhost:3000", + **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop("policies", None) + _policies = kwargs.pop('policies', None) if _policies is None: - _policies = [ - policies.RequestIdPolicy(**kwargs), - self._config.headers_policy, - self._config.user_agent_policy, - self._config.proxy_policy, - policies.ContentDecodePolicy(**kwargs), - AsyncARMAutoResourceProviderRegistrationPolicy(), - self._config.redirect_policy, - self._config.retry_policy, - self._config.authentication_policy, - self._config.custom_hook_policy, - self._config.logging_policy, - policies.DistributedTracingPolicy(**kwargs), - policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, - self._config.http_logging_policy, - ] + _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, "3.0.0" + self._client, self._config, self._serialize, self._deserialize, '3.0.0' ) + def _send_request( - self, request: HttpRequest, *, stream: bool = False, **kwargs: Any + self, + request: HttpRequest, *, stream: bool = False, + **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py index 9c6fd05bb21..d0b3400218b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py @@ -15,11 +15,12 @@ from ..._serialization import Deserializer, Serializer - -class MultiapiServiceClientMixinABC(ABC): +class MultiapiServiceClientMixinABC( + ABC +): """DO NOT use this class. It is for internal typing use only.""" - _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py index 33560f2557a..6aaa304b8a7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py index c33a78c1d80..98c53aed427 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py @@ -11,14 +11,7 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -27,25 +20,19 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operations import ( - build_multiapi_service_test_different_calls_request, - build_multiapi_service_test_paging_request, - build_operation_group_one_test_operation_group_paging_request, - build_operation_group_one_test_two_request, - build_operation_group_two_test_five_request, - build_operation_group_two_test_four_request, -) +from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_paging_request, build_operation_group_one_test_operation_group_paging_request, build_operation_group_one_test_two_request, build_operation_group_two_test_five_request, build_operation_group_two_test_four_request from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -53,7 +40,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -63,19 +53,17 @@ def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -85,25 +73,22 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -111,7 +96,9 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -121,7 +108,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @distributed_trace_async async def test_different_calls( @@ -145,21 +136,19 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -172,7 +161,9 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, stream=_stream, **kwargs + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -183,10 +174,10 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -206,8 +197,14 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -217,19 +214,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.M _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -239,33 +234,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -275,7 +269,11 @@ async def get_next(next_link=None): return pipeline_response - return AsyncItemPaged(get_next, extract_data) + + return AsyncItemPaged( + get_next, extract_data + ) + @overload async def test_two( @@ -299,7 +297,11 @@ async def test_two( @overload async def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -313,9 +315,12 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async async def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -327,19 +332,18 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -348,7 +352,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -363,8 +367,10 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -374,15 +380,18 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -402,9 +411,16 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload async def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -420,7 +436,11 @@ async def test_four( @overload async def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -435,8 +455,13 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace_async - async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: + async def test_four( + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any + ) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -447,19 +472,18 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -467,7 +491,7 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -483,8 +507,10 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -495,10 +521,15 @@ async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace_async - async def test_five(self, **kwargs: Any) -> None: + async def test_five( + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -506,19 +537,19 @@ async def test_five(self, **kwargs: Any) -> None: :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -527,8 +558,10 @@ async def test_five(self, **kwargs: Any) -> None: _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -539,4 +572,6 @@ async def test_five(self, **kwargs: Any) -> None: raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py index 63672cad01d..a21d2c11b05 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py @@ -26,13 +26,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "Error", - "ModelThree", - "PagingResult", - "SourcePath", - "ContentType", + 'Error', + 'ModelThree', + 'PagingResult', + 'SourcePath', + 'ContentType', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py index 9305829f310..7db31c76bef 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py @@ -6,14 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional from ... import _serialization -if TYPE_CHECKING: - from .. import models as _models - - class Error(_serialization.Model): """Error. @@ -28,7 +24,13 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + status: Optional[int] = None, + message: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword status: :paramtype status: int @@ -39,7 +41,6 @@ def __init__(self, *, status: Optional[int] = None, message: Optional[str] = Non self.status = status self.message = message - class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -51,7 +52,12 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + optional_property: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -59,7 +65,6 @@ def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> super().__init__(**kwargs) self.optional_property = optional_property - class PagingResult(_serialization.Model): """PagingResult. @@ -75,7 +80,11 @@ class PagingResult(_serialization.Model): } def __init__( - self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any + self, + *, + values: Optional[List["_models.ModelThree"]] = None, + next_link: Optional[str] = None, + **kwargs: Any ) -> None: """ :keyword values: @@ -87,7 +96,6 @@ def __init__( self.values = values self.next_link = next_link - class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -96,14 +104,19 @@ class SourcePath(_serialization.Model): """ _validation = { - "source": {"max_length": 2048}, + 'source': {'max_length': 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: + def __init__( + self, + *, + source: Optional[str] = None, + **kwargs: Any + ) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py index 7179ffb9c14..853a409c981 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py @@ -11,7 +11,8 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload.""" + """Content type for upload. + """ APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py index 33560f2557a..6aaa304b8a7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py @@ -19,11 +19,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk - __all__ = [ - "MultiapiServiceClientOperationsMixin", - "OperationGroupOneOperations", - "OperationGroupTwoOperations", + 'MultiapiServiceClientOperationsMixin', + 'OperationGroupOneOperations', + 'OperationGroupTwoOperations', ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py index 624e1d0b4b0..45d5174de3a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py @@ -10,14 +10,7 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ( - ClientAuthenticationError, - HttpResponseError, - ResourceExistsError, - ResourceNotFoundError, - ResourceNotModifiedError, - map_error, -) +from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -33,25 +26,32 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar("T") +T = TypeVar('T') ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_multiapi_service_test_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_multiapi_service_test_paging_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long @@ -59,111 +59,147 @@ def build_multiapi_service_test_different_calls_request( # pylint: disable=name greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') if greeting_in_chinese is not None: - _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') if greeting_in_french is not None: - _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) def build_operation_group_one_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any, + **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop("Accept", "application/json") + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + headers=_headers, + **kwargs + ) -def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_one_test_two_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="GET", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_operation_group_two_test_four_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_two_test_four_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers if content_type is not None: - _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + return HttpRequest( + method="POST", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) -def build_operation_group_two_test_five_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long +def build_operation_group_two_test_five_request( # pylint: disable=name-too-long + **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) - accept = _headers.pop("Accept", "application/json") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) + accept = _headers.pop('Accept', "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") + _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') # Construct headers - _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - - return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) - - -class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): + _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + + return HttpRequest( + method="PUT", + url=_url, + params=_params, + headers=_headers, + **kwargs + ) + +class MultiapiServiceClientOperationsMixin( + MultiapiServiceClientMixinABC +): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -171,7 +207,10 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -181,19 +220,17 @@ def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -203,33 +240,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -239,7 +275,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -248,7 +288,7 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any, + **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -263,21 +303,19 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop( - "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None ) - cls: ClsType[None] = kwargs.pop("cls", None) + _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -289,8 +327,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -301,10 +341,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -324,8 +364,14 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @distributed_trace - def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging( + self, + **kwargs: Any + ) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -335,19 +381,17 @@ def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelT _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) + cls: ClsType[_models.PagingResult] = kwargs.pop( + 'cls', None + ) error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) - + error_map.update(kwargs.pop('error_map', {}) or {}) def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -357,33 +401,32 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict( - { - key: [urllib.parse.quote(v) for v in value] - for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() - } - ) + _next_request_params = case_insensitive_dict({ + key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) _next_request_params["api-version"] = self._api_version - _request = HttpRequest( - "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params - ) + _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize("PagingResult", pipeline_response) + deserialized = self._deserialize( + "PagingResult", + pipeline_response + ) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -393,7 +436,11 @@ def get_next(next_link=None): return pipeline_response - return ItemPaged(get_next, extract_data) + + return ItemPaged( + get_next, extract_data + ) + @overload def test_two( @@ -401,7 +448,7 @@ def test_two( parameter_one: Optional[_models.ModelThree] = None, *, content_type: str = "application/json", - **kwargs: Any, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -417,7 +464,11 @@ def test_two( @overload def test_two( - self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any + self, + parameter_one: Optional[IO[bytes]] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -431,9 +482,12 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_two( - self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any + self, + parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, + **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -445,19 +499,18 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[_models.ModelThree] = kwargs.pop( + 'cls', None + ) content_type = content_type or "application/json" _json = None @@ -466,7 +519,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, "ModelThree") + _json = self._serialize.body(parameter_one, 'ModelThree') else: _json = None @@ -481,8 +534,10 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -492,15 +547,18 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize("ModelThree", pipeline_response.http_response) + deserialized = self._deserialize( + 'ModelThree', + pipeline_response.http_response + ) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -520,9 +578,16 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") + + + @overload def test_four( - self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any + self, + input: Optional[_models.SourcePath] = None, + *, + content_type: str = "application/json", + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -538,7 +603,11 @@ def test_four( @overload def test_four( - self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any + self, + input: Optional[IO[bytes]] = None, + *, + content_type: Optional[str] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -553,9 +622,12 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ + @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any + self, + input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, + **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -567,19 +639,18 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) _json = None _content = None @@ -587,7 +658,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, "SourcePath") + _json = self._serialize.body(input, 'SourcePath') else: _json = None content_type = content_type or "application/json" @@ -603,8 +674,10 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -615,10 +688,15 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + @distributed_trace - def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements + def test_five( # pylint: disable=inconsistent-return-statements + self, + **kwargs: Any + ) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -626,19 +704,19 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, - 404: ResourceNotFoundError, - 409: ResourceExistsError, - 304: ResourceNotModifiedError, + 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError } - error_map.update(kwargs.pop("error_map", {}) or {}) + error_map.update(kwargs.pop('error_map', {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop("cls", None) + api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop( + 'cls', None + ) + _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -647,8 +725,10 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, stream=_stream, **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, + stream=_stream, + **kwargs ) response = pipeline_response.http_response @@ -659,4 +739,6 @@ def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore + + diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py index a0a1d0d6aa6..b6f7560c84d 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py @@ -8,13 +8,10 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Optional, Union import msrest.serialization -if TYPE_CHECKING: - from .. import models as _models - class ArrayWrapper(msrest.serialization.Model): """ArrayWrapper. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py index d684fe1f617..5d43068368e 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Optional, TYPE_CHECKING, Union +from typing import Any, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Paths14Hl8BdFormsdataurlencodedPetAddPetidPostRequestbodyContentApplicationXWwwFormUrlencodedSchema( _serialization.Model diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py index 3f916138cc1..b0a67c1f5c8 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Literal, Optional, TYPE_CHECKING, Union +from typing import Any, Literal, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class ModelAsStringNoRequiredOneValueDefault(_serialization.Model): """ModelAsStringNoRequiredOneValueDefault. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py index 0721c41ba08..8f5c8887e95 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Error(_serialization.Model): """The error object. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py index 34a5f6ea52b..030adf044f8 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Optional, TYPE_CHECKING, Union +from typing import Any, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class Pet(_serialization.Model): """Pet. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py index 1473e8fb5fa..7ba62aeb64c 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING +from typing import Any, Dict, List, Literal, Optional from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class BaseProduct(_serialization.Model): """The product documentation. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/__init__.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/__init__.py index a0e664d7104..32afad42897 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/__init__.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/__init__.py @@ -12,6 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import +from ._no_operations_service_client import NoOperationsServiceClient # type: ignore from ._version import VERSION __version__ = VERSION @@ -23,7 +24,9 @@ _patch_all = [] from ._patch import patch_sdk as _patch_sdk -__all__ = [] +__all__ = [ + "NoOperationsServiceClient", +] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_configuration.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_configuration.py new file mode 100644 index 00000000000..574b48abb0b --- /dev/null +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_configuration.py @@ -0,0 +1,38 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from typing import Any + +from azure.core.pipeline import policies + +from ._version import VERSION + + +class NoOperationsServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for NoOperationsServiceClient. + + Note that all parameters used to create this instance are saved as instance + attributes. + """ + + def __init__(self, **kwargs: Any) -> None: + + kwargs.setdefault("sdk_moniker", "nooperationsserviceclient/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) + self._configure(**kwargs) + + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_no_operations_service_client.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_no_operations_service_client.py new file mode 100644 index 00000000000..2f162c51cc3 --- /dev/null +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/NoOperations/nooperations/_no_operations_service_client.py @@ -0,0 +1,82 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from copy import deepcopy +from typing import Any +from typing_extensions import Self + +from azure.core import PipelineClient +from azure.core.pipeline import policies +from azure.core.rest import HttpRequest, HttpResponse + +from . import models as _models +from ._configuration import NoOperationsServiceClientConfiguration +from ._serialization import Deserializer, Serializer + + +class NoOperationsServiceClient: # pylint: disable=client-accepts-api-version-keyword + """Service client with no operations.""" + + def __init__(self, **kwargs: Any) -> None: # pylint: disable=missing-client-constructor-parameter-credential + _endpoint = "" + self._config = NoOperationsServiceClientConfiguration(**kwargs) + _policies = kwargs.pop("policies", None) + if _policies is None: + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] + self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs) + + client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} + self._serialize = Serializer(client_models) + self._deserialize = Deserializer(client_models) + self._serialize.client_side_validation = False + + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + """Runs the network request through the client's chained policies. + + >>> from azure.core.rest import HttpRequest + >>> request = HttpRequest("GET", "https://www.example.org/") + + >>> response = client._send_request(request) + + + For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request + + :param request: The network request you want to make. Required. + :type request: ~azure.core.rest.HttpRequest + :keyword bool stream: Whether the response payload will be streamed. Defaults to False. + :return: The response of your network call. Does not do error handling on your response. + :rtype: ~azure.core.rest.HttpResponse + """ + + request_copy = deepcopy(request) + request_copy.url = self._client.format_url(request_copy.url) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore + + def close(self) -> None: + self._client.close() + + def __enter__(self) -> Self: + self._client.__enter__() + return self + + def __exit__(self, *exc_details: Any) -> None: + self._client.__exit__(*exc_details) diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py index b5ef0ed3e28..aa5163f7604 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, TYPE_CHECKING +from typing import Any, List, Optional import msrest.serialization -if TYPE_CHECKING: - from .. import models as _models - class ArrayOptionalWrapper(msrest.serialization.Model): """ArrayOptionalWrapper. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py index 9722f402934..ffbadc428e4 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py @@ -6,13 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Literal, Optional, TYPE_CHECKING +from typing import Any, List, Literal, Optional import msrest.serialization -if TYPE_CHECKING: - from .. import models as _models - class ChildProduct(msrest.serialization.Model): """The product documentation. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py index 6b8562c0332..2e4a7cda59f 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py @@ -8,13 +8,10 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class AccessPolicy(_serialization.Model): """An Access policy. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py index 6398e7624ff..40f3f1cb254 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py @@ -72,7 +72,7 @@ async def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[_models.Pe 404: cast( Type[HttpResponseError], lambda response: ResourceNotFoundError( - response=response, model=self._deserialize(_models.NotFoundErrorBase, response) + response=response, model=self._deserialize(NotFoundErrorBase, response) ), ), 501: HttpResponseError, @@ -129,7 +129,7 @@ async def do_something(self, what_action: str, **kwargs: Any) -> _models.PetActi 500: cast( Type[HttpResponseError], lambda response: HttpResponseError( - response=response, model=self._deserialize(_models.PetActionError, response) + response=response, model=self._deserialize(PetActionError, response) ), ), } @@ -186,7 +186,7 @@ async def has_models_param(self, models: str = "value1", **kwargs: Any) -> None: 500: cast( Type[HttpResponseError], lambda response: HttpResponseError( - response=response, model=self._deserialize(_models.PetActionError, response) + response=response, model=self._deserialize(PetActionError, response) ), ), } diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py index 1b50a5a433b..cbe689df1b7 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py @@ -129,7 +129,7 @@ def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[_models.Pet]: 404: cast( Type[HttpResponseError], lambda response: ResourceNotFoundError( - response=response, model=self._deserialize(_models.NotFoundErrorBase, response) + response=response, model=self._deserialize(NotFoundErrorBase, response) ), ), 501: HttpResponseError, @@ -186,7 +186,7 @@ def do_something(self, what_action: str, **kwargs: Any) -> _models.PetAction: 500: cast( Type[HttpResponseError], lambda response: HttpResponseError( - response=response, model=self._deserialize(_models.PetActionError, response) + response=response, model=self._deserialize(PetActionError, response) ), ), } @@ -245,7 +245,7 @@ def has_models_param( # pylint: disable=inconsistent-return-statements 500: cast( Type[HttpResponseError], lambda response: HttpResponseError( - response=response, model=self._deserialize(_models.PetActionError, response) + response=response, model=self._deserialize(PetActionError, response) ), ), } diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormDataVersionTolerant/bodyformdataversiontolerant/aio/operations/_operations.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormDataVersionTolerant/bodyformdataversiontolerant/aio/operations/_operations.py index 7fbf76c9ec5..adb85fc687a 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormDataVersionTolerant/bodyformdataversiontolerant/aio/operations/_operations.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormDataVersionTolerant/bodyformdataversiontolerant/aio/operations/_operations.py @@ -23,8 +23,8 @@ from azure.core.tracing.decorator_async import distributed_trace_async from azure.core.utils import case_insensitive_dict +from ..._vendor import raise_if_not_implemented from ...operations._operations import build_formdata_upload_file_via_body_request -from .._vendor import raise_if_not_implemented if sys.version_info >= (3, 9): from collections.abc import MutableMapping diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormUrlEncodedDataVersionTolerant/bodyformurlencodeddataversiontolerant/aio/operations/_operations.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormUrlEncodedDataVersionTolerant/bodyformurlencodeddataversiontolerant/aio/operations/_operations.py index ebdb6efae6f..c9e6adda8e9 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormUrlEncodedDataVersionTolerant/bodyformurlencodeddataversiontolerant/aio/operations/_operations.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/BodyFormUrlEncodedDataVersionTolerant/bodyformurlencodeddataversiontolerant/aio/operations/_operations.py @@ -5,7 +5,7 @@ # Code generated by Microsoft (R) AutoRest Code Generator. # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from .._vendor import raise_if_not_implemented +from ..._vendor import raise_if_not_implemented class FormdataurlencodedOperations: # pylint: disable=abstract-class-instantiated diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/MediaTypesVersionTolerant/mediatypesversiontolerant/aio/_operations/_operations.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/MediaTypesVersionTolerant/mediatypesversiontolerant/aio/_operations/_operations.py index 899603572b2..6212956c805 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/MediaTypesVersionTolerant/mediatypesversiontolerant/aio/_operations/_operations.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/MediaTypesVersionTolerant/mediatypesversiontolerant/aio/_operations/_operations.py @@ -30,7 +30,8 @@ build_media_types_content_type_with_encoding_request, build_media_types_put_text_and_json_body_request, ) -from .._vendor import MediaTypesClientMixinABC, raise_if_not_implemented +from ..._vendor import raise_if_not_implemented +from .._vendor import MediaTypesClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/__init__.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/__init__.py index a0e664d7104..697675d2f6f 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/__init__.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/__init__.py @@ -12,6 +12,7 @@ if TYPE_CHECKING: from ._patch import * # pylint: disable=unused-wildcard-import +from ._client import NoOperationsServiceClient # type: ignore from ._version import VERSION __version__ = VERSION @@ -23,7 +24,9 @@ _patch_all = [] from ._patch import patch_sdk as _patch_sdk -__all__ = [] +__all__ = [ + "NoOperationsServiceClient", +] __all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_client.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_client.py new file mode 100644 index 00000000000..b983492743e --- /dev/null +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_client.py @@ -0,0 +1,80 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from copy import deepcopy +from typing import Any +from typing_extensions import Self + +from azure.core import PipelineClient +from azure.core.pipeline import policies +from azure.core.rest import HttpRequest, HttpResponse + +from ._configuration import NoOperationsServiceClientConfiguration +from ._serialization import Deserializer, Serializer + + +class NoOperationsServiceClient: # pylint: disable=client-accepts-api-version-keyword + """Service client with no operations.""" + + def __init__(self, **kwargs: Any) -> None: # pylint: disable=missing-client-constructor-parameter-credential + _endpoint = "" + self._config = NoOperationsServiceClientConfiguration(**kwargs) + _policies = kwargs.pop("policies", None) + if _policies is None: + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] + self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs) + + self._serialize = Serializer() + self._deserialize = Deserializer() + self._serialize.client_side_validation = False + + def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: + """Runs the network request through the client's chained policies. + + >>> from azure.core.rest import HttpRequest + >>> request = HttpRequest("GET", "https://www.example.org/") + + >>> response = client.send_request(request) + + + For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request + + :param request: The network request you want to make. Required. + :type request: ~azure.core.rest.HttpRequest + :keyword bool stream: Whether the response payload will be streamed. Defaults to False. + :return: The response of your network call. Does not do error handling on your response. + :rtype: ~azure.core.rest.HttpResponse + """ + + request_copy = deepcopy(request) + request_copy.url = self._client.format_url(request_copy.url) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore + + def close(self) -> None: + self._client.close() + + def __enter__(self) -> Self: + self._client.__enter__() + return self + + def __exit__(self, *exc_details: Any) -> None: + self._client.__exit__(*exc_details) diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_configuration.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_configuration.py new file mode 100644 index 00000000000..574b48abb0b --- /dev/null +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/NoOperationsVersionTolerant/nooperationsversiontolerant/_configuration.py @@ -0,0 +1,38 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from typing import Any + +from azure.core.pipeline import policies + +from ._version import VERSION + + +class NoOperationsServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + """Configuration for NoOperationsServiceClient. + + Note that all parameters used to create this instance are saved as instance + attributes. + """ + + def __init__(self, **kwargs: Any) -> None: + + kwargs.setdefault("sdk_moniker", "nooperationsserviceclient/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) + self._configure(**kwargs) + + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/ReservedWordsVersionTolerant/reservedwordsversiontolerant/aio/operations/_operations.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/ReservedWordsVersionTolerant/reservedwordsversiontolerant/aio/operations/_operations.py index dedde817193..10056e05947 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/ReservedWordsVersionTolerant/reservedwordsversiontolerant/aio/operations/_operations.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/ReservedWordsVersionTolerant/reservedwordsversiontolerant/aio/operations/_operations.py @@ -21,6 +21,7 @@ from azure.core.tracing.decorator_async import distributed_trace_async from azure.core.utils import case_insensitive_dict +from ..._vendor import raise_if_not_implemented from ...operations._operations import ( build_import_operations_operation_one_request, build_reserved_words_operation_with_content_param_request, @@ -28,7 +29,7 @@ build_reserved_words_operation_with_url_request, build_reserved_words_reserved_enum_request, ) -from .._vendor import ReservedWordsClientMixinABC, raise_if_not_implemented +from .._vendor import ReservedWordsClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py index 5ddfca1b641..b4fee3cdc46 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py @@ -8,13 +8,10 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Optional, Union from .. import _serialization -if TYPE_CHECKING: - from .. import models as _models - class AccessPolicy(_serialization.Model): """An Access policy. From d02c496fb2aed8304dc9e06e86a8afedf5a34359 Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 11:30:29 +0000 Subject: [PATCH 4/8] Update dependencies (2024-12-06 11:30:29) --- packages/autorest.python/package.json | 2 +- packages/typespec-python/package.json | 2 +- pnpm-lock.yaml | 14 +++++++------- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/packages/autorest.python/package.json b/packages/autorest.python/package.json index ef030eddca6..bf84e9fbe71 100644 --- a/packages/autorest.python/package.json +++ b/packages/autorest.python/package.json @@ -29,7 +29,7 @@ }, "homepage": "https://github.com/Azure/autorest.python/blob/main/README.md", "dependencies": { - "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", + "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", "@autorest/system-requirements": "~1.0.2", "fs-extra": "~11.2.0", "tsx": "~4.19.1" diff --git a/packages/typespec-python/package.json b/packages/typespec-python/package.json index 61a1e0d39f1..e43b2678aa9 100644 --- a/packages/typespec-python/package.json +++ b/packages/typespec-python/package.json @@ -60,7 +60,7 @@ "js-yaml": "~4.1.0", "semver": "~7.6.2", "tsx": "~4.19.1", - "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", + "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", "fs-extra": "~11.2.0" }, "devDependencies": { diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml index d5e70613a06..cf4709e454b 100644 --- a/pnpm-lock.yaml +++ b/pnpm-lock.yaml @@ -54,8 +54,8 @@ importers: specifier: ~1.0.2 version: 1.0.2 '@typespec/http-client-python': - specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz - version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) + specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz + version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) fs-extra: specifier: ~11.2.0 version: 11.2.0 @@ -76,8 +76,8 @@ importers: packages/typespec-python: dependencies: '@typespec/http-client-python': - specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz - version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) + specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz + version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) fs-extra: specifier: ~11.2.0 version: 11.2.0 @@ -1233,8 +1233,8 @@ packages: engines: {node: '>=18.0.0'} hasBin: true - '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz': - resolution: {tarball: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz} + '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz': + resolution: {tarball: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz} version: 0.3.12 engines: {node: '>=18.0.0'} peerDependencies: @@ -4919,7 +4919,7 @@ snapshots: yaml: 2.5.1 yargs: 17.7.2 - '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MDc1Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi)': + '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi)': dependencies: '@azure-tools/typespec-autorest': 0.48.0(2qbrnlizjwf7gxxublducwg43e) '@azure-tools/typespec-azure-core': 0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))) From d48e4cf7999b883f07c6be37e9c0c81435c6fb20 Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 11:34:58 +0000 Subject: [PATCH 5/8] Regenerate for typespec-python (2024-12-06 11:34:58) --- .../apikey => }/apiview_mapping_python.json | 0 .../authentication/apikey/_model_base.py | 16 + .../apikey/_operations/_operations.py | 15 +- .../apikey/aio/_operations/_operations.py | 15 +- .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_api_key.py | 31 ++ .../generated_tests/test_api_key_async.py | 32 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../custom => }/apiview_mapping_python.json | 0 .../authentication/http/custom/_model_base.py | 16 + .../http/custom/_operations/_operations.py | 15 +- .../custom/aio/_operations/_operations.py | 15 +- .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_custom.py | 31 ++ .../generated_tests/test_custom_async.py | 32 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../oauth2 => }/apiview_mapping_python.json | 0 .../authentication/oauth2/_model_base.py | 16 + .../oauth2/_operations/_operations.py | 15 +- .../oauth2/aio/_operations/_operations.py | 15 +- .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_oauth2.py | 31 ++ .../generated_tests/test_oauth2_async.py | 32 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../union => }/apiview_mapping_python.json | 0 .../authentication/union/_model_base.py | 16 + .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_union.py | 31 ++ .../generated_tests/test_union_async.py | 32 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../access => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ ...test_access_public_operation_operations.py | 35 ++ ...ccess_public_operation_operations_async.py | 36 ++ ...ss_shared_model_in_operation_operations.py | 24 ++ ...red_model_in_operation_operations_async.py | 25 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../core/access/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 ++ .../generated_tests/test_flatten_property.py | 35 ++ .../test_flatten_property_async.py | 36 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../core/flattenproperty/_model_base.py | 16 + .../usage => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ ...est_usage_model_in_operation_operations.py | 44 ++ ...age_model_in_operation_operations_async.py | 45 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../clientgenerator/core/usage/_model_base.py | 16 + .../basic => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_basic.py | 91 ++++ .../generated_tests/test_basic_async.py | 92 +++++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../specs/azure/core/basic/_model_base.py | 16 + .../standard => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_standard.py | 48 +++ .../generated_tests/test_standard_async.py | 55 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../azure/core/lro/standard/_model_base.py | 16 + .../model => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ ..._azure_core_embedding_vector_operations.py | 44 ++ ..._core_embedding_vector_operations_async.py | 45 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../specs/azure/core/model/_model_base.py | 16 + .../page => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_page.py | 42 ++ .../generated_tests/test_page_async.py | 43 ++ ...page_two_models_as_page_item_operations.py | 31 ++ ...wo_models_as_page_item_operations_async.py | 32 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../specs/azure/core/page/_model_base.py | 16 + .../scalar => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ ...scalar_azure_location_scalar_operations.py | 67 +++ ..._azure_location_scalar_operations_async.py | 68 +++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../specs/azure/core/scalar/_model_base.py | 16 + .../traits => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_traits.py | 37 ++ .../generated_tests/test_traits_async.py | 38 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../specs/azure/core/traits/_model_base.py | 16 + .../basic => }/apiview_mapping_python.json | 0 .../generated_samples/basic.py | 37 ++ .../generated_tests/conftest.py | 37 ++ .../generated_tests/test_azure_example.py | 31 ++ .../test_azure_example_async.py | 32 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../specs/azure/example/basic/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../commonproperties/_model_base.py | 16 + .../aio/operations/_operations.py | 8 +- .../operations/_operations.py | 8 +- .../generated_tests/conftest.py | 35 ++ ..._properties_managed_identity_operations.py | 96 +++++ ...rties_managed_identity_operations_async.py | 97 +++++ .../apiview_mapping_python.json | 0 .../resourcemanager/resources/_model_base.py | 16 + .../resources/aio/operations/_operations.py | 34 +- .../resources/operations/_operations.py | 34 +- .../generated_tests/conftest.py | 35 ++ .../test_resources_nested_operations.py | 106 +++++ .../test_resources_nested_operations_async.py | 113 +++++ .../test_resources_singleton_operations.py | 91 ++++ ...st_resources_singleton_operations_async.py | 94 +++++ .../test_resources_top_level_operations.py | 125 ++++++ ...st_resources_top_level_operations_async.py | 132 ++++++ .../apiview_mapping_python.json | 0 .../xmsclientrequestid/_model_base.py | 16 + .../generated_tests/conftest.py | 37 ++ .../generated_tests/test_xms_request_id.py | 22 + .../test_xms_request_id_async.py | 23 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../lro/rpc => }/apiview_mapping_python.json | 0 .../azurecore/lro/rpc/_model_base.py | 16 + .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_rpc.py | 24 ++ .../generated_tests/test_rpc_async.py | 27 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../naming => }/apiview_mapping_python.json | 0 .../client/naming/_model_base.py | 16 + .../client-naming/generated_tests/conftest.py | 35 ++ .../generated_tests/test_naming.py | 86 ++++ .../generated_tests/test_naming_async.py | 87 ++++ .../test_naming_client_model_operations.py | 35 ++ ...st_naming_client_model_operations_async.py | 36 ++ .../test_naming_union_enum_operations.py | 37 ++ ...test_naming_union_enum_operations_async.py | 38 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../service => }/apiview_mapping_python.json | 0 .../client/structure/service/_model_base.py | 16 + .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_service.py | 31 ++ .../generated_tests/test_service_async.py | 32 ++ .../test_service_bar_operations.py | 31 ++ .../test_service_bar_operations_async.py | 32 ++ .../test_service_baz_operations.py | 22 + .../test_service_baz_operations_async.py | 23 ++ .../test_service_foo_operations.py | 31 ++ .../test_service_foo_operations_async.py | 32 ++ .../test_service_qux_operations.py | 31 ++ .../test_service_qux_operations_async.py | 32 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../apiview_mapping_python.json | 0 .../structure/multiclient/_model_base.py | 16 + .../generated_tests/conftest.py | 44 ++ .../generated_tests/test_a.py | 40 ++ .../generated_tests/test_aasync.py | 41 ++ .../generated_tests/test_b.py | 40 ++ .../generated_tests/test_basync.py | 41 ++ .../generated_tests/testpreparer.py | 38 ++ .../generated_tests/testpreparer_async.py | 31 ++ .../apiview_mapping_python.json | 0 .../structure/renamedoperation/_model_base.py | 16 + .../generated_tests/conftest.py | 39 ++ .../generated_tests/test_renamed_operation.py | 40 ++ .../test_renamed_operation_async.py | 41 ++ ...test_renamed_operation_group_operations.py | 40 ++ ...enamed_operation_group_operations_async.py | 41 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../apiview_mapping_python.json | 0 .../twooperationgroup/_model_base.py | 16 + .../generated_tests/conftest.py | 39 ++ ...t_two_operation_group_group1_operations.py | 40 ++ ...operation_group_group1_operations_async.py | 41 ++ ...t_two_operation_group_group2_operations.py | 40 ++ ...operation_group_group2_operations_async.py | 41 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../bytes => }/apiview_mapping_python.json | 0 .../encode-bytes/encode/bytes/_model_base.py | 16 + .../encode-bytes/generated_tests/conftest.py | 35 ++ .../test_bytes_header_operations.py | 57 +++ .../test_bytes_header_operations_async.py | 58 +++ .../test_bytes_property_operations.py | 57 +++ .../test_bytes_property_operations_async.py | 58 +++ .../test_bytes_query_operations.py | 57 +++ .../test_bytes_query_operations_async.py | 58 +++ .../test_bytes_request_body_operations.py | 73 ++++ ...est_bytes_request_body_operations_async.py | 74 ++++ .../test_bytes_response_body_operations.py | 58 +++ ...st_bytes_response_body_operations_async.py | 59 +++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../datetime => }/apiview_mapping_python.json | 0 .../encode/datetime/_model_base.py | 16 + .../generated_tests/conftest.py | 35 ++ .../test_datetime_header_operations.py | 68 +++ .../test_datetime_header_operations_async.py | 69 ++++ .../test_datetime_property_operations.py | 68 +++ ...test_datetime_property_operations_async.py | 69 ++++ .../test_datetime_query_operations.py | 68 +++ .../test_datetime_query_operations_async.py | 69 ++++ ...est_datetime_response_header_operations.py | 49 +++ ...tetime_response_header_operations_async.py | 50 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../duration => }/apiview_mapping_python.json | 0 .../encode/duration/_model_base.py | 16 + .../generated_tests/conftest.py | 35 ++ .../test_duration_header_operations.py | 79 ++++ .../test_duration_header_operations_async.py | 80 ++++ .../test_duration_property_operations.py | 79 ++++ ...test_duration_property_operations_async.py | 80 ++++ .../test_duration_query_operations.py | 79 ++++ .../test_duration_query_operations_async.py | 80 ++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../numeric => }/apiview_mapping_python.json | 0 .../encode/numeric/_model_base.py | 16 + .../generated_tests/conftest.py | 35 ++ .../test_numeric_property_operations.py | 46 +++ .../test_numeric_property_operations_async.py | 47 +++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_visibility.py | 90 ++++ .../generated_tests/test_visibility_async.py | 91 ++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../headasbooleanfalse/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_visibility.py | 90 ++++ .../generated_tests/test_visibility_async.py | 91 ++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../headasbooleantrue/_model_base.py | 16 + .../basic => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_basic_explicit_body_operations.py | 24 ++ ...st_basic_explicit_body_operations_async.py | 25 ++ .../test_basic_implicit_body_operations.py | 25 ++ ...st_basic_implicit_body_operations_async.py | 26 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../parameters/basic/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 ++ .../generated_tests/test_body_optionality.py | 36 ++ .../test_body_optionality_async.py | 37 ++ ...ptionality_optional_explicit_operations.py | 31 ++ ...lity_optional_explicit_operations_async.py | 32 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../parameters/bodyoptionality/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 ++ ...est_collection_format_header_operations.py | 24 ++ ...llection_format_header_operations_async.py | 25 ++ ...test_collection_format_query_operations.py | 68 +++ ...ollection_format_query_operations_async.py | 69 ++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../collectionformat/_model_base.py | 16 + .../spread => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_spread_alias_operations.py | 83 ++++ .../test_spread_alias_operations_async.py | 84 ++++ .../test_spread_model_operations.py | 75 ++++ .../test_spread_model_operations_async.py | 76 ++++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../parameters/spread/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 ++ ...t_negotiation_different_body_operations.py | 35 ++ ...tiation_different_body_operations_async.py | 36 ++ ...ontent_negotiation_same_body_operations.py | 35 ++ ..._negotiation_same_body_operations_async.py | 36 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../payload/contentnegotiation/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 ++ .../generated_tests/test_json_merge_patch.py | 61 +++ .../test_json_merge_patch_async.py | 62 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../payload/jsonmergepatch/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_media_type_string_body_operations.py | 55 +++ ...media_type_string_body_operations_async.py | 56 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../payload/mediatype/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_multi_part_form_data_operations.py | 152 +++++++ ...t_multi_part_form_data_operations_async.py | 153 +++++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../payload/multipart/_model_base.py | 16 + .../pageable => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_pageable.py | 22 + .../generated_tests/test_pageable_async.py | 23 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../payload/pageable/_model_base.py | 16 + .../xml => }/apiview_mapping_python.json | 0 .../payload-xml/generated_tests/conftest.py | 35 ++ ...el_with_array_of_model_value_operations.py | 34 ++ ...h_array_of_model_value_operations_async.py | 35 ++ ..._model_with_attributes_value_operations.py | 34 ++ ..._with_attributes_value_operations_async.py | 35 ++ ..._model_with_dictionary_value_operations.py | 34 ++ ..._with_dictionary_value_operations_async.py | 35 ++ ...model_with_empty_array_value_operations.py | 34 ++ ...with_empty_array_value_operations_async.py | 35 ++ ...del_with_encoded_names_value_operations.py | 34 ++ ...th_encoded_names_value_operations_async.py | 35 ++ ...el_with_optional_field_value_operations.py | 34 ++ ...h_optional_field_value_operations_async.py | 35 ++ ...el_with_renamed_arrays_value_operations.py | 34 ++ ...h_renamed_arrays_value_operations_async.py | 35 ++ ...el_with_renamed_fields_value_operations.py | 34 ++ ...h_renamed_fields_value_operations_async.py | 35 ++ ...del_with_simple_arrays_value_operations.py | 34 ++ ...th_simple_arrays_value_operations_async.py | 35 ++ ...st_xml_model_with_text_value_operations.py | 34 ++ ..._model_with_text_value_operations_async.py | 35 ++ ...l_with_unwrapped_array_value_operations.py | 34 ++ ..._unwrapped_array_value_operations_async.py | 35 ++ .../test_xml_simple_model_value_operations.py | 34 ++ ...xml_simple_model_value_operations_async.py | 35 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../payload-xml/payload/xml/_model_base.py | 16 + .../driven1 => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 47 +++ .../test_resiliency_service_driven.py | 42 ++ .../test_resiliency_service_driven_async.py | 43 ++ .../generated_tests/testpreparer.py | 28 ++ .../generated_tests/testpreparer_async.py | 20 + .../resiliency/srv/driven1/_model_base.py | 16 + .../driven2 => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 47 +++ .../test_resiliency_service_driven.py | 51 +++ .../test_resiliency_service_driven_async.py | 52 +++ .../generated_tests/testpreparer.py | 28 ++ .../generated_tests/testpreparer_async.py | 20 + .../resiliency/srv/driven2/_model_base.py | 16 + .../{routes => }/apiview_mapping_python.json | 0 .../routes/generated_tests/conftest.py | 35 ++ .../routes/generated_tests/test_routes.py | 22 + .../generated_tests/test_routes_async.py | 23 ++ .../test_routes_in_interface_operations.py | 22 + ...st_routes_in_interface_operations_async.py | 23 ++ .../test_routes_path_parameters_operations.py | 332 +++++++++++++++ ...routes_path_parameters_operations_async.py | 333 +++++++++++++++ ...test_routes_query_parameters_operations.py | 178 ++++++++ ...outes_query_parameters_operations_async.py | 179 ++++++++ .../routes/generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../generated/routes/routes/_model_base.py | 16 + .../json => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_json_property_operations.py | 33 ++ .../test_json_property_operations_async.py | 34 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../encodedname/json/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_not_defined.py | 22 + .../generated_tests/test_not_defined_async.py | 23 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../server/endpoint/notdefined/_model_base.py | 16 + .../multiple => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_multiple.py | 33 ++ .../generated_tests/test_multiple_async.py | 34 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../server/path/multiple/_model_base.py | 16 + .../single => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_single.py | 22 + .../generated_tests/test_single_async.py | 23 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../server/path/single/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 37 ++ .../generated_tests/test_not_versioned.py | 44 ++ .../test_not_versioned_async.py | 45 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../versions/notversioned/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_versioned.py | 49 +++ .../generated_tests/test_versioned_async.py | 50 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../server/versions/versioned/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 ++ .../test_conditional_request.py | 49 +++ .../test_conditional_request_async.py | 50 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../conditionalrequest/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 37 ++ .../generated_tests/test_repeatability.py | 22 + .../test_repeatability_async.py | 23 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../repeatability/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../special-words/generated_tests/conftest.py | 37 ++ ...ecial_words_model_properties_operations.py | 24 ++ ...words_model_properties_operations_async.py | 25 ++ .../test_special_words_models_operations.py | 376 +++++++++++++++++ ...t_special_words_models_operations_async.py | 377 +++++++++++++++++ .../test_special_words_operations.py | 310 ++++++++++++++ .../test_special_words_operations_async.py | 311 ++++++++++++++ ...est_special_words_parameters_operations.py | 387 +++++++++++++++++ ...ecial_words_parameters_operations_async.py | 388 ++++++++++++++++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../special-words/specialwords/_model_base.py | 16 + .../array => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_array_boolean_value_operations.py | 33 ++ ...st_array_boolean_value_operations_async.py | 34 ++ .../test_array_datetime_value_operations.py | 33 ++ ...t_array_datetime_value_operations_async.py | 34 ++ .../test_array_duration_value_operations.py | 33 ++ ...t_array_duration_value_operations_async.py | 34 ++ .../test_array_float32_value_operations.py | 33 ++ ...st_array_float32_value_operations_async.py | 34 ++ .../test_array_int32_value_operations.py | 33 ++ ...test_array_int32_value_operations_async.py | 34 ++ .../test_array_int64_value_operations.py | 33 ++ ...test_array_int64_value_operations_async.py | 34 ++ .../test_array_model_value_operations.py | 33 ++ ...test_array_model_value_operations_async.py | 34 ++ ...array_nullable_boolean_value_operations.py | 33 ++ ...nullable_boolean_value_operations_async.py | 34 ++ ...t_array_nullable_float_value_operations.py | 33 ++ ...y_nullable_float_value_operations_async.py | 34 ++ ...t_array_nullable_int32_value_operations.py | 33 ++ ...y_nullable_int32_value_operations_async.py | 34 ++ ...t_array_nullable_model_value_operations.py | 33 ++ ...y_nullable_model_value_operations_async.py | 34 ++ ..._array_nullable_string_value_operations.py | 33 ++ ..._nullable_string_value_operations_async.py | 34 ++ .../test_array_string_value_operations.py | 33 ++ ...est_array_string_value_operations_async.py | 34 ++ .../test_array_unknown_value_operations.py | 33 ++ ...st_array_unknown_value_operations_async.py | 34 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/array/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ ...est_dictionary_boolean_value_operations.py | 33 ++ ...ctionary_boolean_value_operations_async.py | 34 ++ ...st_dictionary_datetime_value_operations.py | 33 ++ ...tionary_datetime_value_operations_async.py | 34 ++ ...st_dictionary_duration_value_operations.py | 33 ++ ...tionary_duration_value_operations_async.py | 34 ++ ...est_dictionary_float32_value_operations.py | 33 ++ ...ctionary_float32_value_operations_async.py | 34 ++ .../test_dictionary_int32_value_operations.py | 33 ++ ...dictionary_int32_value_operations_async.py | 34 ++ .../test_dictionary_int64_value_operations.py | 33 ++ ...dictionary_int64_value_operations_async.py | 34 ++ .../test_dictionary_model_value_operations.py | 33 ++ ...dictionary_model_value_operations_async.py | 34 ++ ...tionary_nullable_float_value_operations.py | 33 ++ ...y_nullable_float_value_operations_async.py | 34 ++ ...ionary_recursive_model_value_operations.py | 33 ++ ..._recursive_model_value_operations_async.py | 34 ++ ...test_dictionary_string_value_operations.py | 33 ++ ...ictionary_string_value_operations_async.py | 34 ++ ...est_dictionary_unknown_value_operations.py | 33 ++ ...ctionary_unknown_value_operations_async.py | 34 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/dictionary/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_extensible_string_operations.py | 55 +++ ...test_extensible_string_operations_async.py | 56 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/enum/extensible/_model_base.py | 16 + .../fixed => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_fixed_string_operations.py | 46 +++ .../test_fixed_string_operations_async.py | 47 +++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/enum/fixed/_model_base.py | 16 + .../empty => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_empty.py | 44 ++ .../generated_tests/test_empty_async.py | 45 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/model/empty/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 ++ .../test_enum_discriminator.py | 89 ++++ .../test_enum_discriminator_async.py | 90 ++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../model/enumdiscriminator/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 ++ .../test_nested_discriminator.py | 71 ++++ .../test_nested_discriminator_async.py | 72 ++++ .../generated_tests/testpreparer.py | 28 ++ .../generated_tests/testpreparer_async.py | 20 + .../model/nesteddiscriminator/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 ++ .../generated_tests/test_not_discriminated.py | 44 ++ .../test_not_discriminated_async.py | 45 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../model/notdiscriminated/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_recursive.py | 33 ++ .../generated_tests/test_recursive_async.py | 34 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/model/recursive/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 43 ++ .../test_single_discriminator.py | 80 ++++ .../test_single_discriminator_async.py | 81 ++++ .../generated_tests/testpreparer.py | 28 ++ .../generated_tests/testpreparer_async.py | 20 + .../model/singlediscriminator/_model_base.py | 16 + .../usage => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_usage.py | 44 ++ .../generated_tests/test_usage_async.py | 45 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/model/usage/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_visibility.py | 90 ++++ .../generated_tests/test_visibility_async.py | 91 ++++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/model/visibility/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 45 ++ ...tends_different_spread_float_operations.py | 33 ++ ...different_spread_float_operations_async.py | 34 ++ ...different_spread_model_array_operations.py | 33 ++ ...ent_spread_model_array_operations_async.py | 34 ++ ...tends_different_spread_model_operations.py | 33 ++ ...different_spread_model_operations_async.py | 34 ++ ...ends_different_spread_string_operations.py | 33 ++ ...ifferent_spread_string_operations_async.py | 34 ++ ...nal_properties_extends_float_operations.py | 33 ++ ...operties_extends_float_operations_async.py | 34 ++ ...operties_extends_model_array_operations.py | 33 ++ ...es_extends_model_array_operations_async.py | 34 ++ ...nal_properties_extends_model_operations.py | 33 ++ ...operties_extends_model_operations_async.py | 34 ++ ...al_properties_extends_string_operations.py | 33 ++ ...perties_extends_string_operations_async.py | 34 ++ ...ties_extends_unknown_derived_operations.py | 33 ++ ...xtends_unknown_derived_operations_async.py | 34 ++ ...xtends_unknown_discriminated_operations.py | 33 ++ ..._unknown_discriminated_operations_async.py | 34 ++ ...l_properties_extends_unknown_operations.py | 33 ++ ...erties_extends_unknown_operations_async.py | 34 ++ ...ditional_properties_is_float_operations.py | 33 ++ ...al_properties_is_float_operations_async.py | 34 ++ ...al_properties_is_model_array_operations.py | 33 ++ ...perties_is_model_array_operations_async.py | 34 ++ ...ditional_properties_is_model_operations.py | 33 ++ ...al_properties_is_model_operations_async.py | 34 ++ ...itional_properties_is_string_operations.py | 33 ++ ...l_properties_is_string_operations_async.py | 34 ++ ...roperties_is_unknown_derived_operations.py | 33 ++ ...ies_is_unknown_derived_operations_async.py | 34 ++ ...ies_is_unknown_discriminated_operations.py | 33 ++ ..._unknown_discriminated_operations_async.py | 34 ++ ...tional_properties_is_unknown_operations.py | 33 ++ ..._properties_is_unknown_operations_async.py | 34 ++ ...l_properties_multiple_spread_operations.py | 33 ++ ...erties_multiple_spread_operations_async.py | 34 ++ ...rties_spread_different_float_operations.py | 33 ++ ...spread_different_float_operations_async.py | 34 ++ ...spread_different_model_array_operations.py | 33 ++ ..._different_model_array_operations_async.py | 34 ++ ...rties_spread_different_model_operations.py | 33 ++ ...spread_different_model_operations_async.py | 34 ++ ...ties_spread_different_string_operations.py | 33 ++ ...pread_different_string_operations_async.py | 34 ++ ...onal_properties_spread_float_operations.py | 33 ++ ...roperties_spread_float_operations_async.py | 34 ++ ...roperties_spread_model_array_operations.py | 33 ++ ...ies_spread_model_array_operations_async.py | 34 ++ ...onal_properties_spread_model_operations.py | 33 ++ ...roperties_spread_model_operations_async.py | 34 ++ ...d_record_discriminated_union_operations.py | 33 ++ ...rd_discriminated_union_operations_async.py | 34 ++ ...ord_non_discriminated_union2_operations.py | 33 ++ ...n_discriminated_union2_operations_async.py | 36 ++ ...ord_non_discriminated_union3_operations.py | 33 ++ ...n_discriminated_union3_operations_async.py | 36 ++ ...cord_non_discriminated_union_operations.py | 33 ++ ...on_discriminated_union_operations_async.py | 34 ++ ...operties_spread_record_union_operations.py | 33 ++ ...es_spread_record_union_operations_async.py | 34 ++ ...nal_properties_spread_string_operations.py | 33 ++ ...operties_spread_string_operations_async.py | 34 ++ .../generated_tests/testpreparer.py | 28 ++ .../generated_tests/testpreparer_async.py | 20 + .../additionalproperties/_model_base.py | 16 + .../nullable => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_nullable_bytes_operations.py | 53 +++ .../test_nullable_bytes_operations_async.py | 54 +++ ...st_nullable_collections_byte_operations.py | 53 +++ ...lable_collections_byte_operations_async.py | 54 +++ ...t_nullable_collections_model_operations.py | 53 +++ ...able_collections_model_operations_async.py | 54 +++ ..._nullable_collections_string_operations.py | 53 +++ ...ble_collections_string_operations_async.py | 54 +++ .../test_nullable_datetime_operations.py | 53 +++ ...test_nullable_datetime_operations_async.py | 54 +++ .../test_nullable_duration_operations.py | 53 +++ ...test_nullable_duration_operations_async.py | 54 +++ .../test_nullable_string_operations.py | 53 +++ .../test_nullable_string_operations_async.py | 54 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/property/nullable/_model_base.py | 16 + .../optional => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ ...est_optional_boolean_literal_operations.py | 53 +++ ...tional_boolean_literal_operations_async.py | 54 +++ .../test_optional_bytes_operations.py | 53 +++ .../test_optional_bytes_operations_async.py | 54 +++ ...st_optional_collections_byte_operations.py | 53 +++ ...ional_collections_byte_operations_async.py | 54 +++ ...t_optional_collections_model_operations.py | 53 +++ ...onal_collections_model_operations_async.py | 54 +++ .../test_optional_datetime_operations.py | 53 +++ ...test_optional_datetime_operations_async.py | 54 +++ .../test_optional_duration_operations.py | 53 +++ ...test_optional_duration_operations_async.py | 54 +++ .../test_optional_float_literal_operations.py | 53 +++ ...optional_float_literal_operations_async.py | 54 +++ .../test_optional_int_literal_operations.py | 53 +++ ...t_optional_int_literal_operations_async.py | 54 +++ .../test_optional_plain_date_operations.py | 53 +++ ...st_optional_plain_date_operations_async.py | 54 +++ .../test_optional_plain_time_operations.py | 53 +++ ...st_optional_plain_time_operations_async.py | 54 +++ ...tional_required_and_optional_operations.py | 53 +++ ..._required_and_optional_operations_async.py | 54 +++ ...test_optional_string_literal_operations.py | 53 +++ ...ptional_string_literal_operations_async.py | 54 +++ .../test_optional_string_operations.py | 53 +++ .../test_optional_string_operations_async.py | 54 +++ ...optional_union_float_literal_operations.py | 53 +++ ...al_union_float_literal_operations_async.py | 54 +++ ...t_optional_union_int_literal_operations.py | 53 +++ ...onal_union_int_literal_operations_async.py | 54 +++ ...ptional_union_string_literal_operations.py | 53 +++ ...l_union_string_literal_operations_async.py | 54 +++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/property/optional/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ ..._value_types_boolean_literal_operations.py | 33 ++ ..._types_boolean_literal_operations_async.py | 34 ++ .../test_value_types_boolean_operations.py | 33 ++ ...st_value_types_boolean_operations_async.py | 34 ++ .../test_value_types_bytes_operations.py | 33 ++ ...test_value_types_bytes_operations_async.py | 34 ++ ..._value_types_collections_int_operations.py | 33 ++ ..._types_collections_int_operations_async.py | 34 ++ ...alue_types_collections_model_operations.py | 33 ++ ...ypes_collections_model_operations_async.py | 34 ++ ...lue_types_collections_string_operations.py | 33 ++ ...pes_collections_string_operations_async.py | 34 ++ .../test_value_types_datetime_operations.py | 33 ++ ...t_value_types_datetime_operations_async.py | 34 ++ .../test_value_types_decimal128_operations.py | 33 ++ ...value_types_decimal128_operations_async.py | 34 ++ .../test_value_types_decimal_operations.py | 33 ++ ...st_value_types_decimal_operations_async.py | 34 ++ ...alue_types_dictionary_string_operations.py | 33 ++ ...ypes_dictionary_string_operations_async.py | 34 ++ .../test_value_types_duration_operations.py | 33 ++ ...t_value_types_duration_operations_async.py | 34 ++ .../test_value_types_enum_operations.py | 33 ++ .../test_value_types_enum_operations_async.py | 34 ++ ..._value_types_extensible_enum_operations.py | 33 ++ ..._types_extensible_enum_operations_async.py | 34 ++ ...st_value_types_float_literal_operations.py | 33 ++ ...ue_types_float_literal_operations_async.py | 34 ++ .../test_value_types_float_operations.py | 33 ++ ...test_value_types_float_operations_async.py | 34 ++ ...test_value_types_int_literal_operations.py | 33 ++ ...alue_types_int_literal_operations_async.py | 34 ++ .../test_value_types_int_operations.py | 33 ++ .../test_value_types_int_operations_async.py | 34 ++ .../test_value_types_model_operations.py | 33 ++ ...test_value_types_model_operations_async.py | 34 ++ .../test_value_types_never_operations.py | 33 ++ ...test_value_types_never_operations_async.py | 34 ++ ...t_value_types_string_literal_operations.py | 33 ++ ...e_types_string_literal_operations_async.py | 34 ++ .../test_value_types_string_operations.py | 33 ++ ...est_value_types_string_operations_async.py | 34 ++ ...value_types_union_enum_value_operations.py | 33 ++ ...types_union_enum_value_operations_async.py | 34 ++ ...ue_types_union_float_literal_operations.py | 33 ++ ...es_union_float_literal_operations_async.py | 34 ++ ...alue_types_union_int_literal_operations.py | 33 ++ ...ypes_union_int_literal_operations_async.py | 34 ++ ...e_types_union_string_literal_operations.py | 33 ++ ...s_union_string_literal_operations_async.py | 34 ++ ...st_value_types_unknown_array_operations.py | 33 ++ ...ue_types_unknown_array_operations_async.py | 34 ++ ...est_value_types_unknown_dict_operations.py | 33 ++ ...lue_types_unknown_dict_operations_async.py | 34 ++ ...test_value_types_unknown_int_operations.py | 33 ++ ...alue_types_unknown_int_operations_async.py | 34 ++ ...t_value_types_unknown_string_operations.py | 33 ++ ...e_types_unknown_string_operations_async.py | 34 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../property/valuetypes/_model_base.py | 16 + .../scalar => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_scalar_boolean_operations.py | 34 ++ .../test_scalar_boolean_operations_async.py | 35 ++ .../test_scalar_decimal128_type_operations.py | 45 ++ ...scalar_decimal128_type_operations_async.py | 46 +++ ...est_scalar_decimal128_verify_operations.py | 34 ++ ...alar_decimal128_verify_operations_async.py | 35 ++ .../test_scalar_decimal_type_operations.py | 45 ++ ...st_scalar_decimal_type_operations_async.py | 46 +++ .../test_scalar_decimal_verify_operations.py | 34 ++ ..._scalar_decimal_verify_operations_async.py | 35 ++ .../test_scalar_string_operations.py | 34 ++ .../test_scalar_string_operations_async.py | 35 ++ .../test_scalar_unknown_operations.py | 34 ++ .../test_scalar_unknown_operations_async.py | 35 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/scalar/_model_base.py | 16 + .../union => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../test_union_enums_only_operations.py | 34 ++ .../test_union_enums_only_operations_async.py | 35 ++ .../test_union_floats_only_operations.py | 34 ++ ...test_union_floats_only_operations_async.py | 35 ++ .../test_union_ints_only_operations.py | 34 ++ .../test_union_ints_only_operations_async.py | 35 ++ .../test_union_mixed_literals_operations.py | 34 ++ ...t_union_mixed_literals_operations_async.py | 35 ++ .../test_union_mixed_types_operations.py | 48 +++ ...test_union_mixed_types_operations_async.py | 49 +++ .../test_union_models_only_operations.py | 34 ++ ...test_union_models_only_operations_async.py | 35 ++ .../test_union_string_and_array_operations.py | 34 ++ ...union_string_and_array_operations_async.py | 35 ++ ...nion_string_extensible_named_operations.py | 34 ++ ...tring_extensible_named_operations_async.py | 35 ++ ...test_union_string_extensible_operations.py | 34 ++ ...nion_string_extensible_operations_async.py | 35 ++ .../test_union_strings_only_operations.py | 34 ++ ...est_union_strings_only_operations_async.py | 35 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../typetest/union/_model_base.py | 16 + .../added => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_added.py | 36 ++ .../generated_tests/test_added_async.py | 37 ++ .../test_added_interface_v2_operations.py | 24 ++ ...est_added_interface_v2_operations_async.py | 25 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../versioning/added/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 37 ++ .../generated_tests/test_made_optional.py | 24 ++ .../test_made_optional_async.py | 25 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../versioning/madeoptional/_model_base.py | 16 + .../removed => }/apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_removed.py | 35 ++ .../generated_tests/test_removed_async.py | 36 ++ .../generated_tests/testpreparer.py | 24 ++ .../generated_tests/testpreparer_async.py | 20 + .../versioning/removed/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 35 ++ .../generated_tests/test_renamed_from.py | 25 ++ .../test_renamed_from_async.py | 26 ++ ...t_renamed_from_new_interface_operations.py | 24 ++ ...med_from_new_interface_operations_async.py | 25 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../versioning/renamedfrom/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 45 ++ .../test_return_type_changed_from.py | 25 ++ .../test_return_type_changed_from_async.py | 26 ++ .../generated_tests/testpreparer.py | 28 ++ .../generated_tests/testpreparer_async.py | 20 + .../returntypechangedfrom/_model_base.py | 16 + .../apiview_mapping_python.json | 0 .../generated_tests/conftest.py | 39 ++ .../generated_tests/test_type_changed_from.py | 25 ++ .../test_type_changed_from_async.py | 26 ++ .../generated_tests/testpreparer.py | 26 ++ .../generated_tests/testpreparer_async.py | 20 + .../versioning/typechangedfrom/_model_base.py | 16 + .../authentication/apikey/_model_base.py | 16 + .../apikey/_operations/_operations.py | 15 +- .../apikey/aio/_operations/_operations.py | 15 +- .../authentication/http/custom/_model_base.py | 16 + .../http/custom/_operations/_operations.py | 15 +- .../custom/aio/_operations/_operations.py | 15 +- .../authentication/oauth2/_model_base.py | 16 + .../oauth2/_operations/_operations.py | 15 +- .../oauth2/aio/_operations/_operations.py | 15 +- .../authentication/union/_model_base.py | 16 + .../client/naming/_model_base.py | 16 + .../client/structure/service/_model_base.py | 16 + .../structure/multiclient/_model_base.py | 16 + .../structure/renamedoperation/_model_base.py | 16 + .../twooperationgroup/_model_base.py | 16 + .../encode-bytes/encode/bytes/_model_base.py | 16 + .../encode/datetime/_model_base.py | 16 + .../encode/duration/_model_base.py | 16 + .../encode/numeric/_model_base.py | 16 + .../headasbooleanfalse/_model_base.py | 16 + .../headasbooleantrue/_model_base.py | 16 + .../parameters/basic/_model_base.py | 16 + .../parameters/bodyoptionality/_model_base.py | 16 + .../collectionformat/_model_base.py | 16 + .../parameters/spread/_model_base.py | 16 + .../payload/contentnegotiation/_model_base.py | 16 + .../payload/jsonmergepatch/_model_base.py | 16 + .../payload/mediatype/_model_base.py | 16 + .../payload/multipart/_model_base.py | 16 + .../payload/pageable/_model_base.py | 16 + .../payload-xml/payload/xml/_model_base.py | 16 + .../resiliency/srv/driven1/_model_base.py | 16 + .../resiliency/srv/driven2/_model_base.py | 16 + .../generated/routes/routes/_model_base.py | 16 + .../encodedname/json/_model_base.py | 16 + .../server/endpoint/notdefined/_model_base.py | 16 + .../server/path/multiple/_model_base.py | 16 + .../server/path/single/_model_base.py | 16 + .../versions/notversioned/_model_base.py | 16 + .../server/versions/versioned/_model_base.py | 16 + .../conditionalrequest/_model_base.py | 16 + .../repeatability/_model_base.py | 16 + .../special-words/specialwords/_model_base.py | 16 + .../typetest/array/_model_base.py | 16 + .../typetest/dictionary/_model_base.py | 16 + .../typetest/enum/extensible/_model_base.py | 16 + .../typetest/enum/fixed/_model_base.py | 16 + .../typetest/model/empty/_model_base.py | 16 + .../model/enumdiscriminator/_model_base.py | 16 + .../model/nesteddiscriminator/_model_base.py | 16 + .../model/notdiscriminated/_model_base.py | 16 + .../typetest/model/recursive/_model_base.py | 16 + .../model/singlediscriminator/_model_base.py | 16 + .../typetest/model/usage/_model_base.py | 16 + .../typetest/model/visibility/_model_base.py | 16 + .../additionalproperties/_model_base.py | 16 + .../typetest/property/nullable/_model_base.py | 16 + .../typetest/property/optional/_model_base.py | 16 + .../property/valuetypes/_model_base.py | 16 + .../typetest/scalar/_model_base.py | 16 + .../typetest/union/_model_base.py | 16 + .../versioning/added/_model_base.py | 16 + .../versioning/madeoptional/_model_base.py | 16 + .../versioning/removed/_model_base.py | 16 + .../versioning/renamedfrom/_model_base.py | 16 + .../returntypechangedfrom/_model_base.py | 16 + .../versioning/typechangedfrom/_model_base.py | 16 + 927 files changed, 31315 insertions(+), 150 deletions(-) rename packages/typespec-python/test/azure/generated/authentication-api-key/{authentication/apikey => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/authentication-http-custom/{authentication/http/custom => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/authentication-oauth2/{authentication/oauth2 => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/authentication-union/{authentication/union => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-client-generator-core-access/{specs/azure/clientgenerator/core/access => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/{specs/azure/clientgenerator/core/flattenproperty => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/{specs/azure/clientgenerator/core/usage => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-basic/{specs/azure/core/basic => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-lro-standard/{specs/azure/core/lro/standard => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-model/{specs/azure/core/model => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-page/{specs/azure/core/page => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-scalar/{specs/azure/core/scalar => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-core-traits/{specs/azure/core/traits => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-example-basic/{specs/azure/example/basic => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py create mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py create mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/{azure/resourcemanager/commonproperties => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py rename packages/typespec-python/test/azure/generated/azure-resource-manager-resources/{azure/resourcemanager/resources => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py create mode 100644 packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py rename packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/{azure/specialheaders/xmsclientrequestid => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py create mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py create mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/azurecore-lro-rpc/{azurecore/lro/rpc => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py create mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py create mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-naming/{client/naming => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-default/{client/structure/service => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-multiclient/{client/structure/multiclient => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-renamedoperation/{client/structure/renamedoperation => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/{client/structure/twooperationgroup => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-bytes/{encode/bytes => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-datetime/{encode/datetime => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-duration/{encode/duration => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/encode-numeric/{encode/numeric => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py create mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/headasbooleanfalse/{headasbooleanfalse => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/headasbooleantrue/{headasbooleantrue => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-basic/{parameters/basic => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-body-optionality/{parameters/bodyoptionality => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-collection-format/{parameters/collectionformat => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/parameters-spread/{parameters/spread => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-content-negotiation/{payload/contentnegotiation => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-json-merge-patch/{payload/jsonmergepatch => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py create mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-media-type/{payload/mediatype => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-multipart/{payload/multipart => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-pageable/{payload/pageable => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py create mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/payload-xml/{payload/xml => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/resiliency-srv-driven1/{resiliency/srv/driven1 => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/resiliency-srv-driven2/{resiliency/srv/driven2 => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/routes/{routes => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/serialization-encoded-name-json/{serialization/encodedname/json => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py create mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-endpoint-not-defined/{server/endpoint/notdefined => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py create mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py create mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-path-multiple/{server/path/multiple => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-path-single/{server/path/single => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-versions-not-versioned/{server/versions/notversioned => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/server-versions-versioned/{server/versions/versioned => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/special-headers-conditional-request/{specialheaders/conditionalrequest => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/special-headers-repeatability/{specialheaders/repeatability => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/special-words/{specialwords => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-array/{typetest/array => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-dictionary/{typetest/dictionary => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-enum-extensible/{typetest/enum/extensible => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-enum-fixed/{typetest/enum/fixed => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-empty/{typetest/model/empty => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/{typetest/model/enumdiscriminator => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/{typetest/model/nesteddiscriminator => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/{typetest/model/notdiscriminated => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-recursive/{typetest/model/recursive => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/{typetest/model/singlediscriminator => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-usage/{typetest/model/usage => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-model-visibility/{typetest/model/visibility => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/{typetest/property/additionalproperties => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-nullable/{typetest/property/nullable => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-optional/{typetest/property/optional => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-property-valuetypes/{typetest/property/valuetypes => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-scalar/{typetest/scalar => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/typetest-union/{typetest/union => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-added/{versioning/added => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-madeoptional/{versioning/madeoptional => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-removed/{versioning/removed => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-renamedfrom/{versioning/renamedfrom => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/{versioning/returntypechangedfrom => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py rename packages/typespec-python/test/azure/generated/versioning-typechangedfrom/{versioning/typechangedfrom => }/apiview_mapping_python.json (100%) create mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py create mode 100644 packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-api-key/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-api-key/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_model_base.py b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_model_base.py +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_operations/_operations.py b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_operations/_operations.py index 326ccdcc4f7..07bcdba930d 100644 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -22,7 +22,7 @@ from azure.core.utils import case_insensitive_dict from .. import models as _models -from .._model_base import _deserialize +from .._model_base import _failsafe_deserialize from .._serialization import Serializer from .._vendor import ApiKeyClientMixinABC @@ -117,12 +117,6 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -149,7 +143,10 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py index c1c1b85fc4f..f9520a8eb3a 100644 --- a/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -21,7 +21,7 @@ from azure.core.tracing.decorator_async import distributed_trace_async from ... import models as _models -from ..._model_base import _deserialize +from ..._model_base import _failsafe_deserialize from ..._operations._operations import build_api_key_invalid_request, build_api_key_valid_request from .._vendor import ApiKeyClientMixinABC @@ -92,12 +92,6 @@ async def invalid(self, **kwargs: Any) -> None: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -124,7 +118,10 @@ async def invalid(self, **kwargs: Any) -> None: if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py new file mode 100644 index 00000000000..8082748a92a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + apikey_subscription_id = os.environ.get("APIKEY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + apikey_tenant_id = os.environ.get("APIKEY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + apikey_client_id = os.environ.get("APIKEY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + apikey_client_secret = os.environ.get("APIKEY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=apikey_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=apikey_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=apikey_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=apikey_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py new file mode 100644 index 00000000000..2916e7b9498 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ApiKeyClientTestBase, ApiKeyPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestApiKey(ApiKeyClientTestBase): + @ApiKeyPreparer() + @recorded_by_proxy + def test_valid(self, apikey_endpoint): + client = self.create_client(endpoint=apikey_endpoint) + response = client.valid() + + # please add some check logic here by yourself + # ... + + @ApiKeyPreparer() + @recorded_by_proxy + def test_invalid(self, apikey_endpoint): + client = self.create_client(endpoint=apikey_endpoint) + response = client.invalid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py new file mode 100644 index 00000000000..214be1ee608 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/test_api_key_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ApiKeyPreparer +from testpreparer_async import ApiKeyClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestApiKeyAsync(ApiKeyClientTestBaseAsync): + @ApiKeyPreparer() + @recorded_by_proxy_async + async def test_valid(self, apikey_endpoint): + client = self.create_async_client(endpoint=apikey_endpoint) + response = await client.valid() + + # please add some check logic here by yourself + # ... + + @ApiKeyPreparer() + @recorded_by_proxy_async + async def test_invalid(self, apikey_endpoint): + client = self.create_async_client(endpoint=apikey_endpoint) + response = await client.invalid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py new file mode 100644 index 00000000000..5af4c923ab1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.apikey import ApiKeyClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class ApiKeyClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ApiKeyClient) + return self.create_client_from_credential( + ApiKeyClient, + credential=credential, + endpoint=endpoint, + ) + + +ApiKeyPreparer = functools.partial(PowerShellPreparer, "apikey", apikey_endpoint="https://fake_apikey_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..09c26f3fe60 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-api-key/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.apikey.aio import ApiKeyClient +from devtools_testutils import AzureRecordedTestCase + + +class ApiKeyClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ApiKeyClient, is_async=True) + return self.create_client_from_credential( + ApiKeyClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-http-custom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-http-custom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_model_base.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_model_base.py +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py index 9e8a461c1c0..07eecd4a0ff 100644 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -22,7 +22,7 @@ from azure.core.utils import case_insensitive_dict from .. import models as _models -from .._model_base import _deserialize +from .._model_base import _failsafe_deserialize from .._serialization import Serializer from .._vendor import CustomClientMixinABC @@ -117,12 +117,6 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -149,7 +143,10 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py index 83c3f3acf3e..7aa51e5da9e 100644 --- a/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -21,7 +21,7 @@ from azure.core.tracing.decorator_async import distributed_trace_async from ... import models as _models -from ..._model_base import _deserialize +from ..._model_base import _failsafe_deserialize from ..._operations._operations import build_custom_invalid_request, build_custom_valid_request from .._vendor import CustomClientMixinABC @@ -92,12 +92,6 @@ async def invalid(self, **kwargs: Any) -> None: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -124,7 +118,10 @@ async def invalid(self, **kwargs: Any) -> None: if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py new file mode 100644 index 00000000000..fac3e2bc399 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + custom_subscription_id = os.environ.get("CUSTOM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + custom_tenant_id = os.environ.get("CUSTOM_TENANT_ID", "00000000-0000-0000-0000-000000000000") + custom_client_id = os.environ.get("CUSTOM_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + custom_client_secret = os.environ.get("CUSTOM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=custom_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=custom_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=custom_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=custom_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py new file mode 100644 index 00000000000..d87f2c0f154 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import CustomClientTestBase, CustomPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCustom(CustomClientTestBase): + @CustomPreparer() + @recorded_by_proxy + def test_valid(self, custom_endpoint): + client = self.create_client(endpoint=custom_endpoint) + response = client.valid() + + # please add some check logic here by yourself + # ... + + @CustomPreparer() + @recorded_by_proxy + def test_invalid(self, custom_endpoint): + client = self.create_client(endpoint=custom_endpoint) + response = client.invalid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py new file mode 100644 index 00000000000..97407d6ad8b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/test_custom_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import CustomPreparer +from testpreparer_async import CustomClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCustomAsync(CustomClientTestBaseAsync): + @CustomPreparer() + @recorded_by_proxy_async + async def test_valid(self, custom_endpoint): + client = self.create_async_client(endpoint=custom_endpoint) + response = await client.valid() + + # please add some check logic here by yourself + # ... + + @CustomPreparer() + @recorded_by_proxy_async + async def test_invalid(self, custom_endpoint): + client = self.create_async_client(endpoint=custom_endpoint) + response = await client.invalid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py new file mode 100644 index 00000000000..5d6664b2552 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.http.custom import CustomClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class CustomClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(CustomClient) + return self.create_client_from_credential( + CustomClient, + credential=credential, + endpoint=endpoint, + ) + + +CustomPreparer = functools.partial(PowerShellPreparer, "custom", custom_endpoint="https://fake_custom_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..3355fa69b2b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-http-custom/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.http.custom.aio import CustomClient +from devtools_testutils import AzureRecordedTestCase + + +class CustomClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(CustomClient, is_async=True) + return self.create_client_from_credential( + CustomClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-oauth2/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-oauth2/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_model_base.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_model_base.py +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py index 32e92fbc6fd..69d66aedf8c 100644 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -22,7 +22,7 @@ from azure.core.utils import case_insensitive_dict from .. import models as _models -from .._model_base import _deserialize +from .._model_base import _failsafe_deserialize from .._serialization import Serializer from .._vendor import OAuth2ClientMixinABC @@ -117,12 +117,6 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -149,7 +143,10 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py index c5d01164158..4bbc165cab4 100644 --- a/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -21,7 +21,7 @@ from azure.core.tracing.decorator_async import distributed_trace_async from ... import models as _models -from ..._model_base import _deserialize +from ..._model_base import _failsafe_deserialize from ..._operations._operations import build_oauth2_invalid_request, build_oauth2_valid_request from .._vendor import OAuth2ClientMixinABC @@ -92,12 +92,6 @@ async def invalid(self, **kwargs: Any) -> None: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -124,7 +118,10 @@ async def invalid(self, **kwargs: Any) -> None: if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py new file mode 100644 index 00000000000..f8a1115d861 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + oauth2_subscription_id = os.environ.get("OAUTH2_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + oauth2_tenant_id = os.environ.get("OAUTH2_TENANT_ID", "00000000-0000-0000-0000-000000000000") + oauth2_client_id = os.environ.get("OAUTH2_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + oauth2_client_secret = os.environ.get("OAUTH2_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=oauth2_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=oauth2_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=oauth2_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=oauth2_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py new file mode 100644 index 00000000000..d1c8fee03c3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OAuth2ClientTestBase, OAuth2Preparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOAuth2(OAuth2ClientTestBase): + @OAuth2Preparer() + @recorded_by_proxy + def test_valid(self, oauth2_endpoint): + client = self.create_client(endpoint=oauth2_endpoint) + response = client.valid() + + # please add some check logic here by yourself + # ... + + @OAuth2Preparer() + @recorded_by_proxy + def test_invalid(self, oauth2_endpoint): + client = self.create_client(endpoint=oauth2_endpoint) + response = client.invalid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py new file mode 100644 index 00000000000..f7f88060283 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/test_oauth2_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OAuth2Preparer +from testpreparer_async import OAuth2ClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOAuth2Async(OAuth2ClientTestBaseAsync): + @OAuth2Preparer() + @recorded_by_proxy_async + async def test_valid(self, oauth2_endpoint): + client = self.create_async_client(endpoint=oauth2_endpoint) + response = await client.valid() + + # please add some check logic here by yourself + # ... + + @OAuth2Preparer() + @recorded_by_proxy_async + async def test_invalid(self, oauth2_endpoint): + client = self.create_async_client(endpoint=oauth2_endpoint) + response = await client.invalid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py new file mode 100644 index 00000000000..24fb53e3ec1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.oauth2 import OAuth2Client +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class OAuth2ClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(OAuth2Client) + return self.create_client_from_credential( + OAuth2Client, + credential=credential, + endpoint=endpoint, + ) + + +OAuth2Preparer = functools.partial(PowerShellPreparer, "oauth2", oauth2_endpoint="https://fake_oauth2_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..6eff123aef9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-oauth2/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.oauth2.aio import OAuth2Client +from devtools_testutils import AzureRecordedTestCase + + +class OAuth2ClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(OAuth2Client, is_async=True) + return self.create_client_from_credential( + OAuth2Client, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/authentication-union/authentication/union/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/authentication-union/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/authentication-union/authentication/union/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/authentication-union/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/authentication-union/authentication/union/_model_base.py b/packages/typespec-python/test/azure/generated/authentication-union/authentication/union/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/authentication-union/authentication/union/_model_base.py +++ b/packages/typespec-python/test/azure/generated/authentication-union/authentication/union/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py new file mode 100644 index 00000000000..5e28d6377c8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + union_subscription_id = os.environ.get("UNION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + union_tenant_id = os.environ.get("UNION_TENANT_ID", "00000000-0000-0000-0000-000000000000") + union_client_id = os.environ.get("UNION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + union_client_secret = os.environ.get("UNION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py new file mode 100644 index 00000000000..fdfea7e808b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnion(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_valid_key(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.valid_key() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_valid_token(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.valid_token() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py new file mode 100644 index 00000000000..7efe4a658f5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/test_union_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_valid_key(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.valid_key() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_valid_token(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.valid_token() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py new file mode 100644 index 00000000000..70237769077 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.union import UnionClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class UnionClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(UnionClient) + return self.create_client_from_credential( + UnionClient, + credential=credential, + endpoint=endpoint, + ) + + +UnionPreparer = functools.partial(PowerShellPreparer, "union", union_endpoint="https://fake_union_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..7b184e26233 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/authentication-union/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from authentication.union.aio import UnionClient +from devtools_testutils import AzureRecordedTestCase + + +class UnionClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(UnionClient, is_async=True) + return self.create_client_from_credential( + UnionClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-client-generator-core-access/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py new file mode 100644 index 00000000000..33880e8e0f8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + access_subscription_id = os.environ.get("ACCESS_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + access_tenant_id = os.environ.get("ACCESS_TENANT_ID", "00000000-0000-0000-0000-000000000000") + access_client_id = os.environ.get("ACCESS_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + access_client_secret = os.environ.get("ACCESS_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=access_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=access_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=access_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=access_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py new file mode 100644 index 00000000000..bf07380e8fa --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AccessClientTestBase, AccessPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAccessPublicOperationOperations(AccessClientTestBase): + @AccessPreparer() + @recorded_by_proxy + def test_public_operation_no_decorator_in_public(self, access_endpoint): + client = self.create_client(endpoint=access_endpoint) + response = client.public_operation.no_decorator_in_public( + name="str", + ) + + # please add some check logic here by yourself + # ... + + @AccessPreparer() + @recorded_by_proxy + def test_public_operation_public_decorator_in_public(self, access_endpoint): + client = self.create_client(endpoint=access_endpoint) + response = client.public_operation.public_decorator_in_public( + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py new file mode 100644 index 00000000000..2396a14d56f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_public_operation_operations_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AccessPreparer +from testpreparer_async import AccessClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAccessPublicOperationOperationsAsync(AccessClientTestBaseAsync): + @AccessPreparer() + @recorded_by_proxy_async + async def test_public_operation_no_decorator_in_public(self, access_endpoint): + client = self.create_async_client(endpoint=access_endpoint) + response = await client.public_operation.no_decorator_in_public( + name="str", + ) + + # please add some check logic here by yourself + # ... + + @AccessPreparer() + @recorded_by_proxy_async + async def test_public_operation_public_decorator_in_public(self, access_endpoint): + client = self.create_async_client(endpoint=access_endpoint) + response = await client.public_operation.public_decorator_in_public( + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py new file mode 100644 index 00000000000..df2954375fd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AccessClientTestBase, AccessPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAccessSharedModelInOperationOperations(AccessClientTestBase): + @AccessPreparer() + @recorded_by_proxy + def test_shared_model_in_operation_public(self, access_endpoint): + client = self.create_client(endpoint=access_endpoint) + response = client.shared_model_in_operation.public( + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py new file mode 100644 index 00000000000..a4ce11fd46b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/test_access_shared_model_in_operation_operations_async.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AccessPreparer +from testpreparer_async import AccessClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAccessSharedModelInOperationOperationsAsync(AccessClientTestBaseAsync): + @AccessPreparer() + @recorded_by_proxy_async + async def test_shared_model_in_operation_public(self, access_endpoint): + client = self.create_async_client(endpoint=access_endpoint) + response = await client.shared_model_in_operation.public( + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py new file mode 100644 index 00000000000..7deee53be35 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.clientgenerator.core.access import AccessClient + + +class AccessClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(AccessClient) + return self.create_client_from_credential( + AccessClient, + credential=credential, + endpoint=endpoint, + ) + + +AccessPreparer = functools.partial(PowerShellPreparer, "access", access_endpoint="https://fake_access_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..7d4e8352e2b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.clientgenerator.core.access.aio import AccessClient + + +class AccessClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(AccessClient, is_async=True) + return self.create_client_from_credential( + AccessClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/_model_base.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-access/specs/azure/clientgenerator/core/access/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py new file mode 100644 index 00000000000..77400d8847a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + flattenproperty_subscription_id = os.environ.get( + "FLATTENPROPERTY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + flattenproperty_tenant_id = os.environ.get("FLATTENPROPERTY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + flattenproperty_client_id = os.environ.get("FLATTENPROPERTY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + flattenproperty_client_secret = os.environ.get( + "FLATTENPROPERTY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=flattenproperty_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=flattenproperty_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=flattenproperty_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=flattenproperty_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py new file mode 100644 index 00000000000..8563ebaa3b4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import FlattenPropertyClientTestBase, FlattenPropertyPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestFlattenProperty(FlattenPropertyClientTestBase): + @FlattenPropertyPreparer() + @recorded_by_proxy + def test_put_flatten_model(self, flattenproperty_endpoint): + client = self.create_client(endpoint=flattenproperty_endpoint) + response = client.put_flatten_model( + input={"name": "str", "properties": {"age": 0, "description": "str"}}, + ) + + # please add some check logic here by yourself + # ... + + @FlattenPropertyPreparer() + @recorded_by_proxy + def test_put_nested_flatten_model(self, flattenproperty_endpoint): + client = self.create_client(endpoint=flattenproperty_endpoint) + response = client.put_nested_flatten_model( + input={"name": "str", "properties": {"properties": {"age": 0, "description": "str"}, "summary": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py new file mode 100644 index 00000000000..93a1b1b5eff --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/test_flatten_property_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import FlattenPropertyPreparer +from testpreparer_async import FlattenPropertyClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestFlattenPropertyAsync(FlattenPropertyClientTestBaseAsync): + @FlattenPropertyPreparer() + @recorded_by_proxy_async + async def test_put_flatten_model(self, flattenproperty_endpoint): + client = self.create_async_client(endpoint=flattenproperty_endpoint) + response = await client.put_flatten_model( + input={"name": "str", "properties": {"age": 0, "description": "str"}}, + ) + + # please add some check logic here by yourself + # ... + + @FlattenPropertyPreparer() + @recorded_by_proxy_async + async def test_put_nested_flatten_model(self, flattenproperty_endpoint): + client = self.create_async_client(endpoint=flattenproperty_endpoint) + response = await client.put_nested_flatten_model( + input={"name": "str", "properties": {"properties": {"age": 0, "description": "str"}, "summary": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py new file mode 100644 index 00000000000..739c5ff2de4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.clientgenerator.core.flattenproperty import FlattenPropertyClient + + +class FlattenPropertyClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(FlattenPropertyClient) + return self.create_client_from_credential( + FlattenPropertyClient, + credential=credential, + endpoint=endpoint, + ) + + +FlattenPropertyPreparer = functools.partial( + PowerShellPreparer, "flattenproperty", flattenproperty_endpoint="https://fake_flattenproperty_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..f66cc7610b9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.clientgenerator.core.flattenproperty.aio import FlattenPropertyClient + + +class FlattenPropertyClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(FlattenPropertyClient, is_async=True) + return self.create_client_from_credential( + FlattenPropertyClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/_model_base.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-flatten-property/specs/azure/clientgenerator/core/flattenproperty/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py new file mode 100644 index 00000000000..6a990b92bd7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + usage_subscription_id = os.environ.get("USAGE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + usage_tenant_id = os.environ.get("USAGE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + usage_client_id = os.environ.get("USAGE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + usage_client_secret = os.environ.get("USAGE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py new file mode 100644 index 00000000000..1594a7225fd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations.py @@ -0,0 +1,44 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UsageClientTestBase, UsagePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUsageModelInOperationOperations(UsageClientTestBase): + @UsagePreparer() + @recorded_by_proxy + def test_model_in_operation_input_to_input_output(self, usage_endpoint): + client = self.create_client(endpoint=usage_endpoint) + response = client.model_in_operation.input_to_input_output( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy + def test_model_in_operation_output_to_input_output(self, usage_endpoint): + client = self.create_client(endpoint=usage_endpoint) + response = client.model_in_operation.output_to_input_output() + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy + def test_model_in_operation_model_in_read_only_property(self, usage_endpoint): + client = self.create_client(endpoint=usage_endpoint) + response = client.model_in_operation.model_in_read_only_property( + body={"result": {"name": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py new file mode 100644 index 00000000000..c0b04a06e4d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/test_usage_model_in_operation_operations_async.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UsagePreparer +from testpreparer_async import UsageClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUsageModelInOperationOperationsAsync(UsageClientTestBaseAsync): + @UsagePreparer() + @recorded_by_proxy_async + async def test_model_in_operation_input_to_input_output(self, usage_endpoint): + client = self.create_async_client(endpoint=usage_endpoint) + response = await client.model_in_operation.input_to_input_output( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy_async + async def test_model_in_operation_output_to_input_output(self, usage_endpoint): + client = self.create_async_client(endpoint=usage_endpoint) + response = await client.model_in_operation.output_to_input_output() + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy_async + async def test_model_in_operation_model_in_read_only_property(self, usage_endpoint): + client = self.create_async_client(endpoint=usage_endpoint) + response = await client.model_in_operation.model_in_read_only_property( + body={"result": {"name": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py new file mode 100644 index 00000000000..6d1b1d9481f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.clientgenerator.core.usage import UsageClient + + +class UsageClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(UsageClient) + return self.create_client_from_credential( + UsageClient, + credential=credential, + endpoint=endpoint, + ) + + +UsagePreparer = functools.partial(PowerShellPreparer, "usage", usage_endpoint="https://fake_usage_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..d27978d8baa --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.clientgenerator.core.usage.aio import UsageClient + + +class UsageClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(UsageClient, is_async=True) + return self.create_client_from_credential( + UsageClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/_model_base.py b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-client-generator-core-usage/specs/azure/clientgenerator/core/usage/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-basic/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-basic/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py new file mode 100644 index 00000000000..635e5035817 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + basic_subscription_id = os.environ.get("BASIC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + basic_tenant_id = os.environ.get("BASIC_TENANT_ID", "00000000-0000-0000-0000-000000000000") + basic_client_id = os.environ.get("BASIC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + basic_client_secret = os.environ.get("BASIC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py new file mode 100644 index 00000000000..4e1621dc840 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic.py @@ -0,0 +1,91 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BasicClientTestBase, BasicPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBasic(BasicClientTestBase): + @BasicPreparer() + @recorded_by_proxy + def test_create_or_update(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.create_or_update( + id=0, + resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy + def test_create_or_replace(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.create_or_replace( + id=0, + resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy + def test_get(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.get( + id=0, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy + def test_list(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.list() + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy + def test_delete(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.delete( + id=0, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy + def test_export(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.export( + id=0, + format="str", + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy + def test_export_all_users(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.export_all_users( + format="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py new file mode 100644 index 00000000000..c3c7f05a6cf --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/test_basic_async.py @@ -0,0 +1,92 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BasicPreparer +from testpreparer_async import BasicClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBasicAsync(BasicClientTestBaseAsync): + @BasicPreparer() + @recorded_by_proxy_async + async def test_create_or_update(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.create_or_update( + id=0, + resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy_async + async def test_create_or_replace(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.create_or_replace( + id=0, + resource={"etag": "str", "id": 0, "name": "str", "orders": [{"detail": "str", "id": 0, "userId": 0}]}, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy_async + async def test_get(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.get( + id=0, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy_async + async def test_list(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = client.list() + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy_async + async def test_delete(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.delete( + id=0, + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy_async + async def test_export(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.export( + id=0, + format="str", + ) + + # please add some check logic here by yourself + # ... + + @BasicPreparer() + @recorded_by_proxy_async + async def test_export_all_users(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.export_all_users( + format="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py new file mode 100644 index 00000000000..b662845bf16 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.core.basic import BasicClient + + +class BasicClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(BasicClient) + return self.create_client_from_credential( + BasicClient, + credential=credential, + endpoint=endpoint, + ) + + +BasicPreparer = functools.partial(PowerShellPreparer, "basic", basic_endpoint="https://fake_basic_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..2b63ad3c8f4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-basic/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.core.basic.aio import BasicClient + + +class BasicClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(BasicClient, is_async=True) + return self.create_client_from_credential( + BasicClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/_model_base.py b/packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-core-basic/specs/azure/core/basic/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-lro-standard/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py new file mode 100644 index 00000000000..420b04f8797 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + standard_subscription_id = os.environ.get("STANDARD_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + standard_tenant_id = os.environ.get("STANDARD_TENANT_ID", "00000000-0000-0000-0000-000000000000") + standard_client_id = os.environ.get("STANDARD_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + standard_client_secret = os.environ.get("STANDARD_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=standard_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=standard_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=standard_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=standard_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py new file mode 100644 index 00000000000..e697fd1634b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard.py @@ -0,0 +1,48 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import StandardClientTestBase, StandardPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestStandard(StandardClientTestBase): + @StandardPreparer() + @recorded_by_proxy + def test_begin_create_or_replace(self, standard_endpoint): + client = self.create_client(endpoint=standard_endpoint) + response = client.begin_create_or_replace( + name="str", + resource={"name": "str", "role": "str"}, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @StandardPreparer() + @recorded_by_proxy + def test_begin_delete(self, standard_endpoint): + client = self.create_client(endpoint=standard_endpoint) + response = client.begin_delete( + name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @StandardPreparer() + @recorded_by_proxy + def test_begin_export(self, standard_endpoint): + client = self.create_client(endpoint=standard_endpoint) + response = client.begin_export( + name="str", + format="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py new file mode 100644 index 00000000000..5b8c4cb2479 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/test_standard_async.py @@ -0,0 +1,55 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import StandardPreparer +from testpreparer_async import StandardClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestStandardAsync(StandardClientTestBaseAsync): + @StandardPreparer() + @recorded_by_proxy_async + async def test_begin_create_or_replace(self, standard_endpoint): + client = self.create_async_client(endpoint=standard_endpoint) + response = await ( + await client.begin_create_or_replace( + name="str", + resource={"name": "str", "role": "str"}, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @StandardPreparer() + @recorded_by_proxy_async + async def test_begin_delete(self, standard_endpoint): + client = self.create_async_client(endpoint=standard_endpoint) + response = await ( + await client.begin_delete( + name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @StandardPreparer() + @recorded_by_proxy_async + async def test_begin_export(self, standard_endpoint): + client = self.create_async_client(endpoint=standard_endpoint) + response = await ( + await client.begin_export( + name="str", + format="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py new file mode 100644 index 00000000000..e17f19a3271 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.core.lro.standard import StandardClient + + +class StandardClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(StandardClient) + return self.create_client_from_credential( + StandardClient, + credential=credential, + endpoint=endpoint, + ) + + +StandardPreparer = functools.partial( + PowerShellPreparer, "standard", standard_endpoint="https://fake_standard_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..9b1a3cbea5c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.core.lro.standard.aio import StandardClient + + +class StandardClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(StandardClient, is_async=True) + return self.create_client_from_credential( + StandardClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/_model_base.py b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-core-lro-standard/specs/azure/core/lro/standard/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-model/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-model/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py new file mode 100644 index 00000000000..b11e17f60f7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + model_subscription_id = os.environ.get("MODEL_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + model_tenant_id = os.environ.get("MODEL_TENANT_ID", "00000000-0000-0000-0000-000000000000") + model_client_id = os.environ.get("MODEL_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + model_client_secret = os.environ.get("MODEL_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=model_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=model_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=model_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=model_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py new file mode 100644 index 00000000000..824eea481e1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations.py @@ -0,0 +1,44 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ModelClientTestBase, ModelPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestModelAzureCoreEmbeddingVectorOperations(ModelClientTestBase): + @ModelPreparer() + @recorded_by_proxy + def test_azure_core_embedding_vector_get(self, model_endpoint): + client = self.create_client(endpoint=model_endpoint) + response = client.azure_core_embedding_vector.get() + + # please add some check logic here by yourself + # ... + + @ModelPreparer() + @recorded_by_proxy + def test_azure_core_embedding_vector_put(self, model_endpoint): + client = self.create_client(endpoint=model_endpoint) + response = client.azure_core_embedding_vector.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... + + @ModelPreparer() + @recorded_by_proxy + def test_azure_core_embedding_vector_post(self, model_endpoint): + client = self.create_client(endpoint=model_endpoint) + response = client.azure_core_embedding_vector.post( + body={"embedding": [0]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py new file mode 100644 index 00000000000..b7f1463844b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/test_model_azure_core_embedding_vector_operations_async.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ModelPreparer +from testpreparer_async import ModelClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestModelAzureCoreEmbeddingVectorOperationsAsync(ModelClientTestBaseAsync): + @ModelPreparer() + @recorded_by_proxy_async + async def test_azure_core_embedding_vector_get(self, model_endpoint): + client = self.create_async_client(endpoint=model_endpoint) + response = await client.azure_core_embedding_vector.get() + + # please add some check logic here by yourself + # ... + + @ModelPreparer() + @recorded_by_proxy_async + async def test_azure_core_embedding_vector_put(self, model_endpoint): + client = self.create_async_client(endpoint=model_endpoint) + response = await client.azure_core_embedding_vector.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... + + @ModelPreparer() + @recorded_by_proxy_async + async def test_azure_core_embedding_vector_post(self, model_endpoint): + client = self.create_async_client(endpoint=model_endpoint) + response = await client.azure_core_embedding_vector.post( + body={"embedding": [0]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py new file mode 100644 index 00000000000..e8743f4445d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.core.model import ModelClient + + +class ModelClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ModelClient) + return self.create_client_from_credential( + ModelClient, + credential=credential, + endpoint=endpoint, + ) + + +ModelPreparer = functools.partial(PowerShellPreparer, "model", model_endpoint="https://fake_model_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..6eafdc7e92f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-model/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.core.model.aio import ModelClient + + +class ModelClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ModelClient, is_async=True) + return self.create_client_from_credential( + ModelClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/_model_base.py b/packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-core-model/specs/azure/core/model/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-page/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-page/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py new file mode 100644 index 00000000000..76efbc92590 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + page_subscription_id = os.environ.get("PAGE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + page_tenant_id = os.environ.get("PAGE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + page_client_id = os.environ.get("PAGE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + page_client_secret = os.environ.get("PAGE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=page_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=page_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=page_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=page_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py new file mode 100644 index 00000000000..053081e9ec6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page.py @@ -0,0 +1,42 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import PageClientTestBase, PagePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestPage(PageClientTestBase): + @PagePreparer() + @recorded_by_proxy + def test_list_with_page(self, page_endpoint): + client = self.create_client(endpoint=page_endpoint) + response = client.list_with_page() + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @PagePreparer() + @recorded_by_proxy + def test_list_with_parameters(self, page_endpoint): + client = self.create_client(endpoint=page_endpoint) + response = client.list_with_parameters( + body_input={"inputName": "str"}, + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @PagePreparer() + @recorded_by_proxy + def test_list_with_custom_page_model(self, page_endpoint): + client = self.create_client(endpoint=page_endpoint) + response = client.list_with_custom_page_model() + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py new file mode 100644 index 00000000000..3bfc3a98048 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_async.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import PagePreparer +from testpreparer_async import PageClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestPageAsync(PageClientTestBaseAsync): + @PagePreparer() + @recorded_by_proxy_async + async def test_list_with_page(self, page_endpoint): + client = self.create_async_client(endpoint=page_endpoint) + response = client.list_with_page() + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @PagePreparer() + @recorded_by_proxy_async + async def test_list_with_parameters(self, page_endpoint): + client = self.create_async_client(endpoint=page_endpoint) + response = client.list_with_parameters( + body_input={"inputName": "str"}, + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @PagePreparer() + @recorded_by_proxy_async + async def test_list_with_custom_page_model(self, page_endpoint): + client = self.create_async_client(endpoint=page_endpoint) + response = client.list_with_custom_page_model() + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py new file mode 100644 index 00000000000..e3b9755a70d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import PageClientTestBase, PagePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestPageTwoModelsAsPageItemOperations(PageClientTestBase): + @PagePreparer() + @recorded_by_proxy + def test_two_models_as_page_item_list_first_item(self, page_endpoint): + client = self.create_client(endpoint=page_endpoint) + response = client.two_models_as_page_item.list_first_item() + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @PagePreparer() + @recorded_by_proxy + def test_two_models_as_page_item_list_second_item(self, page_endpoint): + client = self.create_client(endpoint=page_endpoint) + response = client.two_models_as_page_item.list_second_item() + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py new file mode 100644 index 00000000000..6ce3f441718 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/test_page_two_models_as_page_item_operations_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import PagePreparer +from testpreparer_async import PageClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestPageTwoModelsAsPageItemOperationsAsync(PageClientTestBaseAsync): + @PagePreparer() + @recorded_by_proxy_async + async def test_two_models_as_page_item_list_first_item(self, page_endpoint): + client = self.create_async_client(endpoint=page_endpoint) + response = client.two_models_as_page_item.list_first_item() + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @PagePreparer() + @recorded_by_proxy_async + async def test_two_models_as_page_item_list_second_item(self, page_endpoint): + client = self.create_async_client(endpoint=page_endpoint) + response = client.two_models_as_page_item.list_second_item() + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py new file mode 100644 index 00000000000..921bc83a56c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.core.page import PageClient + + +class PageClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(PageClient) + return self.create_client_from_credential( + PageClient, + credential=credential, + endpoint=endpoint, + ) + + +PagePreparer = functools.partial(PowerShellPreparer, "page", page_endpoint="https://fake_page_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..fa9ad45c103 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-page/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.core.page.aio import PageClient + + +class PageClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(PageClient, is_async=True) + return self.create_client_from_credential( + PageClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/_model_base.py b/packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-core-page/specs/azure/core/page/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-scalar/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-scalar/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py new file mode 100644 index 00000000000..f7252bc7a1b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + scalar_subscription_id = os.environ.get("SCALAR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + scalar_tenant_id = os.environ.get("SCALAR_TENANT_ID", "00000000-0000-0000-0000-000000000000") + scalar_client_id = os.environ.get("SCALAR_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + scalar_client_secret = os.environ.get("SCALAR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py new file mode 100644 index 00000000000..234ea6351a1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations.py @@ -0,0 +1,67 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarAzureLocationScalarOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_azure_location_scalar_get(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.azure_location_scalar.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_azure_location_scalar_put(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.azure_location_scalar.put( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_azure_location_scalar_post(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.azure_location_scalar.post( + body={"location": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_azure_location_scalar_header(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.azure_location_scalar.header( + region="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_azure_location_scalar_query(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.azure_location_scalar.query( + region="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py new file mode 100644 index 00000000000..38cff3d36b5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/test_scalar_azure_location_scalar_operations_async.py @@ -0,0 +1,68 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarAzureLocationScalarOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_azure_location_scalar_get(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.azure_location_scalar.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_azure_location_scalar_put(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.azure_location_scalar.put( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_azure_location_scalar_post(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.azure_location_scalar.post( + body={"location": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_azure_location_scalar_header(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.azure_location_scalar.header( + region="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_azure_location_scalar_query(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.azure_location_scalar.query( + region="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py new file mode 100644 index 00000000000..cb3c43946e6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.core.scalar import ScalarClient + + +class ScalarClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ScalarClient) + return self.create_client_from_credential( + ScalarClient, + credential=credential, + endpoint=endpoint, + ) + + +ScalarPreparer = functools.partial(PowerShellPreparer, "scalar", scalar_endpoint="https://fake_scalar_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..5d70e7f1fa2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-scalar/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.core.scalar.aio import ScalarClient + + +class ScalarClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ScalarClient, is_async=True) + return self.create_client_from_credential( + ScalarClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/_model_base.py b/packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-core-scalar/specs/azure/core/scalar/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-core-traits/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-core-traits/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py new file mode 100644 index 00000000000..fd65dd2df68 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + traits_subscription_id = os.environ.get("TRAITS_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + traits_tenant_id = os.environ.get("TRAITS_TENANT_ID", "00000000-0000-0000-0000-000000000000") + traits_client_id = os.environ.get("TRAITS_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + traits_client_secret = os.environ.get("TRAITS_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=traits_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=traits_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=traits_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=traits_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py new file mode 100644 index 00000000000..eca09365ca5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import TraitsClientTestBase, TraitsPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTraits(TraitsClientTestBase): + @TraitsPreparer() + @recorded_by_proxy + def test_smoke_test(self, traits_endpoint): + client = self.create_client(endpoint=traits_endpoint) + response = client.smoke_test( + id=0, + foo="str", + ) + + # please add some check logic here by yourself + # ... + + @TraitsPreparer() + @recorded_by_proxy + def test_repeatable_action(self, traits_endpoint): + client = self.create_client(endpoint=traits_endpoint) + response = client.repeatable_action( + id=0, + body={"userActionValue": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py new file mode 100644 index 00000000000..7188bd6c1e7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/test_traits_async.py @@ -0,0 +1,38 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import TraitsPreparer +from testpreparer_async import TraitsClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTraitsAsync(TraitsClientTestBaseAsync): + @TraitsPreparer() + @recorded_by_proxy_async + async def test_smoke_test(self, traits_endpoint): + client = self.create_async_client(endpoint=traits_endpoint) + response = await client.smoke_test( + id=0, + foo="str", + ) + + # please add some check logic here by yourself + # ... + + @TraitsPreparer() + @recorded_by_proxy_async + async def test_repeatable_action(self, traits_endpoint): + client = self.create_async_client(endpoint=traits_endpoint) + response = await client.repeatable_action( + id=0, + body={"userActionValue": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py new file mode 100644 index 00000000000..661fdabcd75 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.core.traits import TraitsClient + + +class TraitsClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(TraitsClient) + return self.create_client_from_credential( + TraitsClient, + credential=credential, + endpoint=endpoint, + ) + + +TraitsPreparer = functools.partial(PowerShellPreparer, "traits", traits_endpoint="https://fake_traits_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..ee53fde0dfb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-core-traits/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.core.traits.aio import TraitsClient + + +class TraitsClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(TraitsClient, is_async=True) + return self.create_client_from_credential( + TraitsClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/_model_base.py b/packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-core-traits/specs/azure/core/traits/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-example-basic/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-example-basic/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py new file mode 100644 index 00000000000..8fc73742343 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from specs.azure.example.basic import AzureExampleClient + +""" +# PREREQUISITES + pip install specs-azure-example-basic +# USAGE + python basic.py +""" + + +def main(): + client = AzureExampleClient() + + response = client.basic_action( + body={ + "arrayProperty": ["item"], + "modelProperty": {"enumProperty": "EnumValue1", "float32Property": 1.5, "int32Property": 1}, + "recordProperty": {"record": "value"}, + "stringProperty": "text", + }, + query_param="query", + header_param="header", + ) + print(response) + + +# x-ms-original-file: 2022-12-01-preview/basic.json +if __name__ == "__main__": + main() diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py new file mode 100644 index 00000000000..dfc189b22aa --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/conftest.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + azureexample_subscription_id = os.environ.get( + "AZUREEXAMPLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + azureexample_tenant_id = os.environ.get("AZUREEXAMPLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + azureexample_client_id = os.environ.get("AZUREEXAMPLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + azureexample_client_secret = os.environ.get("AZUREEXAMPLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=azureexample_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=azureexample_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=azureexample_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=azureexample_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py new file mode 100644 index 00000000000..9d66b6994ae --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AzureExampleClientTestBase, AzureExamplePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAzureExample(AzureExampleClientTestBase): + @AzureExamplePreparer() + @recorded_by_proxy + def test_basic_action(self, azureexample_endpoint): + client = self.create_client(endpoint=azureexample_endpoint) + response = client.basic_action( + body={ + "stringProperty": "str", + "arrayProperty": ["str"], + "modelProperty": {"enumProperty": "str", "float32Property": 0.0, "int32Property": 0}, + "recordProperty": {"str": "str"}, + }, + query_param="str", + header_param="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py new file mode 100644 index 00000000000..ec5969f68e3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/test_azure_example_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AzureExamplePreparer +from testpreparer_async import AzureExampleClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAzureExampleAsync(AzureExampleClientTestBaseAsync): + @AzureExamplePreparer() + @recorded_by_proxy_async + async def test_basic_action(self, azureexample_endpoint): + client = self.create_async_client(endpoint=azureexample_endpoint) + response = await client.basic_action( + body={ + "stringProperty": "str", + "arrayProperty": ["str"], + "modelProperty": {"enumProperty": "str", "float32Property": 0.0, "int32Property": 0}, + "recordProperty": {"str": "str"}, + }, + query_param="str", + header_param="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py new file mode 100644 index 00000000000..6039e937d65 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specs.azure.example.basic import AzureExampleClient + + +class AzureExampleClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(AzureExampleClient) + return self.create_client_from_credential( + AzureExampleClient, + credential=credential, + endpoint=endpoint, + ) + + +AzureExamplePreparer = functools.partial( + PowerShellPreparer, "azureexample", azureexample_endpoint="https://fake_azureexample_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..64243921636 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specs.azure.example.basic.aio import AzureExampleClient + + +class AzureExampleClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(AzureExampleClient, is_async=True) + return self.create_client_from_credential( + AzureExampleClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/_model_base.py b/packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-example-basic/specs/azure/example/basic/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/_model_base.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/aio/operations/_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/aio/operations/_operations.py index becd47a4266..f0fa524ded4 100644 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/aio/operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/aio/operations/_operations.py @@ -27,7 +27,7 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ..._model_base import SdkJSONEncoder, _deserialize +from ..._model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from ...operations._operations import ( build_managed_identity_create_with_system_assigned_request, build_managed_identity_get_request, @@ -116,7 +116,7 @@ async def get( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -287,7 +287,7 @@ async def create_with_system_assigned( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -458,7 +458,7 @@ async def update_with_user_assigned_and_system_assigned( # pylint: disable=name except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/operations/_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/operations/_operations.py index f4cd08bb366..c1f07c0062e 100644 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/azure/resourcemanager/commonproperties/operations/_operations.py @@ -27,7 +27,7 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from .. import models as _models -from .._model_base import SdkJSONEncoder, _deserialize +from .._model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from .._serialization import Serializer if sys.version_info >= (3, 9): @@ -211,7 +211,7 @@ def get( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -382,7 +382,7 @@ def create_with_system_assigned( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -553,7 +553,7 @@ def update_with_user_assigned_and_system_assigned( # pylint: disable=name-too-l except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py new file mode 100644 index 00000000000..49852579a99 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + commonproperties_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + commonproperties_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + commonproperties_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + commonproperties_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=commonproperties_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=commonproperties_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=commonproperties_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=commonproperties_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py new file mode 100644 index 00000000000..92226f4ed8f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations.py @@ -0,0 +1,96 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.commonproperties import CommonPropertiesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCommonPropertiesManagedIdentityOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(CommonPropertiesClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_managed_identity_get(self, resource_group): + response = self.client.managed_identity.get( + resource_group_name=resource_group.name, + managed_identity_tracked_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_managed_identity_create_with_system_assigned(self, resource_group): + response = self.client.managed_identity.create_with_system_assigned( + resource_group_name=resource_group.name, + managed_identity_tracked_resource_name="str", + resource={ + "location": "str", + "id": "str", + "identity": { + "type": "str", + "principalId": "str", + "tenantId": "str", + "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, + }, + "name": "str", + "properties": {"provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_managed_identity_update_with_user_assigned_and_system_assigned(self, resource_group): + response = self.client.managed_identity.update_with_user_assigned_and_system_assigned( + resource_group_name=resource_group.name, + managed_identity_tracked_resource_name="str", + properties={ + "location": "str", + "id": "str", + "identity": { + "type": "str", + "principalId": "str", + "tenantId": "str", + "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, + }, + "name": "str", + "properties": {"provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py new file mode 100644 index 00000000000..5b0776d5e57 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-common-properties/generated_tests/test_common_properties_managed_identity_operations_async.py @@ -0,0 +1,97 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.commonproperties.aio import CommonPropertiesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCommonPropertiesManagedIdentityOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(CommonPropertiesClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_managed_identity_get(self, resource_group): + response = await self.client.managed_identity.get( + resource_group_name=resource_group.name, + managed_identity_tracked_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_managed_identity_create_with_system_assigned(self, resource_group): + response = await self.client.managed_identity.create_with_system_assigned( + resource_group_name=resource_group.name, + managed_identity_tracked_resource_name="str", + resource={ + "location": "str", + "id": "str", + "identity": { + "type": "str", + "principalId": "str", + "tenantId": "str", + "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, + }, + "name": "str", + "properties": {"provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_managed_identity_update_with_user_assigned_and_system_assigned(self, resource_group): + response = await self.client.managed_identity.update_with_user_assigned_and_system_assigned( + resource_group_name=resource_group.name, + managed_identity_tracked_resource_name="str", + properties={ + "location": "str", + "id": "str", + "identity": { + "type": "str", + "principalId": "str", + "tenantId": "str", + "userAssignedIdentities": {"str": {"clientId": "str", "principalId": "str"}}, + }, + "name": "str", + "properties": {"provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-resource-manager-resources/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/_model_base.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/aio/operations/_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/aio/operations/_operations.py index de77b48f74f..7a47b8cb6ce 100644 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/aio/operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/aio/operations/_operations.py @@ -33,7 +33,7 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ..._model_base import SdkJSONEncoder, _deserialize +from ..._model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from ...operations._operations import ( build_nested_create_or_replace_request, build_nested_delete_request, @@ -134,7 +134,7 @@ async def get( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -203,7 +203,7 @@ async def _create_or_replace_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -440,7 +440,7 @@ async def _update_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -661,7 +661,7 @@ async def _delete_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -819,7 +819,7 @@ async def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response @@ -904,7 +904,7 @@ async def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response @@ -1059,7 +1059,7 @@ async def action_sync( if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: @@ -1145,7 +1145,7 @@ async def get( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -1216,7 +1216,7 @@ async def _create_or_replace_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -1468,7 +1468,7 @@ async def _update_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -1707,7 +1707,7 @@ async def _delete_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -1875,7 +1875,7 @@ async def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response @@ -1951,7 +1951,7 @@ async def get_by_resource_group(self, resource_group_name: str, **kwargs: Any) - except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -2018,7 +2018,7 @@ async def _create_or_update_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -2311,7 +2311,7 @@ async def update( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -2408,7 +2408,7 @@ async def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/operations/_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/operations/_operations.py index 3a4d2d872ee..5495acbbd88 100644 --- a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/operations/_operations.py +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/azure/resourcemanager/resources/operations/_operations.py @@ -32,7 +32,7 @@ from azure.mgmt.core.polling.arm_polling import ARMPolling from .. import models as _models -from .._model_base import SdkJSONEncoder, _deserialize +from .._model_base import SdkJSONEncoder, _deserialize, _failsafe_deserialize from .._serialization import Serializer if sys.version_info >= (3, 9): @@ -619,7 +619,7 @@ def get( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -688,7 +688,7 @@ def _create_or_replace_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -925,7 +925,7 @@ def _update_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -1146,7 +1146,7 @@ def _delete_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -1304,7 +1304,7 @@ def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response @@ -1389,7 +1389,7 @@ def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response @@ -1544,7 +1544,7 @@ def action_sync( # pylint: disable=inconsistent-return-statements if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: @@ -1630,7 +1630,7 @@ def get( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -1701,7 +1701,7 @@ def _create_or_replace_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -1953,7 +1953,7 @@ def _update_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -2192,7 +2192,7 @@ def _delete_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -2360,7 +2360,7 @@ def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response @@ -2436,7 +2436,7 @@ def get_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> _mod except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -2503,7 +2503,7 @@ def _create_or_update_initial( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) response_headers = {} @@ -2796,7 +2796,7 @@ def update( except (StreamConsumedError, StreamClosedError): pass map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if _stream: @@ -2893,7 +2893,7 @@ def get_next(next_link=None): if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = _deserialize(_models.ErrorResponse, response.json()) + error = _failsafe_deserialize(_models.ErrorResponse, response.json()) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) return pipeline_response diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py new file mode 100644 index 00000000000..fe8753d01f3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + resources_subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + resources_tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + resources_client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + resources_client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=resources_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=resources_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=resources_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=resources_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py new file mode 100644 index 00000000000..b60fadc33d8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations.py @@ -0,0 +1,106 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.resources import ResourcesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResourcesNestedOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ResourcesClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_nested_get(self, resource_group): + response = self.client.nested.get( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_nested_begin_create_or_replace(self, resource_group): + response = self.client.nested.begin_create_or_replace( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + resource={ + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_nested_begin_update(self, resource_group): + response = self.client.nested.begin_update( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + properties={ + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_nested_begin_delete(self, resource_group): + response = self.client.nested.begin_delete( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_nested_list_by_top_level_tracked_resource(self, resource_group): + response = self.client.nested.list_by_top_level_tracked_resource( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py new file mode 100644 index 00000000000..bb86a6ff911 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_nested_operations_async.py @@ -0,0 +1,113 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.resources.aio import ResourcesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResourcesNestedOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ResourcesClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_nested_get(self, resource_group): + response = await self.client.nested.get( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_nested_begin_create_or_replace(self, resource_group): + response = await ( + await self.client.nested.begin_create_or_replace( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + resource={ + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_nested_begin_update(self, resource_group): + response = await ( + await self.client.nested.begin_update( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + properties={ + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_nested_begin_delete(self, resource_group): + response = await ( + await self.client.nested.begin_delete( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + nexted_proxy_resource_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_nested_list_by_top_level_tracked_resource(self, resource_group): + response = self.client.nested.list_by_top_level_tracked_resource( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py new file mode 100644 index 00000000000..bef8b7bbe35 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations.py @@ -0,0 +1,91 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.resources import ResourcesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResourcesSingletonOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ResourcesClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_singleton_get_by_resource_group(self, resource_group): + response = self.client.singleton.get_by_resource_group( + resource_group_name=resource_group.name, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_singleton_begin_create_or_update(self, resource_group): + response = self.client.singleton.begin_create_or_update( + resource_group_name=resource_group.name, + resource={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_singleton_update(self, resource_group): + response = self.client.singleton.update( + resource_group_name=resource_group.name, + properties={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_singleton_list_by_resource_group(self, resource_group): + response = self.client.singleton.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py new file mode 100644 index 00000000000..dc973f54204 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_singleton_operations_async.py @@ -0,0 +1,94 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.resources.aio import ResourcesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResourcesSingletonOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ResourcesClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_singleton_get_by_resource_group(self, resource_group): + response = await self.client.singleton.get_by_resource_group( + resource_group_name=resource_group.name, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_singleton_begin_create_or_update(self, resource_group): + response = await ( + await self.client.singleton.begin_create_or_update( + resource_group_name=resource_group.name, + resource={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_singleton_update(self, resource_group): + response = await self.client.singleton.update( + resource_group_name=resource_group.name, + properties={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_singleton_list_by_resource_group(self, resource_group): + response = self.client.singleton.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py new file mode 100644 index 00000000000..8c685fd5000 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations.py @@ -0,0 +1,125 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.resources import ResourcesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer, recorded_by_proxy + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResourcesTopLevelOperations(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ResourcesClient) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_top_level_get(self, resource_group): + response = self.client.top_level.get( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_top_level_begin_create_or_replace(self, resource_group): + response = self.client.top_level.begin_create_or_replace( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + resource={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_top_level_begin_update(self, resource_group): + response = self.client.top_level.begin_update( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + properties={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_top_level_begin_delete(self, resource_group): + response = self.client.top_level.begin_delete( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_top_level_list_by_resource_group(self, resource_group): + response = self.client.top_level.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_top_level_list_by_subscription(self, resource_group): + response = self.client.top_level.list_by_subscription() + result = [r for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy + def test_top_level_action_sync(self, resource_group): + response = self.client.top_level.action_sync( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + body={"message": "str", "urgent": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py new file mode 100644 index 00000000000..c3cec3db79a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-resource-manager-resources/generated_tests/test_resources_top_level_operations_async.py @@ -0,0 +1,132 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from azure.resourcemanager.resources.aio import ResourcesClient + +from devtools_testutils import AzureMgmtRecordedTestCase, RandomNameResourceGroupPreparer +from devtools_testutils.aio import recorded_by_proxy_async + +AZURE_LOCATION = "eastus" + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResourcesTopLevelOperationsAsync(AzureMgmtRecordedTestCase): + def setup_method(self, method): + self.client = self.create_mgmt_client(ResourcesClient, is_async=True) + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_top_level_get(self, resource_group): + response = await self.client.top_level.get( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + ) + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_top_level_begin_create_or_replace(self, resource_group): + response = await ( + await self.client.top_level.begin_create_or_replace( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + resource={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_top_level_begin_update(self, resource_group): + response = await ( + await self.client.top_level.begin_update( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + properties={ + "location": "str", + "id": "str", + "name": "str", + "properties": {"description": "str", "provisioningState": "str"}, + "systemData": { + "createdAt": "2020-02-20 00:00:00", + "createdBy": "str", + "createdByType": "str", + "lastModifiedAt": "2020-02-20 00:00:00", + "lastModifiedBy": "str", + "lastModifiedByType": "str", + }, + "tags": {"str": "str"}, + "type": "str", + }, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_top_level_begin_delete(self, resource_group): + response = await ( + await self.client.top_level.begin_delete( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_top_level_list_by_resource_group(self, resource_group): + response = self.client.top_level.list_by_resource_group( + resource_group_name=resource_group.name, + ) + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_top_level_list_by_subscription(self, resource_group): + response = self.client.top_level.list_by_subscription() + result = [r async for r in response] + # please add some check logic here by yourself + # ... + + @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) + @recorded_by_proxy_async + async def test_top_level_action_sync(self, resource_group): + response = await self.client.top_level.action_sync( + resource_group_name=resource_group.name, + top_level_tracked_resource_name="str", + body={"message": "str", "urgent": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/_model_base.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/azure/specialheaders/xmsclientrequestid/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py new file mode 100644 index 00000000000..81b0a14c303 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/conftest.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + xmsrequestid_subscription_id = os.environ.get( + "XMSREQUESTID_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + xmsrequestid_tenant_id = os.environ.get("XMSREQUESTID_TENANT_ID", "00000000-0000-0000-0000-000000000000") + xmsrequestid_client_id = os.environ.get("XMSREQUESTID_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + xmsrequestid_client_secret = os.environ.get("XMSREQUESTID_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xmsrequestid_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xmsrequestid_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xmsrequestid_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xmsrequestid_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py new file mode 100644 index 00000000000..adb49afe641 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmsClientRequestIdClientTestBase, XmsRequestIdPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmsRequestId(XmsClientRequestIdClientTestBase): + @XmsRequestIdPreparer() + @recorded_by_proxy + def test_get(self, xmsrequestid_endpoint): + client = self.create_client(endpoint=xmsrequestid_endpoint) + response = client.get() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py new file mode 100644 index 00000000000..8830328ad72 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/test_xms_request_id_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmsRequestIdPreparer +from testpreparer_async import XmsClientRequestIdClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmsRequestIdAsync(XmsClientRequestIdClientTestBaseAsync): + @XmsRequestIdPreparer() + @recorded_by_proxy_async + async def test_get(self, xmsrequestid_endpoint): + client = self.create_async_client(endpoint=xmsrequestid_endpoint) + response = await client.get() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py new file mode 100644 index 00000000000..975eab0459d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from azure.specialheaders.xmsclientrequestid import XmsClientRequestIdClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class XmsClientRequestIdClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(XmsClientRequestIdClient) + return self.create_client_from_credential( + XmsClientRequestIdClient, + credential=credential, + endpoint=endpoint, + ) + + +XmsRequestIdPreparer = functools.partial( + PowerShellPreparer, "xmsrequestid", xmsrequestid_endpoint="https://fake_xmsrequestid_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..df7d540a17b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azure-special-headers-client-request-id/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from azure.specialheaders.xmsclientrequestid.aio import XmsClientRequestIdClient +from devtools_testutils import AzureRecordedTestCase + + +class XmsClientRequestIdClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(XmsClientRequestIdClient, is_async=True) + return self.create_client_from_credential( + XmsClientRequestIdClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/azurecore-lro-rpc/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/_model_base.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/_model_base.py +++ b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/azurecore/lro/rpc/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py new file mode 100644 index 00000000000..066375380f8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + rpc_subscription_id = os.environ.get("RPC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + rpc_tenant_id = os.environ.get("RPC_TENANT_ID", "00000000-0000-0000-0000-000000000000") + rpc_client_id = os.environ.get("RPC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + rpc_client_secret = os.environ.get("RPC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=rpc_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=rpc_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=rpc_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=rpc_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py new file mode 100644 index 00000000000..1f3b18e28ab --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RpcClientTestBase, RpcPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRpc(RpcClientTestBase): + @RpcPreparer() + @recorded_by_proxy + def test_begin_long_running_rpc(self, rpc_endpoint): + client = self.create_client(endpoint=rpc_endpoint) + response = client.begin_long_running_rpc( + body={"prompt": "str"}, + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py new file mode 100644 index 00000000000..9d55e7fa603 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/test_rpc_async.py @@ -0,0 +1,27 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RpcPreparer +from testpreparer_async import RpcClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRpcAsync(RpcClientTestBaseAsync): + @RpcPreparer() + @recorded_by_proxy_async + async def test_begin_long_running_rpc(self, rpc_endpoint): + client = self.create_async_client(endpoint=rpc_endpoint) + response = await ( + await client.begin_long_running_rpc( + body={"prompt": "str"}, + ) + ).result() # call '.result()' to poll until service return final result + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py new file mode 100644 index 00000000000..b3da2928846 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from azurecore.lro.rpc import RpcClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class RpcClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(RpcClient) + return self.create_client_from_credential( + RpcClient, + credential=credential, + endpoint=endpoint, + ) + + +RpcPreparer = functools.partial(PowerShellPreparer, "rpc", rpc_endpoint="https://fake_rpc_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..c9b185a7228 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/azurecore-lro-rpc/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from azurecore.lro.rpc.aio import RpcClient +from devtools_testutils import AzureRecordedTestCase + + +class RpcClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(RpcClient, is_async=True) + return self.create_client_from_credential( + RpcClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/client-naming/client/naming/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-naming/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-naming/client/naming/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-naming/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-naming/client/naming/_model_base.py b/packages/typespec-python/test/azure/generated/client-naming/client/naming/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/client-naming/client/naming/_model_base.py +++ b/packages/typespec-python/test/azure/generated/client-naming/client/naming/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py new file mode 100644 index 00000000000..e1548f36b47 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + naming_subscription_id = os.environ.get("NAMING_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + naming_tenant_id = os.environ.get("NAMING_TENANT_ID", "00000000-0000-0000-0000-000000000000") + naming_client_id = os.environ.get("NAMING_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + naming_client_secret = os.environ.get("NAMING_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=naming_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=naming_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=naming_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=naming_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py new file mode 100644 index 00000000000..d61a3d6139f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming.py @@ -0,0 +1,86 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NamingClientTestBase, NamingPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNaming(NamingClientTestBase): + @NamingPreparer() + @recorded_by_proxy + def test_client_name(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.client_name() + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_parameter(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.parameter( + client_name="str", + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_client(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.client( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_language(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.language( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_compatible_with_encoded_name(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.compatible_with_encoded_name( + body={"wireName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_request(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.request( + client_name="str", + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_response(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.response() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py new file mode 100644 index 00000000000..8e359563300 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_async.py @@ -0,0 +1,87 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NamingPreparer +from testpreparer_async import NamingClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNamingAsync(NamingClientTestBaseAsync): + @NamingPreparer() + @recorded_by_proxy_async + async def test_client_name(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.client_name() + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_parameter(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.parameter( + client_name="str", + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_client(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.client( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_language(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.language( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_compatible_with_encoded_name(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.compatible_with_encoded_name( + body={"wireName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_request(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.request( + client_name="str", + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_response(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.response() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py new file mode 100644 index 00000000000..854de9bc234 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NamingClientTestBase, NamingPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNamingClientModelOperations(NamingClientTestBase): + @NamingPreparer() + @recorded_by_proxy + def test_client_model_client(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.client_model.client( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_client_model_language(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.client_model.language( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py new file mode 100644 index 00000000000..ef1c0109021 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_client_model_operations_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NamingPreparer +from testpreparer_async import NamingClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNamingClientModelOperationsAsync(NamingClientTestBaseAsync): + @NamingPreparer() + @recorded_by_proxy_async + async def test_client_model_client(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.client_model.client( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_client_model_language(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.client_model.language( + body={"defaultName": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py new file mode 100644 index 00000000000..45e18f9ddc8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NamingClientTestBase, NamingPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNamingUnionEnumOperations(NamingClientTestBase): + @NamingPreparer() + @recorded_by_proxy + def test_union_enum_union_enum_name(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.union_enum.union_enum_name( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy + def test_union_enum_union_enum_member_name(self, naming_endpoint): + client = self.create_client(endpoint=naming_endpoint) + response = client.union_enum.union_enum_member_name( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py new file mode 100644 index 00000000000..141e125b601 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/test_naming_union_enum_operations_async.py @@ -0,0 +1,38 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NamingPreparer +from testpreparer_async import NamingClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNamingUnionEnumOperationsAsync(NamingClientTestBaseAsync): + @NamingPreparer() + @recorded_by_proxy_async + async def test_union_enum_union_enum_name(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.union_enum.union_enum_name( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @NamingPreparer() + @recorded_by_proxy_async + async def test_union_enum_union_enum_member_name(self, naming_endpoint): + client = self.create_async_client(endpoint=naming_endpoint) + response = await client.union_enum.union_enum_member_name( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py new file mode 100644 index 00000000000..0b6e8178c61 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.naming import NamingClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class NamingClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(NamingClient) + return self.create_client_from_credential( + NamingClient, + credential=credential, + endpoint=endpoint, + ) + + +NamingPreparer = functools.partial(PowerShellPreparer, "naming", naming_endpoint="https://fake_naming_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..f306afdbed6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-naming/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.naming.aio import NamingClient +from devtools_testutils import AzureRecordedTestCase + + +class NamingClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(NamingClient, is_async=True) + return self.create_client_from_credential( + NamingClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-default/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-default/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/_model_base.py b/packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/_model_base.py +++ b/packages/typespec-python/test/azure/generated/client-structure-default/client/structure/service/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py new file mode 100644 index 00000000000..91256e0344e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + service_subscription_id = os.environ.get("SERVICE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + service_tenant_id = os.environ.get("SERVICE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + service_client_id = os.environ.get("SERVICE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + service_client_secret = os.environ.get("SERVICE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=service_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=service_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=service_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=service_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py new file mode 100644 index 00000000000..1e7ce2517d5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ServiceClientTestBase, ServicePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestService(ServiceClientTestBase): + @ServicePreparer() + @recorded_by_proxy + def test_one(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.one() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy + def test_two(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.two() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py new file mode 100644 index 00000000000..2947d4222c5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ServicePreparer +from testpreparer_async import ServiceClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceAsync(ServiceClientTestBaseAsync): + @ServicePreparer() + @recorded_by_proxy_async + async def test_one(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.one() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy_async + async def test_two(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.two() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py new file mode 100644 index 00000000000..ca4600e7c2c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ServiceClientTestBase, ServicePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceBarOperations(ServiceClientTestBase): + @ServicePreparer() + @recorded_by_proxy + def test_bar_five(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.bar.five() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy + def test_bar_six(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.bar.six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py new file mode 100644 index 00000000000..563b02c2ac8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_bar_operations_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ServicePreparer +from testpreparer_async import ServiceClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceBarOperationsAsync(ServiceClientTestBaseAsync): + @ServicePreparer() + @recorded_by_proxy_async + async def test_bar_five(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.bar.five() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy_async + async def test_bar_six(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.bar.six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py new file mode 100644 index 00000000000..25e68492cd5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ServiceClientTestBase, ServicePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceBazOperations(ServiceClientTestBase): + @ServicePreparer() + @recorded_by_proxy + def test_baz_foo_seven(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.baz.foo.seven() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py new file mode 100644 index 00000000000..6fc9529df7e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_baz_operations_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ServicePreparer +from testpreparer_async import ServiceClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceBazOperationsAsync(ServiceClientTestBaseAsync): + @ServicePreparer() + @recorded_by_proxy_async + async def test_baz_foo_seven(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.baz.foo.seven() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py new file mode 100644 index 00000000000..5a62fa3eed0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ServiceClientTestBase, ServicePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceFooOperations(ServiceClientTestBase): + @ServicePreparer() + @recorded_by_proxy + def test_foo_three(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.foo.three() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy + def test_foo_four(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.foo.four() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py new file mode 100644 index 00000000000..547fc58c746 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_foo_operations_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ServicePreparer +from testpreparer_async import ServiceClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceFooOperationsAsync(ServiceClientTestBaseAsync): + @ServicePreparer() + @recorded_by_proxy_async + async def test_foo_three(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.foo.three() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy_async + async def test_foo_four(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.foo.four() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py new file mode 100644 index 00000000000..19079ec2a92 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ServiceClientTestBase, ServicePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceQuxOperations(ServiceClientTestBase): + @ServicePreparer() + @recorded_by_proxy + def test_qux_eight(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.qux.eight() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy + def test_qux_bar_nine(self, service_endpoint): + client = self.create_client(endpoint=service_endpoint) + response = client.qux.bar.nine() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py new file mode 100644 index 00000000000..4eb28c55ad0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/test_service_qux_operations_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ServicePreparer +from testpreparer_async import ServiceClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestServiceQuxOperationsAsync(ServiceClientTestBaseAsync): + @ServicePreparer() + @recorded_by_proxy_async + async def test_qux_eight(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.qux.eight() + + # please add some check logic here by yourself + # ... + + @ServicePreparer() + @recorded_by_proxy_async + async def test_qux_bar_nine(self, service_endpoint): + client = self.create_async_client(endpoint=service_endpoint) + response = await client.qux.bar.nine() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py new file mode 100644 index 00000000000..957dd158a79 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.service import ServiceClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class ServiceClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ServiceClient) + return self.create_client_from_credential( + ServiceClient, + credential=credential, + endpoint=endpoint, + ) + + +ServicePreparer = functools.partial(PowerShellPreparer, "service", service_endpoint="https://fake_service_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..5ff957026f2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-default/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.service.aio import ServiceClient +from devtools_testutils import AzureRecordedTestCase + + +class ServiceClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ServiceClient, is_async=True) + return self.create_client_from_credential( + ServiceClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-multiclient/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-multiclient/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py new file mode 100644 index 00000000000..89cc9a005d6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/conftest.py @@ -0,0 +1,44 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + a_subscription_id = os.environ.get("A_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + a_tenant_id = os.environ.get("A_TENANT_ID", "00000000-0000-0000-0000-000000000000") + a_client_id = os.environ.get("A_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + a_client_secret = os.environ.get("A_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=a_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=a_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=a_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=a_client_secret, value="00000000-0000-0000-0000-000000000000") + + b_subscription_id = os.environ.get("B_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + b_tenant_id = os.environ.get("B_TENANT_ID", "00000000-0000-0000-0000-000000000000") + b_client_id = os.environ.get("B_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + b_client_secret = os.environ.get("B_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=b_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=b_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=b_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=b_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py new file mode 100644 index 00000000000..758a7aa3087 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_a.py @@ -0,0 +1,40 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import APreparer, ClientAClientTestBase + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestA(ClientAClientTestBase): + @APreparer() + @recorded_by_proxy + def test_renamed_one(self, a_endpoint): + client = self.create_client(endpoint=a_endpoint) + response = client.renamed_one() + + # please add some check logic here by yourself + # ... + + @APreparer() + @recorded_by_proxy + def test_renamed_three(self, a_endpoint): + client = self.create_client(endpoint=a_endpoint) + response = client.renamed_three() + + # please add some check logic here by yourself + # ... + + @APreparer() + @recorded_by_proxy + def test_renamed_five(self, a_endpoint): + client = self.create_client(endpoint=a_endpoint) + response = client.renamed_five() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py new file mode 100644 index 00000000000..350a49c78cd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_aasync.py @@ -0,0 +1,41 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import APreparer +from testpreparer_async import ClientAClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAAsync(ClientAClientTestBaseAsync): + @APreparer() + @recorded_by_proxy_async + async def test_renamed_one(self, a_endpoint): + client = self.create_async_client(endpoint=a_endpoint) + response = await client.renamed_one() + + # please add some check logic here by yourself + # ... + + @APreparer() + @recorded_by_proxy_async + async def test_renamed_three(self, a_endpoint): + client = self.create_async_client(endpoint=a_endpoint) + response = await client.renamed_three() + + # please add some check logic here by yourself + # ... + + @APreparer() + @recorded_by_proxy_async + async def test_renamed_five(self, a_endpoint): + client = self.create_async_client(endpoint=a_endpoint) + response = await client.renamed_five() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py new file mode 100644 index 00000000000..0c35b7bfba4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_b.py @@ -0,0 +1,40 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BPreparer, ClientBClientTestBase + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestB(ClientBClientTestBase): + @BPreparer() + @recorded_by_proxy + def test_renamed_two(self, b_endpoint): + client = self.create_client(endpoint=b_endpoint) + response = client.renamed_two() + + # please add some check logic here by yourself + # ... + + @BPreparer() + @recorded_by_proxy + def test_renamed_four(self, b_endpoint): + client = self.create_client(endpoint=b_endpoint) + response = client.renamed_four() + + # please add some check logic here by yourself + # ... + + @BPreparer() + @recorded_by_proxy + def test_renamed_six(self, b_endpoint): + client = self.create_client(endpoint=b_endpoint) + response = client.renamed_six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py new file mode 100644 index 00000000000..93e20b2c5f5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/test_basync.py @@ -0,0 +1,41 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BPreparer +from testpreparer_async import ClientBClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBAsync(ClientBClientTestBaseAsync): + @BPreparer() + @recorded_by_proxy_async + async def test_renamed_two(self, b_endpoint): + client = self.create_async_client(endpoint=b_endpoint) + response = await client.renamed_two() + + # please add some check logic here by yourself + # ... + + @BPreparer() + @recorded_by_proxy_async + async def test_renamed_four(self, b_endpoint): + client = self.create_async_client(endpoint=b_endpoint) + response = await client.renamed_four() + + # please add some check logic here by yourself + # ... + + @BPreparer() + @recorded_by_proxy_async + async def test_renamed_six(self, b_endpoint): + client = self.create_async_client(endpoint=b_endpoint) + response = await client.renamed_six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py new file mode 100644 index 00000000000..de2a68557e5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer.py @@ -0,0 +1,38 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.multiclient import ClientAClient, ClientBClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class ClientAClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ClientAClient) + return self.create_client_from_credential( + ClientAClient, + credential=credential, + endpoint=endpoint, + ) + + +APreparer = functools.partial(PowerShellPreparer, "a", a_endpoint="https://fake_a_endpoint.com") + + +class ClientBClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ClientBClient) + return self.create_client_from_credential( + ClientBClient, + credential=credential, + endpoint=endpoint, + ) + + +BPreparer = functools.partial(PowerShellPreparer, "b", b_endpoint="https://fake_b_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..c8d0f416c46 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-multiclient/generated_tests/testpreparer_async.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.multiclient.aio import ClientAClient, ClientBClient +from devtools_testutils import AzureRecordedTestCase + + +class ClientAClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ClientAClient, is_async=True) + return self.create_client_from_credential( + ClientAClient, + credential=credential, + endpoint=endpoint, + ) + + +class ClientBClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ClientBClient, is_async=True) + return self.create_client_from_credential( + ClientBClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-renamedoperation/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py new file mode 100644 index 00000000000..351cda6c0b8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + renamedoperation_subscription_id = os.environ.get( + "RENAMEDOPERATION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + renamedoperation_tenant_id = os.environ.get("RENAMEDOPERATION_TENANT_ID", "00000000-0000-0000-0000-000000000000") + renamedoperation_client_id = os.environ.get("RENAMEDOPERATION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + renamedoperation_client_secret = os.environ.get( + "RENAMEDOPERATION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=renamedoperation_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=renamedoperation_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=renamedoperation_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=renamedoperation_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py new file mode 100644 index 00000000000..e62c8f801d3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation.py @@ -0,0 +1,40 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RenamedOperationClientTestBase, RenamedOperationPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedOperation(RenamedOperationClientTestBase): + @RenamedOperationPreparer() + @recorded_by_proxy + def test_renamed_one(self, renamedoperation_endpoint): + client = self.create_client(endpoint=renamedoperation_endpoint) + response = client.renamed_one() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy + def test_renamed_three(self, renamedoperation_endpoint): + client = self.create_client(endpoint=renamedoperation_endpoint) + response = client.renamed_three() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy + def test_renamed_five(self, renamedoperation_endpoint): + client = self.create_client(endpoint=renamedoperation_endpoint) + response = client.renamed_five() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py new file mode 100644 index 00000000000..ce2bd7c8e4c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_async.py @@ -0,0 +1,41 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RenamedOperationPreparer +from testpreparer_async import RenamedOperationClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedOperationAsync(RenamedOperationClientTestBaseAsync): + @RenamedOperationPreparer() + @recorded_by_proxy_async + async def test_renamed_one(self, renamedoperation_endpoint): + client = self.create_async_client(endpoint=renamedoperation_endpoint) + response = await client.renamed_one() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy_async + async def test_renamed_three(self, renamedoperation_endpoint): + client = self.create_async_client(endpoint=renamedoperation_endpoint) + response = await client.renamed_three() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy_async + async def test_renamed_five(self, renamedoperation_endpoint): + client = self.create_async_client(endpoint=renamedoperation_endpoint) + response = await client.renamed_five() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py new file mode 100644 index 00000000000..bba0c3a932d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations.py @@ -0,0 +1,40 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RenamedOperationClientTestBase, RenamedOperationPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedOperationGroupOperations(RenamedOperationClientTestBase): + @RenamedOperationPreparer() + @recorded_by_proxy + def test_group_renamed_two(self, renamedoperation_endpoint): + client = self.create_client(endpoint=renamedoperation_endpoint) + response = client.group.renamed_two() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy + def test_group_renamed_four(self, renamedoperation_endpoint): + client = self.create_client(endpoint=renamedoperation_endpoint) + response = client.group.renamed_four() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy + def test_group_renamed_six(self, renamedoperation_endpoint): + client = self.create_client(endpoint=renamedoperation_endpoint) + response = client.group.renamed_six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py new file mode 100644 index 00000000000..5ce2eb78017 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/test_renamed_operation_group_operations_async.py @@ -0,0 +1,41 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RenamedOperationPreparer +from testpreparer_async import RenamedOperationClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedOperationGroupOperationsAsync(RenamedOperationClientTestBaseAsync): + @RenamedOperationPreparer() + @recorded_by_proxy_async + async def test_group_renamed_two(self, renamedoperation_endpoint): + client = self.create_async_client(endpoint=renamedoperation_endpoint) + response = await client.group.renamed_two() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy_async + async def test_group_renamed_four(self, renamedoperation_endpoint): + client = self.create_async_client(endpoint=renamedoperation_endpoint) + response = await client.group.renamed_four() + + # please add some check logic here by yourself + # ... + + @RenamedOperationPreparer() + @recorded_by_proxy_async + async def test_group_renamed_six(self, renamedoperation_endpoint): + client = self.create_async_client(endpoint=renamedoperation_endpoint) + response = await client.group.renamed_six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py new file mode 100644 index 00000000000..5a277b43b52 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.renamedoperation import RenamedOperationClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class RenamedOperationClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(RenamedOperationClient) + return self.create_client_from_credential( + RenamedOperationClient, + credential=credential, + endpoint=endpoint, + ) + + +RenamedOperationPreparer = functools.partial( + PowerShellPreparer, "renamedoperation", renamedoperation_endpoint="https://fake_renamedoperation_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..b44022452e6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-renamedoperation/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.renamedoperation.aio import RenamedOperationClient +from devtools_testutils import AzureRecordedTestCase + + +class RenamedOperationClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(RenamedOperationClient, is_async=True) + return self.create_client_from_credential( + RenamedOperationClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py new file mode 100644 index 00000000000..c7671c2b0f6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + twooperationgroup_subscription_id = os.environ.get( + "TWOOPERATIONGROUP_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + twooperationgroup_tenant_id = os.environ.get("TWOOPERATIONGROUP_TENANT_ID", "00000000-0000-0000-0000-000000000000") + twooperationgroup_client_id = os.environ.get("TWOOPERATIONGROUP_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + twooperationgroup_client_secret = os.environ.get( + "TWOOPERATIONGROUP_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=twooperationgroup_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=twooperationgroup_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=twooperationgroup_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=twooperationgroup_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py new file mode 100644 index 00000000000..c3c32ec56f0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations.py @@ -0,0 +1,40 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import TwoOperationGroupClientTestBase, TwoOperationGroupPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTwoOperationGroupGroup1Operations(TwoOperationGroupClientTestBase): + @TwoOperationGroupPreparer() + @recorded_by_proxy + def test_group1_one(self, twooperationgroup_endpoint): + client = self.create_client(endpoint=twooperationgroup_endpoint) + response = client.group1.one() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy + def test_group1_three(self, twooperationgroup_endpoint): + client = self.create_client(endpoint=twooperationgroup_endpoint) + response = client.group1.three() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy + def test_group1_four(self, twooperationgroup_endpoint): + client = self.create_client(endpoint=twooperationgroup_endpoint) + response = client.group1.four() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py new file mode 100644 index 00000000000..539bfde4da6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group1_operations_async.py @@ -0,0 +1,41 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import TwoOperationGroupPreparer +from testpreparer_async import TwoOperationGroupClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTwoOperationGroupGroup1OperationsAsync(TwoOperationGroupClientTestBaseAsync): + @TwoOperationGroupPreparer() + @recorded_by_proxy_async + async def test_group1_one(self, twooperationgroup_endpoint): + client = self.create_async_client(endpoint=twooperationgroup_endpoint) + response = await client.group1.one() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy_async + async def test_group1_three(self, twooperationgroup_endpoint): + client = self.create_async_client(endpoint=twooperationgroup_endpoint) + response = await client.group1.three() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy_async + async def test_group1_four(self, twooperationgroup_endpoint): + client = self.create_async_client(endpoint=twooperationgroup_endpoint) + response = await client.group1.four() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py new file mode 100644 index 00000000000..0e4194269bb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations.py @@ -0,0 +1,40 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import TwoOperationGroupClientTestBase, TwoOperationGroupPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTwoOperationGroupGroup2Operations(TwoOperationGroupClientTestBase): + @TwoOperationGroupPreparer() + @recorded_by_proxy + def test_group2_two(self, twooperationgroup_endpoint): + client = self.create_client(endpoint=twooperationgroup_endpoint) + response = client.group2.two() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy + def test_group2_five(self, twooperationgroup_endpoint): + client = self.create_client(endpoint=twooperationgroup_endpoint) + response = client.group2.five() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy + def test_group2_six(self, twooperationgroup_endpoint): + client = self.create_client(endpoint=twooperationgroup_endpoint) + response = client.group2.six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py new file mode 100644 index 00000000000..2097a90a553 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/test_two_operation_group_group2_operations_async.py @@ -0,0 +1,41 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import TwoOperationGroupPreparer +from testpreparer_async import TwoOperationGroupClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTwoOperationGroupGroup2OperationsAsync(TwoOperationGroupClientTestBaseAsync): + @TwoOperationGroupPreparer() + @recorded_by_proxy_async + async def test_group2_two(self, twooperationgroup_endpoint): + client = self.create_async_client(endpoint=twooperationgroup_endpoint) + response = await client.group2.two() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy_async + async def test_group2_five(self, twooperationgroup_endpoint): + client = self.create_async_client(endpoint=twooperationgroup_endpoint) + response = await client.group2.five() + + # please add some check logic here by yourself + # ... + + @TwoOperationGroupPreparer() + @recorded_by_proxy_async + async def test_group2_six(self, twooperationgroup_endpoint): + client = self.create_async_client(endpoint=twooperationgroup_endpoint) + response = await client.group2.six() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py new file mode 100644 index 00000000000..a015a7808f7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.twooperationgroup import TwoOperationGroupClient +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools + + +class TwoOperationGroupClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(TwoOperationGroupClient) + return self.create_client_from_credential( + TwoOperationGroupClient, + credential=credential, + endpoint=endpoint, + ) + + +TwoOperationGroupPreparer = functools.partial( + PowerShellPreparer, "twooperationgroup", twooperationgroup_endpoint="https://fake_twooperationgroup_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..4f86f927c06 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/client-structure-twooperationgroup/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from client.structure.twooperationgroup.aio import TwoOperationGroupClient +from devtools_testutils import AzureRecordedTestCase + + +class TwoOperationGroupClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(TwoOperationGroupClient, is_async=True) + return self.create_client_from_credential( + TwoOperationGroupClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-bytes/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-bytes/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/_model_base.py b/packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/_model_base.py +++ b/packages/typespec-python/test/azure/generated/encode-bytes/encode/bytes/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py new file mode 100644 index 00000000000..9238da04a06 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + bytes_subscription_id = os.environ.get("BYTES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + bytes_tenant_id = os.environ.get("BYTES_TENANT_ID", "00000000-0000-0000-0000-000000000000") + bytes_client_id = os.environ.get("BYTES_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + bytes_client_secret = os.environ.get("BYTES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=bytes_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=bytes_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=bytes_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=bytes_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py new file mode 100644 index 00000000000..616fd7ecfa3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations.py @@ -0,0 +1,57 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BytesClientTestBase, BytesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesHeaderOperations(BytesClientTestBase): + @BytesPreparer() + @recorded_by_proxy + def test_header_default(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.header.default( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_header_base64(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.header.base64( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_header_base64_url(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.header.base64_url( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_header_base64_url_array(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.header.base64_url_array( + value=[bytes("bytes", encoding="utf-8")], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py new file mode 100644 index 00000000000..8a957f4dbfc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_header_operations_async.py @@ -0,0 +1,58 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BytesPreparer +from testpreparer_async import BytesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesHeaderOperationsAsync(BytesClientTestBaseAsync): + @BytesPreparer() + @recorded_by_proxy_async + async def test_header_default(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.header.default( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_header_base64(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.header.base64( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_header_base64_url(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.header.base64_url( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_header_base64_url_array(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.header.base64_url_array( + value=[bytes("bytes", encoding="utf-8")], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py new file mode 100644 index 00000000000..a797ba0131d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations.py @@ -0,0 +1,57 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BytesClientTestBase, BytesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesPropertyOperations(BytesClientTestBase): + @BytesPreparer() + @recorded_by_proxy + def test_property_default(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.property.default( + body={"value": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_property_base64(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.property.base64( + body={"value": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_property_base64_url(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.property.base64_url( + body={"value": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_property_base64_url_array(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.property.base64_url_array( + body={"value": [bytes("bytes", encoding="utf-8")]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py new file mode 100644 index 00000000000..fd1c0f97a12 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_property_operations_async.py @@ -0,0 +1,58 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BytesPreparer +from testpreparer_async import BytesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesPropertyOperationsAsync(BytesClientTestBaseAsync): + @BytesPreparer() + @recorded_by_proxy_async + async def test_property_default(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.property.default( + body={"value": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_property_base64(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.property.base64( + body={"value": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_property_base64_url(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.property.base64_url( + body={"value": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_property_base64_url_array(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.property.base64_url_array( + body={"value": [bytes("bytes", encoding="utf-8")]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py new file mode 100644 index 00000000000..49f4b3ff3db --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations.py @@ -0,0 +1,57 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BytesClientTestBase, BytesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesQueryOperations(BytesClientTestBase): + @BytesPreparer() + @recorded_by_proxy + def test_query_default(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.query.default( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_query_base64(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.query.base64( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_query_base64_url(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.query.base64_url( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_query_base64_url_array(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.query.base64_url_array( + value=[bytes("bytes", encoding="utf-8")], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py new file mode 100644 index 00000000000..b108afe5b12 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_query_operations_async.py @@ -0,0 +1,58 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BytesPreparer +from testpreparer_async import BytesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesQueryOperationsAsync(BytesClientTestBaseAsync): + @BytesPreparer() + @recorded_by_proxy_async + async def test_query_default(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.query.default( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_query_base64(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.query.base64( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_query_base64_url(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.query.base64_url( + value=bytes("bytes", encoding="utf-8"), + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_query_base64_url_array(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.query.base64_url_array( + value=[bytes("bytes", encoding="utf-8")], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py new file mode 100644 index 00000000000..1d083b7c7b3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations.py @@ -0,0 +1,73 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BytesClientTestBase, BytesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesRequestBodyOperations(BytesClientTestBase): + @BytesPreparer() + @recorded_by_proxy + def test_request_body_default(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.request_body.default( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_request_body_octet_stream(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.request_body.octet_stream( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_request_body_custom_content_type(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.request_body.custom_content_type( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_request_body_base64(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.request_body.base64( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_request_body_base64_url(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.request_body.base64_url( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py new file mode 100644 index 00000000000..a02a0e071e2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_request_body_operations_async.py @@ -0,0 +1,74 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BytesPreparer +from testpreparer_async import BytesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesRequestBodyOperationsAsync(BytesClientTestBaseAsync): + @BytesPreparer() + @recorded_by_proxy_async + async def test_request_body_default(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.request_body.default( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_request_body_octet_stream(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.request_body.octet_stream( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_request_body_custom_content_type(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.request_body.custom_content_type( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_request_body_base64(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.request_body.base64( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_request_body_base64_url(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.request_body.base64_url( + value=bytes("bytes", encoding="utf-8"), + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py new file mode 100644 index 00000000000..1bb3aafaef9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations.py @@ -0,0 +1,58 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BytesClientTestBase, BytesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesResponseBodyOperations(BytesClientTestBase): + @BytesPreparer() + @recorded_by_proxy + def test_response_body_default(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.response_body.default() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_response_body_octet_stream(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.response_body.octet_stream() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_response_body_custom_content_type(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.response_body.custom_content_type() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_response_body_base64(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.response_body.base64() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy + def test_response_body_base64_url(self, bytes_endpoint): + client = self.create_client(endpoint=bytes_endpoint) + response = client.response_body.base64_url() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py new file mode 100644 index 00000000000..8e702b41e39 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/test_bytes_response_body_operations_async.py @@ -0,0 +1,59 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BytesPreparer +from testpreparer_async import BytesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBytesResponseBodyOperationsAsync(BytesClientTestBaseAsync): + @BytesPreparer() + @recorded_by_proxy_async + async def test_response_body_default(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.response_body.default() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_response_body_octet_stream(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.response_body.octet_stream() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_response_body_custom_content_type(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.response_body.custom_content_type() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_response_body_base64(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.response_body.base64() + + # please add some check logic here by yourself + # ... + + @BytesPreparer() + @recorded_by_proxy_async + async def test_response_body_base64_url(self, bytes_endpoint): + client = self.create_async_client(endpoint=bytes_endpoint) + response = await client.response_body.base64_url() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py new file mode 100644 index 00000000000..0885f387943 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +from encode.bytes import BytesClient +import functools + + +class BytesClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(BytesClient) + return self.create_client_from_credential( + BytesClient, + credential=credential, + endpoint=endpoint, + ) + + +BytesPreparer = functools.partial(PowerShellPreparer, "bytes", bytes_endpoint="https://fake_bytes_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..8d0f600ec98 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-bytes/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from encode.bytes.aio import BytesClient + + +class BytesClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(BytesClient, is_async=True) + return self.create_client_from_credential( + BytesClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-datetime/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-datetime/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/_model_base.py b/packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/_model_base.py +++ b/packages/typespec-python/test/azure/generated/encode-datetime/encode/datetime/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py new file mode 100644 index 00000000000..f3723e069e8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + datetime_subscription_id = os.environ.get("DATETIME_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + datetime_tenant_id = os.environ.get("DATETIME_TENANT_ID", "00000000-0000-0000-0000-000000000000") + datetime_client_id = os.environ.get("DATETIME_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + datetime_client_secret = os.environ.get("DATETIME_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=datetime_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=datetime_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=datetime_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=datetime_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py new file mode 100644 index 00000000000..db1045ed06e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations.py @@ -0,0 +1,68 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DatetimeClientTestBase, DatetimePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimeHeaderOperations(DatetimeClientTestBase): + @DatetimePreparer() + @recorded_by_proxy + def test_header_default(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.header.default( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_header_rfc3339(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.header.rfc3339( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_header_rfc7231(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.header.rfc7231( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_header_unix_timestamp(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.header.unix_timestamp( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_header_unix_timestamp_array(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.header.unix_timestamp_array( + value=["2020-02-20 00:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py new file mode 100644 index 00000000000..bdf46c102fb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_header_operations_async.py @@ -0,0 +1,69 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DatetimePreparer +from testpreparer_async import DatetimeClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimeHeaderOperationsAsync(DatetimeClientTestBaseAsync): + @DatetimePreparer() + @recorded_by_proxy_async + async def test_header_default(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.header.default( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_header_rfc3339(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.header.rfc3339( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_header_rfc7231(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.header.rfc7231( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_header_unix_timestamp(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.header.unix_timestamp( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_header_unix_timestamp_array(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.header.unix_timestamp_array( + value=["2020-02-20 00:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py new file mode 100644 index 00000000000..53447582c24 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations.py @@ -0,0 +1,68 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DatetimeClientTestBase, DatetimePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimePropertyOperations(DatetimeClientTestBase): + @DatetimePreparer() + @recorded_by_proxy + def test_property_default(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.property.default( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_property_rfc3339(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.property.rfc3339( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_property_rfc7231(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.property.rfc7231( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_property_unix_timestamp(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.property.unix_timestamp( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_property_unix_timestamp_array(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.property.unix_timestamp_array( + body={"value": ["2020-02-20 00:00:00"]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py new file mode 100644 index 00000000000..c7062cd1d6f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_property_operations_async.py @@ -0,0 +1,69 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DatetimePreparer +from testpreparer_async import DatetimeClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimePropertyOperationsAsync(DatetimeClientTestBaseAsync): + @DatetimePreparer() + @recorded_by_proxy_async + async def test_property_default(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.property.default( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_property_rfc3339(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.property.rfc3339( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_property_rfc7231(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.property.rfc7231( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_property_unix_timestamp(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.property.unix_timestamp( + body={"value": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_property_unix_timestamp_array(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.property.unix_timestamp_array( + body={"value": ["2020-02-20 00:00:00"]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py new file mode 100644 index 00000000000..43419883b18 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations.py @@ -0,0 +1,68 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DatetimeClientTestBase, DatetimePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimeQueryOperations(DatetimeClientTestBase): + @DatetimePreparer() + @recorded_by_proxy + def test_query_default(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.query.default( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_query_rfc3339(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.query.rfc3339( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_query_rfc7231(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.query.rfc7231( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_query_unix_timestamp(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.query.unix_timestamp( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_query_unix_timestamp_array(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.query.unix_timestamp_array( + value=["2020-02-20 00:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py new file mode 100644 index 00000000000..d2ea13edbfd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_query_operations_async.py @@ -0,0 +1,69 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DatetimePreparer +from testpreparer_async import DatetimeClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimeQueryOperationsAsync(DatetimeClientTestBaseAsync): + @DatetimePreparer() + @recorded_by_proxy_async + async def test_query_default(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.query.default( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_query_rfc3339(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.query.rfc3339( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_query_rfc7231(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.query.rfc7231( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_query_unix_timestamp(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.query.unix_timestamp( + value="2020-02-20 00:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_query_unix_timestamp_array(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.query.unix_timestamp_array( + value=["2020-02-20 00:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py new file mode 100644 index 00000000000..7facfe78744 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations.py @@ -0,0 +1,49 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DatetimeClientTestBase, DatetimePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimeResponseHeaderOperations(DatetimeClientTestBase): + @DatetimePreparer() + @recorded_by_proxy + def test_response_header_default(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.response_header.default() + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_response_header_rfc3339(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.response_header.rfc3339() + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_response_header_rfc7231(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.response_header.rfc7231() + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy + def test_response_header_unix_timestamp(self, datetime_endpoint): + client = self.create_client(endpoint=datetime_endpoint) + response = client.response_header.unix_timestamp() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py new file mode 100644 index 00000000000..304b646609e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/test_datetime_response_header_operations_async.py @@ -0,0 +1,50 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DatetimePreparer +from testpreparer_async import DatetimeClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDatetimeResponseHeaderOperationsAsync(DatetimeClientTestBaseAsync): + @DatetimePreparer() + @recorded_by_proxy_async + async def test_response_header_default(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.response_header.default() + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_response_header_rfc3339(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.response_header.rfc3339() + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_response_header_rfc7231(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.response_header.rfc7231() + + # please add some check logic here by yourself + # ... + + @DatetimePreparer() + @recorded_by_proxy_async + async def test_response_header_unix_timestamp(self, datetime_endpoint): + client = self.create_async_client(endpoint=datetime_endpoint) + response = await client.response_header.unix_timestamp() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py new file mode 100644 index 00000000000..b05b3815661 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +from encode.datetime import DatetimeClient +import functools + + +class DatetimeClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(DatetimeClient) + return self.create_client_from_credential( + DatetimeClient, + credential=credential, + endpoint=endpoint, + ) + + +DatetimePreparer = functools.partial( + PowerShellPreparer, "datetime", datetime_endpoint="https://fake_datetime_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..fa3fcd42461 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-datetime/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from encode.datetime.aio import DatetimeClient + + +class DatetimeClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(DatetimeClient, is_async=True) + return self.create_client_from_credential( + DatetimeClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/encode-duration/encode/duration/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-duration/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-duration/encode/duration/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-duration/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-duration/encode/duration/_model_base.py b/packages/typespec-python/test/azure/generated/encode-duration/encode/duration/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/encode-duration/encode/duration/_model_base.py +++ b/packages/typespec-python/test/azure/generated/encode-duration/encode/duration/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py new file mode 100644 index 00000000000..7c8e921ac05 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + duration_subscription_id = os.environ.get("DURATION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + duration_tenant_id = os.environ.get("DURATION_TENANT_ID", "00000000-0000-0000-0000-000000000000") + duration_client_id = os.environ.get("DURATION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + duration_client_secret = os.environ.get("DURATION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=duration_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=duration_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=duration_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=duration_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py new file mode 100644 index 00000000000..f3bf725bed1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations.py @@ -0,0 +1,79 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DurationClientTestBase, DurationPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDurationHeaderOperations(DurationClientTestBase): + @DurationPreparer() + @recorded_by_proxy + def test_header_default(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.header.default( + duration="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_header_iso8601(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.header.iso8601( + duration="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_header_iso8601_array(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.header.iso8601_array( + duration=["1 day, 0:00:00"], + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_header_int32_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.header.int32_seconds( + duration=0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_header_float_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.header.float_seconds( + duration=0.0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_header_float64_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.header.float64_seconds( + duration=0.0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py new file mode 100644 index 00000000000..8437b4e599c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_header_operations_async.py @@ -0,0 +1,80 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DurationPreparer +from testpreparer_async import DurationClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDurationHeaderOperationsAsync(DurationClientTestBaseAsync): + @DurationPreparer() + @recorded_by_proxy_async + async def test_header_default(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.header.default( + duration="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_header_iso8601(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.header.iso8601( + duration="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_header_iso8601_array(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.header.iso8601_array( + duration=["1 day, 0:00:00"], + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_header_int32_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.header.int32_seconds( + duration=0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_header_float_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.header.float_seconds( + duration=0.0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_header_float64_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.header.float64_seconds( + duration=0.0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py new file mode 100644 index 00000000000..696c38e6897 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations.py @@ -0,0 +1,79 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DurationClientTestBase, DurationPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDurationPropertyOperations(DurationClientTestBase): + @DurationPreparer() + @recorded_by_proxy + def test_property_default(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.property.default( + body={"value": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_property_iso8601(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.property.iso8601( + body={"value": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_property_int32_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.property.int32_seconds( + body={"value": 0}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_property_float_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.property.float_seconds( + body={"value": 0.0}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_property_float64_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.property.float64_seconds( + body={"value": 0.0}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_property_float_seconds_array(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.property.float_seconds_array( + body={"value": [0.0]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py new file mode 100644 index 00000000000..8b6db79b5d8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_property_operations_async.py @@ -0,0 +1,80 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DurationPreparer +from testpreparer_async import DurationClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDurationPropertyOperationsAsync(DurationClientTestBaseAsync): + @DurationPreparer() + @recorded_by_proxy_async + async def test_property_default(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.property.default( + body={"value": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_property_iso8601(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.property.iso8601( + body={"value": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_property_int32_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.property.int32_seconds( + body={"value": 0}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_property_float_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.property.float_seconds( + body={"value": 0.0}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_property_float64_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.property.float64_seconds( + body={"value": 0.0}, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_property_float_seconds_array(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.property.float_seconds_array( + body={"value": [0.0]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py new file mode 100644 index 00000000000..f01cfb4019e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations.py @@ -0,0 +1,79 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DurationClientTestBase, DurationPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDurationQueryOperations(DurationClientTestBase): + @DurationPreparer() + @recorded_by_proxy + def test_query_default(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.query.default( + input="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_query_iso8601(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.query.iso8601( + input="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_query_int32_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.query.int32_seconds( + input=0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_query_float_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.query.float_seconds( + input=0.0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_query_float64_seconds(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.query.float64_seconds( + input=0.0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy + def test_query_int32_seconds_array(self, duration_endpoint): + client = self.create_client(endpoint=duration_endpoint) + response = client.query.int32_seconds_array( + input=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py new file mode 100644 index 00000000000..ce3da7ce8e0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/test_duration_query_operations_async.py @@ -0,0 +1,80 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DurationPreparer +from testpreparer_async import DurationClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDurationQueryOperationsAsync(DurationClientTestBaseAsync): + @DurationPreparer() + @recorded_by_proxy_async + async def test_query_default(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.query.default( + input="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_query_iso8601(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.query.iso8601( + input="1 day, 0:00:00", + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_query_int32_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.query.int32_seconds( + input=0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_query_float_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.query.float_seconds( + input=0.0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_query_float64_seconds(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.query.float64_seconds( + input=0.0, + ) + + # please add some check logic here by yourself + # ... + + @DurationPreparer() + @recorded_by_proxy_async + async def test_query_int32_seconds_array(self, duration_endpoint): + client = self.create_async_client(endpoint=duration_endpoint) + response = await client.query.int32_seconds_array( + input=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py new file mode 100644 index 00000000000..92daf94d4fc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +from encode.duration import DurationClient +import functools + + +class DurationClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(DurationClient) + return self.create_client_from_credential( + DurationClient, + credential=credential, + endpoint=endpoint, + ) + + +DurationPreparer = functools.partial( + PowerShellPreparer, "duration", duration_endpoint="https://fake_duration_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..9024e6891fc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-duration/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from encode.duration.aio import DurationClient + + +class DurationClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(DurationClient, is_async=True) + return self.create_client_from_credential( + DurationClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/encode-numeric/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/encode-numeric/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/_model_base.py b/packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/_model_base.py +++ b/packages/typespec-python/test/azure/generated/encode-numeric/encode/numeric/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py new file mode 100644 index 00000000000..4d23d2b7057 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + numeric_subscription_id = os.environ.get("NUMERIC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + numeric_tenant_id = os.environ.get("NUMERIC_TENANT_ID", "00000000-0000-0000-0000-000000000000") + numeric_client_id = os.environ.get("NUMERIC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + numeric_client_secret = os.environ.get("NUMERIC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=numeric_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=numeric_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=numeric_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=numeric_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py new file mode 100644 index 00000000000..fffcf2d5f4e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations.py @@ -0,0 +1,46 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NumericClientTestBase, NumericPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNumericPropertyOperations(NumericClientTestBase): + @NumericPreparer() + @recorded_by_proxy + def test_property_safeint_as_string(self, numeric_endpoint): + client = self.create_client(endpoint=numeric_endpoint) + response = client.property.safeint_as_string( + value={"value": 0}, + ) + + # please add some check logic here by yourself + # ... + + @NumericPreparer() + @recorded_by_proxy + def test_property_uint32_as_string_optional(self, numeric_endpoint): + client = self.create_client(endpoint=numeric_endpoint) + response = client.property.uint32_as_string_optional( + value={"value": 0}, + ) + + # please add some check logic here by yourself + # ... + + @NumericPreparer() + @recorded_by_proxy + def test_property_uint8_as_string(self, numeric_endpoint): + client = self.create_client(endpoint=numeric_endpoint) + response = client.property.uint8_as_string( + value={"value": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py new file mode 100644 index 00000000000..8de4f2ad73e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/test_numeric_property_operations_async.py @@ -0,0 +1,47 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NumericPreparer +from testpreparer_async import NumericClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNumericPropertyOperationsAsync(NumericClientTestBaseAsync): + @NumericPreparer() + @recorded_by_proxy_async + async def test_property_safeint_as_string(self, numeric_endpoint): + client = self.create_async_client(endpoint=numeric_endpoint) + response = await client.property.safeint_as_string( + value={"value": 0}, + ) + + # please add some check logic here by yourself + # ... + + @NumericPreparer() + @recorded_by_proxy_async + async def test_property_uint32_as_string_optional(self, numeric_endpoint): + client = self.create_async_client(endpoint=numeric_endpoint) + response = await client.property.uint32_as_string_optional( + value={"value": 0}, + ) + + # please add some check logic here by yourself + # ... + + @NumericPreparer() + @recorded_by_proxy_async + async def test_property_uint8_as_string(self, numeric_endpoint): + client = self.create_async_client(endpoint=numeric_endpoint) + response = await client.property.uint8_as_string( + value={"value": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py new file mode 100644 index 00000000000..f69f6a84bd9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +from encode.numeric import NumericClient +import functools + + +class NumericClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(NumericClient) + return self.create_client_from_credential( + NumericClient, + credential=credential, + endpoint=endpoint, + ) + + +NumericPreparer = functools.partial(PowerShellPreparer, "numeric", numeric_endpoint="https://fake_numeric_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..1f2469a6268 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/encode-numeric/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from encode.numeric.aio import NumericClient + + +class NumericClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(NumericClient, is_async=True) + return self.create_client_from_credential( + NumericClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/headasbooleanfalse/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/headasbooleanfalse/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py new file mode 100644 index 00000000000..3fc4942cf79 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + visibility_subscription_id = os.environ.get("VISIBILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + visibility_tenant_id = os.environ.get("VISIBILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + visibility_client_id = os.environ.get("VISIBILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + visibility_client_secret = os.environ.get("VISIBILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py new file mode 100644 index 00000000000..7ed8cd7db1d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility.py @@ -0,0 +1,90 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import VisibilityClientTestBase, VisibilityPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVisibility(VisibilityClientTestBase): + @VisibilityPreparer() + @recorded_by_proxy + def test_get_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.get_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_head_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.head_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_put_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.put_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_patch_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.patch_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_post_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.post_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_delete_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.delete_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_put_read_only_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.put_read_only_model( + input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py new file mode 100644 index 00000000000..9909a3a67eb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/test_visibility_async.py @@ -0,0 +1,91 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import VisibilityPreparer +from testpreparer_async import VisibilityClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVisibilityAsync(VisibilityClientTestBaseAsync): + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_get_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.get_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_head_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.head_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_put_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.put_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_patch_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.patch_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_post_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.post_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_delete_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.delete_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_put_read_only_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.put_read_only_model( + input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py new file mode 100644 index 00000000000..877033c71e3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from headasbooleanfalse import VisibilityClient + + +class VisibilityClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(VisibilityClient) + return self.create_client_from_credential( + VisibilityClient, + credential=credential, + endpoint=endpoint, + ) + + +VisibilityPreparer = functools.partial( + PowerShellPreparer, "visibility", visibility_endpoint="https://fake_visibility_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..2d82231904d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleanfalse/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from headasbooleanfalse.aio import VisibilityClient + + +class VisibilityClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(VisibilityClient, is_async=True) + return self.create_client_from_credential( + VisibilityClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py b/packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py +++ b/packages/typespec-python/test/azure/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/headasbooleantrue/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/headasbooleantrue/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py new file mode 100644 index 00000000000..3fc4942cf79 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + visibility_subscription_id = os.environ.get("VISIBILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + visibility_tenant_id = os.environ.get("VISIBILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + visibility_client_id = os.environ.get("VISIBILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + visibility_client_secret = os.environ.get("VISIBILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py new file mode 100644 index 00000000000..7ed8cd7db1d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility.py @@ -0,0 +1,90 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import VisibilityClientTestBase, VisibilityPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVisibility(VisibilityClientTestBase): + @VisibilityPreparer() + @recorded_by_proxy + def test_get_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.get_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_head_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.head_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_put_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.put_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_patch_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.patch_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_post_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.post_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_delete_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.delete_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_put_read_only_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.put_read_only_model( + input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py new file mode 100644 index 00000000000..9909a3a67eb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/test_visibility_async.py @@ -0,0 +1,91 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import VisibilityPreparer +from testpreparer_async import VisibilityClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVisibilityAsync(VisibilityClientTestBaseAsync): + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_get_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.get_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_head_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.head_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_put_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.put_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_patch_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.patch_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_post_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.post_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_delete_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.delete_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_put_read_only_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.put_read_only_model( + input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py new file mode 100644 index 00000000000..370f4302de3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from headasbooleantrue import VisibilityClient + + +class VisibilityClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(VisibilityClient) + return self.create_client_from_credential( + VisibilityClient, + credential=credential, + endpoint=endpoint, + ) + + +VisibilityPreparer = functools.partial( + PowerShellPreparer, "visibility", visibility_endpoint="https://fake_visibility_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..816ff818734 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/headasbooleantrue/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from headasbooleantrue.aio import VisibilityClient + + +class VisibilityClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(VisibilityClient, is_async=True) + return self.create_client_from_credential( + VisibilityClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/_model_base.py b/packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/_model_base.py +++ b/packages/typespec-python/test/azure/generated/headasbooleantrue/headasbooleantrue/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-basic/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-basic/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py new file mode 100644 index 00000000000..635e5035817 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + basic_subscription_id = os.environ.get("BASIC_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + basic_tenant_id = os.environ.get("BASIC_TENANT_ID", "00000000-0000-0000-0000-000000000000") + basic_client_id = os.environ.get("BASIC_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + basic_client_secret = os.environ.get("BASIC_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=basic_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py new file mode 100644 index 00000000000..a764b421795 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BasicClientTestBase, BasicPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBasicExplicitBodyOperations(BasicClientTestBase): + @BasicPreparer() + @recorded_by_proxy + def test_explicit_body_simple(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.explicit_body.simple( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py new file mode 100644 index 00000000000..bef8dd97225 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_explicit_body_operations_async.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BasicPreparer +from testpreparer_async import BasicClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBasicExplicitBodyOperationsAsync(BasicClientTestBaseAsync): + @BasicPreparer() + @recorded_by_proxy_async + async def test_explicit_body_simple(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.explicit_body.simple( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py new file mode 100644 index 00000000000..525c35f6dd0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BasicClientTestBase, BasicPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBasicImplicitBodyOperations(BasicClientTestBase): + @BasicPreparer() + @recorded_by_proxy + def test_implicit_body_simple(self, basic_endpoint): + client = self.create_client(endpoint=basic_endpoint) + response = client.implicit_body.simple( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py new file mode 100644 index 00000000000..83322a2b281 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/test_basic_implicit_body_operations_async.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BasicPreparer +from testpreparer_async import BasicClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBasicImplicitBodyOperationsAsync(BasicClientTestBaseAsync): + @BasicPreparer() + @recorded_by_proxy_async + async def test_implicit_body_simple(self, basic_endpoint): + client = self.create_async_client(endpoint=basic_endpoint) + response = await client.implicit_body.simple( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py new file mode 100644 index 00000000000..7b6afef93f7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from parameters.basic import BasicClient + + +class BasicClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(BasicClient) + return self.create_client_from_credential( + BasicClient, + credential=credential, + endpoint=endpoint, + ) + + +BasicPreparer = functools.partial(PowerShellPreparer, "basic", basic_endpoint="https://fake_basic_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..24f41a78773 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-basic/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from parameters.basic.aio import BasicClient + + +class BasicClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(BasicClient, is_async=True) + return self.create_client_from_credential( + BasicClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/_model_base.py b/packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/_model_base.py +++ b/packages/typespec-python/test/azure/generated/parameters-basic/parameters/basic/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-body-optionality/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-body-optionality/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py new file mode 100644 index 00000000000..a125385319e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + bodyoptionality_subscription_id = os.environ.get( + "BODYOPTIONALITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + bodyoptionality_tenant_id = os.environ.get("BODYOPTIONALITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + bodyoptionality_client_id = os.environ.get("BODYOPTIONALITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + bodyoptionality_client_secret = os.environ.get( + "BODYOPTIONALITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=bodyoptionality_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=bodyoptionality_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=bodyoptionality_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=bodyoptionality_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py new file mode 100644 index 00000000000..ef56ba5b5a8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BodyOptionalityClientTestBase, BodyOptionalityPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBodyOptionality(BodyOptionalityClientTestBase): + @BodyOptionalityPreparer() + @recorded_by_proxy + def test_required_explicit(self, bodyoptionality_endpoint): + client = self.create_client(endpoint=bodyoptionality_endpoint) + response = client.required_explicit( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @BodyOptionalityPreparer() + @recorded_by_proxy + def test_required_implicit(self, bodyoptionality_endpoint): + client = self.create_client(endpoint=bodyoptionality_endpoint) + response = client.required_implicit( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py new file mode 100644 index 00000000000..5024fbdff9d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_async.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BodyOptionalityPreparer +from testpreparer_async import BodyOptionalityClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBodyOptionalityAsync(BodyOptionalityClientTestBaseAsync): + @BodyOptionalityPreparer() + @recorded_by_proxy_async + async def test_required_explicit(self, bodyoptionality_endpoint): + client = self.create_async_client(endpoint=bodyoptionality_endpoint) + response = await client.required_explicit( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @BodyOptionalityPreparer() + @recorded_by_proxy_async + async def test_required_implicit(self, bodyoptionality_endpoint): + client = self.create_async_client(endpoint=bodyoptionality_endpoint) + response = await client.required_implicit( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py new file mode 100644 index 00000000000..5dfe0e95ed4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations.py @@ -0,0 +1,31 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import BodyOptionalityClientTestBase, BodyOptionalityPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBodyOptionalityOptionalExplicitOperations(BodyOptionalityClientTestBase): + @BodyOptionalityPreparer() + @recorded_by_proxy + def test_optional_explicit_set(self, bodyoptionality_endpoint): + client = self.create_client(endpoint=bodyoptionality_endpoint) + response = client.optional_explicit.set() + + # please add some check logic here by yourself + # ... + + @BodyOptionalityPreparer() + @recorded_by_proxy + def test_optional_explicit_omit(self, bodyoptionality_endpoint): + client = self.create_client(endpoint=bodyoptionality_endpoint) + response = client.optional_explicit.omit() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py new file mode 100644 index 00000000000..aa6ee862b35 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/test_body_optionality_optional_explicit_operations_async.py @@ -0,0 +1,32 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import BodyOptionalityPreparer +from testpreparer_async import BodyOptionalityClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestBodyOptionalityOptionalExplicitOperationsAsync(BodyOptionalityClientTestBaseAsync): + @BodyOptionalityPreparer() + @recorded_by_proxy_async + async def test_optional_explicit_set(self, bodyoptionality_endpoint): + client = self.create_async_client(endpoint=bodyoptionality_endpoint) + response = await client.optional_explicit.set() + + # please add some check logic here by yourself + # ... + + @BodyOptionalityPreparer() + @recorded_by_proxy_async + async def test_optional_explicit_omit(self, bodyoptionality_endpoint): + client = self.create_async_client(endpoint=bodyoptionality_endpoint) + response = await client.optional_explicit.omit() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py new file mode 100644 index 00000000000..0e733a2adbb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from parameters.bodyoptionality import BodyOptionalityClient + + +class BodyOptionalityClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(BodyOptionalityClient) + return self.create_client_from_credential( + BodyOptionalityClient, + credential=credential, + endpoint=endpoint, + ) + + +BodyOptionalityPreparer = functools.partial( + PowerShellPreparer, "bodyoptionality", bodyoptionality_endpoint="https://fake_bodyoptionality_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..0b22395b14c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from parameters.bodyoptionality.aio import BodyOptionalityClient + + +class BodyOptionalityClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(BodyOptionalityClient, is_async=True) + return self.create_client_from_credential( + BodyOptionalityClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py b/packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py +++ b/packages/typespec-python/test/azure/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-collection-format/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-collection-format/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py new file mode 100644 index 00000000000..d97918929d3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + collectionformat_subscription_id = os.environ.get( + "COLLECTIONFORMAT_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + collectionformat_tenant_id = os.environ.get("COLLECTIONFORMAT_TENANT_ID", "00000000-0000-0000-0000-000000000000") + collectionformat_client_id = os.environ.get("COLLECTIONFORMAT_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + collectionformat_client_secret = os.environ.get( + "COLLECTIONFORMAT_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=collectionformat_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=collectionformat_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=collectionformat_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=collectionformat_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py new file mode 100644 index 00000000000..8d22f920b98 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import CollectionFormatClientTestBase, CollectionFormatPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCollectionFormatHeaderOperations(CollectionFormatClientTestBase): + @CollectionFormatPreparer() + @recorded_by_proxy + def test_header_csv(self, collectionformat_endpoint): + client = self.create_client(endpoint=collectionformat_endpoint) + response = client.header.csv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py new file mode 100644 index 00000000000..3b596202fb2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_header_operations_async.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import CollectionFormatPreparer +from testpreparer_async import CollectionFormatClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCollectionFormatHeaderOperationsAsync(CollectionFormatClientTestBaseAsync): + @CollectionFormatPreparer() + @recorded_by_proxy_async + async def test_header_csv(self, collectionformat_endpoint): + client = self.create_async_client(endpoint=collectionformat_endpoint) + response = await client.header.csv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py new file mode 100644 index 00000000000..12397560740 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations.py @@ -0,0 +1,68 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import CollectionFormatClientTestBase, CollectionFormatPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCollectionFormatQueryOperations(CollectionFormatClientTestBase): + @CollectionFormatPreparer() + @recorded_by_proxy + def test_query_multi(self, collectionformat_endpoint): + client = self.create_client(endpoint=collectionformat_endpoint) + response = client.query.multi( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy + def test_query_ssv(self, collectionformat_endpoint): + client = self.create_client(endpoint=collectionformat_endpoint) + response = client.query.ssv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy + def test_query_tsv(self, collectionformat_endpoint): + client = self.create_client(endpoint=collectionformat_endpoint) + response = client.query.tsv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy + def test_query_pipes(self, collectionformat_endpoint): + client = self.create_client(endpoint=collectionformat_endpoint) + response = client.query.pipes( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy + def test_query_csv(self, collectionformat_endpoint): + client = self.create_client(endpoint=collectionformat_endpoint) + response = client.query.csv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py new file mode 100644 index 00000000000..4a73943c601 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/test_collection_format_query_operations_async.py @@ -0,0 +1,69 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import CollectionFormatPreparer +from testpreparer_async import CollectionFormatClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestCollectionFormatQueryOperationsAsync(CollectionFormatClientTestBaseAsync): + @CollectionFormatPreparer() + @recorded_by_proxy_async + async def test_query_multi(self, collectionformat_endpoint): + client = self.create_async_client(endpoint=collectionformat_endpoint) + response = await client.query.multi( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy_async + async def test_query_ssv(self, collectionformat_endpoint): + client = self.create_async_client(endpoint=collectionformat_endpoint) + response = await client.query.ssv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy_async + async def test_query_tsv(self, collectionformat_endpoint): + client = self.create_async_client(endpoint=collectionformat_endpoint) + response = await client.query.tsv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy_async + async def test_query_pipes(self, collectionformat_endpoint): + client = self.create_async_client(endpoint=collectionformat_endpoint) + response = await client.query.pipes( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... + + @CollectionFormatPreparer() + @recorded_by_proxy_async + async def test_query_csv(self, collectionformat_endpoint): + client = self.create_async_client(endpoint=collectionformat_endpoint) + response = await client.query.csv( + colors=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py new file mode 100644 index 00000000000..94fd34f3df5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from parameters.collectionformat import CollectionFormatClient + + +class CollectionFormatClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(CollectionFormatClient) + return self.create_client_from_credential( + CollectionFormatClient, + credential=credential, + endpoint=endpoint, + ) + + +CollectionFormatPreparer = functools.partial( + PowerShellPreparer, "collectionformat", collectionformat_endpoint="https://fake_collectionformat_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..fd91ec8b8cb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from parameters.collectionformat.aio import CollectionFormatClient + + +class CollectionFormatClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(CollectionFormatClient, is_async=True) + return self.create_client_from_credential( + CollectionFormatClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/_model_base.py b/packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/_model_base.py +++ b/packages/typespec-python/test/azure/generated/parameters-collection-format/parameters/collectionformat/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/parameters-spread/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/parameters-spread/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py new file mode 100644 index 00000000000..67d88cfd75b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + spread_subscription_id = os.environ.get("SPREAD_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + spread_tenant_id = os.environ.get("SPREAD_TENANT_ID", "00000000-0000-0000-0000-000000000000") + spread_client_id = os.environ.get("SPREAD_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + spread_client_secret = os.environ.get("SPREAD_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=spread_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=spread_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=spread_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=spread_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py new file mode 100644 index 00000000000..c1be4f1f508 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations.py @@ -0,0 +1,83 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SpreadClientTestBase, SpreadPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpreadAliasOperations(SpreadClientTestBase): + @SpreadPreparer() + @recorded_by_proxy + def test_alias_spread_as_request_body(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.alias.spread_as_request_body( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_alias_spread_parameter_with_inner_model(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.alias.spread_parameter_with_inner_model( + id="str", + body={"name": "str"}, + x_ms_test_header="str", + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_alias_spread_as_request_parameter(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.alias.spread_as_request_parameter( + id="str", + body={"name": "str"}, + x_ms_test_header="str", + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_alias_spread_with_multiple_parameters(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.alias.spread_with_multiple_parameters( + id="str", + body={"requiredIntList": [0], "requiredString": "str", "optionalInt": 0, "optionalStringList": ["str"]}, + x_ms_test_header="str", + required_string="str", + required_int_list=[0], + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_alias_spread_parameter_with_inner_alias(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.alias.spread_parameter_with_inner_alias( + id="str", + body={"age": 0, "name": "str"}, + x_ms_test_header="str", + name="str", + age=0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py new file mode 100644 index 00000000000..e54868a73d8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_alias_operations_async.py @@ -0,0 +1,84 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SpreadPreparer +from testpreparer_async import SpreadClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpreadAliasOperationsAsync(SpreadClientTestBaseAsync): + @SpreadPreparer() + @recorded_by_proxy_async + async def test_alias_spread_as_request_body(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.alias.spread_as_request_body( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_alias_spread_parameter_with_inner_model(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.alias.spread_parameter_with_inner_model( + id="str", + body={"name": "str"}, + x_ms_test_header="str", + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_alias_spread_as_request_parameter(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.alias.spread_as_request_parameter( + id="str", + body={"name": "str"}, + x_ms_test_header="str", + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_alias_spread_with_multiple_parameters(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.alias.spread_with_multiple_parameters( + id="str", + body={"requiredIntList": [0], "requiredString": "str", "optionalInt": 0, "optionalStringList": ["str"]}, + x_ms_test_header="str", + required_string="str", + required_int_list=[0], + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_alias_spread_parameter_with_inner_alias(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.alias.spread_parameter_with_inner_alias( + id="str", + body={"age": 0, "name": "str"}, + x_ms_test_header="str", + name="str", + age=0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py new file mode 100644 index 00000000000..e223de65762 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations.py @@ -0,0 +1,75 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SpreadClientTestBase, SpreadPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpreadModelOperations(SpreadClientTestBase): + @SpreadPreparer() + @recorded_by_proxy + def test_model_spread_as_request_body(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.model.spread_as_request_body( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_model_spread_composite_request_only_with_body(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.model.spread_composite_request_only_with_body( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_model_spread_composite_request_without_body(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.model.spread_composite_request_without_body( + name="str", + test_header="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_model_spread_composite_request(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.model.spread_composite_request( + name="str", + body={"name": "str"}, + test_header="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy + def test_model_spread_composite_request_mix(self, spread_endpoint): + client = self.create_client(endpoint=spread_endpoint) + response = client.model.spread_composite_request_mix( + name="str", + body={"prop": "str"}, + test_header="str", + prop="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py new file mode 100644 index 00000000000..d8919fbfd48 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/test_spread_model_operations_async.py @@ -0,0 +1,76 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SpreadPreparer +from testpreparer_async import SpreadClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpreadModelOperationsAsync(SpreadClientTestBaseAsync): + @SpreadPreparer() + @recorded_by_proxy_async + async def test_model_spread_as_request_body(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.model.spread_as_request_body( + body={"name": "str"}, + name="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_model_spread_composite_request_only_with_body(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.model.spread_composite_request_only_with_body( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_model_spread_composite_request_without_body(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.model.spread_composite_request_without_body( + name="str", + test_header="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_model_spread_composite_request(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.model.spread_composite_request( + name="str", + body={"name": "str"}, + test_header="str", + ) + + # please add some check logic here by yourself + # ... + + @SpreadPreparer() + @recorded_by_proxy_async + async def test_model_spread_composite_request_mix(self, spread_endpoint): + client = self.create_async_client(endpoint=spread_endpoint) + response = await client.model.spread_composite_request_mix( + name="str", + body={"prop": "str"}, + test_header="str", + prop="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py new file mode 100644 index 00000000000..2901c04003f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from parameters.spread import SpreadClient + + +class SpreadClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(SpreadClient) + return self.create_client_from_credential( + SpreadClient, + credential=credential, + endpoint=endpoint, + ) + + +SpreadPreparer = functools.partial(PowerShellPreparer, "spread", spread_endpoint="https://fake_spread_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..1416ad2c18b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/parameters-spread/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from parameters.spread.aio import SpreadClient + + +class SpreadClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(SpreadClient, is_async=True) + return self.create_client_from_credential( + SpreadClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/_model_base.py b/packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/_model_base.py +++ b/packages/typespec-python/test/azure/generated/parameters-spread/parameters/spread/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-content-negotiation/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-content-negotiation/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py new file mode 100644 index 00000000000..acf7adc84bf --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/conftest.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + contentnegotiation_subscription_id = os.environ.get( + "CONTENTNEGOTIATION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + contentnegotiation_tenant_id = os.environ.get( + "CONTENTNEGOTIATION_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + contentnegotiation_client_id = os.environ.get( + "CONTENTNEGOTIATION_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + contentnegotiation_client_secret = os.environ.get( + "CONTENTNEGOTIATION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=contentnegotiation_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=contentnegotiation_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=contentnegotiation_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=contentnegotiation_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py new file mode 100644 index 00000000000..da9558d9c13 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ContentNegotiationClientTestBase, ContentNegotiationPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestContentNegotiationDifferentBodyOperations(ContentNegotiationClientTestBase): + @ContentNegotiationPreparer() + @recorded_by_proxy + def test_different_body_get_avatar_as_png(self, contentnegotiation_endpoint): + client = self.create_client(endpoint=contentnegotiation_endpoint) + response = client.different_body.get_avatar_as_png( + accept="image/png", + ) + + # please add some check logic here by yourself + # ... + + @ContentNegotiationPreparer() + @recorded_by_proxy + def test_different_body_get_avatar_as_json(self, contentnegotiation_endpoint): + client = self.create_client(endpoint=contentnegotiation_endpoint) + response = client.different_body.get_avatar_as_json( + accept="application/json", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py new file mode 100644 index 00000000000..62d47e75f10 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_different_body_operations_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ContentNegotiationPreparer +from testpreparer_async import ContentNegotiationClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestContentNegotiationDifferentBodyOperationsAsync(ContentNegotiationClientTestBaseAsync): + @ContentNegotiationPreparer() + @recorded_by_proxy_async + async def test_different_body_get_avatar_as_png(self, contentnegotiation_endpoint): + client = self.create_async_client(endpoint=contentnegotiation_endpoint) + response = await client.different_body.get_avatar_as_png( + accept="image/png", + ) + + # please add some check logic here by yourself + # ... + + @ContentNegotiationPreparer() + @recorded_by_proxy_async + async def test_different_body_get_avatar_as_json(self, contentnegotiation_endpoint): + client = self.create_async_client(endpoint=contentnegotiation_endpoint) + response = await client.different_body.get_avatar_as_json( + accept="application/json", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py new file mode 100644 index 00000000000..84186b5709d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ContentNegotiationClientTestBase, ContentNegotiationPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestContentNegotiationSameBodyOperations(ContentNegotiationClientTestBase): + @ContentNegotiationPreparer() + @recorded_by_proxy + def test_same_body_get_avatar_as_png(self, contentnegotiation_endpoint): + client = self.create_client(endpoint=contentnegotiation_endpoint) + response = client.same_body.get_avatar_as_png( + accept="image/png", + ) + + # please add some check logic here by yourself + # ... + + @ContentNegotiationPreparer() + @recorded_by_proxy + def test_same_body_get_avatar_as_jpeg(self, contentnegotiation_endpoint): + client = self.create_client(endpoint=contentnegotiation_endpoint) + response = client.same_body.get_avatar_as_jpeg( + accept="image/jpeg", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py new file mode 100644 index 00000000000..2ece9daf85f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/test_content_negotiation_same_body_operations_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ContentNegotiationPreparer +from testpreparer_async import ContentNegotiationClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestContentNegotiationSameBodyOperationsAsync(ContentNegotiationClientTestBaseAsync): + @ContentNegotiationPreparer() + @recorded_by_proxy_async + async def test_same_body_get_avatar_as_png(self, contentnegotiation_endpoint): + client = self.create_async_client(endpoint=contentnegotiation_endpoint) + response = await client.same_body.get_avatar_as_png( + accept="image/png", + ) + + # please add some check logic here by yourself + # ... + + @ContentNegotiationPreparer() + @recorded_by_proxy_async + async def test_same_body_get_avatar_as_jpeg(self, contentnegotiation_endpoint): + client = self.create_async_client(endpoint=contentnegotiation_endpoint) + response = await client.same_body.get_avatar_as_jpeg( + accept="image/jpeg", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py new file mode 100644 index 00000000000..da4060d62c8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from payload.contentnegotiation import ContentNegotiationClient + + +class ContentNegotiationClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ContentNegotiationClient) + return self.create_client_from_credential( + ContentNegotiationClient, + credential=credential, + endpoint=endpoint, + ) + + +ContentNegotiationPreparer = functools.partial( + PowerShellPreparer, "contentnegotiation", contentnegotiation_endpoint="https://fake_contentnegotiation_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..7133bd2e095 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from payload.contentnegotiation.aio import ContentNegotiationClient + + +class ContentNegotiationClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ContentNegotiationClient, is_async=True) + return self.create_client_from_credential( + ContentNegotiationClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py b/packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py +++ b/packages/typespec-python/test/azure/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-json-merge-patch/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py new file mode 100644 index 00000000000..687cbc1cd18 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + jsonmergepatch_subscription_id = os.environ.get( + "JSONMERGEPATCH_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + jsonmergepatch_tenant_id = os.environ.get("JSONMERGEPATCH_TENANT_ID", "00000000-0000-0000-0000-000000000000") + jsonmergepatch_client_id = os.environ.get("JSONMERGEPATCH_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + jsonmergepatch_client_secret = os.environ.get( + "JSONMERGEPATCH_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=jsonmergepatch_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=jsonmergepatch_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=jsonmergepatch_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=jsonmergepatch_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py new file mode 100644 index 00000000000..5719033ace2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch.py @@ -0,0 +1,61 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import JsonMergePatchClientTestBase, JsonMergePatchPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestJsonMergePatch(JsonMergePatchClientTestBase): + @JsonMergePatchPreparer() + @recorded_by_proxy + def test_create_resource(self, jsonmergepatch_endpoint): + client = self.create_client(endpoint=jsonmergepatch_endpoint) + response = client.create_resource( + body={ + "name": "str", + "array": [{"description": "str", "name": "str"}], + "description": "str", + "floatValue": 0.0, + "innerModel": {"description": "str", "name": "str"}, + "intArray": [0], + "intValue": 0, + "map": {"str": {"description": "str", "name": "str"}}, + }, + ) + + # please add some check logic here by yourself + # ... + + @JsonMergePatchPreparer() + @recorded_by_proxy + def test_update_resource(self, jsonmergepatch_endpoint): + client = self.create_client(endpoint=jsonmergepatch_endpoint) + response = client.update_resource( + body={ + "array": [{"description": "str", "name": "str"}], + "description": "str", + "floatValue": 0.0, + "innerModel": {"description": "str", "name": "str"}, + "intArray": [0], + "intValue": 0, + "map": {"str": {"description": "str", "name": "str"}}, + }, + ) + + # please add some check logic here by yourself + # ... + + @JsonMergePatchPreparer() + @recorded_by_proxy + def test_update_optional_resource(self, jsonmergepatch_endpoint): + client = self.create_client(endpoint=jsonmergepatch_endpoint) + response = client.update_optional_resource() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py new file mode 100644 index 00000000000..bfc931ddf4b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/test_json_merge_patch_async.py @@ -0,0 +1,62 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import JsonMergePatchPreparer +from testpreparer_async import JsonMergePatchClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestJsonMergePatchAsync(JsonMergePatchClientTestBaseAsync): + @JsonMergePatchPreparer() + @recorded_by_proxy_async + async def test_create_resource(self, jsonmergepatch_endpoint): + client = self.create_async_client(endpoint=jsonmergepatch_endpoint) + response = await client.create_resource( + body={ + "name": "str", + "array": [{"description": "str", "name": "str"}], + "description": "str", + "floatValue": 0.0, + "innerModel": {"description": "str", "name": "str"}, + "intArray": [0], + "intValue": 0, + "map": {"str": {"description": "str", "name": "str"}}, + }, + ) + + # please add some check logic here by yourself + # ... + + @JsonMergePatchPreparer() + @recorded_by_proxy_async + async def test_update_resource(self, jsonmergepatch_endpoint): + client = self.create_async_client(endpoint=jsonmergepatch_endpoint) + response = await client.update_resource( + body={ + "array": [{"description": "str", "name": "str"}], + "description": "str", + "floatValue": 0.0, + "innerModel": {"description": "str", "name": "str"}, + "intArray": [0], + "intValue": 0, + "map": {"str": {"description": "str", "name": "str"}}, + }, + ) + + # please add some check logic here by yourself + # ... + + @JsonMergePatchPreparer() + @recorded_by_proxy_async + async def test_update_optional_resource(self, jsonmergepatch_endpoint): + client = self.create_async_client(endpoint=jsonmergepatch_endpoint) + response = await client.update_optional_resource() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py new file mode 100644 index 00000000000..4be2411f2f8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from payload.jsonmergepatch import JsonMergePatchClient + + +class JsonMergePatchClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(JsonMergePatchClient) + return self.create_client_from_credential( + JsonMergePatchClient, + credential=credential, + endpoint=endpoint, + ) + + +JsonMergePatchPreparer = functools.partial( + PowerShellPreparer, "jsonmergepatch", jsonmergepatch_endpoint="https://fake_jsonmergepatch_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..fcc202c7605 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from payload.jsonmergepatch.aio import JsonMergePatchClient + + +class JsonMergePatchClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(JsonMergePatchClient, is_async=True) + return self.create_client_from_credential( + JsonMergePatchClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py +++ b/packages/typespec-python/test/azure/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-media-type/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-media-type/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py new file mode 100644 index 00000000000..9c6c4e96dc8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + mediatype_subscription_id = os.environ.get("MEDIATYPE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + mediatype_tenant_id = os.environ.get("MEDIATYPE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + mediatype_client_id = os.environ.get("MEDIATYPE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + mediatype_client_secret = os.environ.get("MEDIATYPE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mediatype_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mediatype_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mediatype_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=mediatype_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py new file mode 100644 index 00000000000..f3f77dafaf9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations.py @@ -0,0 +1,55 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import MediaTypeClientTestBase, MediaTypePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMediaTypeStringBodyOperations(MediaTypeClientTestBase): + @MediaTypePreparer() + @recorded_by_proxy + def test_string_body_send_as_text(self, mediatype_endpoint): + client = self.create_client(endpoint=mediatype_endpoint) + response = client.string_body.send_as_text( + text="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @MediaTypePreparer() + @recorded_by_proxy + def test_string_body_get_as_text(self, mediatype_endpoint): + client = self.create_client(endpoint=mediatype_endpoint) + response = client.string_body.get_as_text() + + # please add some check logic here by yourself + # ... + + @MediaTypePreparer() + @recorded_by_proxy + def test_string_body_send_as_json(self, mediatype_endpoint): + client = self.create_client(endpoint=mediatype_endpoint) + response = client.string_body.send_as_json( + text="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @MediaTypePreparer() + @recorded_by_proxy + def test_string_body_get_as_json(self, mediatype_endpoint): + client = self.create_client(endpoint=mediatype_endpoint) + response = client.string_body.get_as_json() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py new file mode 100644 index 00000000000..12547676b6f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/test_media_type_string_body_operations_async.py @@ -0,0 +1,56 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import MediaTypePreparer +from testpreparer_async import MediaTypeClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMediaTypeStringBodyOperationsAsync(MediaTypeClientTestBaseAsync): + @MediaTypePreparer() + @recorded_by_proxy_async + async def test_string_body_send_as_text(self, mediatype_endpoint): + client = self.create_async_client(endpoint=mediatype_endpoint) + response = await client.string_body.send_as_text( + text="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @MediaTypePreparer() + @recorded_by_proxy_async + async def test_string_body_get_as_text(self, mediatype_endpoint): + client = self.create_async_client(endpoint=mediatype_endpoint) + response = await client.string_body.get_as_text() + + # please add some check logic here by yourself + # ... + + @MediaTypePreparer() + @recorded_by_proxy_async + async def test_string_body_send_as_json(self, mediatype_endpoint): + client = self.create_async_client(endpoint=mediatype_endpoint) + response = await client.string_body.send_as_json( + text="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @MediaTypePreparer() + @recorded_by_proxy_async + async def test_string_body_get_as_json(self, mediatype_endpoint): + client = self.create_async_client(endpoint=mediatype_endpoint) + response = await client.string_body.get_as_json() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py new file mode 100644 index 00000000000..f32364a5010 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from payload.mediatype import MediaTypeClient + + +class MediaTypeClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(MediaTypeClient) + return self.create_client_from_credential( + MediaTypeClient, + credential=credential, + endpoint=endpoint, + ) + + +MediaTypePreparer = functools.partial( + PowerShellPreparer, "mediatype", mediatype_endpoint="https://fake_mediatype_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..34e76d5eeaa --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-media-type/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from payload.mediatype.aio import MediaTypeClient + + +class MediaTypeClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(MediaTypeClient, is_async=True) + return self.create_client_from_credential( + MediaTypeClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/_model_base.py b/packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/_model_base.py +++ b/packages/typespec-python/test/azure/generated/payload-media-type/payload/mediatype/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-multipart/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-multipart/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py new file mode 100644 index 00000000000..7b1948a5cb1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + multipart_subscription_id = os.environ.get("MULTIPART_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + multipart_tenant_id = os.environ.get("MULTIPART_TENANT_ID", "00000000-0000-0000-0000-000000000000") + multipart_client_id = os.environ.get("MULTIPART_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + multipart_client_secret = os.environ.get("MULTIPART_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multipart_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multipart_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multipart_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multipart_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py new file mode 100644 index 00000000000..c0787c2ae85 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations.py @@ -0,0 +1,152 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import MultiPartClientTestBase, MultiPartPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMultiPartFormDataOperations(MultiPartClientTestBase): + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_basic(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.basic( + body={"id": "str", "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_file_array_and_basic(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.file_array_and_basic( + body={"address": {"city": "str"}, "id": "str", "pictures": ["filetype"], "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_json_part(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.json_part( + body={"address": {"city": "str"}, "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_binary_array_parts(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.binary_array_parts( + body={"id": "str", "pictures": ["filetype"]}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_multi_binary_parts(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.multi_binary_parts( + body={"profileImage": "filetype", "picture": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_check_file_name_and_content_type(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.check_file_name_and_content_type( + body={"id": "str", "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_anonymous_model(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.anonymous_model( + body={"profileImage": "filetype"}, + profile_image="filetype", + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_http_parts_json_array_and_file_array(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.http_parts.json_array_and_file_array( + body={ + "address": {"city": "str"}, + "id": "str", + "pictures": ["filetype"], + "previousAddresses": [{"city": "str"}], + "profileImage": "filetype", + }, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_http_parts_content_type_image_jpeg_content_type(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.http_parts.content_type.image_jpeg_content_type( + body={"profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_http_parts_content_type_required_content_type(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.http_parts.content_type.required_content_type( + body={"profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_http_parts_content_type_optional_content_type(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.http_parts.content_type.optional_content_type( + body={"profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy + def test_form_data_http_parts_non_string_float(self, multipart_endpoint): + client = self.create_client(endpoint=multipart_endpoint) + response = client.form_data.http_parts.non_string.float( + body={"temperature": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py new file mode 100644 index 00000000000..e030fa8600a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/test_multi_part_form_data_operations_async.py @@ -0,0 +1,153 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import MultiPartPreparer +from testpreparer_async import MultiPartClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMultiPartFormDataOperationsAsync(MultiPartClientTestBaseAsync): + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_basic(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.basic( + body={"id": "str", "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_file_array_and_basic(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.file_array_and_basic( + body={"address": {"city": "str"}, "id": "str", "pictures": ["filetype"], "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_json_part(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.json_part( + body={"address": {"city": "str"}, "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_binary_array_parts(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.binary_array_parts( + body={"id": "str", "pictures": ["filetype"]}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_multi_binary_parts(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.multi_binary_parts( + body={"profileImage": "filetype", "picture": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_check_file_name_and_content_type(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.check_file_name_and_content_type( + body={"id": "str", "profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_anonymous_model(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.anonymous_model( + body={"profileImage": "filetype"}, + profile_image="filetype", + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_http_parts_json_array_and_file_array(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.http_parts.json_array_and_file_array( + body={ + "address": {"city": "str"}, + "id": "str", + "pictures": ["filetype"], + "previousAddresses": [{"city": "str"}], + "profileImage": "filetype", + }, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_http_parts_content_type_image_jpeg_content_type(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.http_parts.content_type.image_jpeg_content_type( + body={"profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_http_parts_content_type_required_content_type(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.http_parts.content_type.required_content_type( + body={"profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_http_parts_content_type_optional_content_type(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.http_parts.content_type.optional_content_type( + body={"profileImage": "filetype"}, + ) + + # please add some check logic here by yourself + # ... + + @MultiPartPreparer() + @recorded_by_proxy_async + async def test_form_data_http_parts_non_string_float(self, multipart_endpoint): + client = self.create_async_client(endpoint=multipart_endpoint) + response = await client.form_data.http_parts.non_string.float( + body={"temperature": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py new file mode 100644 index 00000000000..ad58b84f2c2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from payload.multipart import MultiPartClient + + +class MultiPartClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(MultiPartClient) + return self.create_client_from_credential( + MultiPartClient, + credential=credential, + endpoint=endpoint, + ) + + +MultiPartPreparer = functools.partial( + PowerShellPreparer, "multipart", multipart_endpoint="https://fake_multipart_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..98b451ec0c0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-multipart/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from payload.multipart.aio import MultiPartClient + + +class MultiPartClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(MultiPartClient, is_async=True) + return self.create_client_from_credential( + MultiPartClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/_model_base.py b/packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/_model_base.py +++ b/packages/typespec-python/test/azure/generated/payload-multipart/payload/multipart/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-pageable/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-pageable/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py new file mode 100644 index 00000000000..164dcd89702 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + pageable_subscription_id = os.environ.get("PAGEABLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + pageable_tenant_id = os.environ.get("PAGEABLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + pageable_client_id = os.environ.get("PAGEABLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + pageable_client_secret = os.environ.get("PAGEABLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=pageable_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=pageable_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=pageable_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=pageable_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py new file mode 100644 index 00000000000..e66879ed799 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import PageableClientTestBase, PageablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestPageable(PageableClientTestBase): + @PageablePreparer() + @recorded_by_proxy + def test_list(self, pageable_endpoint): + client = self.create_client(endpoint=pageable_endpoint) + response = client.list() + result = [r for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py new file mode 100644 index 00000000000..44faaa807bd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/test_pageable_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import PageablePreparer +from testpreparer_async import PageableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestPageableAsync(PageableClientTestBaseAsync): + @PageablePreparer() + @recorded_by_proxy_async + async def test_list(self, pageable_endpoint): + client = self.create_async_client(endpoint=pageable_endpoint) + response = client.list() + result = [r async for r in response] + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py new file mode 100644 index 00000000000..9f5844c2124 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from payload.pageable import PageableClient + + +class PageableClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(PageableClient) + return self.create_client_from_credential( + PageableClient, + credential=credential, + endpoint=endpoint, + ) + + +PageablePreparer = functools.partial( + PowerShellPreparer, "pageable", pageable_endpoint="https://fake_pageable_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..21379ef4880 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-pageable/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from payload.pageable.aio import PageableClient + + +class PageableClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(PageableClient, is_async=True) + return self.create_client_from_credential( + PageableClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/_model_base.py b/packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/_model_base.py +++ b/packages/typespec-python/test/azure/generated/payload-pageable/payload/pageable/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/payload-xml/payload/xml/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/payload-xml/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/payload-xml/payload/xml/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/payload-xml/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py new file mode 100644 index 00000000000..b7edb687c98 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + xml_subscription_id = os.environ.get("XML_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + xml_tenant_id = os.environ.get("XML_TENANT_ID", "00000000-0000-0000-0000-000000000000") + xml_client_id = os.environ.get("XML_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + xml_client_secret = os.environ.get("XML_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xml_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xml_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xml_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=xml_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py new file mode 100644 index 00000000000..513a9286363 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithArrayOfModelValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_array_of_model_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_array_of_model_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_array_of_model_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_array_of_model_value.put( + input={"items": [{"age": 0, "name": "str"}]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py new file mode 100644 index 00000000000..7f8831671cf --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_array_of_model_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithArrayOfModelValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_array_of_model_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_array_of_model_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_array_of_model_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_array_of_model_value.put( + input={"items": [{"age": 0, "name": "str"}]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py new file mode 100644 index 00000000000..75477c169a0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithAttributesValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_attributes_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_attributes_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_attributes_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_attributes_value.put( + input={"enabled": bool, "id1": 0, "id2": "str"}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py new file mode 100644 index 00000000000..38e116b8bf0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_attributes_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithAttributesValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_attributes_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_attributes_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_attributes_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_attributes_value.put( + input={"enabled": bool, "id1": 0, "id2": "str"}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py new file mode 100644 index 00000000000..16928d7c7ab --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithDictionaryValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_dictionary_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_dictionary_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_dictionary_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_dictionary_value.put( + input={"metadata": {"str": "str"}}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py new file mode 100644 index 00000000000..9bf17851386 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_dictionary_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithDictionaryValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_dictionary_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_dictionary_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_dictionary_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_dictionary_value.put( + input={"metadata": {"str": "str"}}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py new file mode 100644 index 00000000000..ad415cb3ba3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithEmptyArrayValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_empty_array_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_empty_array_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_empty_array_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_empty_array_value.put( + input={"items": [{"age": 0, "name": "str"}]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py new file mode 100644 index 00000000000..15c4211942d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_empty_array_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithEmptyArrayValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_empty_array_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_empty_array_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_empty_array_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_empty_array_value.put( + input={"items": [{"age": 0, "name": "str"}]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py new file mode 100644 index 00000000000..d5af0600ff8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithEncodedNamesValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_encoded_names_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_encoded_names_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_encoded_names_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_encoded_names_value.put( + input={"colors": ["str"], "modelData": {"age": 0, "name": "str"}}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py new file mode 100644 index 00000000000..26b306b521e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_encoded_names_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithEncodedNamesValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_encoded_names_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_encoded_names_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_encoded_names_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_encoded_names_value.put( + input={"colors": ["str"], "modelData": {"age": 0, "name": "str"}}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py new file mode 100644 index 00000000000..3390b3458d8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithOptionalFieldValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_optional_field_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_optional_field_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_optional_field_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_optional_field_value.put( + input={"item": "str", "value": 0}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py new file mode 100644 index 00000000000..4ce92bfc3e5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_optional_field_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithOptionalFieldValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_optional_field_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_optional_field_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_optional_field_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_optional_field_value.put( + input={"item": "str", "value": 0}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py new file mode 100644 index 00000000000..1616afab05b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithRenamedArraysValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_renamed_arrays_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_renamed_arrays_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_renamed_arrays_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_renamed_arrays_value.put( + input={"colors": ["str"], "counts": [0]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py new file mode 100644 index 00000000000..85c8109461f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_arrays_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithRenamedArraysValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_renamed_arrays_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_renamed_arrays_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_renamed_arrays_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_renamed_arrays_value.put( + input={"colors": ["str"], "counts": [0]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py new file mode 100644 index 00000000000..55fa6ec8b3a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithRenamedFieldsValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_renamed_fields_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_renamed_fields_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_renamed_fields_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_renamed_fields_value.put( + input={"inputData": {"age": 0, "name": "str"}, "outputData": {"age": 0, "name": "str"}}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py new file mode 100644 index 00000000000..7ab55fa7257 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_renamed_fields_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithRenamedFieldsValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_renamed_fields_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_renamed_fields_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_renamed_fields_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_renamed_fields_value.put( + input={"inputData": {"age": 0, "name": "str"}, "outputData": {"age": 0, "name": "str"}}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py new file mode 100644 index 00000000000..097a57230ca --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithSimpleArraysValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_simple_arrays_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_simple_arrays_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_simple_arrays_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_simple_arrays_value.put( + input={"colors": ["str"], "counts": [0]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py new file mode 100644 index 00000000000..4d4d4a54266 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_simple_arrays_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithSimpleArraysValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_simple_arrays_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_simple_arrays_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_simple_arrays_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_simple_arrays_value.put( + input={"colors": ["str"], "counts": [0]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py new file mode 100644 index 00000000000..c19e8ce8aba --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithTextValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_text_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_text_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_text_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_text_value.put( + input={"content": "str", "language": "str"}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py new file mode 100644 index 00000000000..65eab4fe6b5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_text_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithTextValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_text_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_text_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_text_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_text_value.put( + input={"content": "str", "language": "str"}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py new file mode 100644 index 00000000000..06dfa2d6fa9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithUnwrappedArrayValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_model_with_unwrapped_array_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_unwrapped_array_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_model_with_unwrapped_array_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.model_with_unwrapped_array_value.put( + input={"colors": ["str"], "counts": [0]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py new file mode 100644 index 00000000000..7057cddd2c0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_model_with_unwrapped_array_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlModelWithUnwrappedArrayValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_unwrapped_array_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_unwrapped_array_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_model_with_unwrapped_array_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.model_with_unwrapped_array_value.put( + input={"colors": ["str"], "counts": [0]}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py new file mode 100644 index 00000000000..a8376712a5d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import XmlClientTestBase, XmlPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlSimpleModelValueOperations(XmlClientTestBase): + @XmlPreparer() + @recorded_by_proxy + def test_simple_model_value_get(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.simple_model_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy + def test_simple_model_value_put(self, xml_endpoint): + client = self.create_client(endpoint=xml_endpoint) + response = client.simple_model_value.put( + input={"age": 0, "name": "str"}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py new file mode 100644 index 00000000000..0c42bd7d059 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/test_xml_simple_model_value_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import XmlPreparer +from testpreparer_async import XmlClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestXmlSimpleModelValueOperationsAsync(XmlClientTestBaseAsync): + @XmlPreparer() + @recorded_by_proxy_async + async def test_simple_model_value_get(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.simple_model_value.get() + + # please add some check logic here by yourself + # ... + + @XmlPreparer() + @recorded_by_proxy_async + async def test_simple_model_value_put(self, xml_endpoint): + client = self.create_async_client(endpoint=xml_endpoint) + response = await client.simple_model_value.put( + input={"age": 0, "name": "str"}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py new file mode 100644 index 00000000000..7470b7afb70 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from payload.xml import XmlClient + + +class XmlClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(XmlClient) + return self.create_client_from_credential( + XmlClient, + credential=credential, + endpoint=endpoint, + ) + + +XmlPreparer = functools.partial(PowerShellPreparer, "xml", xml_endpoint="https://fake_xml_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..d993ba8306c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/payload-xml/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from payload.xml.aio import XmlClient + + +class XmlClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(XmlClient, is_async=True) + return self.create_client_from_credential( + XmlClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/payload-xml/payload/xml/_model_base.py b/packages/typespec-python/test/azure/generated/payload-xml/payload/xml/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/payload-xml/payload/xml/_model_base.py +++ b/packages/typespec-python/test/azure/generated/payload-xml/payload/xml/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/resiliency-srv-driven1/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py new file mode 100644 index 00000000000..a2a87317d0a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/conftest.py @@ -0,0 +1,47 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + resiliencyservicedriven_subscription_id = os.environ.get( + "RESILIENCYSERVICEDRIVEN_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + resiliencyservicedriven_tenant_id = os.environ.get( + "RESILIENCYSERVICEDRIVEN_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + resiliencyservicedriven_client_id = os.environ.get( + "RESILIENCYSERVICEDRIVEN_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + resiliencyservicedriven_client_secret = os.environ.get( + "RESILIENCYSERVICEDRIVEN_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=resiliencyservicedriven_subscription_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=resiliencyservicedriven_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=resiliencyservicedriven_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer( + regex=resiliencyservicedriven_client_secret, value="00000000-0000-0000-0000-000000000000" + ) + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py new file mode 100644 index 00000000000..39eb11630ef --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven.py @@ -0,0 +1,42 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ResiliencyServiceDrivenClientTestBase, ResiliencyServiceDrivenPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResiliencyServiceDriven(ResiliencyServiceDrivenClientTestBase): + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy + def test_from_none(self, resiliencyservicedriven_endpoint): + client = self.create_client(endpoint=resiliencyservicedriven_endpoint) + response = client.from_none() + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy + def test_from_one_required(self, resiliencyservicedriven_endpoint): + client = self.create_client(endpoint=resiliencyservicedriven_endpoint) + response = client.from_one_required( + parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy + def test_from_one_optional(self, resiliencyservicedriven_endpoint): + client = self.create_client(endpoint=resiliencyservicedriven_endpoint) + response = client.from_one_optional() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py new file mode 100644 index 00000000000..dd7d11a9487 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/test_resiliency_service_driven_async.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ResiliencyServiceDrivenPreparer +from testpreparer_async import ResiliencyServiceDrivenClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResiliencyServiceDrivenAsync(ResiliencyServiceDrivenClientTestBaseAsync): + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy_async + async def test_from_none(self, resiliencyservicedriven_endpoint): + client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) + response = await client.from_none() + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy_async + async def test_from_one_required(self, resiliencyservicedriven_endpoint): + client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) + response = await client.from_one_required( + parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy_async + async def test_from_one_optional(self, resiliencyservicedriven_endpoint): + client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) + response = await client.from_one_optional() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py new file mode 100644 index 00000000000..d657bd047dc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from resiliency.srv.driven1 import ResiliencyServiceDrivenClient + + +class ResiliencyServiceDrivenClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ResiliencyServiceDrivenClient) + return self.create_client_from_credential( + ResiliencyServiceDrivenClient, + credential=credential, + endpoint=endpoint, + ) + + +ResiliencyServiceDrivenPreparer = functools.partial( + PowerShellPreparer, + "resiliencyservicedriven", + resiliencyservicedriven_endpoint="https://fake_resiliencyservicedriven_endpoint.com", +) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..bb886d1ab0a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from resiliency.srv.driven1.aio import ResiliencyServiceDrivenClient + + +class ResiliencyServiceDrivenClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ResiliencyServiceDrivenClient, is_async=True) + return self.create_client_from_credential( + ResiliencyServiceDrivenClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/resiliency-srv-driven2/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py new file mode 100644 index 00000000000..a2a87317d0a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/conftest.py @@ -0,0 +1,47 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + resiliencyservicedriven_subscription_id = os.environ.get( + "RESILIENCYSERVICEDRIVEN_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + resiliencyservicedriven_tenant_id = os.environ.get( + "RESILIENCYSERVICEDRIVEN_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + resiliencyservicedriven_client_id = os.environ.get( + "RESILIENCYSERVICEDRIVEN_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + resiliencyservicedriven_client_secret = os.environ.get( + "RESILIENCYSERVICEDRIVEN_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=resiliencyservicedriven_subscription_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=resiliencyservicedriven_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=resiliencyservicedriven_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer( + regex=resiliencyservicedriven_client_secret, value="00000000-0000-0000-0000-000000000000" + ) + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py new file mode 100644 index 00000000000..a827df42fc3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven.py @@ -0,0 +1,51 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ResiliencyServiceDrivenClientTestBase, ResiliencyServiceDrivenPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResiliencyServiceDriven(ResiliencyServiceDrivenClientTestBase): + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy + def test_add_operation(self, resiliencyservicedriven_endpoint): + client = self.create_client(endpoint=resiliencyservicedriven_endpoint) + response = client.add_operation() + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy + def test_from_none(self, resiliencyservicedriven_endpoint): + client = self.create_client(endpoint=resiliencyservicedriven_endpoint) + response = client.from_none() + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy + def test_from_one_required(self, resiliencyservicedriven_endpoint): + client = self.create_client(endpoint=resiliencyservicedriven_endpoint) + response = client.from_one_required( + parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy + def test_from_one_optional(self, resiliencyservicedriven_endpoint): + client = self.create_client(endpoint=resiliencyservicedriven_endpoint) + response = client.from_one_optional() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py new file mode 100644 index 00000000000..05d9bcc89f6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/test_resiliency_service_driven_async.py @@ -0,0 +1,52 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ResiliencyServiceDrivenPreparer +from testpreparer_async import ResiliencyServiceDrivenClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestResiliencyServiceDrivenAsync(ResiliencyServiceDrivenClientTestBaseAsync): + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy_async + async def test_add_operation(self, resiliencyservicedriven_endpoint): + client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) + response = await client.add_operation() + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy_async + async def test_from_none(self, resiliencyservicedriven_endpoint): + client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) + response = await client.from_none() + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy_async + async def test_from_one_required(self, resiliencyservicedriven_endpoint): + client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) + response = await client.from_one_required( + parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @ResiliencyServiceDrivenPreparer() + @recorded_by_proxy_async + async def test_from_one_optional(self, resiliencyservicedriven_endpoint): + client = self.create_async_client(endpoint=resiliencyservicedriven_endpoint) + response = await client.from_one_optional() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py new file mode 100644 index 00000000000..8ce298e478a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from resiliency.srv.driven2 import ResiliencyServiceDrivenClient + + +class ResiliencyServiceDrivenClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ResiliencyServiceDrivenClient) + return self.create_client_from_credential( + ResiliencyServiceDrivenClient, + credential=credential, + endpoint=endpoint, + ) + + +ResiliencyServiceDrivenPreparer = functools.partial( + PowerShellPreparer, + "resiliencyservicedriven", + resiliencyservicedriven_endpoint="https://fake_resiliencyservicedriven_endpoint.com", +) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..e94f4e5ea90 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from resiliency.srv.driven2.aio import ResiliencyServiceDrivenClient + + +class ResiliencyServiceDrivenClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ResiliencyServiceDrivenClient, is_async=True) + return self.create_client_from_credential( + ResiliencyServiceDrivenClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py +++ b/packages/typespec-python/test/azure/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/routes/routes/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/routes/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/routes/routes/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/routes/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py new file mode 100644 index 00000000000..0416b7b5099 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + routes_subscription_id = os.environ.get("ROUTES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + routes_tenant_id = os.environ.get("ROUTES_TENANT_ID", "00000000-0000-0000-0000-000000000000") + routes_client_id = os.environ.get("ROUTES_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + routes_client_secret = os.environ.get("ROUTES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=routes_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=routes_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=routes_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=routes_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py new file mode 100644 index 00000000000..76ea321422e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RoutesClientTestBase, RoutesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutes(RoutesClientTestBase): + @RoutesPreparer() + @recorded_by_proxy + def test_fixed(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.fixed() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py new file mode 100644 index 00000000000..705ab5d1955 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RoutesPreparer +from testpreparer_async import RoutesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutesAsync(RoutesClientTestBaseAsync): + @RoutesPreparer() + @recorded_by_proxy_async + async def test_fixed(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.fixed() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py new file mode 100644 index 00000000000..4c2c9a7be14 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RoutesClientTestBase, RoutesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutesInInterfaceOperations(RoutesClientTestBase): + @RoutesPreparer() + @recorded_by_proxy + def test_in_interface_fixed(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.in_interface.fixed() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py new file mode 100644 index 00000000000..1fac7cb10fe --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_in_interface_operations_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RoutesPreparer +from testpreparer_async import RoutesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutesInInterfaceOperationsAsync(RoutesClientTestBaseAsync): + @RoutesPreparer() + @recorded_by_proxy_async + async def test_in_interface_fixed(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.in_interface.fixed() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py new file mode 100644 index 00000000000..20cf78b6765 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations.py @@ -0,0 +1,332 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RoutesClientTestBase, RoutesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutesPathParametersOperations(RoutesClientTestBase): + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_template_only(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.template_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_explicit(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.explicit( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_annotation_only(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.annotation_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_reserved_expansion_template(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.reserved_expansion.template( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_reserved_expansion_annotation(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.reserved_expansion.annotation( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_simple_expansion_standard_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.simple_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_simple_expansion_standard_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.simple_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_simple_expansion_standard_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.simple_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_simple_expansion_explode_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.simple_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_simple_expansion_explode_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.simple_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_simple_expansion_explode_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.simple_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_path_expansion_standard_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.path_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_path_expansion_standard_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.path_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_path_expansion_standard_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.path_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_path_expansion_explode_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.path_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_path_expansion_explode_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.path_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_path_expansion_explode_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.path_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_label_expansion_standard_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.label_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_label_expansion_standard_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.label_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_label_expansion_standard_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.label_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_label_expansion_explode_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.label_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_label_expansion_explode_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.label_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_label_expansion_explode_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.label_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_matrix_expansion_standard_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.matrix_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_matrix_expansion_standard_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.matrix_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_matrix_expansion_standard_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.matrix_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_matrix_expansion_explode_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.matrix_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_matrix_expansion_explode_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.matrix_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_path_parameters_matrix_expansion_explode_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.path_parameters.matrix_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py new file mode 100644 index 00000000000..c73a227f87e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_path_parameters_operations_async.py @@ -0,0 +1,333 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RoutesPreparer +from testpreparer_async import RoutesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutesPathParametersOperationsAsync(RoutesClientTestBaseAsync): + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_template_only(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.template_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_explicit(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.explicit( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_annotation_only(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.annotation_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_reserved_expansion_template(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.reserved_expansion.template( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_reserved_expansion_annotation(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.reserved_expansion.annotation( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_simple_expansion_standard_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.simple_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_simple_expansion_standard_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.simple_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_simple_expansion_standard_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.simple_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_simple_expansion_explode_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.simple_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_simple_expansion_explode_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.simple_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_simple_expansion_explode_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.simple_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_path_expansion_standard_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.path_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_path_expansion_standard_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.path_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_path_expansion_standard_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.path_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_path_expansion_explode_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.path_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_path_expansion_explode_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.path_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_path_expansion_explode_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.path_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_label_expansion_standard_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.label_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_label_expansion_standard_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.label_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_label_expansion_standard_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.label_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_label_expansion_explode_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.label_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_label_expansion_explode_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.label_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_label_expansion_explode_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.label_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_matrix_expansion_standard_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.matrix_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_matrix_expansion_standard_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.matrix_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_matrix_expansion_standard_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.matrix_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_matrix_expansion_explode_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.matrix_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_matrix_expansion_explode_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.matrix_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_path_parameters_matrix_expansion_explode_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.path_parameters.matrix_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py new file mode 100644 index 00000000000..566ff148af5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations.py @@ -0,0 +1,178 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RoutesClientTestBase, RoutesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutesQueryParametersOperations(RoutesClientTestBase): + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_template_only(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.template_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_explicit(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.explicit( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_annotation_only(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.annotation_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_expansion_standard_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_expansion_standard_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_expansion_standard_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_expansion_explode_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_expansion_explode_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_expansion_explode_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_continuation_standard_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_continuation.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_continuation_standard_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_continuation.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_continuation_standard_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_continuation.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_continuation_explode_primitive(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_continuation.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_continuation_explode_array(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_continuation.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy + def test_query_parameters_query_continuation_explode_record(self, routes_endpoint): + client = self.create_client(endpoint=routes_endpoint) + response = client.query_parameters.query_continuation.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py new file mode 100644 index 00000000000..c85cfc7ebd7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/test_routes_query_parameters_operations_async.py @@ -0,0 +1,179 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RoutesPreparer +from testpreparer_async import RoutesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRoutesQueryParametersOperationsAsync(RoutesClientTestBaseAsync): + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_template_only(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.template_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_explicit(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.explicit( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_annotation_only(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.annotation_only( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_expansion_standard_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_expansion.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_expansion_standard_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_expansion.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_expansion_standard_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_expansion.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_expansion_explode_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_expansion.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_expansion_explode_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_expansion.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_expansion_explode_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_expansion.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_continuation_standard_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_continuation.standard.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_continuation_standard_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_continuation.standard.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_continuation_standard_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_continuation.standard.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_continuation_explode_primitive(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_continuation.explode.primitive( + param="str", + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_continuation_explode_array(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_continuation.explode.array( + param=["str"], + ) + + # please add some check logic here by yourself + # ... + + @RoutesPreparer() + @recorded_by_proxy_async + async def test_query_parameters_query_continuation_explode_record(self, routes_endpoint): + client = self.create_async_client(endpoint=routes_endpoint) + response = await client.query_parameters.query_continuation.explode.record( + param={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py new file mode 100644 index 00000000000..baf88af09ab --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from routes import RoutesClient + + +class RoutesClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(RoutesClient) + return self.create_client_from_credential( + RoutesClient, + credential=credential, + endpoint=endpoint, + ) + + +RoutesPreparer = functools.partial(PowerShellPreparer, "routes", routes_endpoint="https://fake_routes_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..3d66391fd7d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/routes/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from routes.aio import RoutesClient + + +class RoutesClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(RoutesClient, is_async=True) + return self.create_client_from_credential( + RoutesClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/routes/routes/_model_base.py b/packages/typespec-python/test/azure/generated/routes/routes/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/routes/routes/_model_base.py +++ b/packages/typespec-python/test/azure/generated/routes/routes/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/serialization-encoded-name-json/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py new file mode 100644 index 00000000000..9e4e68059bf --- /dev/null +++ b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + json_subscription_id = os.environ.get("JSON_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + json_tenant_id = os.environ.get("JSON_TENANT_ID", "00000000-0000-0000-0000-000000000000") + json_client_id = os.environ.get("JSON_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + json_client_secret = os.environ.get("JSON_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=json_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=json_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=json_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=json_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py new file mode 100644 index 00000000000..ffa148b4644 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import JsonClientTestBase, JsonPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestJsonPropertyOperations(JsonClientTestBase): + @JsonPreparer() + @recorded_by_proxy + def test_property_send(self, json_endpoint): + client = self.create_client(endpoint=json_endpoint) + response = client.property.send( + body={"wireName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @JsonPreparer() + @recorded_by_proxy + def test_property_get(self, json_endpoint): + client = self.create_client(endpoint=json_endpoint) + response = client.property.get() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py new file mode 100644 index 00000000000..44de344c6f5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/test_json_property_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import JsonPreparer +from testpreparer_async import JsonClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestJsonPropertyOperationsAsync(JsonClientTestBaseAsync): + @JsonPreparer() + @recorded_by_proxy_async + async def test_property_send(self, json_endpoint): + client = self.create_async_client(endpoint=json_endpoint) + response = await client.property.send( + body={"wireName": bool}, + ) + + # please add some check logic here by yourself + # ... + + @JsonPreparer() + @recorded_by_proxy_async + async def test_property_get(self, json_endpoint): + client = self.create_async_client(endpoint=json_endpoint) + response = await client.property.get() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py new file mode 100644 index 00000000000..c451c83f0f7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from serialization.encodedname.json import JsonClient + + +class JsonClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(JsonClient) + return self.create_client_from_credential( + JsonClient, + credential=credential, + endpoint=endpoint, + ) + + +JsonPreparer = functools.partial(PowerShellPreparer, "json", json_endpoint="https://fake_json_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..ede52b11d00 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from serialization.encodedname.json.aio import JsonClient + + +class JsonClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(JsonClient, is_async=True) + return self.create_client_from_credential( + JsonClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py +++ b/packages/typespec-python/test/azure/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-endpoint-not-defined/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py new file mode 100644 index 00000000000..0ab89c65aa8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + notdefined_subscription_id = os.environ.get("NOTDEFINED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + notdefined_tenant_id = os.environ.get("NOTDEFINED_TENANT_ID", "00000000-0000-0000-0000-000000000000") + notdefined_client_id = os.environ.get("NOTDEFINED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + notdefined_client_secret = os.environ.get("NOTDEFINED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notdefined_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notdefined_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notdefined_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notdefined_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py new file mode 100644 index 00000000000..d8ac72c4999 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NotDefinedClientTestBase, NotDefinedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNotDefined(NotDefinedClientTestBase): + @NotDefinedPreparer() + @recorded_by_proxy + def test_valid(self, notdefined_endpoint): + client = self.create_client(endpoint=notdefined_endpoint) + response = client.valid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py new file mode 100644 index 00000000000..debae5f767c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/test_not_defined_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NotDefinedPreparer +from testpreparer_async import NotDefinedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNotDefinedAsync(NotDefinedClientTestBaseAsync): + @NotDefinedPreparer() + @recorded_by_proxy_async + async def test_valid(self, notdefined_endpoint): + client = self.create_async_client(endpoint=notdefined_endpoint) + response = await client.valid() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py new file mode 100644 index 00000000000..d376459f692 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from server.endpoint.notdefined import NotDefinedClient + + +class NotDefinedClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(NotDefinedClient) + return self.create_client_from_credential( + NotDefinedClient, + credential=credential, + endpoint=endpoint, + ) + + +NotDefinedPreparer = functools.partial( + PowerShellPreparer, "notdefined", notdefined_endpoint="https://fake_notdefined_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..18259ec1835 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from server.endpoint.notdefined.aio import NotDefinedClient + + +class NotDefinedClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(NotDefinedClient, is_async=True) + return self.create_client_from_credential( + NotDefinedClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py +++ b/packages/typespec-python/test/azure/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-path-multiple/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-path-multiple/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py new file mode 100644 index 00000000000..ecb5f89d885 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + multiple_subscription_id = os.environ.get("MULTIPLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + multiple_tenant_id = os.environ.get("MULTIPLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + multiple_client_id = os.environ.get("MULTIPLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + multiple_client_secret = os.environ.get("MULTIPLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multiple_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multiple_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multiple_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=multiple_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py new file mode 100644 index 00000000000..ff7fdbce871 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import MultipleClientTestBase, MultiplePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMultiple(MultipleClientTestBase): + @MultiplePreparer() + @recorded_by_proxy + def test_no_operation_params(self, multiple_endpoint): + client = self.create_client(endpoint=multiple_endpoint) + response = client.no_operation_params() + + # please add some check logic here by yourself + # ... + + @MultiplePreparer() + @recorded_by_proxy + def test_with_operation_path_param(self, multiple_endpoint): + client = self.create_client(endpoint=multiple_endpoint) + response = client.with_operation_path_param( + keyword="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py new file mode 100644 index 00000000000..af09dbc3f87 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/test_multiple_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import MultiplePreparer +from testpreparer_async import MultipleClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMultipleAsync(MultipleClientTestBaseAsync): + @MultiplePreparer() + @recorded_by_proxy_async + async def test_no_operation_params(self, multiple_endpoint): + client = self.create_async_client(endpoint=multiple_endpoint) + response = await client.no_operation_params() + + # please add some check logic here by yourself + # ... + + @MultiplePreparer() + @recorded_by_proxy_async + async def test_with_operation_path_param(self, multiple_endpoint): + client = self.create_async_client(endpoint=multiple_endpoint) + response = await client.with_operation_path_param( + keyword="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py new file mode 100644 index 00000000000..058cc89c40c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from server.path.multiple import MultipleClient + + +class MultipleClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(MultipleClient) + return self.create_client_from_credential( + MultipleClient, + credential=credential, + endpoint=endpoint, + ) + + +MultiplePreparer = functools.partial( + PowerShellPreparer, "multiple", multiple_endpoint="https://fake_multiple_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..3b5a7b2c131 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-multiple/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from server.path.multiple.aio import MultipleClient + + +class MultipleClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(MultipleClient, is_async=True) + return self.create_client_from_credential( + MultipleClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/_model_base.py b/packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/_model_base.py +++ b/packages/typespec-python/test/azure/generated/server-path-multiple/server/path/multiple/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/server-path-single/server/path/single/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-path-single/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-path-single/server/path/single/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-path-single/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py new file mode 100644 index 00000000000..2b689de1153 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + single_subscription_id = os.environ.get("SINGLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + single_tenant_id = os.environ.get("SINGLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + single_client_id = os.environ.get("SINGLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + single_client_secret = os.environ.get("SINGLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=single_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=single_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=single_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=single_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py new file mode 100644 index 00000000000..e6c45a6e182 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SingleClientTestBase, SinglePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSingle(SingleClientTestBase): + @SinglePreparer() + @recorded_by_proxy + def test_my_op(self, single_endpoint): + client = self.create_client(endpoint=single_endpoint) + response = client.my_op() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py new file mode 100644 index 00000000000..b841788ad37 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/test_single_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SinglePreparer +from testpreparer_async import SingleClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSingleAsync(SingleClientTestBaseAsync): + @SinglePreparer() + @recorded_by_proxy_async + async def test_my_op(self, single_endpoint): + client = self.create_async_client(endpoint=single_endpoint) + response = await client.my_op() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py new file mode 100644 index 00000000000..37982e18354 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from server.path.single import SingleClient + + +class SingleClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(SingleClient) + return self.create_client_from_credential( + SingleClient, + credential=credential, + endpoint=endpoint, + ) + + +SinglePreparer = functools.partial(PowerShellPreparer, "single", single_endpoint="https://fake_single_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..131dc39a8ce --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-path-single/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from server.path.single.aio import SingleClient + + +class SingleClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(SingleClient, is_async=True) + return self.create_client_from_credential( + SingleClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/server-path-single/server/path/single/_model_base.py b/packages/typespec-python/test/azure/generated/server-path-single/server/path/single/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/server-path-single/server/path/single/_model_base.py +++ b/packages/typespec-python/test/azure/generated/server-path-single/server/path/single/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-versions-not-versioned/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py new file mode 100644 index 00000000000..113408d780d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/conftest.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + notversioned_subscription_id = os.environ.get( + "NOTVERSIONED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + notversioned_tenant_id = os.environ.get("NOTVERSIONED_TENANT_ID", "00000000-0000-0000-0000-000000000000") + notversioned_client_id = os.environ.get("NOTVERSIONED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + notversioned_client_secret = os.environ.get("NOTVERSIONED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notversioned_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notversioned_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notversioned_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notversioned_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py new file mode 100644 index 00000000000..0168620cb05 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned.py @@ -0,0 +1,44 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NotVersionedClientTestBase, NotVersionedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNotVersioned(NotVersionedClientTestBase): + @NotVersionedPreparer() + @recorded_by_proxy + def test_without_api_version(self, notversioned_endpoint): + client = self.create_client(endpoint=notversioned_endpoint) + response = client.without_api_version() + + # please add some check logic here by yourself + # ... + + @NotVersionedPreparer() + @recorded_by_proxy + def test_with_query_api_version(self, notversioned_endpoint): + client = self.create_client(endpoint=notversioned_endpoint) + response = client.with_query_api_version( + api_version="str", + ) + + # please add some check logic here by yourself + # ... + + @NotVersionedPreparer() + @recorded_by_proxy + def test_with_path_api_version(self, notversioned_endpoint): + client = self.create_client(endpoint=notversioned_endpoint) + response = client.with_path_api_version( + api_version="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py new file mode 100644 index 00000000000..10167440b12 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/test_not_versioned_async.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NotVersionedPreparer +from testpreparer_async import NotVersionedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNotVersionedAsync(NotVersionedClientTestBaseAsync): + @NotVersionedPreparer() + @recorded_by_proxy_async + async def test_without_api_version(self, notversioned_endpoint): + client = self.create_async_client(endpoint=notversioned_endpoint) + response = await client.without_api_version() + + # please add some check logic here by yourself + # ... + + @NotVersionedPreparer() + @recorded_by_proxy_async + async def test_with_query_api_version(self, notversioned_endpoint): + client = self.create_async_client(endpoint=notversioned_endpoint) + response = await client.with_query_api_version( + api_version="str", + ) + + # please add some check logic here by yourself + # ... + + @NotVersionedPreparer() + @recorded_by_proxy_async + async def test_with_path_api_version(self, notversioned_endpoint): + client = self.create_async_client(endpoint=notversioned_endpoint) + response = await client.with_path_api_version( + api_version="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py new file mode 100644 index 00000000000..d60bf7abed6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from server.versions.notversioned import NotVersionedClient + + +class NotVersionedClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(NotVersionedClient) + return self.create_client_from_credential( + NotVersionedClient, + credential=credential, + endpoint=endpoint, + ) + + +NotVersionedPreparer = functools.partial( + PowerShellPreparer, "notversioned", notversioned_endpoint="https://fake_notversioned_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..4d083176898 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from server.versions.notversioned.aio import NotVersionedClient + + +class NotVersionedClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(NotVersionedClient, is_async=True) + return self.create_client_from_credential( + NotVersionedClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py +++ b/packages/typespec-python/test/azure/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/server-versions-versioned/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/server-versions-versioned/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py new file mode 100644 index 00000000000..b21f87d9301 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + versioned_subscription_id = os.environ.get("VERSIONED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + versioned_tenant_id = os.environ.get("VERSIONED_TENANT_ID", "00000000-0000-0000-0000-000000000000") + versioned_client_id = os.environ.get("VERSIONED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + versioned_client_secret = os.environ.get("VERSIONED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=versioned_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=versioned_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=versioned_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=versioned_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py new file mode 100644 index 00000000000..c3377064708 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned.py @@ -0,0 +1,49 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import VersionedClientTestBase, VersionedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVersioned(VersionedClientTestBase): + @VersionedPreparer() + @recorded_by_proxy + def test_without_api_version(self, versioned_endpoint): + client = self.create_client(endpoint=versioned_endpoint) + response = client.without_api_version() + + # please add some check logic here by yourself + # ... + + @VersionedPreparer() + @recorded_by_proxy + def test_with_query_api_version(self, versioned_endpoint): + client = self.create_client(endpoint=versioned_endpoint) + response = client.with_query_api_version() + + # please add some check logic here by yourself + # ... + + @VersionedPreparer() + @recorded_by_proxy + def test_with_path_api_version(self, versioned_endpoint): + client = self.create_client(endpoint=versioned_endpoint) + response = client.with_path_api_version() + + # please add some check logic here by yourself + # ... + + @VersionedPreparer() + @recorded_by_proxy + def test_with_query_old_api_version(self, versioned_endpoint): + client = self.create_client(endpoint=versioned_endpoint) + response = client.with_query_old_api_version() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py new file mode 100644 index 00000000000..b5a2f83d581 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/test_versioned_async.py @@ -0,0 +1,50 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import VersionedPreparer +from testpreparer_async import VersionedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVersionedAsync(VersionedClientTestBaseAsync): + @VersionedPreparer() + @recorded_by_proxy_async + async def test_without_api_version(self, versioned_endpoint): + client = self.create_async_client(endpoint=versioned_endpoint) + response = await client.without_api_version() + + # please add some check logic here by yourself + # ... + + @VersionedPreparer() + @recorded_by_proxy_async + async def test_with_query_api_version(self, versioned_endpoint): + client = self.create_async_client(endpoint=versioned_endpoint) + response = await client.with_query_api_version() + + # please add some check logic here by yourself + # ... + + @VersionedPreparer() + @recorded_by_proxy_async + async def test_with_path_api_version(self, versioned_endpoint): + client = self.create_async_client(endpoint=versioned_endpoint) + response = await client.with_path_api_version() + + # please add some check logic here by yourself + # ... + + @VersionedPreparer() + @recorded_by_proxy_async + async def test_with_query_old_api_version(self, versioned_endpoint): + client = self.create_async_client(endpoint=versioned_endpoint) + response = await client.with_query_old_api_version() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py new file mode 100644 index 00000000000..8d7f9d4965f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from server.versions.versioned import VersionedClient + + +class VersionedClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(VersionedClient) + return self.create_client_from_credential( + VersionedClient, + credential=credential, + endpoint=endpoint, + ) + + +VersionedPreparer = functools.partial( + PowerShellPreparer, "versioned", versioned_endpoint="https://fake_versioned_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..b0610e33752 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/server-versions-versioned/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from server.versions.versioned.aio import VersionedClient + + +class VersionedClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(VersionedClient, is_async=True) + return self.create_client_from_credential( + VersionedClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/_model_base.py b/packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/_model_base.py +++ b/packages/typespec-python/test/azure/generated/server-versions-versioned/server/versions/versioned/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/special-headers-conditional-request/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py new file mode 100644 index 00000000000..732c2c5da0d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/conftest.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + conditionalrequest_subscription_id = os.environ.get( + "CONDITIONALREQUEST_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + conditionalrequest_tenant_id = os.environ.get( + "CONDITIONALREQUEST_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + conditionalrequest_client_id = os.environ.get( + "CONDITIONALREQUEST_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + conditionalrequest_client_secret = os.environ.get( + "CONDITIONALREQUEST_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=conditionalrequest_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=conditionalrequest_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=conditionalrequest_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=conditionalrequest_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py new file mode 100644 index 00000000000..3850898ec82 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request.py @@ -0,0 +1,49 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ConditionalRequestClientTestBase, ConditionalRequestPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestConditionalRequest(ConditionalRequestClientTestBase): + @ConditionalRequestPreparer() + @recorded_by_proxy + def test_post_if_match(self, conditionalrequest_endpoint): + client = self.create_client(endpoint=conditionalrequest_endpoint) + response = client.post_if_match() + + # please add some check logic here by yourself + # ... + + @ConditionalRequestPreparer() + @recorded_by_proxy + def test_post_if_none_match(self, conditionalrequest_endpoint): + client = self.create_client(endpoint=conditionalrequest_endpoint) + response = client.post_if_none_match() + + # please add some check logic here by yourself + # ... + + @ConditionalRequestPreparer() + @recorded_by_proxy + def test_head_if_modified_since(self, conditionalrequest_endpoint): + client = self.create_client(endpoint=conditionalrequest_endpoint) + response = client.head_if_modified_since() + + # please add some check logic here by yourself + # ... + + @ConditionalRequestPreparer() + @recorded_by_proxy + def test_post_if_unmodified_since(self, conditionalrequest_endpoint): + client = self.create_client(endpoint=conditionalrequest_endpoint) + response = client.post_if_unmodified_since() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py new file mode 100644 index 00000000000..714becb95a0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/test_conditional_request_async.py @@ -0,0 +1,50 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ConditionalRequestPreparer +from testpreparer_async import ConditionalRequestClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestConditionalRequestAsync(ConditionalRequestClientTestBaseAsync): + @ConditionalRequestPreparer() + @recorded_by_proxy_async + async def test_post_if_match(self, conditionalrequest_endpoint): + client = self.create_async_client(endpoint=conditionalrequest_endpoint) + response = await client.post_if_match() + + # please add some check logic here by yourself + # ... + + @ConditionalRequestPreparer() + @recorded_by_proxy_async + async def test_post_if_none_match(self, conditionalrequest_endpoint): + client = self.create_async_client(endpoint=conditionalrequest_endpoint) + response = await client.post_if_none_match() + + # please add some check logic here by yourself + # ... + + @ConditionalRequestPreparer() + @recorded_by_proxy_async + async def test_head_if_modified_since(self, conditionalrequest_endpoint): + client = self.create_async_client(endpoint=conditionalrequest_endpoint) + response = await client.head_if_modified_since() + + # please add some check logic here by yourself + # ... + + @ConditionalRequestPreparer() + @recorded_by_proxy_async + async def test_post_if_unmodified_since(self, conditionalrequest_endpoint): + client = self.create_async_client(endpoint=conditionalrequest_endpoint) + response = await client.post_if_unmodified_since() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py new file mode 100644 index 00000000000..147f64006bd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specialheaders.conditionalrequest import ConditionalRequestClient + + +class ConditionalRequestClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ConditionalRequestClient) + return self.create_client_from_credential( + ConditionalRequestClient, + credential=credential, + endpoint=endpoint, + ) + + +ConditionalRequestPreparer = functools.partial( + PowerShellPreparer, "conditionalrequest", conditionalrequest_endpoint="https://fake_conditionalrequest_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..9d23fc53553 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specialheaders.conditionalrequest.aio import ConditionalRequestClient + + +class ConditionalRequestClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ConditionalRequestClient, is_async=True) + return self.create_client_from_credential( + ConditionalRequestClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py +++ b/packages/typespec-python/test/azure/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/special-headers-repeatability/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/special-headers-repeatability/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py new file mode 100644 index 00000000000..420e0fefcb5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/conftest.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + repeatability_subscription_id = os.environ.get( + "REPEATABILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + repeatability_tenant_id = os.environ.get("REPEATABILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + repeatability_client_id = os.environ.get("REPEATABILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + repeatability_client_secret = os.environ.get("REPEATABILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=repeatability_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=repeatability_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=repeatability_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=repeatability_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py new file mode 100644 index 00000000000..f90fb946847 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability.py @@ -0,0 +1,22 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RepeatabilityClientTestBase, RepeatabilityPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRepeatability(RepeatabilityClientTestBase): + @RepeatabilityPreparer() + @recorded_by_proxy + def test_immediate_success(self, repeatability_endpoint): + client = self.create_client(endpoint=repeatability_endpoint) + response = client.immediate_success() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py new file mode 100644 index 00000000000..96a827ca7de --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/test_repeatability_async.py @@ -0,0 +1,23 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RepeatabilityPreparer +from testpreparer_async import RepeatabilityClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRepeatabilityAsync(RepeatabilityClientTestBaseAsync): + @RepeatabilityPreparer() + @recorded_by_proxy_async + async def test_immediate_success(self, repeatability_endpoint): + client = self.create_async_client(endpoint=repeatability_endpoint) + response = await client.immediate_success() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py new file mode 100644 index 00000000000..20f2aa53457 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specialheaders.repeatability import RepeatabilityClient + + +class RepeatabilityClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(RepeatabilityClient) + return self.create_client_from_credential( + RepeatabilityClient, + credential=credential, + endpoint=endpoint, + ) + + +RepeatabilityPreparer = functools.partial( + PowerShellPreparer, "repeatability", repeatability_endpoint="https://fake_repeatability_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..1aee05e0694 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-headers-repeatability/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specialheaders.repeatability.aio import RepeatabilityClient + + +class RepeatabilityClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(RepeatabilityClient, is_async=True) + return self.create_client_from_credential( + RepeatabilityClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py b/packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py +++ b/packages/typespec-python/test/azure/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/special-words/specialwords/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/special-words/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/special-words/specialwords/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/special-words/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py new file mode 100644 index 00000000000..299af9f3cf3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/conftest.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + specialwords_subscription_id = os.environ.get( + "SPECIALWORDS_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + specialwords_tenant_id = os.environ.get("SPECIALWORDS_TENANT_ID", "00000000-0000-0000-0000-000000000000") + specialwords_client_id = os.environ.get("SPECIALWORDS_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + specialwords_client_secret = os.environ.get("SPECIALWORDS_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=specialwords_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=specialwords_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=specialwords_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=specialwords_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py new file mode 100644 index 00000000000..b0d93f73c52 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsModelPropertiesOperations(SpecialWordsClientTestBase): + @SpecialWordsPreparer() + @recorded_by_proxy + def test_model_properties_same_as_model(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.model_properties.same_as_model( + body={"SameAsModel": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py new file mode 100644 index 00000000000..fe0b9e2887b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_model_properties_operations_async.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SpecialWordsPreparer +from testpreparer_async import SpecialWordsClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsModelPropertiesOperationsAsync(SpecialWordsClientTestBaseAsync): + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_model_properties_same_as_model(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.model_properties.same_as_model( + body={"SameAsModel": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py new file mode 100644 index 00000000000..1e4b1cce356 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations.py @@ -0,0 +1,376 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsModelsOperations(SpecialWordsClientTestBase): + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_and(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_and( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_as(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_as( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_assert(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_assert( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_async(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_async( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_await(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_await( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_break(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_break( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_class(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_class( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_constructor(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_constructor( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_continue(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_continue( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_def(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_def( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_del(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_del( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_elif(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_elif( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_else(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_else( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_except(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_except( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_exec(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_exec( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_finally(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_finally( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_for(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_for( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_from(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_from( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_global(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_global( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_if(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_if( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_import(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_import( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_in(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_in( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_is(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_is( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_lambda(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_lambda( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_not(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_not( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_or(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_or( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_pass(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_pass( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_raise(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_raise( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_return(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_return( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_try(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_try( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_while(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_while( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_with(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_with( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_models_with_yield(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.models.with_yield( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py new file mode 100644 index 00000000000..36c089dc3a4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_models_operations_async.py @@ -0,0 +1,377 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SpecialWordsPreparer +from testpreparer_async import SpecialWordsClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsModelsOperationsAsync(SpecialWordsClientTestBaseAsync): + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_and(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_and( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_as(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_as( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_assert(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_assert( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_async(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_async( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_await(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_await( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_break(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_break( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_class(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_class( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_constructor(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_constructor( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_continue(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_continue( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_def(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_def( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_del(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_del( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_elif(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_elif( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_else(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_else( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_except(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_except( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_exec(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_exec( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_finally(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_finally( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_for(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_for( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_from(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_from( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_global(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_global( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_if(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_if( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_import(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_import( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_in(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_in( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_is(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_is( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_lambda(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_lambda( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_not(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_not( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_or(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_or( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_pass(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_pass( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_raise(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_raise( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_return(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_return( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_try(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_try( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_while(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_while( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_with(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_with( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_models_with_yield(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.models.with_yield( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py new file mode 100644 index 00000000000..e73807af358 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations.py @@ -0,0 +1,310 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsOperations(SpecialWordsClientTestBase): + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_and_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.and_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_as_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.as_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_assert_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.assert_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_async_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.async_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_await_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.await_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_break_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.break_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_class_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.class_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_constructor(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.constructor() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_continue_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.continue_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_def_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.def_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_del_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.del_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_elif_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.elif_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_else_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.else_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_except_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.except_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_exec_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.exec_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_finally_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.finally_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_for_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.for_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_from_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.from_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_global_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.global_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_if_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.if_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_import_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.import_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_in_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.in_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_is_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.is_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_lambda_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.lambda_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_not_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.not_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_or_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.or_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_pass_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.pass_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_raise_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.raise_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_return_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.return_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_try_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.try_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_while_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.while_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_with_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.with_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_operations_yield_method(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.operations.yield_method() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py new file mode 100644 index 00000000000..1404d0b1ef8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_operations_async.py @@ -0,0 +1,311 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SpecialWordsPreparer +from testpreparer_async import SpecialWordsClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsOperationsAsync(SpecialWordsClientTestBaseAsync): + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_and_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.and_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_as_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.as_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_assert_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.assert_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_async_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.async_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_await_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.await_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_break_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.break_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_class_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.class_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_constructor(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.constructor() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_continue_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.continue_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_def_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.def_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_del_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.del_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_elif_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.elif_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_else_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.else_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_except_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.except_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_exec_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.exec_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_finally_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.finally_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_for_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.for_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_from_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.from_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_global_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.global_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_if_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.if_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_import_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.import_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_in_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.in_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_is_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.is_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_lambda_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.lambda_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_not_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.not_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_or_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.or_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_pass_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.pass_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_raise_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.raise_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_return_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.return_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_try_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.try_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_while_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.while_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_with_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.with_method() + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_operations_yield_method(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.operations.yield_method() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py new file mode 100644 index 00000000000..3096a60a28e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations.py @@ -0,0 +1,387 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SpecialWordsClientTestBase, SpecialWordsPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsParametersOperations(SpecialWordsClientTestBase): + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_and(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_and( + and_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_as(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_as( + as_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_assert(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_assert( + assert_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_async(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_async( + async_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_await(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_await( + await_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_break(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_break( + break_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_class(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_class( + class_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_constructor(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_constructor( + constructor="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_continue(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_continue( + continue_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_def(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_def( + def_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_del(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_del( + del_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_elif(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_elif( + elif_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_else(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_else( + else_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_except(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_except( + except_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_exec(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_exec( + exec_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_finally(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_finally( + finally_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_for(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_for( + for_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_from(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_from( + from_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_global(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_global( + global_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_if(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_if( + if_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_import(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_import( + import_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_in(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_in( + in_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_is(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_is( + is_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_lambda(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_lambda( + lambda_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_not(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_not( + not_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_or(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_or( + or_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_pass(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_pass( + pass_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_raise(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_raise( + raise_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_return(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_return( + return_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_try(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_try( + try_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_while(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_while( + while_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_with(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_with( + with_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_yield(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_yield( + yield_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy + def test_parameters_with_cancellation_token(self, specialwords_endpoint): + client = self.create_client(endpoint=specialwords_endpoint) + response = client.parameters.with_cancellation_token( + cancellation_token="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py new file mode 100644 index 00000000000..94ca98849b1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/test_special_words_parameters_operations_async.py @@ -0,0 +1,388 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SpecialWordsPreparer +from testpreparer_async import SpecialWordsClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSpecialWordsParametersOperationsAsync(SpecialWordsClientTestBaseAsync): + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_and(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_and( + and_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_as(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_as( + as_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_assert(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_assert( + assert_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_async(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_async( + async_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_await(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_await( + await_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_break(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_break( + break_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_class(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_class( + class_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_constructor(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_constructor( + constructor="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_continue(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_continue( + continue_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_def(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_def( + def_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_del(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_del( + del_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_elif(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_elif( + elif_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_else(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_else( + else_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_except(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_except( + except_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_exec(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_exec( + exec_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_finally(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_finally( + finally_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_for(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_for( + for_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_from(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_from( + from_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_global(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_global( + global_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_if(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_if( + if_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_import(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_import( + import_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_in(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_in( + in_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_is(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_is( + is_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_lambda(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_lambda( + lambda_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_not(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_not( + not_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_or(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_or( + or_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_pass(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_pass( + pass_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_raise(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_raise( + raise_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_return(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_return( + return_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_try(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_try( + try_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_while(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_while( + while_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_with(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_with( + with_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_yield(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_yield( + yield_parameter="str", + ) + + # please add some check logic here by yourself + # ... + + @SpecialWordsPreparer() + @recorded_by_proxy_async + async def test_parameters_with_cancellation_token(self, specialwords_endpoint): + client = self.create_async_client(endpoint=specialwords_endpoint) + response = await client.parameters.with_cancellation_token( + cancellation_token="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py new file mode 100644 index 00000000000..09070149dec --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from specialwords import SpecialWordsClient + + +class SpecialWordsClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(SpecialWordsClient) + return self.create_client_from_credential( + SpecialWordsClient, + credential=credential, + endpoint=endpoint, + ) + + +SpecialWordsPreparer = functools.partial( + PowerShellPreparer, "specialwords", specialwords_endpoint="https://fake_specialwords_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..877ce52b4e9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/special-words/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from specialwords.aio import SpecialWordsClient + + +class SpecialWordsClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(SpecialWordsClient, is_async=True) + return self.create_client_from_credential( + SpecialWordsClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/special-words/specialwords/_model_base.py b/packages/typespec-python/test/azure/generated/special-words/specialwords/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/special-words/specialwords/_model_base.py +++ b/packages/typespec-python/test/azure/generated/special-words/specialwords/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-array/typetest/array/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-array/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-array/typetest/array/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-array/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py new file mode 100644 index 00000000000..76591609ab0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + array_subscription_id = os.environ.get("ARRAY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + array_tenant_id = os.environ.get("ARRAY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + array_client_id = os.environ.get("ARRAY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + array_client_secret = os.environ.get("ARRAY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=array_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=array_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=array_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=array_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py new file mode 100644 index 00000000000..a92d8a1de47 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayBooleanValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_boolean_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.boolean_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_boolean_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.boolean_value.put( + body=[bool], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py new file mode 100644 index 00000000000..dbb00a2f7d6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_boolean_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayBooleanValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_boolean_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.boolean_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_boolean_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.boolean_value.put( + body=[bool], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py new file mode 100644 index 00000000000..51c85df3067 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayDatetimeValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_datetime_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.datetime_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_datetime_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.datetime_value.put( + body=["2020-02-20 00:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py new file mode 100644 index 00000000000..c5143d890c3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_datetime_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayDatetimeValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_datetime_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.datetime_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_datetime_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.datetime_value.put( + body=["2020-02-20 00:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py new file mode 100644 index 00000000000..e5f5d6f5398 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayDurationValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_duration_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.duration_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_duration_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.duration_value.put( + body=["1 day, 0:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py new file mode 100644 index 00000000000..d6dca6b97bb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_duration_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayDurationValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_duration_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.duration_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_duration_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.duration_value.put( + body=["1 day, 0:00:00"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py new file mode 100644 index 00000000000..7667d59ab18 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayFloat32ValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_float32_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.float32_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_float32_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.float32_value.put( + body=[0.0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py new file mode 100644 index 00000000000..32d2941545f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_float32_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayFloat32ValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_float32_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.float32_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_float32_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.float32_value.put( + body=[0.0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py new file mode 100644 index 00000000000..5d3d8e7b084 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayInt32ValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_int32_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.int32_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_int32_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.int32_value.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py new file mode 100644 index 00000000000..945de66811c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int32_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayInt32ValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_int32_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.int32_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_int32_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.int32_value.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py new file mode 100644 index 00000000000..f7e0c27b1c4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayInt64ValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_int64_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.int64_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_int64_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.int64_value.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py new file mode 100644 index 00000000000..0a65a3409bc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_int64_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayInt64ValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_int64_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.int64_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_int64_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.int64_value.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py new file mode 100644 index 00000000000..902a16cc99b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayModelValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_model_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.model_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_model_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.model_value.put( + body=[{"property": "str", "children": [...]}], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py new file mode 100644 index 00000000000..3f3fd3c8302 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_model_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayModelValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_model_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.model_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_model_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.model_value.put( + body=[{"property": "str", "children": [...]}], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py new file mode 100644 index 00000000000..424edfcac74 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableBooleanValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_boolean_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_boolean_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_boolean_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_boolean_value.put( + body=[bool], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py new file mode 100644 index 00000000000..00f98afecd6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_boolean_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableBooleanValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_boolean_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_boolean_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_boolean_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_boolean_value.put( + body=[bool], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py new file mode 100644 index 00000000000..22e6eb6f081 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableFloatValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_float_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_float_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_float_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_float_value.put( + body=[0.0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py new file mode 100644 index 00000000000..51eb6323a14 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_float_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableFloatValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_float_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_float_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_float_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_float_value.put( + body=[0.0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py new file mode 100644 index 00000000000..b15da9d989f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableInt32ValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_int32_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_int32_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_int32_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_int32_value.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py new file mode 100644 index 00000000000..99ab79992ad --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_int32_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableInt32ValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_int32_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_int32_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_int32_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_int32_value.put( + body=[0], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py new file mode 100644 index 00000000000..5af02b802ea --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableModelValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_model_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_model_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_model_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_model_value.put( + body=[{"property": "str", "children": [...]}], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py new file mode 100644 index 00000000000..add8c12e990 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_model_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableModelValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_model_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_model_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_model_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_model_value.put( + body=[{"property": "str", "children": [...]}], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py new file mode 100644 index 00000000000..6f822baef0c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableStringValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_string_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_string_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_nullable_string_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.nullable_string_value.put( + body=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py new file mode 100644 index 00000000000..872997efda8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_nullable_string_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayNullableStringValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_string_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_string_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_nullable_string_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.nullable_string_value.put( + body=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py new file mode 100644 index 00000000000..88918304f26 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayStringValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_string_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.string_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_string_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.string_value.put( + body=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py new file mode 100644 index 00000000000..fd99dcd54fc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_string_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayStringValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_string_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.string_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_string_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.string_value.put( + body=["str"], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py new file mode 100644 index 00000000000..2a018ba6fa0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ArrayClientTestBase, ArrayPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayUnknownValueOperations(ArrayClientTestBase): + @ArrayPreparer() + @recorded_by_proxy + def test_unknown_value_get(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.unknown_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy + def test_unknown_value_put(self, array_endpoint): + client = self.create_client(endpoint=array_endpoint) + response = client.unknown_value.put( + body=[{}], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py new file mode 100644 index 00000000000..ffbc2d2c7df --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/test_array_unknown_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ArrayPreparer +from testpreparer_async import ArrayClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestArrayUnknownValueOperationsAsync(ArrayClientTestBaseAsync): + @ArrayPreparer() + @recorded_by_proxy_async + async def test_unknown_value_get(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.unknown_value.get() + + # please add some check logic here by yourself + # ... + + @ArrayPreparer() + @recorded_by_proxy_async + async def test_unknown_value_put(self, array_endpoint): + client = self.create_async_client(endpoint=array_endpoint) + response = await client.unknown_value.put( + body=[{}], + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py new file mode 100644 index 00000000000..7e657f1d6f7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.array import ArrayClient + + +class ArrayClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ArrayClient) + return self.create_client_from_credential( + ArrayClient, + credential=credential, + endpoint=endpoint, + ) + + +ArrayPreparer = functools.partial(PowerShellPreparer, "array", array_endpoint="https://fake_array_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..b0e2f641713 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-array/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.array.aio import ArrayClient + + +class ArrayClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ArrayClient, is_async=True) + return self.create_client_from_credential( + ArrayClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-array/typetest/array/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-array/typetest/array/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-array/typetest/array/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-array/typetest/array/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-dictionary/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-dictionary/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py new file mode 100644 index 00000000000..694a0f94554 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + dictionary_subscription_id = os.environ.get("DICTIONARY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + dictionary_tenant_id = os.environ.get("DICTIONARY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + dictionary_client_id = os.environ.get("DICTIONARY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + dictionary_client_secret = os.environ.get("DICTIONARY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=dictionary_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=dictionary_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=dictionary_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=dictionary_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py new file mode 100644 index 00000000000..066f228c89a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryBooleanValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_boolean_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.boolean_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_boolean_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.boolean_value.put( + body={"str": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py new file mode 100644 index 00000000000..ea0403b733d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_boolean_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryBooleanValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_boolean_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.boolean_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_boolean_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.boolean_value.put( + body={"str": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py new file mode 100644 index 00000000000..30fb62304ce --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryDatetimeValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_datetime_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.datetime_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_datetime_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.datetime_value.put( + body={"str": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py new file mode 100644 index 00000000000..3e83eefc827 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_datetime_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryDatetimeValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_datetime_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.datetime_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_datetime_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.datetime_value.put( + body={"str": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py new file mode 100644 index 00000000000..b0b3e702188 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryDurationValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_duration_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.duration_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_duration_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.duration_value.put( + body={"str": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py new file mode 100644 index 00000000000..0d5394008dd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_duration_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryDurationValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_duration_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.duration_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_duration_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.duration_value.put( + body={"str": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py new file mode 100644 index 00000000000..859239e7348 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryFloat32ValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_float32_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.float32_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_float32_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.float32_value.put( + body={"str": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py new file mode 100644 index 00000000000..a52a03e857f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_float32_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryFloat32ValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_float32_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.float32_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_float32_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.float32_value.put( + body={"str": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py new file mode 100644 index 00000000000..a2199a45778 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryInt32ValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_int32_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.int32_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_int32_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.int32_value.put( + body={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py new file mode 100644 index 00000000000..77d4bb4577d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int32_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryInt32ValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_int32_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.int32_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_int32_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.int32_value.put( + body={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py new file mode 100644 index 00000000000..6f7322982d7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryInt64ValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_int64_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.int64_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_int64_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.int64_value.put( + body={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py new file mode 100644 index 00000000000..12b6e15ef77 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_int64_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryInt64ValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_int64_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.int64_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_int64_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.int64_value.put( + body={"str": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py new file mode 100644 index 00000000000..040afe8e20b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryModelValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_model_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.model_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_model_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.model_value.put( + body={"str": {"property": "str", "children": {"str": ...}}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py new file mode 100644 index 00000000000..54fbc74db5b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_model_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryModelValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_model_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.model_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_model_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.model_value.put( + body={"str": {"property": "str", "children": {"str": ...}}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py new file mode 100644 index 00000000000..3f17f0f6982 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryNullableFloatValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_nullable_float_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.nullable_float_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_nullable_float_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.nullable_float_value.put( + body={"str": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py new file mode 100644 index 00000000000..1e8f5f07b35 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_nullable_float_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryNullableFloatValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_nullable_float_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.nullable_float_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_nullable_float_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.nullable_float_value.put( + body={"str": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py new file mode 100644 index 00000000000..fe00987385c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryRecursiveModelValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_recursive_model_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.recursive_model_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_recursive_model_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.recursive_model_value.put( + body={"str": {"property": "str", "children": {"str": ...}}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py new file mode 100644 index 00000000000..63ed9c3ca9f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_recursive_model_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryRecursiveModelValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_recursive_model_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.recursive_model_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_recursive_model_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.recursive_model_value.put( + body={"str": {"property": "str", "children": {"str": ...}}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py new file mode 100644 index 00000000000..6309558db05 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryStringValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_string_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.string_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_string_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.string_value.put( + body={"str": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py new file mode 100644 index 00000000000..91fdb59db9f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_string_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryStringValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_string_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.string_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_string_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.string_value.put( + body={"str": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py new file mode 100644 index 00000000000..e84f0564e91 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import DictionaryClientTestBase, DictionaryPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryUnknownValueOperations(DictionaryClientTestBase): + @DictionaryPreparer() + @recorded_by_proxy + def test_unknown_value_get(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.unknown_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy + def test_unknown_value_put(self, dictionary_endpoint): + client = self.create_client(endpoint=dictionary_endpoint) + response = client.unknown_value.put( + body={"str": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py new file mode 100644 index 00000000000..6bd7a2bf28a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/test_dictionary_unknown_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import DictionaryPreparer +from testpreparer_async import DictionaryClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestDictionaryUnknownValueOperationsAsync(DictionaryClientTestBaseAsync): + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_unknown_value_get(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.unknown_value.get() + + # please add some check logic here by yourself + # ... + + @DictionaryPreparer() + @recorded_by_proxy_async + async def test_unknown_value_put(self, dictionary_endpoint): + client = self.create_async_client(endpoint=dictionary_endpoint) + response = await client.unknown_value.put( + body={"str": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py new file mode 100644 index 00000000000..8f9f9085cc3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.dictionary import DictionaryClient + + +class DictionaryClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(DictionaryClient) + return self.create_client_from_credential( + DictionaryClient, + credential=credential, + endpoint=endpoint, + ) + + +DictionaryPreparer = functools.partial( + PowerShellPreparer, "dictionary", dictionary_endpoint="https://fake_dictionary_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..d6149b7d469 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.dictionary.aio import DictionaryClient + + +class DictionaryClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(DictionaryClient, is_async=True) + return self.create_client_from_credential( + DictionaryClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-dictionary/typetest/dictionary/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-enum-extensible/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py new file mode 100644 index 00000000000..ef77f38e7b8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + extensible_subscription_id = os.environ.get("EXTENSIBLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + extensible_tenant_id = os.environ.get("EXTENSIBLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + extensible_client_id = os.environ.get("EXTENSIBLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + extensible_client_secret = os.environ.get("EXTENSIBLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=extensible_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=extensible_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=extensible_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=extensible_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py new file mode 100644 index 00000000000..be910b162cb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations.py @@ -0,0 +1,55 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ExtensibleClientTestBase, ExtensiblePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestExtensibleStringOperations(ExtensibleClientTestBase): + @ExtensiblePreparer() + @recorded_by_proxy + def test_string_get_known_value(self, extensible_endpoint): + client = self.create_client(endpoint=extensible_endpoint) + response = client.string.get_known_value() + + # please add some check logic here by yourself + # ... + + @ExtensiblePreparer() + @recorded_by_proxy + def test_string_get_unknown_value(self, extensible_endpoint): + client = self.create_client(endpoint=extensible_endpoint) + response = client.string.get_unknown_value() + + # please add some check logic here by yourself + # ... + + @ExtensiblePreparer() + @recorded_by_proxy + def test_string_put_known_value(self, extensible_endpoint): + client = self.create_client(endpoint=extensible_endpoint) + response = client.string.put_known_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ExtensiblePreparer() + @recorded_by_proxy + def test_string_put_unknown_value(self, extensible_endpoint): + client = self.create_client(endpoint=extensible_endpoint) + response = client.string.put_unknown_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py new file mode 100644 index 00000000000..3e12585bf55 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/test_extensible_string_operations_async.py @@ -0,0 +1,56 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ExtensiblePreparer +from testpreparer_async import ExtensibleClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestExtensibleStringOperationsAsync(ExtensibleClientTestBaseAsync): + @ExtensiblePreparer() + @recorded_by_proxy_async + async def test_string_get_known_value(self, extensible_endpoint): + client = self.create_async_client(endpoint=extensible_endpoint) + response = await client.string.get_known_value() + + # please add some check logic here by yourself + # ... + + @ExtensiblePreparer() + @recorded_by_proxy_async + async def test_string_get_unknown_value(self, extensible_endpoint): + client = self.create_async_client(endpoint=extensible_endpoint) + response = await client.string.get_unknown_value() + + # please add some check logic here by yourself + # ... + + @ExtensiblePreparer() + @recorded_by_proxy_async + async def test_string_put_known_value(self, extensible_endpoint): + client = self.create_async_client(endpoint=extensible_endpoint) + response = await client.string.put_known_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ExtensiblePreparer() + @recorded_by_proxy_async + async def test_string_put_unknown_value(self, extensible_endpoint): + client = self.create_async_client(endpoint=extensible_endpoint) + response = await client.string.put_unknown_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py new file mode 100644 index 00000000000..67cc900aa4c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.enum.extensible import ExtensibleClient + + +class ExtensibleClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ExtensibleClient) + return self.create_client_from_credential( + ExtensibleClient, + credential=credential, + endpoint=endpoint, + ) + + +ExtensiblePreparer = functools.partial( + PowerShellPreparer, "extensible", extensible_endpoint="https://fake_extensible_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..2ffefb6d6b1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.enum.extensible.aio import ExtensibleClient + + +class ExtensibleClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ExtensibleClient, is_async=True) + return self.create_client_from_credential( + ExtensibleClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-enum-fixed/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py new file mode 100644 index 00000000000..4dd457b05f4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + fixed_subscription_id = os.environ.get("FIXED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + fixed_tenant_id = os.environ.get("FIXED_TENANT_ID", "00000000-0000-0000-0000-000000000000") + fixed_client_id = os.environ.get("FIXED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + fixed_client_secret = os.environ.get("FIXED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=fixed_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=fixed_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=fixed_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=fixed_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py new file mode 100644 index 00000000000..01f766ccb9c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations.py @@ -0,0 +1,46 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import FixedClientTestBase, FixedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestFixedStringOperations(FixedClientTestBase): + @FixedPreparer() + @recorded_by_proxy + def test_string_get_known_value(self, fixed_endpoint): + client = self.create_client(endpoint=fixed_endpoint) + response = client.string.get_known_value() + + # please add some check logic here by yourself + # ... + + @FixedPreparer() + @recorded_by_proxy + def test_string_put_known_value(self, fixed_endpoint): + client = self.create_client(endpoint=fixed_endpoint) + response = client.string.put_known_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @FixedPreparer() + @recorded_by_proxy + def test_string_put_unknown_value(self, fixed_endpoint): + client = self.create_client(endpoint=fixed_endpoint) + response = client.string.put_unknown_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py new file mode 100644 index 00000000000..fd4a6d20c4c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/test_fixed_string_operations_async.py @@ -0,0 +1,47 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import FixedPreparer +from testpreparer_async import FixedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestFixedStringOperationsAsync(FixedClientTestBaseAsync): + @FixedPreparer() + @recorded_by_proxy_async + async def test_string_get_known_value(self, fixed_endpoint): + client = self.create_async_client(endpoint=fixed_endpoint) + response = await client.string.get_known_value() + + # please add some check logic here by yourself + # ... + + @FixedPreparer() + @recorded_by_proxy_async + async def test_string_put_known_value(self, fixed_endpoint): + client = self.create_async_client(endpoint=fixed_endpoint) + response = await client.string.put_known_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @FixedPreparer() + @recorded_by_proxy_async + async def test_string_put_unknown_value(self, fixed_endpoint): + client = self.create_async_client(endpoint=fixed_endpoint) + response = await client.string.put_unknown_value( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py new file mode 100644 index 00000000000..5a5bb459bc4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.enum.fixed import FixedClient + + +class FixedClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(FixedClient) + return self.create_client_from_credential( + FixedClient, + credential=credential, + endpoint=endpoint, + ) + + +FixedPreparer = functools.partial(PowerShellPreparer, "fixed", fixed_endpoint="https://fake_fixed_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..ec4217913e8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.enum.fixed.aio import FixedClient + + +class FixedClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(FixedClient, is_async=True) + return self.create_client_from_credential( + FixedClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-empty/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-empty/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py new file mode 100644 index 00000000000..2a895bea492 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + empty_subscription_id = os.environ.get("EMPTY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + empty_tenant_id = os.environ.get("EMPTY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + empty_client_id = os.environ.get("EMPTY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + empty_client_secret = os.environ.get("EMPTY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=empty_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=empty_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=empty_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=empty_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py new file mode 100644 index 00000000000..cdadb9d894a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty.py @@ -0,0 +1,44 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import EmptyClientTestBase, EmptyPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestEmpty(EmptyClientTestBase): + @EmptyPreparer() + @recorded_by_proxy + def test_put_empty(self, empty_endpoint): + client = self.create_client(endpoint=empty_endpoint) + response = client.put_empty( + input={}, + ) + + # please add some check logic here by yourself + # ... + + @EmptyPreparer() + @recorded_by_proxy + def test_get_empty(self, empty_endpoint): + client = self.create_client(endpoint=empty_endpoint) + response = client.get_empty() + + # please add some check logic here by yourself + # ... + + @EmptyPreparer() + @recorded_by_proxy + def test_post_round_trip_empty(self, empty_endpoint): + client = self.create_client(endpoint=empty_endpoint) + response = client.post_round_trip_empty( + body={}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py new file mode 100644 index 00000000000..ed450cbb2e3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/test_empty_async.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import EmptyPreparer +from testpreparer_async import EmptyClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestEmptyAsync(EmptyClientTestBaseAsync): + @EmptyPreparer() + @recorded_by_proxy_async + async def test_put_empty(self, empty_endpoint): + client = self.create_async_client(endpoint=empty_endpoint) + response = await client.put_empty( + input={}, + ) + + # please add some check logic here by yourself + # ... + + @EmptyPreparer() + @recorded_by_proxy_async + async def test_get_empty(self, empty_endpoint): + client = self.create_async_client(endpoint=empty_endpoint) + response = await client.get_empty() + + # please add some check logic here by yourself + # ... + + @EmptyPreparer() + @recorded_by_proxy_async + async def test_post_round_trip_empty(self, empty_endpoint): + client = self.create_async_client(endpoint=empty_endpoint) + response = await client.post_round_trip_empty( + body={}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py new file mode 100644 index 00000000000..697114f9f4b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.empty import EmptyClient + + +class EmptyClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(EmptyClient) + return self.create_client_from_credential( + EmptyClient, + credential=credential, + endpoint=endpoint, + ) + + +EmptyPreparer = functools.partial(PowerShellPreparer, "empty", empty_endpoint="https://fake_empty_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..de9919c1869 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-empty/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.empty.aio import EmptyClient + + +class EmptyClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(EmptyClient, is_async=True) + return self.create_client_from_credential( + EmptyClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-empty/typetest/model/empty/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py new file mode 100644 index 00000000000..8f844ad92c7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + enumdiscriminator_subscription_id = os.environ.get( + "ENUMDISCRIMINATOR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + enumdiscriminator_tenant_id = os.environ.get("ENUMDISCRIMINATOR_TENANT_ID", "00000000-0000-0000-0000-000000000000") + enumdiscriminator_client_id = os.environ.get("ENUMDISCRIMINATOR_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + enumdiscriminator_client_secret = os.environ.get( + "ENUMDISCRIMINATOR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=enumdiscriminator_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=enumdiscriminator_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=enumdiscriminator_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=enumdiscriminator_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py new file mode 100644 index 00000000000..efc5b841192 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator.py @@ -0,0 +1,89 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import EnumDiscriminatorClientTestBase, EnumDiscriminatorPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestEnumDiscriminator(EnumDiscriminatorClientTestBase): + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_get_extensible_model(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.get_extensible_model() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_put_extensible_model(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.put_extensible_model( + input={"kind": "golden", "weight": 0}, + ) + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_get_extensible_model_missing_discriminator(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.get_extensible_model_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_get_extensible_model_wrong_discriminator(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.get_extensible_model_wrong_discriminator() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_get_fixed_model(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.get_fixed_model() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_put_fixed_model(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.put_fixed_model( + input={"kind": "cobra", "length": 0}, + ) + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_get_fixed_model_missing_discriminator(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.get_fixed_model_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy + def test_get_fixed_model_wrong_discriminator(self, enumdiscriminator_endpoint): + client = self.create_client(endpoint=enumdiscriminator_endpoint) + response = client.get_fixed_model_wrong_discriminator() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py new file mode 100644 index 00000000000..d4c0c170943 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/test_enum_discriminator_async.py @@ -0,0 +1,90 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import EnumDiscriminatorPreparer +from testpreparer_async import EnumDiscriminatorClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestEnumDiscriminatorAsync(EnumDiscriminatorClientTestBaseAsync): + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_extensible_model(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.get_extensible_model() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_put_extensible_model(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.put_extensible_model( + input={"kind": "golden", "weight": 0}, + ) + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_extensible_model_missing_discriminator(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.get_extensible_model_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_extensible_model_wrong_discriminator(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.get_extensible_model_wrong_discriminator() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_fixed_model(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.get_fixed_model() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_put_fixed_model(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.put_fixed_model( + input={"kind": "cobra", "length": 0}, + ) + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_fixed_model_missing_discriminator(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.get_fixed_model_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @EnumDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_fixed_model_wrong_discriminator(self, enumdiscriminator_endpoint): + client = self.create_async_client(endpoint=enumdiscriminator_endpoint) + response = await client.get_fixed_model_wrong_discriminator() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py new file mode 100644 index 00000000000..99473c8b5b6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.enumdiscriminator import EnumDiscriminatorClient + + +class EnumDiscriminatorClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(EnumDiscriminatorClient) + return self.create_client_from_credential( + EnumDiscriminatorClient, + credential=credential, + endpoint=endpoint, + ) + + +EnumDiscriminatorPreparer = functools.partial( + PowerShellPreparer, "enumdiscriminator", enumdiscriminator_endpoint="https://fake_enumdiscriminator_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..479d42f79ee --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.enumdiscriminator.aio import EnumDiscriminatorClient + + +class EnumDiscriminatorClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(EnumDiscriminatorClient, is_async=True) + return self.create_client_from_credential( + EnumDiscriminatorClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py new file mode 100644 index 00000000000..8ba1f6f463b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/conftest.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + nesteddiscriminator_subscription_id = os.environ.get( + "NESTEDDISCRIMINATOR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + nesteddiscriminator_tenant_id = os.environ.get( + "NESTEDDISCRIMINATOR_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + nesteddiscriminator_client_id = os.environ.get( + "NESTEDDISCRIMINATOR_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + nesteddiscriminator_client_secret = os.environ.get( + "NESTEDDISCRIMINATOR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=nesteddiscriminator_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=nesteddiscriminator_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=nesteddiscriminator_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=nesteddiscriminator_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py new file mode 100644 index 00000000000..76206b37b92 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator.py @@ -0,0 +1,71 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NestedDiscriminatorClientTestBase, NestedDiscriminatorPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNestedDiscriminator(NestedDiscriminatorClientTestBase): + @NestedDiscriminatorPreparer() + @recorded_by_proxy + def test_get_model(self, nesteddiscriminator_endpoint): + client = self.create_client(endpoint=nesteddiscriminator_endpoint) + response = client.get_model() + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy + def test_put_model(self, nesteddiscriminator_endpoint): + client = self.create_client(endpoint=nesteddiscriminator_endpoint) + response = client.put_model( + input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, + ) + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy + def test_get_recursive_model(self, nesteddiscriminator_endpoint): + client = self.create_client(endpoint=nesteddiscriminator_endpoint) + response = client.get_recursive_model() + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy + def test_put_recursive_model(self, nesteddiscriminator_endpoint): + client = self.create_client(endpoint=nesteddiscriminator_endpoint) + response = client.put_recursive_model( + input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, + ) + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy + def test_get_missing_discriminator(self, nesteddiscriminator_endpoint): + client = self.create_client(endpoint=nesteddiscriminator_endpoint) + response = client.get_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy + def test_get_wrong_discriminator(self, nesteddiscriminator_endpoint): + client = self.create_client(endpoint=nesteddiscriminator_endpoint) + response = client.get_wrong_discriminator() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py new file mode 100644 index 00000000000..08270488ab8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/test_nested_discriminator_async.py @@ -0,0 +1,72 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NestedDiscriminatorPreparer +from testpreparer_async import NestedDiscriminatorClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNestedDiscriminatorAsync(NestedDiscriminatorClientTestBaseAsync): + @NestedDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_model(self, nesteddiscriminator_endpoint): + client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) + response = await client.get_model() + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_put_model(self, nesteddiscriminator_endpoint): + client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) + response = await client.put_model( + input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, + ) + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_recursive_model(self, nesteddiscriminator_endpoint): + client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) + response = await client.get_recursive_model() + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_put_recursive_model(self, nesteddiscriminator_endpoint): + client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) + response = await client.put_recursive_model( + input={"age": 0, "kind": "salmon", "friends": ["fish"], "hate": {"str": "fish"}, "partner": "fish"}, + ) + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_missing_discriminator(self, nesteddiscriminator_endpoint): + client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) + response = await client.get_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @NestedDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_wrong_discriminator(self, nesteddiscriminator_endpoint): + client = self.create_async_client(endpoint=nesteddiscriminator_endpoint) + response = await client.get_wrong_discriminator() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py new file mode 100644 index 00000000000..11eb32cb7e3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.nesteddiscriminator import NestedDiscriminatorClient + + +class NestedDiscriminatorClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(NestedDiscriminatorClient) + return self.create_client_from_credential( + NestedDiscriminatorClient, + credential=credential, + endpoint=endpoint, + ) + + +NestedDiscriminatorPreparer = functools.partial( + PowerShellPreparer, + "nesteddiscriminator", + nesteddiscriminator_endpoint="https://fake_nesteddiscriminator_endpoint.com", +) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..7913db8a24b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.nesteddiscriminator.aio import NestedDiscriminatorClient + + +class NestedDiscriminatorClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(NestedDiscriminatorClient, is_async=True) + return self.create_client_from_credential( + NestedDiscriminatorClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py new file mode 100644 index 00000000000..f6b00219293 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + notdiscriminated_subscription_id = os.environ.get( + "NOTDISCRIMINATED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + notdiscriminated_tenant_id = os.environ.get("NOTDISCRIMINATED_TENANT_ID", "00000000-0000-0000-0000-000000000000") + notdiscriminated_client_id = os.environ.get("NOTDISCRIMINATED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + notdiscriminated_client_secret = os.environ.get( + "NOTDISCRIMINATED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=notdiscriminated_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notdiscriminated_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notdiscriminated_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=notdiscriminated_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py new file mode 100644 index 00000000000..8cae28e36be --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated.py @@ -0,0 +1,44 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NotDiscriminatedClientTestBase, NotDiscriminatedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNotDiscriminated(NotDiscriminatedClientTestBase): + @NotDiscriminatedPreparer() + @recorded_by_proxy + def test_post_valid(self, notdiscriminated_endpoint): + client = self.create_client(endpoint=notdiscriminated_endpoint) + response = client.post_valid( + input={"age": 0, "name": "str", "smart": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NotDiscriminatedPreparer() + @recorded_by_proxy + def test_get_valid(self, notdiscriminated_endpoint): + client = self.create_client(endpoint=notdiscriminated_endpoint) + response = client.get_valid() + + # please add some check logic here by yourself + # ... + + @NotDiscriminatedPreparer() + @recorded_by_proxy + def test_put_valid(self, notdiscriminated_endpoint): + client = self.create_client(endpoint=notdiscriminated_endpoint) + response = client.put_valid( + input={"age": 0, "name": "str", "smart": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py new file mode 100644 index 00000000000..a9df6bb895b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/test_not_discriminated_async.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NotDiscriminatedPreparer +from testpreparer_async import NotDiscriminatedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNotDiscriminatedAsync(NotDiscriminatedClientTestBaseAsync): + @NotDiscriminatedPreparer() + @recorded_by_proxy_async + async def test_post_valid(self, notdiscriminated_endpoint): + client = self.create_async_client(endpoint=notdiscriminated_endpoint) + response = await client.post_valid( + input={"age": 0, "name": "str", "smart": bool}, + ) + + # please add some check logic here by yourself + # ... + + @NotDiscriminatedPreparer() + @recorded_by_proxy_async + async def test_get_valid(self, notdiscriminated_endpoint): + client = self.create_async_client(endpoint=notdiscriminated_endpoint) + response = await client.get_valid() + + # please add some check logic here by yourself + # ... + + @NotDiscriminatedPreparer() + @recorded_by_proxy_async + async def test_put_valid(self, notdiscriminated_endpoint): + client = self.create_async_client(endpoint=notdiscriminated_endpoint) + response = await client.put_valid( + input={"age": 0, "name": "str", "smart": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py new file mode 100644 index 00000000000..82ddc70adbe --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.notdiscriminated import NotDiscriminatedClient + + +class NotDiscriminatedClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(NotDiscriminatedClient) + return self.create_client_from_credential( + NotDiscriminatedClient, + credential=credential, + endpoint=endpoint, + ) + + +NotDiscriminatedPreparer = functools.partial( + PowerShellPreparer, "notdiscriminated", notdiscriminated_endpoint="https://fake_notdiscriminated_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..ecfe643396f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.notdiscriminated.aio import NotDiscriminatedClient + + +class NotDiscriminatedClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(NotDiscriminatedClient, is_async=True) + return self.create_client_from_credential( + NotDiscriminatedClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-recursive/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-recursive/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py new file mode 100644 index 00000000000..338abb9869a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + recursive_subscription_id = os.environ.get("RECURSIVE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + recursive_tenant_id = os.environ.get("RECURSIVE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + recursive_client_id = os.environ.get("RECURSIVE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + recursive_client_secret = os.environ.get("RECURSIVE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=recursive_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=recursive_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=recursive_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=recursive_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py new file mode 100644 index 00000000000..11b06922531 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RecursiveClientTestBase, RecursivePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRecursive(RecursiveClientTestBase): + @RecursivePreparer() + @recorded_by_proxy + def test_put(self, recursive_endpoint): + client = self.create_client(endpoint=recursive_endpoint) + response = client.put( + input={"level": 0, "extension": [...]}, + ) + + # please add some check logic here by yourself + # ... + + @RecursivePreparer() + @recorded_by_proxy + def test_get(self, recursive_endpoint): + client = self.create_client(endpoint=recursive_endpoint) + response = client.get() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py new file mode 100644 index 00000000000..a09a896c425 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/test_recursive_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RecursivePreparer +from testpreparer_async import RecursiveClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRecursiveAsync(RecursiveClientTestBaseAsync): + @RecursivePreparer() + @recorded_by_proxy_async + async def test_put(self, recursive_endpoint): + client = self.create_async_client(endpoint=recursive_endpoint) + response = await client.put( + input={"level": 0, "extension": [...]}, + ) + + # please add some check logic here by yourself + # ... + + @RecursivePreparer() + @recorded_by_proxy_async + async def test_get(self, recursive_endpoint): + client = self.create_async_client(endpoint=recursive_endpoint) + response = await client.get() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py new file mode 100644 index 00000000000..33dbd65de15 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.recursive import RecursiveClient + + +class RecursiveClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(RecursiveClient) + return self.create_client_from_credential( + RecursiveClient, + credential=credential, + endpoint=endpoint, + ) + + +RecursivePreparer = functools.partial( + PowerShellPreparer, "recursive", recursive_endpoint="https://fake_recursive_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..e57630973f5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-recursive/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.recursive.aio import RecursiveClient + + +class RecursiveClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(RecursiveClient, is_async=True) + return self.create_client_from_credential( + RecursiveClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py new file mode 100644 index 00000000000..633ac1e52f0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/conftest.py @@ -0,0 +1,43 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + singlediscriminator_subscription_id = os.environ.get( + "SINGLEDISCRIMINATOR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + singlediscriminator_tenant_id = os.environ.get( + "SINGLEDISCRIMINATOR_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + singlediscriminator_client_id = os.environ.get( + "SINGLEDISCRIMINATOR_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + singlediscriminator_client_secret = os.environ.get( + "SINGLEDISCRIMINATOR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=singlediscriminator_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=singlediscriminator_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=singlediscriminator_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=singlediscriminator_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py new file mode 100644 index 00000000000..efc2bac6ce8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator.py @@ -0,0 +1,80 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import SingleDiscriminatorClientTestBase, SingleDiscriminatorPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSingleDiscriminator(SingleDiscriminatorClientTestBase): + @SingleDiscriminatorPreparer() + @recorded_by_proxy + def test_get_model(self, singlediscriminator_endpoint): + client = self.create_client(endpoint=singlediscriminator_endpoint) + response = client.get_model() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy + def test_put_model(self, singlediscriminator_endpoint): + client = self.create_client(endpoint=singlediscriminator_endpoint) + response = client.put_model( + input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, + ) + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy + def test_get_recursive_model(self, singlediscriminator_endpoint): + client = self.create_client(endpoint=singlediscriminator_endpoint) + response = client.get_recursive_model() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy + def test_put_recursive_model(self, singlediscriminator_endpoint): + client = self.create_client(endpoint=singlediscriminator_endpoint) + response = client.put_recursive_model( + input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, + ) + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy + def test_get_missing_discriminator(self, singlediscriminator_endpoint): + client = self.create_client(endpoint=singlediscriminator_endpoint) + response = client.get_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy + def test_get_wrong_discriminator(self, singlediscriminator_endpoint): + client = self.create_client(endpoint=singlediscriminator_endpoint) + response = client.get_wrong_discriminator() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy + def test_get_legacy_model(self, singlediscriminator_endpoint): + client = self.create_client(endpoint=singlediscriminator_endpoint) + response = client.get_legacy_model() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py new file mode 100644 index 00000000000..96fc035088f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/test_single_discriminator_async.py @@ -0,0 +1,81 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import SingleDiscriminatorPreparer +from testpreparer_async import SingleDiscriminatorClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestSingleDiscriminatorAsync(SingleDiscriminatorClientTestBaseAsync): + @SingleDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_model(self, singlediscriminator_endpoint): + client = self.create_async_client(endpoint=singlediscriminator_endpoint) + response = await client.get_model() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_put_model(self, singlediscriminator_endpoint): + client = self.create_async_client(endpoint=singlediscriminator_endpoint) + response = await client.put_model( + input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, + ) + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_recursive_model(self, singlediscriminator_endpoint): + client = self.create_async_client(endpoint=singlediscriminator_endpoint) + response = await client.get_recursive_model() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_put_recursive_model(self, singlediscriminator_endpoint): + client = self.create_async_client(endpoint=singlediscriminator_endpoint) + response = await client.put_recursive_model( + input={"kind": "eagle", "wingspan": 0, "friends": ["bird"], "hate": {"str": "bird"}, "partner": "bird"}, + ) + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_missing_discriminator(self, singlediscriminator_endpoint): + client = self.create_async_client(endpoint=singlediscriminator_endpoint) + response = await client.get_missing_discriminator() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_wrong_discriminator(self, singlediscriminator_endpoint): + client = self.create_async_client(endpoint=singlediscriminator_endpoint) + response = await client.get_wrong_discriminator() + + # please add some check logic here by yourself + # ... + + @SingleDiscriminatorPreparer() + @recorded_by_proxy_async + async def test_get_legacy_model(self, singlediscriminator_endpoint): + client = self.create_async_client(endpoint=singlediscriminator_endpoint) + response = await client.get_legacy_model() + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py new file mode 100644 index 00000000000..97b81d355e1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.singlediscriminator import SingleDiscriminatorClient + + +class SingleDiscriminatorClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(SingleDiscriminatorClient) + return self.create_client_from_credential( + SingleDiscriminatorClient, + credential=credential, + endpoint=endpoint, + ) + + +SingleDiscriminatorPreparer = functools.partial( + PowerShellPreparer, + "singlediscriminator", + singlediscriminator_endpoint="https://fake_singlediscriminator_endpoint.com", +) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..71110d7df51 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.singlediscriminator.aio import SingleDiscriminatorClient + + +class SingleDiscriminatorClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(SingleDiscriminatorClient, is_async=True) + return self.create_client_from_credential( + SingleDiscriminatorClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-usage/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-usage/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py new file mode 100644 index 00000000000..6a990b92bd7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + usage_subscription_id = os.environ.get("USAGE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + usage_tenant_id = os.environ.get("USAGE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + usage_client_id = os.environ.get("USAGE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + usage_client_secret = os.environ.get("USAGE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=usage_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py new file mode 100644 index 00000000000..7514cae181d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage.py @@ -0,0 +1,44 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UsageClientTestBase, UsagePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUsage(UsageClientTestBase): + @UsagePreparer() + @recorded_by_proxy + def test_input(self, usage_endpoint): + client = self.create_client(endpoint=usage_endpoint) + response = client.input( + input={"requiredProp": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy + def test_output(self, usage_endpoint): + client = self.create_client(endpoint=usage_endpoint) + response = client.output() + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy + def test_input_and_output(self, usage_endpoint): + client = self.create_client(endpoint=usage_endpoint) + response = client.input_and_output( + body={"requiredProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py new file mode 100644 index 00000000000..c2a66f7642b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/test_usage_async.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UsagePreparer +from testpreparer_async import UsageClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUsageAsync(UsageClientTestBaseAsync): + @UsagePreparer() + @recorded_by_proxy_async + async def test_input(self, usage_endpoint): + client = self.create_async_client(endpoint=usage_endpoint) + response = await client.input( + input={"requiredProp": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy_async + async def test_output(self, usage_endpoint): + client = self.create_async_client(endpoint=usage_endpoint) + response = await client.output() + + # please add some check logic here by yourself + # ... + + @UsagePreparer() + @recorded_by_proxy_async + async def test_input_and_output(self, usage_endpoint): + client = self.create_async_client(endpoint=usage_endpoint) + response = await client.input_and_output( + body={"requiredProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py new file mode 100644 index 00000000000..1c43d7f3f9e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.usage import UsageClient + + +class UsageClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(UsageClient) + return self.create_client_from_credential( + UsageClient, + credential=credential, + endpoint=endpoint, + ) + + +UsagePreparer = functools.partial(PowerShellPreparer, "usage", usage_endpoint="https://fake_usage_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..feabf6b2b2e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-usage/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.usage.aio import UsageClient + + +class UsageClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(UsageClient, is_async=True) + return self.create_client_from_credential( + UsageClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-usage/typetest/model/usage/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-model-visibility/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-model-visibility/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py new file mode 100644 index 00000000000..3fc4942cf79 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + visibility_subscription_id = os.environ.get("VISIBILITY_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + visibility_tenant_id = os.environ.get("VISIBILITY_TENANT_ID", "00000000-0000-0000-0000-000000000000") + visibility_client_id = os.environ.get("VISIBILITY_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + visibility_client_secret = os.environ.get("VISIBILITY_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=visibility_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py new file mode 100644 index 00000000000..7ed8cd7db1d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility.py @@ -0,0 +1,90 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import VisibilityClientTestBase, VisibilityPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVisibility(VisibilityClientTestBase): + @VisibilityPreparer() + @recorded_by_proxy + def test_get_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.get_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_head_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.head_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_put_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.put_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_patch_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.patch_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_post_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.post_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_delete_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.delete_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy + def test_put_read_only_model(self, visibility_endpoint): + client = self.create_client(endpoint=visibility_endpoint) + response = client.put_read_only_model( + input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py new file mode 100644 index 00000000000..9909a3a67eb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/test_visibility_async.py @@ -0,0 +1,91 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import VisibilityPreparer +from testpreparer_async import VisibilityClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestVisibilityAsync(VisibilityClientTestBaseAsync): + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_get_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.get_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_head_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.head_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_put_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.put_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_patch_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.patch_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_post_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.post_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_delete_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.delete_model( + input={"createProp": ["str"], "deleteProp": bool, "queryProp": 0, "readProp": "str", "updateProp": [0]}, + ) + + # please add some check logic here by yourself + # ... + + @VisibilityPreparer() + @recorded_by_proxy_async + async def test_put_read_only_model(self, visibility_endpoint): + client = self.create_async_client(endpoint=visibility_endpoint) + response = await client.put_read_only_model( + input={"optionalNullableIntList": [0], "optionalStringRecord": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py new file mode 100644 index 00000000000..5b050ddfa6b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.model.visibility import VisibilityClient + + +class VisibilityClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(VisibilityClient) + return self.create_client_from_credential( + VisibilityClient, + credential=credential, + endpoint=endpoint, + ) + + +VisibilityPreparer = functools.partial( + PowerShellPreparer, "visibility", visibility_endpoint="https://fake_visibility_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..d029cadcb66 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-model-visibility/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.model.visibility.aio import VisibilityClient + + +class VisibilityClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(VisibilityClient, is_async=True) + return self.create_client_from_credential( + VisibilityClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py new file mode 100644 index 00000000000..1468d06c958 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/conftest.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + additionalproperties_subscription_id = os.environ.get( + "ADDITIONALPROPERTIES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + additionalproperties_tenant_id = os.environ.get( + "ADDITIONALPROPERTIES_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + additionalproperties_client_id = os.environ.get( + "ADDITIONALPROPERTIES_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + additionalproperties_client_secret = os.environ.get( + "ADDITIONALPROPERTIES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=additionalproperties_subscription_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=additionalproperties_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=additionalproperties_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=additionalproperties_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py new file mode 100644 index 00000000000..bce95a06639 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadFloatOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_float_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_float_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_float.put( + body={"derivedProp": 0.0, "name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py new file mode 100644 index 00000000000..efbdeeb9588 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_float_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_float_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_float_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_float.put( + body={"derivedProp": 0.0, "name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py new file mode 100644 index 00000000000..048d80dbb9e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadModelArrayOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_model_array_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_model_array_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_model_array.put( + body={"derivedProp": [{"state": "str"}], "knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py new file mode 100644 index 00000000000..dad7cd099de --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_array_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_model_array_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_model_array_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_model_array.put( + body={"derivedProp": [{"state": "str"}], "knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py new file mode 100644 index 00000000000..9a3ab0104f4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadModelOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_model_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_model_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_model.put( + body={"derivedProp": {"state": "str"}, "knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py new file mode 100644 index 00000000000..eb9aed8f98e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_model_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_model_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_model_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_model.put( + body={"derivedProp": {"state": "str"}, "knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py new file mode 100644 index 00000000000..c2d8c804b7a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadStringOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_string_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_different_spread_string_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_different_spread_string.put( + body={"derivedProp": "str", "id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py new file mode 100644 index 00000000000..117ae0ce9d0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_different_spread_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsDifferentSpreadStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_string_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_different_spread_string_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_different_spread_string.put( + body={"derivedProp": "str", "id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py new file mode 100644 index 00000000000..9f7581b3d8c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsFloatOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_float_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_float_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_float.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py new file mode 100644 index 00000000000..737ed0bd682 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_float_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_float_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_float_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_float.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py new file mode 100644 index 00000000000..6e539bb68c5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsModelArrayOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_model_array_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_model_array_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_model_array.put( + body={"knownProp": [{"state": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py new file mode 100644 index 00000000000..1c0cb2da785 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_array_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_model_array_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_model_array_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_model_array.put( + body={"knownProp": [{"state": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py new file mode 100644 index 00000000000..71f8be6216c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsModelOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_model_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_model_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_model.put( + body={"knownProp": {"state": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py new file mode 100644 index 00000000000..1ed0b99ca3f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_model_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_model_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_model_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_model.put( + body={"knownProp": {"state": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py new file mode 100644 index 00000000000..ba56033e4af --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsStringOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_string_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_string_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_string.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py new file mode 100644 index 00000000000..58e4e63d8ea --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_string_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_string_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_string.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py new file mode 100644 index 00000000000..3a65da7a619 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsUnknownDerivedOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_unknown_derived_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_unknown_derived.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_unknown_derived_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_unknown_derived.put( + body={"index": 0, "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py new file mode 100644 index 00000000000..ce20df60dc4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_derived_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsUnknownDerivedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_unknown_derived_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_unknown_derived.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_unknown_derived_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_unknown_derived.put( + body={"index": 0, "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py new file mode 100644 index 00000000000..c0a56d2ff1a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsUnknownDiscriminatedOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_unknown_discriminated_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_unknown_discriminated.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_unknown_discriminated_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_unknown_discriminated.put( + body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py new file mode 100644 index 00000000000..ad6020d0192 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_discriminated_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsUnknownDiscriminatedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_unknown_discriminated_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_unknown_discriminated.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_unknown_discriminated_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_unknown_discriminated.put( + body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py new file mode 100644 index 00000000000..31c2bed53f2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsUnknownOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_unknown_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_unknown.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_extends_unknown_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.extends_unknown.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py new file mode 100644 index 00000000000..83136a4fe7e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_extends_unknown_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesExtendsUnknownOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_unknown_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_unknown.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_extends_unknown_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.extends_unknown.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py new file mode 100644 index 00000000000..354263e1857 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsFloatOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_float_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_float_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_float.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py new file mode 100644 index 00000000000..da7825acb80 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_float_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_float_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_float_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_float.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py new file mode 100644 index 00000000000..bdb8bd90138 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsModelArrayOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_model_array_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_model_array_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_model_array.put( + body={"knownProp": [{"state": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py new file mode 100644 index 00000000000..31593031a17 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_array_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_model_array_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_model_array_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_model_array.put( + body={"knownProp": [{"state": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py new file mode 100644 index 00000000000..5ccd7fddce3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsModelOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_model_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_model_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_model.put( + body={"knownProp": {"state": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py new file mode 100644 index 00000000000..1606f5f21cc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_model_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_model_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_model_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_model.put( + body={"knownProp": {"state": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py new file mode 100644 index 00000000000..7fbd6658191 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsStringOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_string_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_string_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_string.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py new file mode 100644 index 00000000000..60854f96eb7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_string_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_string_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_string.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py new file mode 100644 index 00000000000..b3798ae7950 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsUnknownDerivedOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_unknown_derived_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_unknown_derived.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_unknown_derived_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_unknown_derived.put( + body={"index": 0, "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py new file mode 100644 index 00000000000..ff24780610d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_derived_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsUnknownDerivedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_unknown_derived_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_unknown_derived.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_unknown_derived_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_unknown_derived.put( + body={"index": 0, "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py new file mode 100644 index 00000000000..6721020fb1a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsUnknownDiscriminatedOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_unknown_discriminated_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_unknown_discriminated.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_unknown_discriminated_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_unknown_discriminated.put( + body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py new file mode 100644 index 00000000000..baabdf3a52e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_discriminated_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsUnknownDiscriminatedOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_unknown_discriminated_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_unknown_discriminated.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_unknown_discriminated_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_unknown_discriminated.put( + body={"index": 0, "kind": "derived", "name": "str", "age": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py new file mode 100644 index 00000000000..8886d068ecd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsUnknownOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_unknown_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_unknown.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_is_unknown_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.is_unknown.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py new file mode 100644 index 00000000000..a382ceef639 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_is_unknown_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesIsUnknownOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_unknown_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_unknown.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_is_unknown_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.is_unknown.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py new file mode 100644 index 00000000000..05639d3daf0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesMultipleSpreadOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_multiple_spread_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.multiple_spread.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_multiple_spread_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.multiple_spread.put( + body={"flag": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py new file mode 100644 index 00000000000..d252e42b2e6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_multiple_spread_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesMultipleSpreadOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_multiple_spread_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.multiple_spread.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_multiple_spread_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.multiple_spread.put( + body={"flag": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py new file mode 100644 index 00000000000..c49babb04e8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentFloatOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_float_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_float_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_float.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py new file mode 100644 index 00000000000..a809a2f2a04 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_float_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_float_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_float_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_float.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py new file mode 100644 index 00000000000..72f42e0aa6d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentModelArrayOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_model_array_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_model_array_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_model_array.put( + body={"knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py new file mode 100644 index 00000000000..87aa47bc9e0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_array_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_model_array_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_model_array_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_model_array.put( + body={"knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py new file mode 100644 index 00000000000..f98dbe3ac01 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentModelOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_model_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_model_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_model.put( + body={"knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py new file mode 100644 index 00000000000..bf9d2bed264 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_model_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_model_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_model_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_model.put( + body={"knownProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py new file mode 100644 index 00000000000..6dc582546a5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentStringOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_string_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_different_string_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_different_string.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py new file mode 100644 index 00000000000..65018748418 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_different_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadDifferentStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_string_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_different_string_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_different_string.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py new file mode 100644 index 00000000000..6061341f6b0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadFloatOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_float_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_float_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_float.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py new file mode 100644 index 00000000000..429930f2719 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_float_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadFloatOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_float_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_float.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_float_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_float.put( + body={"id": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py new file mode 100644 index 00000000000..ba829536ceb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadModelArrayOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_model_array_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_model_array_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_model_array.put( + body={"knownProp": [{"state": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py new file mode 100644 index 00000000000..2647755f4dd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_array_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadModelArrayOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_model_array_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_model_array.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_model_array_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_model_array.put( + body={"knownProp": [{"state": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py new file mode 100644 index 00000000000..9fe7fb6e2cc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadModelOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_model_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_model_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_model.put( + body={"knownProp": {"state": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py new file mode 100644 index 00000000000..d1987ed0a6d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_model_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadModelOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_model_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_model.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_model_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_model.put( + body={"knownProp": {"state": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py new file mode 100644 index 00000000000..d783fba2a7e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordDiscriminatedUnionOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_discriminated_union_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_discriminated_union.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_discriminated_union_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_discriminated_union.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py new file mode 100644 index 00000000000..f9afa893275 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_discriminated_union_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordDiscriminatedUnionOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_discriminated_union_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_discriminated_union.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_discriminated_union_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_discriminated_union.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py new file mode 100644 index 00000000000..1e2055dfb52 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion2Operations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_non_discriminated_union2_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_non_discriminated_union2.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_non_discriminated_union2_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_non_discriminated_union2.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py new file mode 100644 index 00000000000..b4fc9526888 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union2_operations_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion2OperationsAsync( + AdditionalPropertiesClientTestBaseAsync +): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_non_discriminated_union2_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_non_discriminated_union2.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_non_discriminated_union2_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_non_discriminated_union2.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py new file mode 100644 index 00000000000..d80b30b4424 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion3Operations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_non_discriminated_union3_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_non_discriminated_union3.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_non_discriminated_union3_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_non_discriminated_union3.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py new file mode 100644 index 00000000000..9b6f4cf260e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union3_operations_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnion3OperationsAsync( + AdditionalPropertiesClientTestBaseAsync +): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_non_discriminated_union3_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_non_discriminated_union3.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_non_discriminated_union3_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_non_discriminated_union3.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py new file mode 100644 index 00000000000..fc4e613ba15 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnionOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_non_discriminated_union_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_non_discriminated_union.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_non_discriminated_union_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_non_discriminated_union.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py new file mode 100644 index 00000000000..af876dc59cc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_non_discriminated_union_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordNonDiscriminatedUnionOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_non_discriminated_union_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_non_discriminated_union.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_non_discriminated_union_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_non_discriminated_union.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py new file mode 100644 index 00000000000..a78a589899d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordUnionOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_union_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_union.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_record_union_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_record_union.put( + body={"flag": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py new file mode 100644 index 00000000000..6738840018d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_record_union_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadRecordUnionOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_union_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_union.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_record_union_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_record_union.put( + body={"flag": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py new file mode 100644 index 00000000000..b01e760174d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AdditionalPropertiesClientTestBase, AdditionalPropertiesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadStringOperations(AdditionalPropertiesClientTestBase): + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_string_get(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy + def test_spread_string_put(self, additionalproperties_endpoint): + client = self.create_client(endpoint=additionalproperties_endpoint) + response = client.spread_string.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py new file mode 100644 index 00000000000..6647bf7b020 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/test_additional_properties_spread_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AdditionalPropertiesPreparer +from testpreparer_async import AdditionalPropertiesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdditionalPropertiesSpreadStringOperationsAsync(AdditionalPropertiesClientTestBaseAsync): + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_string_get(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_string.get() + + # please add some check logic here by yourself + # ... + + @AdditionalPropertiesPreparer() + @recorded_by_proxy_async + async def test_spread_string_put(self, additionalproperties_endpoint): + client = self.create_async_client(endpoint=additionalproperties_endpoint) + response = await client.spread_string.put( + body={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py new file mode 100644 index 00000000000..84f49b4c073 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.property.additionalproperties import AdditionalPropertiesClient + + +class AdditionalPropertiesClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(AdditionalPropertiesClient) + return self.create_client_from_credential( + AdditionalPropertiesClient, + credential=credential, + endpoint=endpoint, + ) + + +AdditionalPropertiesPreparer = functools.partial( + PowerShellPreparer, + "additionalproperties", + additionalproperties_endpoint="https://fake_additionalproperties_endpoint.com", +) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..a512953aeed --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.property.additionalproperties.aio import AdditionalPropertiesClient + + +class AdditionalPropertiesClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(AdditionalPropertiesClient, is_async=True) + return self.create_client_from_credential( + AdditionalPropertiesClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-nullable/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-nullable/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py new file mode 100644 index 00000000000..3bb2820941c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + nullable_subscription_id = os.environ.get("NULLABLE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + nullable_tenant_id = os.environ.get("NULLABLE_TENANT_ID", "00000000-0000-0000-0000-000000000000") + nullable_client_id = os.environ.get("NULLABLE_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + nullable_client_secret = os.environ.get("NULLABLE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=nullable_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=nullable_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=nullable_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=nullable_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py new file mode 100644 index 00000000000..b6cb74e9109 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NullableClientTestBase, NullablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableBytesOperations(NullableClientTestBase): + @NullablePreparer() + @recorded_by_proxy + def test_bytes_get_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.bytes.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_bytes_get_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.bytes.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_bytes_patch_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.bytes.patch_non_null( + body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_bytes_patch_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.bytes.patch_null( + body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py new file mode 100644 index 00000000000..792c936c76e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_bytes_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NullablePreparer +from testpreparer_async import NullableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableBytesOperationsAsync(NullableClientTestBaseAsync): + @NullablePreparer() + @recorded_by_proxy_async + async def test_bytes_get_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.bytes.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_bytes_get_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.bytes.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_bytes_patch_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.bytes.patch_non_null( + body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_bytes_patch_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.bytes.patch_null( + body={"nullableProperty": bytes("bytes", encoding="utf-8"), "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py new file mode 100644 index 00000000000..ab22c0a94d7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NullableClientTestBase, NullablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableCollectionsByteOperations(NullableClientTestBase): + @NullablePreparer() + @recorded_by_proxy + def test_collections_byte_get_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_byte.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_byte_get_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_byte.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_byte_patch_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_byte.patch_non_null( + body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_byte_patch_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_byte.patch_null( + body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py new file mode 100644 index 00000000000..b970b004499 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_byte_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NullablePreparer +from testpreparer_async import NullableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableCollectionsByteOperationsAsync(NullableClientTestBaseAsync): + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_byte_get_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_byte.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_byte_get_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_byte.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_byte_patch_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_byte.patch_non_null( + body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_byte_patch_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_byte.patch_null( + body={"nullableProperty": [bytes("bytes", encoding="utf-8")], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py new file mode 100644 index 00000000000..36ead8cceca --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NullableClientTestBase, NullablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableCollectionsModelOperations(NullableClientTestBase): + @NullablePreparer() + @recorded_by_proxy + def test_collections_model_get_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_model.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_model_get_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_model.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_model_patch_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_model.patch_non_null( + body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_model_patch_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_model.patch_null( + body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py new file mode 100644 index 00000000000..dd0a92fa140 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_model_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NullablePreparer +from testpreparer_async import NullableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableCollectionsModelOperationsAsync(NullableClientTestBaseAsync): + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_model_get_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_model.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_model_get_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_model.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_model_patch_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_model.patch_non_null( + body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_model_patch_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_model.patch_null( + body={"nullableProperty": [{"property": "str"}], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py new file mode 100644 index 00000000000..ec73d86cce2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NullableClientTestBase, NullablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableCollectionsStringOperations(NullableClientTestBase): + @NullablePreparer() + @recorded_by_proxy + def test_collections_string_get_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_string.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_string_get_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_string.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_string_patch_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_string.patch_non_null( + body={"nullableProperty": ["str"], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_collections_string_patch_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.collections_string.patch_null( + body={"nullableProperty": ["str"], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py new file mode 100644 index 00000000000..f8e87b572b6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_collections_string_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NullablePreparer +from testpreparer_async import NullableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableCollectionsStringOperationsAsync(NullableClientTestBaseAsync): + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_string_get_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_string.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_string_get_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_string.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_string_patch_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_string.patch_non_null( + body={"nullableProperty": ["str"], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_collections_string_patch_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.collections_string.patch_null( + body={"nullableProperty": ["str"], "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py new file mode 100644 index 00000000000..afdf8b9796a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NullableClientTestBase, NullablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableDatetimeOperations(NullableClientTestBase): + @NullablePreparer() + @recorded_by_proxy + def test_datetime_get_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.datetime.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_datetime_get_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.datetime.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_datetime_patch_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.datetime.patch_non_null( + body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_datetime_patch_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.datetime.patch_null( + body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py new file mode 100644 index 00000000000..0e3f62d7660 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_datetime_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NullablePreparer +from testpreparer_async import NullableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableDatetimeOperationsAsync(NullableClientTestBaseAsync): + @NullablePreparer() + @recorded_by_proxy_async + async def test_datetime_get_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.datetime.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_datetime_get_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.datetime.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_datetime_patch_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.datetime.patch_non_null( + body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_datetime_patch_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.datetime.patch_null( + body={"nullableProperty": "2020-02-20 00:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py new file mode 100644 index 00000000000..00c29045174 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NullableClientTestBase, NullablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableDurationOperations(NullableClientTestBase): + @NullablePreparer() + @recorded_by_proxy + def test_duration_get_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.duration.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_duration_get_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.duration.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_duration_patch_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.duration.patch_non_null( + body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_duration_patch_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.duration.patch_null( + body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py new file mode 100644 index 00000000000..e2fc03a01e0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_duration_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NullablePreparer +from testpreparer_async import NullableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableDurationOperationsAsync(NullableClientTestBaseAsync): + @NullablePreparer() + @recorded_by_proxy_async + async def test_duration_get_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.duration.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_duration_get_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.duration.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_duration_patch_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.duration.patch_non_null( + body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_duration_patch_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.duration.patch_null( + body={"nullableProperty": "1 day, 0:00:00", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py new file mode 100644 index 00000000000..9dcc24d95a3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import NullableClientTestBase, NullablePreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableStringOperations(NullableClientTestBase): + @NullablePreparer() + @recorded_by_proxy + def test_string_get_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.string.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_string_get_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.string.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_string_patch_non_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.string.patch_non_null( + body={"nullableProperty": "str", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy + def test_string_patch_null(self, nullable_endpoint): + client = self.create_client(endpoint=nullable_endpoint) + response = client.string.patch_null( + body={"nullableProperty": "str", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py new file mode 100644 index 00000000000..dbf4e588182 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/test_nullable_string_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import NullablePreparer +from testpreparer_async import NullableClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestNullableStringOperationsAsync(NullableClientTestBaseAsync): + @NullablePreparer() + @recorded_by_proxy_async + async def test_string_get_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.string.get_non_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_string_get_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.string.get_null() + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_string_patch_non_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.string.patch_non_null( + body={"nullableProperty": "str", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @NullablePreparer() + @recorded_by_proxy_async + async def test_string_patch_null(self, nullable_endpoint): + client = self.create_async_client(endpoint=nullable_endpoint) + response = await client.string.patch_null( + body={"nullableProperty": "str", "requiredProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py new file mode 100644 index 00000000000..1a8af063f5a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.property.nullable import NullableClient + + +class NullableClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(NullableClient) + return self.create_client_from_credential( + NullableClient, + credential=credential, + endpoint=endpoint, + ) + + +NullablePreparer = functools.partial( + PowerShellPreparer, "nullable", nullable_endpoint="https://fake_nullable_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..243f7d80b11 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.property.nullable.aio import NullableClient + + +class NullableClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(NullableClient, is_async=True) + return self.create_client_from_credential( + NullableClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-optional/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-optional/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py new file mode 100644 index 00000000000..0361d7a1e81 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + optional_subscription_id = os.environ.get("OPTIONAL_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + optional_tenant_id = os.environ.get("OPTIONAL_TENANT_ID", "00000000-0000-0000-0000-000000000000") + optional_client_id = os.environ.get("OPTIONAL_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + optional_client_secret = os.environ.get("OPTIONAL_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=optional_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=optional_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=optional_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=optional_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py new file mode 100644 index 00000000000..54ed36a824b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalBooleanLiteralOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_boolean_literal_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.boolean_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_boolean_literal_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.boolean_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_boolean_literal_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.boolean_literal.put_all( + body={"property": True}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_boolean_literal_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.boolean_literal.put_default( + body={"property": True}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py new file mode 100644 index 00000000000..30609287dc3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_boolean_literal_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalBooleanLiteralOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_boolean_literal_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.boolean_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_boolean_literal_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.boolean_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_boolean_literal_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.boolean_literal.put_all( + body={"property": True}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_boolean_literal_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.boolean_literal.put_default( + body={"property": True}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py new file mode 100644 index 00000000000..305d2612ee3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalBytesOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_bytes_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.bytes.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_bytes_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.bytes.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_bytes_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.bytes.put_all( + body={"property": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_bytes_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.bytes.put_default( + body={"property": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py new file mode 100644 index 00000000000..ce7a116254f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_bytes_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalBytesOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_bytes_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.bytes.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_bytes_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.bytes.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_bytes_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.bytes.put_all( + body={"property": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_bytes_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.bytes.put_default( + body={"property": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py new file mode 100644 index 00000000000..1f6dcf106f3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalCollectionsByteOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_collections_byte_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_byte.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_collections_byte_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_byte.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_collections_byte_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_byte.put_all( + body={"property": [bytes("bytes", encoding="utf-8")]}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_collections_byte_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_byte.put_default( + body={"property": [bytes("bytes", encoding="utf-8")]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py new file mode 100644 index 00000000000..896f44ea6ed --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_byte_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalCollectionsByteOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_byte_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_byte.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_byte_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_byte.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_byte_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_byte.put_all( + body={"property": [bytes("bytes", encoding="utf-8")]}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_byte_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_byte.put_default( + body={"property": [bytes("bytes", encoding="utf-8")]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py new file mode 100644 index 00000000000..18cc93f8a5a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalCollectionsModelOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_collections_model_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_model.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_collections_model_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_model.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_collections_model_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_model.put_all( + body={"property": [{"property": "str"}]}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_collections_model_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.collections_model.put_default( + body={"property": [{"property": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py new file mode 100644 index 00000000000..06e36e8e9f8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_collections_model_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalCollectionsModelOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_model_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_model.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_model_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_model.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_model_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_model.put_all( + body={"property": [{"property": "str"}]}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_collections_model_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.collections_model.put_default( + body={"property": [{"property": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py new file mode 100644 index 00000000000..bc19d57fef3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalDatetimeOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_datetime_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.datetime.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_datetime_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.datetime.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_datetime_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.datetime.put_all( + body={"property": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_datetime_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.datetime.put_default( + body={"property": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py new file mode 100644 index 00000000000..280a0014040 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_datetime_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalDatetimeOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_datetime_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.datetime.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_datetime_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.datetime.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_datetime_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.datetime.put_all( + body={"property": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_datetime_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.datetime.put_default( + body={"property": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py new file mode 100644 index 00000000000..e920893260a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalDurationOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_duration_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.duration.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_duration_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.duration.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_duration_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.duration.put_all( + body={"property": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_duration_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.duration.put_default( + body={"property": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py new file mode 100644 index 00000000000..c31f449ec75 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_duration_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalDurationOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_duration_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.duration.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_duration_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.duration.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_duration_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.duration.put_all( + body={"property": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_duration_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.duration.put_default( + body={"property": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py new file mode 100644 index 00000000000..f546fda9550 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalFloatLiteralOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_float_literal_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.float_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_float_literal_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.float_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_float_literal_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.float_literal.put_all( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_float_literal_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.float_literal.put_default( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py new file mode 100644 index 00000000000..f10606012a1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_float_literal_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalFloatLiteralOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_float_literal_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.float_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_float_literal_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.float_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_float_literal_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.float_literal.put_all( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_float_literal_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.float_literal.put_default( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py new file mode 100644 index 00000000000..044da751628 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalIntLiteralOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_int_literal_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.int_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_int_literal_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.int_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_int_literal_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.int_literal.put_all( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_int_literal_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.int_literal.put_default( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py new file mode 100644 index 00000000000..a84ba860d26 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_int_literal_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalIntLiteralOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_int_literal_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.int_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_int_literal_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.int_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_int_literal_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.int_literal.put_all( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_int_literal_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.int_literal.put_default( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py new file mode 100644 index 00000000000..8d7331a5528 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalPlainDateOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_plain_date_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_date.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_plain_date_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_date.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_plain_date_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_date.put_all( + body={"property": "2020-02-20"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_plain_date_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_date.put_default( + body={"property": "2020-02-20"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py new file mode 100644 index 00000000000..ecea48e2582 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_date_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalPlainDateOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_date_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_date.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_date_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_date.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_date_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_date.put_all( + body={"property": "2020-02-20"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_date_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_date.put_default( + body={"property": "2020-02-20"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py new file mode 100644 index 00000000000..223b55f35ea --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalPlainTimeOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_plain_time_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_time.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_plain_time_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_time.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_plain_time_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_time.put_all( + body={"property": "12:30:00"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_plain_time_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.plain_time.put_default( + body={"property": "12:30:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py new file mode 100644 index 00000000000..062b2c1a48d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_plain_time_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalPlainTimeOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_time_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_time.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_time_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_time.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_time_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_time.put_all( + body={"property": "12:30:00"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_plain_time_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.plain_time.put_default( + body={"property": "12:30:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py new file mode 100644 index 00000000000..38b641dad1a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalRequiredAndOptionalOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_required_and_optional_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.required_and_optional.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_required_and_optional_get_required_only(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.required_and_optional.get_required_only() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_required_and_optional_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.required_and_optional.put_all( + body={"requiredProperty": 0, "optionalProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_required_and_optional_put_required_only(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.required_and_optional.put_required_only( + body={"requiredProperty": 0, "optionalProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py new file mode 100644 index 00000000000..0d694036bda --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_required_and_optional_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalRequiredAndOptionalOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_required_and_optional_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.required_and_optional.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_required_and_optional_get_required_only(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.required_and_optional.get_required_only() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_required_and_optional_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.required_and_optional.put_all( + body={"requiredProperty": 0, "optionalProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_required_and_optional_put_required_only(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.required_and_optional.put_required_only( + body={"requiredProperty": 0, "optionalProperty": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py new file mode 100644 index 00000000000..3c75efd4b0a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalStringLiteralOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_string_literal_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_string_literal_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_string_literal_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string_literal.put_all( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_string_literal_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string_literal.put_default( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py new file mode 100644 index 00000000000..0f48cbf56a8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_literal_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalStringLiteralOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_literal_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_literal_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_literal_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string_literal.put_all( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_literal_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string_literal.put_default( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py new file mode 100644 index 00000000000..1dcc560ba06 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalStringOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_string_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_string_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_string_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string.put_all( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_string_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.string.put_default( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py new file mode 100644 index 00000000000..d757488c9b5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_string_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalStringOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string.put_all( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_string_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.string.put_default( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py new file mode 100644 index 00000000000..775c2e338e9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalUnionFloatLiteralOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_union_float_literal_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_float_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_float_literal_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_float_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_float_literal_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_float_literal.put_all( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_float_literal_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_float_literal.put_default( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py new file mode 100644 index 00000000000..1c7555ba53e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_float_literal_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalUnionFloatLiteralOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_float_literal_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_float_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_float_literal_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_float_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_float_literal_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_float_literal.put_all( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_float_literal_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_float_literal.put_default( + body={"property": 1.25}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py new file mode 100644 index 00000000000..a301c507aea --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalUnionIntLiteralOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_union_int_literal_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_int_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_int_literal_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_int_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_int_literal_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_int_literal.put_all( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_int_literal_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_int_literal.put_default( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py new file mode 100644 index 00000000000..552dbfca317 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_int_literal_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalUnionIntLiteralOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_int_literal_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_int_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_int_literal_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_int_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_int_literal_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_int_literal.put_all( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_int_literal_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_int_literal.put_default( + body={"property": 1}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py new file mode 100644 index 00000000000..4cde960c6d4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations.py @@ -0,0 +1,53 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import OptionalClientTestBase, OptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalUnionStringLiteralOperations(OptionalClientTestBase): + @OptionalPreparer() + @recorded_by_proxy + def test_union_string_literal_get_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_string_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_string_literal_get_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_string_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_string_literal_put_all(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_string_literal.put_all( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy + def test_union_string_literal_put_default(self, optional_endpoint): + client = self.create_client(endpoint=optional_endpoint) + response = client.union_string_literal.put_default( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py new file mode 100644 index 00000000000..785ec15fdc2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/test_optional_union_string_literal_operations_async.py @@ -0,0 +1,54 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import OptionalPreparer +from testpreparer_async import OptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestOptionalUnionStringLiteralOperationsAsync(OptionalClientTestBaseAsync): + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_string_literal_get_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_string_literal.get_all() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_string_literal_get_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_string_literal.get_default() + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_string_literal_put_all(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_string_literal.put_all( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... + + @OptionalPreparer() + @recorded_by_proxy_async + async def test_union_string_literal_put_default(self, optional_endpoint): + client = self.create_async_client(endpoint=optional_endpoint) + response = await client.union_string_literal.put_default( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py new file mode 100644 index 00000000000..3fea0db046e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.property.optional import OptionalClient + + +class OptionalClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(OptionalClient) + return self.create_client_from_credential( + OptionalClient, + credential=credential, + endpoint=endpoint, + ) + + +OptionalPreparer = functools.partial( + PowerShellPreparer, "optional", optional_endpoint="https://fake_optional_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..617123e80fe --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.property.optional.aio import OptionalClient + + +class OptionalClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(OptionalClient, is_async=True) + return self.create_client_from_credential( + OptionalClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-property-optional/typetest/property/optional/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-property-valuetypes/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py new file mode 100644 index 00000000000..c2d5129d87a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + valuetypes_subscription_id = os.environ.get("VALUETYPES_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + valuetypes_tenant_id = os.environ.get("VALUETYPES_TENANT_ID", "00000000-0000-0000-0000-000000000000") + valuetypes_client_id = os.environ.get("VALUETYPES_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + valuetypes_client_secret = os.environ.get("VALUETYPES_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=valuetypes_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=valuetypes_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=valuetypes_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=valuetypes_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py new file mode 100644 index 00000000000..c3c3d54b963 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesBooleanLiteralOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_boolean_literal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.boolean_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_boolean_literal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.boolean_literal.put( + body={"property": True}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py new file mode 100644 index 00000000000..e8c8189aae4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_literal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesBooleanLiteralOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_boolean_literal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.boolean_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_boolean_literal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.boolean_literal.put( + body={"property": True}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py new file mode 100644 index 00000000000..51309ac9860 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesBooleanOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_boolean_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.boolean.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_boolean_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.boolean.put( + body={"property": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py new file mode 100644 index 00000000000..069b7a93c2b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_boolean_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesBooleanOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_boolean_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.boolean.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_boolean_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.boolean.put( + body={"property": bool}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py new file mode 100644 index 00000000000..3f08982c2f8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesBytesOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_bytes_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.bytes.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_bytes_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.bytes.put( + body={"property": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py new file mode 100644 index 00000000000..dd9dcffc483 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_bytes_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesBytesOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_bytes_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.bytes.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_bytes_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.bytes.put( + body={"property": bytes("bytes", encoding="utf-8")}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py new file mode 100644 index 00000000000..df06ce23f99 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesCollectionsIntOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_collections_int_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.collections_int.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_collections_int_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.collections_int.put( + body={"property": [0]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py new file mode 100644 index 00000000000..51b31ffdd77 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_int_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesCollectionsIntOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_collections_int_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.collections_int.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_collections_int_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.collections_int.put( + body={"property": [0]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py new file mode 100644 index 00000000000..8fcab968ccd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesCollectionsModelOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_collections_model_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.collections_model.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_collections_model_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.collections_model.put( + body={"property": [{"property": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py new file mode 100644 index 00000000000..991461dc3dd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_model_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesCollectionsModelOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_collections_model_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.collections_model.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_collections_model_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.collections_model.put( + body={"property": [{"property": "str"}]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py new file mode 100644 index 00000000000..1ab22215eab --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesCollectionsStringOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_collections_string_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.collections_string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_collections_string_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.collections_string.put( + body={"property": ["str"]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py new file mode 100644 index 00000000000..cb8e324ceb2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_collections_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesCollectionsStringOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_collections_string_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.collections_string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_collections_string_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.collections_string.put( + body={"property": ["str"]}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py new file mode 100644 index 00000000000..a0eaa45c66a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDatetimeOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_datetime_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.datetime.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_datetime_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.datetime.put( + body={"property": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py new file mode 100644 index 00000000000..888caa316a4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_datetime_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDatetimeOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_datetime_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.datetime.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_datetime_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.datetime.put( + body={"property": "2020-02-20 00:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py new file mode 100644 index 00000000000..00676e9f6a1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDecimal128Operations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_decimal128_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.decimal128.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_decimal128_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.decimal128.put( + body={"property": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py new file mode 100644 index 00000000000..605c463e8df --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal128_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDecimal128OperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_decimal128_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.decimal128.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_decimal128_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.decimal128.put( + body={"property": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py new file mode 100644 index 00000000000..f495eea2814 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDecimalOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_decimal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.decimal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_decimal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.decimal.put( + body={"property": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py new file mode 100644 index 00000000000..7153118e60d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_decimal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDecimalOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_decimal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.decimal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_decimal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.decimal.put( + body={"property": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py new file mode 100644 index 00000000000..9e09f1d71f4 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDictionaryStringOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_dictionary_string_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.dictionary_string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_dictionary_string_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.dictionary_string.put( + body={"property": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py new file mode 100644 index 00000000000..f24b8b13007 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_dictionary_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDictionaryStringOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_dictionary_string_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.dictionary_string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_dictionary_string_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.dictionary_string.put( + body={"property": {"str": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py new file mode 100644 index 00000000000..135aa3cb772 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDurationOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_duration_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.duration.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_duration_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.duration.put( + body={"property": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py new file mode 100644 index 00000000000..cac800420e5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_duration_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesDurationOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_duration_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.duration.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_duration_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.duration.put( + body={"property": "1 day, 0:00:00"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py new file mode 100644 index 00000000000..cc503ad2394 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesEnumOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_enum_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.enum.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_enum_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.enum.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py new file mode 100644 index 00000000000..778528c75cf --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_enum_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesEnumOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_enum_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.enum.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_enum_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.enum.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py new file mode 100644 index 00000000000..a095e152b17 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesExtensibleEnumOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_extensible_enum_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.extensible_enum.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_extensible_enum_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.extensible_enum.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py new file mode 100644 index 00000000000..362775ee153 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_extensible_enum_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesExtensibleEnumOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_extensible_enum_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.extensible_enum.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_extensible_enum_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.extensible_enum.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py new file mode 100644 index 00000000000..df265bbaafc --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesFloatLiteralOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_float_literal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.float_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_float_literal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.float_literal.put( + body={"property": 43.125}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py new file mode 100644 index 00000000000..9bffccf770d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_literal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesFloatLiteralOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_float_literal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.float_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_float_literal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.float_literal.put( + body={"property": 43.125}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py new file mode 100644 index 00000000000..b0fe9df2e81 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesFloatOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_float_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.float.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_float_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.float.put( + body={"property": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py new file mode 100644 index 00000000000..18f8789a51f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_float_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesFloatOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_float_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.float.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_float_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.float.put( + body={"property": 0.0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py new file mode 100644 index 00000000000..061f04a21a5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesIntLiteralOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_int_literal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.int_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_int_literal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.int_literal.put( + body={"property": 42}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py new file mode 100644 index 00000000000..46d4783f71f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_literal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesIntLiteralOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_int_literal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.int_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_int_literal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.int_literal.put( + body={"property": 42}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py new file mode 100644 index 00000000000..a59b36e06e1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesIntOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_int_operations_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.int_operations.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_int_operations_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.int_operations.put( + body={"property": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py new file mode 100644 index 00000000000..c694b12be4d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_int_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesIntOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_int_operations_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.int_operations.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_int_operations_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.int_operations.put( + body={"property": 0}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py new file mode 100644 index 00000000000..a65204abb54 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesModelOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_model_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.model.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_model_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.model.put( + body={"property": {"property": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py new file mode 100644 index 00000000000..aa51c3983e5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_model_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesModelOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_model_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.model.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_model_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.model.put( + body={"property": {"property": "str"}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py new file mode 100644 index 00000000000..9271124511a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesNeverOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_never_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.never.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_never_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.never.put( + body={}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py new file mode 100644 index 00000000000..3a5917baf1b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_never_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesNeverOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_never_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.never.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_never_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.never.put( + body={}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py new file mode 100644 index 00000000000..906cdcc0ec9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesStringLiteralOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_string_literal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.string_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_string_literal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.string_literal.put( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py new file mode 100644 index 00000000000..40d3144d7f1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_literal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesStringLiteralOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_string_literal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.string_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_string_literal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.string_literal.put( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py new file mode 100644 index 00000000000..60cec79a37b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesStringOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_string_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_string_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.string.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py new file mode 100644 index 00000000000..4da3d00b3ff --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesStringOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_string_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_string_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.string.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py new file mode 100644 index 00000000000..8f3ada270c8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionEnumValueOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_enum_value_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_enum_value.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_enum_value_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_enum_value.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py new file mode 100644 index 00000000000..356ae6028a7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_enum_value_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionEnumValueOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_enum_value_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_enum_value.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_enum_value_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_enum_value.put( + body={"property": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py new file mode 100644 index 00000000000..014f408a041 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionFloatLiteralOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_float_literal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_float_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_float_literal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_float_literal.put( + body={"property": 43.125}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py new file mode 100644 index 00000000000..a9690ec570c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_float_literal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionFloatLiteralOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_float_literal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_float_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_float_literal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_float_literal.put( + body={"property": 43.125}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py new file mode 100644 index 00000000000..5388367dd02 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionIntLiteralOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_int_literal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_int_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_int_literal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_int_literal.put( + body={"property": 42}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py new file mode 100644 index 00000000000..589cebfb201 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_int_literal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionIntLiteralOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_int_literal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_int_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_int_literal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_int_literal.put( + body={"property": 42}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py new file mode 100644 index 00000000000..89f1206661e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionStringLiteralOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_string_literal_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_string_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_union_string_literal_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.union_string_literal.put( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py new file mode 100644 index 00000000000..b1d0eeb1cea --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_union_string_literal_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnionStringLiteralOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_string_literal_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_string_literal.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_union_string_literal_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.union_string_literal.put( + body={"property": "hello"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py new file mode 100644 index 00000000000..328bdaea272 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownArrayOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_array_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_array.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_array_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_array.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py new file mode 100644 index 00000000000..7fcf40b5904 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_array_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownArrayOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_array_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_array.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_array_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_array.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py new file mode 100644 index 00000000000..f1f1b2d139f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownDictOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_dict_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_dict.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_dict_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_dict.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py new file mode 100644 index 00000000000..d43b1a5e594 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_dict_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownDictOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_dict_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_dict.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_dict_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_dict.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py new file mode 100644 index 00000000000..1fd3a8f292b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownIntOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_int_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_int.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_int_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_int.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py new file mode 100644 index 00000000000..688cf49172d --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_int_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownIntOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_int_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_int.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_int_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_int.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py new file mode 100644 index 00000000000..de51f421a4f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations.py @@ -0,0 +1,33 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ValueTypesClientTestBase, ValueTypesPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownStringOperations(ValueTypesClientTestBase): + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_string_get(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy + def test_unknown_string_put(self, valuetypes_endpoint): + client = self.create_client(endpoint=valuetypes_endpoint) + response = client.unknown_string.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py new file mode 100644 index 00000000000..4c60198bcef --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/test_value_types_unknown_string_operations_async.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ValueTypesPreparer +from testpreparer_async import ValueTypesClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestValueTypesUnknownStringOperationsAsync(ValueTypesClientTestBaseAsync): + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_string_get(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_string.get() + + # please add some check logic here by yourself + # ... + + @ValueTypesPreparer() + @recorded_by_proxy_async + async def test_unknown_string_put(self, valuetypes_endpoint): + client = self.create_async_client(endpoint=valuetypes_endpoint) + response = await client.unknown_string.put( + body={"property": {}}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py new file mode 100644 index 00000000000..8c8476a2489 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.property.valuetypes import ValueTypesClient + + +class ValueTypesClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ValueTypesClient) + return self.create_client_from_credential( + ValueTypesClient, + credential=credential, + endpoint=endpoint, + ) + + +ValueTypesPreparer = functools.partial( + PowerShellPreparer, "valuetypes", valuetypes_endpoint="https://fake_valuetypes_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..08c15283880 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.property.valuetypes.aio import ValueTypesClient + + +class ValueTypesClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ValueTypesClient, is_async=True) + return self.create_client_from_credential( + ValueTypesClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-scalar/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-scalar/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py new file mode 100644 index 00000000000..f7252bc7a1b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + scalar_subscription_id = os.environ.get("SCALAR_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + scalar_tenant_id = os.environ.get("SCALAR_TENANT_ID", "00000000-0000-0000-0000-000000000000") + scalar_client_id = os.environ.get("SCALAR_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + scalar_client_secret = os.environ.get("SCALAR_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=scalar_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py new file mode 100644 index 00000000000..8ed568688c1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarBooleanOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_boolean_get(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.boolean.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_boolean_put(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.boolean.put( + body=bool, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py new file mode 100644 index 00000000000..beb9b7ee726 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_boolean_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarBooleanOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_boolean_get(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.boolean.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_boolean_put(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.boolean.put( + body=bool, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py new file mode 100644 index 00000000000..a66b1a98438 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimal128TypeOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_decimal128_type_response_body(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal128_type.response_body() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_decimal128_type_request_body(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal128_type.request_body( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_decimal128_type_request_parameter(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal128_type.request_parameter( + value=0.0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py new file mode 100644 index 00000000000..8d30937e1da --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_type_operations_async.py @@ -0,0 +1,46 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimal128TypeOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal128_type_response_body(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal128_type.response_body() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal128_type_request_body(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal128_type.request_body( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal128_type_request_parameter(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal128_type.request_parameter( + value=0.0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py new file mode 100644 index 00000000000..575d594f1da --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimal128VerifyOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_decimal128_verify_prepare_verify(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal128_verify.prepare_verify() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_decimal128_verify_verify(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal128_verify.verify( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py new file mode 100644 index 00000000000..c8a29772379 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal128_verify_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimal128VerifyOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal128_verify_prepare_verify(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal128_verify.prepare_verify() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal128_verify_verify(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal128_verify.verify( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py new file mode 100644 index 00000000000..744251a8c93 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimalTypeOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_decimal_type_response_body(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal_type.response_body() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_decimal_type_request_body(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal_type.request_body( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_decimal_type_request_parameter(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal_type.request_parameter( + value=0.0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py new file mode 100644 index 00000000000..8ceb3f201ac --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_type_operations_async.py @@ -0,0 +1,46 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimalTypeOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal_type_response_body(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal_type.response_body() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal_type_request_body(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal_type.request_body( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal_type_request_parameter(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal_type.request_parameter( + value=0.0, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py new file mode 100644 index 00000000000..4b4b954a380 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimalVerifyOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_decimal_verify_prepare_verify(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal_verify.prepare_verify() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_decimal_verify_verify(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.decimal_verify.verify( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py new file mode 100644 index 00000000000..0d649c9cdec --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_decimal_verify_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarDecimalVerifyOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal_verify_prepare_verify(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal_verify.prepare_verify() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_decimal_verify_verify(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.decimal_verify.verify( + body=0.0, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py new file mode 100644 index 00000000000..951f291fca6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarStringOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_string_get(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.string.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_string_put(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.string.put( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py new file mode 100644 index 00000000000..03824836948 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_string_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarStringOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_string_get(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.string.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_string_put(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.string.put( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py new file mode 100644 index 00000000000..9af0d808bc9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ScalarClientTestBase, ScalarPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarUnknownOperations(ScalarClientTestBase): + @ScalarPreparer() + @recorded_by_proxy + def test_unknown_get(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.unknown.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy + def test_unknown_put(self, scalar_endpoint): + client = self.create_client(endpoint=scalar_endpoint) + response = client.unknown.put( + body={}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py new file mode 100644 index 00000000000..fd3aa53ac1b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/test_scalar_unknown_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ScalarPreparer +from testpreparer_async import ScalarClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestScalarUnknownOperationsAsync(ScalarClientTestBaseAsync): + @ScalarPreparer() + @recorded_by_proxy_async + async def test_unknown_get(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.unknown.get() + + # please add some check logic here by yourself + # ... + + @ScalarPreparer() + @recorded_by_proxy_async + async def test_unknown_put(self, scalar_endpoint): + client = self.create_async_client(endpoint=scalar_endpoint) + response = await client.unknown.put( + body={}, + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py new file mode 100644 index 00000000000..3ba91005381 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.scalar import ScalarClient + + +class ScalarClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ScalarClient) + return self.create_client_from_credential( + ScalarClient, + credential=credential, + endpoint=endpoint, + ) + + +ScalarPreparer = functools.partial(PowerShellPreparer, "scalar", scalar_endpoint="https://fake_scalar_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..f3e076a1b41 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.scalar.aio import ScalarClient + + +class ScalarClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ScalarClient, is_async=True) + return self.create_client_from_credential( + ScalarClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-scalar/typetest/scalar/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/typetest-union/typetest/union/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/typetest-union/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/typetest-union/typetest/union/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/typetest-union/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py new file mode 100644 index 00000000000..5e28d6377c8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + union_subscription_id = os.environ.get("UNION_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + union_tenant_id = os.environ.get("UNION_TENANT_ID", "00000000-0000-0000-0000-000000000000") + union_client_id = os.environ.get("UNION_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + union_client_secret = os.environ.get("UNION_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=union_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py new file mode 100644 index 00000000000..e35f531e7af --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionEnumsOnlyOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_enums_only_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.enums_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_enums_only_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.enums_only.send( + body={"prop": {"lr": "left", "ud": "up"}}, + prop={"lr": "left", "ud": "up"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py new file mode 100644 index 00000000000..7927950425b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_enums_only_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionEnumsOnlyOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_enums_only_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.enums_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_enums_only_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.enums_only.send( + body={"prop": {"lr": "left", "ud": "up"}}, + prop={"lr": "left", "ud": "up"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py new file mode 100644 index 00000000000..1f6b8c84fc7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionFloatsOnlyOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_floats_only_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.floats_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_floats_only_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.floats_only.send( + body={"prop": 1.1}, + prop=1.1, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py new file mode 100644 index 00000000000..95c162de097 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_floats_only_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionFloatsOnlyOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_floats_only_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.floats_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_floats_only_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.floats_only.send( + body={"prop": 1.1}, + prop=1.1, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py new file mode 100644 index 00000000000..8eb6efc6d13 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionIntsOnlyOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_ints_only_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.ints_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_ints_only_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.ints_only.send( + body={"prop": 1}, + prop=1, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py new file mode 100644 index 00000000000..876727cdc0c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_ints_only_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionIntsOnlyOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_ints_only_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.ints_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_ints_only_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.ints_only.send( + body={"prop": 1}, + prop=1, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py new file mode 100644 index 00000000000..2b46b3270d9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionMixedLiteralsOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_mixed_literals_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.mixed_literals.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_mixed_literals_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.mixed_literals.send( + body={"prop": {"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}}, + prop={"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py new file mode 100644 index 00000000000..356f7496bf0 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_literals_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionMixedLiteralsOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_mixed_literals_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.mixed_literals.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_mixed_literals_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.mixed_literals.send( + body={"prop": {"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}}, + prop={"booleanLiteral": "a", "floatLiteral": "a", "intLiteral": "a", "stringLiteral": "a"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py new file mode 100644 index 00000000000..325317368c6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations.py @@ -0,0 +1,48 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionMixedTypesOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_mixed_types_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.mixed_types.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_mixed_types_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.mixed_types.send( + body={ + "prop": { + "array": [{"name": "str"}], + "boolean": {"name": "str"}, + "int": {"name": "str"}, + "literal": {"name": "str"}, + "model": {"name": "str"}, + } + }, + prop={ + "array": [{"name": "str"}], + "boolean": {"name": "str"}, + "int": {"name": "str"}, + "literal": {"name": "str"}, + "model": {"name": "str"}, + }, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py new file mode 100644 index 00000000000..b146ea74056 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_mixed_types_operations_async.py @@ -0,0 +1,49 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionMixedTypesOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_mixed_types_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.mixed_types.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_mixed_types_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.mixed_types.send( + body={ + "prop": { + "array": [{"name": "str"}], + "boolean": {"name": "str"}, + "int": {"name": "str"}, + "literal": {"name": "str"}, + "model": {"name": "str"}, + } + }, + prop={ + "array": [{"name": "str"}], + "boolean": {"name": "str"}, + "int": {"name": "str"}, + "literal": {"name": "str"}, + "model": {"name": "str"}, + }, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py new file mode 100644 index 00000000000..e932af10a22 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionModelsOnlyOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_models_only_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.models_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_models_only_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.models_only.send( + body={"prop": {"name": "str"}}, + prop={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py new file mode 100644 index 00000000000..1ee10f9b68e --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_models_only_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionModelsOnlyOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_models_only_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.models_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_models_only_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.models_only.send( + body={"prop": {"name": "str"}}, + prop={"name": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py new file mode 100644 index 00000000000..4e68cad1acd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringAndArrayOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_string_and_array_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.string_and_array.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_string_and_array_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.string_and_array.send( + body={"prop": {"array": "str", "string": "str"}}, + prop={"array": "str", "string": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py new file mode 100644 index 00000000000..7692fd0bd05 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_and_array_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringAndArrayOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_string_and_array_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.string_and_array.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_string_and_array_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.string_and_array.send( + body={"prop": {"array": "str", "string": "str"}}, + prop={"array": "str", "string": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py new file mode 100644 index 00000000000..b05ecc2f0c7 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringExtensibleNamedOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_string_extensible_named_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.string_extensible_named.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_string_extensible_named_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.string_extensible_named.send( + body={"prop": "str"}, + prop="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py new file mode 100644 index 00000000000..63d9457c2a8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_named_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringExtensibleNamedOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_string_extensible_named_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.string_extensible_named.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_string_extensible_named_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.string_extensible_named.send( + body={"prop": "str"}, + prop="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py new file mode 100644 index 00000000000..4514fdc2676 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringExtensibleOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_string_extensible_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.string_extensible.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_string_extensible_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.string_extensible.send( + body={"prop": "b"}, + prop="b", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py new file mode 100644 index 00000000000..7150caa0edd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_string_extensible_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringExtensibleOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_string_extensible_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.string_extensible.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_string_extensible_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.string_extensible.send( + body={"prop": "b"}, + prop="b", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py new file mode 100644 index 00000000000..2181e8a0ced --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations.py @@ -0,0 +1,34 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import UnionClientTestBase, UnionPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringsOnlyOperations(UnionClientTestBase): + @UnionPreparer() + @recorded_by_proxy + def test_strings_only_get(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.strings_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy + def test_strings_only_send(self, union_endpoint): + client = self.create_client(endpoint=union_endpoint) + response = client.strings_only.send( + body={"prop": "a"}, + prop="a", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py new file mode 100644 index 00000000000..89d23916804 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/test_union_strings_only_operations_async.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import UnionPreparer +from testpreparer_async import UnionClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestUnionStringsOnlyOperationsAsync(UnionClientTestBaseAsync): + @UnionPreparer() + @recorded_by_proxy_async + async def test_strings_only_get(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.strings_only.get() + + # please add some check logic here by yourself + # ... + + @UnionPreparer() + @recorded_by_proxy_async + async def test_strings_only_send(self, union_endpoint): + client = self.create_async_client(endpoint=union_endpoint) + response = await client.strings_only.send( + body={"prop": "a"}, + prop="a", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py new file mode 100644 index 00000000000..0d66b2cf3fa --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from typetest.union import UnionClient + + +class UnionClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(UnionClient) + return self.create_client_from_credential( + UnionClient, + credential=credential, + endpoint=endpoint, + ) + + +UnionPreparer = functools.partial(PowerShellPreparer, "union", union_endpoint="https://fake_union_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..d61fa77c079 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/typetest-union/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from typetest.union.aio import UnionClient + + +class UnionClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(UnionClient, is_async=True) + return self.create_client_from_credential( + UnionClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/typetest-union/typetest/union/_model_base.py b/packages/typespec-python/test/azure/generated/typetest-union/typetest/union/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/typetest-union/typetest/union/_model_base.py +++ b/packages/typespec-python/test/azure/generated/typetest-union/typetest/union/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/versioning-added/versioning/added/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-added/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-added/versioning/added/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-added/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py new file mode 100644 index 00000000000..37ef9defab6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + added_subscription_id = os.environ.get("ADDED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + added_tenant_id = os.environ.get("ADDED_TENANT_ID", "00000000-0000-0000-0000-000000000000") + added_client_id = os.environ.get("ADDED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + added_client_secret = os.environ.get("ADDED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=added_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=added_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=added_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=added_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py new file mode 100644 index 00000000000..21f3c1c245f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AddedClientTestBase, AddedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAdded(AddedClientTestBase): + @AddedPreparer() + @recorded_by_proxy + def test_v1(self, added_endpoint): + client = self.create_client(endpoint=added_endpoint) + response = client.v1( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + header_v2="str", + ) + + # please add some check logic here by yourself + # ... + + @AddedPreparer() + @recorded_by_proxy + def test_v2(self, added_endpoint): + client = self.create_client(endpoint=added_endpoint) + response = client.v2( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py new file mode 100644 index 00000000000..c0aa575e088 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_async.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AddedPreparer +from testpreparer_async import AddedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAddedAsync(AddedClientTestBaseAsync): + @AddedPreparer() + @recorded_by_proxy_async + async def test_v1(self, added_endpoint): + client = self.create_async_client(endpoint=added_endpoint) + response = await client.v1( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + header_v2="str", + ) + + # please add some check logic here by yourself + # ... + + @AddedPreparer() + @recorded_by_proxy_async + async def test_v2(self, added_endpoint): + client = self.create_async_client(endpoint=added_endpoint) + response = await client.v2( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py new file mode 100644 index 00000000000..d29deadeedd --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import AddedClientTestBase, AddedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAddedInterfaceV2Operations(AddedClientTestBase): + @AddedPreparer() + @recorded_by_proxy + def test_interface_v2_v2_in_interface(self, added_endpoint): + client = self.create_client(endpoint=added_endpoint) + response = client.interface_v2.v2_in_interface( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py new file mode 100644 index 00000000000..653c57ae97f --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/test_added_interface_v2_operations_async.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import AddedPreparer +from testpreparer_async import AddedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestAddedInterfaceV2OperationsAsync(AddedClientTestBaseAsync): + @AddedPreparer() + @recorded_by_proxy_async + async def test_interface_v2_v2_in_interface(self, added_endpoint): + client = self.create_async_client(endpoint=added_endpoint) + response = await client.interface_v2.v2_in_interface( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py new file mode 100644 index 00000000000..da5e39b6fe6 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from versioning.added import AddedClient + + +class AddedClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(AddedClient) + return self.create_client_from_credential( + AddedClient, + credential=credential, + endpoint=endpoint, + ) + + +AddedPreparer = functools.partial(PowerShellPreparer, "added", added_endpoint="https://fake_added_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..fb7f4883db3 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-added/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from versioning.added.aio import AddedClient + + +class AddedClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(AddedClient, is_async=True) + return self.create_client_from_credential( + AddedClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/versioning-added/versioning/added/_model_base.py b/packages/typespec-python/test/azure/generated/versioning-added/versioning/added/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/versioning-added/versioning/added/_model_base.py +++ b/packages/typespec-python/test/azure/generated/versioning-added/versioning/added/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-madeoptional/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-madeoptional/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py new file mode 100644 index 00000000000..02d95966ca8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/conftest.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + madeoptional_subscription_id = os.environ.get( + "MADEOPTIONAL_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + madeoptional_tenant_id = os.environ.get("MADEOPTIONAL_TENANT_ID", "00000000-0000-0000-0000-000000000000") + madeoptional_client_id = os.environ.get("MADEOPTIONAL_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + madeoptional_client_secret = os.environ.get("MADEOPTIONAL_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=madeoptional_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=madeoptional_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=madeoptional_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=madeoptional_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py new file mode 100644 index 00000000000..13cd7b9ea12 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import MadeOptionalClientTestBase, MadeOptionalPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMadeOptional(MadeOptionalClientTestBase): + @MadeOptionalPreparer() + @recorded_by_proxy + def test_test(self, madeoptional_endpoint): + client = self.create_client(endpoint=madeoptional_endpoint) + response = client.test( + body={"prop": "str", "changedProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py new file mode 100644 index 00000000000..6a6baf3922c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/test_made_optional_async.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import MadeOptionalPreparer +from testpreparer_async import MadeOptionalClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestMadeOptionalAsync(MadeOptionalClientTestBaseAsync): + @MadeOptionalPreparer() + @recorded_by_proxy_async + async def test_test(self, madeoptional_endpoint): + client = self.create_async_client(endpoint=madeoptional_endpoint) + response = await client.test( + body={"prop": "str", "changedProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py new file mode 100644 index 00000000000..e2d307ecbe9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from versioning.madeoptional import MadeOptionalClient + + +class MadeOptionalClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(MadeOptionalClient) + return self.create_client_from_credential( + MadeOptionalClient, + credential=credential, + endpoint=endpoint, + ) + + +MadeOptionalPreparer = functools.partial( + PowerShellPreparer, "madeoptional", madeoptional_endpoint="https://fake_madeoptional_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..394e086e4ac --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-madeoptional/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from versioning.madeoptional.aio import MadeOptionalClient + + +class MadeOptionalClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(MadeOptionalClient, is_async=True) + return self.create_client_from_credential( + MadeOptionalClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py b/packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py +++ b/packages/typespec-python/test/azure/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-removed/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-removed/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py new file mode 100644 index 00000000000..dd62b27f5ea --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + removed_subscription_id = os.environ.get("REMOVED_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + removed_tenant_id = os.environ.get("REMOVED_TENANT_ID", "00000000-0000-0000-0000-000000000000") + removed_client_id = os.environ.get("REMOVED_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + removed_client_secret = os.environ.get("REMOVED_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=removed_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=removed_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=removed_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=removed_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py new file mode 100644 index 00000000000..d3f57f2b16a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RemovedClientTestBase, RemovedPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRemoved(RemovedClientTestBase): + @RemovedPreparer() + @recorded_by_proxy + def test_v2(self, removed_endpoint): + client = self.create_client(endpoint=removed_endpoint) + response = client.v2( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @RemovedPreparer() + @recorded_by_proxy + def test_model_v3(self, removed_endpoint): + client = self.create_client(endpoint=removed_endpoint) + response = client.model_v3( + body={"enumProp": "str", "id": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py new file mode 100644 index 00000000000..0a5c7bee85a --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/test_removed_async.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RemovedPreparer +from testpreparer_async import RemovedClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRemovedAsync(RemovedClientTestBaseAsync): + @RemovedPreparer() + @recorded_by_proxy_async + async def test_v2(self, removed_endpoint): + client = self.create_async_client(endpoint=removed_endpoint) + response = await client.v2( + body={"enumProp": "str", "prop": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... + + @RemovedPreparer() + @recorded_by_proxy_async + async def test_model_v3(self, removed_endpoint): + client = self.create_async_client(endpoint=removed_endpoint) + response = await client.model_v3( + body={"enumProp": "str", "id": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py new file mode 100644 index 00000000000..60f068de6e5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from versioning.removed import RemovedClient + + +class RemovedClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(RemovedClient) + return self.create_client_from_credential( + RemovedClient, + credential=credential, + endpoint=endpoint, + ) + + +RemovedPreparer = functools.partial(PowerShellPreparer, "removed", removed_endpoint="https://fake_removed_endpoint.com") diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..fb44cbf1a9c --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-removed/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from versioning.removed.aio import RemovedClient + + +class RemovedClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(RemovedClient, is_async=True) + return self.create_client_from_credential( + RemovedClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/_model_base.py b/packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/_model_base.py +++ b/packages/typespec-python/test/azure/generated/versioning-removed/versioning/removed/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-renamedfrom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py new file mode 100644 index 00000000000..4a15c18fbd5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/conftest.py @@ -0,0 +1,35 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + renamedfrom_subscription_id = os.environ.get("RENAMEDFROM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000") + renamedfrom_tenant_id = os.environ.get("RENAMEDFROM_TENANT_ID", "00000000-0000-0000-0000-000000000000") + renamedfrom_client_id = os.environ.get("RENAMEDFROM_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + renamedfrom_client_secret = os.environ.get("RENAMEDFROM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=renamedfrom_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=renamedfrom_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=renamedfrom_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=renamedfrom_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py new file mode 100644 index 00000000000..724043b1b94 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RenamedFromClientTestBase, RenamedFromPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedFrom(RenamedFromClientTestBase): + @RenamedFromPreparer() + @recorded_by_proxy + def test_new_op(self, renamedfrom_endpoint): + client = self.create_client(endpoint=renamedfrom_endpoint) + response = client.new_op( + body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, + new_query="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py new file mode 100644 index 00000000000..aac2319e12b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_async.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RenamedFromPreparer +from testpreparer_async import RenamedFromClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedFromAsync(RenamedFromClientTestBaseAsync): + @RenamedFromPreparer() + @recorded_by_proxy_async + async def test_new_op(self, renamedfrom_endpoint): + client = self.create_async_client(endpoint=renamedfrom_endpoint) + response = await client.new_op( + body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, + new_query="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py new file mode 100644 index 00000000000..d3aa2db48ef --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations.py @@ -0,0 +1,24 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import RenamedFromClientTestBase, RenamedFromPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedFromNewInterfaceOperations(RenamedFromClientTestBase): + @RenamedFromPreparer() + @recorded_by_proxy + def test_new_interface_new_op_in_new_interface(self, renamedfrom_endpoint): + client = self.create_client(endpoint=renamedfrom_endpoint) + response = client.new_interface.new_op_in_new_interface( + body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py new file mode 100644 index 00000000000..ded1a984fd9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/test_renamed_from_new_interface_operations_async.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import RenamedFromPreparer +from testpreparer_async import RenamedFromClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestRenamedFromNewInterfaceOperationsAsync(RenamedFromClientTestBaseAsync): + @RenamedFromPreparer() + @recorded_by_proxy_async + async def test_new_interface_new_op_in_new_interface(self, renamedfrom_endpoint): + client = self.create_async_client(endpoint=renamedfrom_endpoint) + response = await client.new_interface.new_op_in_new_interface( + body={"enumProp": "str", "newProp": "str", "unionProp": "str"}, + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py new file mode 100644 index 00000000000..af5459a16cb --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from versioning.renamedfrom import RenamedFromClient + + +class RenamedFromClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(RenamedFromClient) + return self.create_client_from_credential( + RenamedFromClient, + credential=credential, + endpoint=endpoint, + ) + + +RenamedFromPreparer = functools.partial( + PowerShellPreparer, "renamedfrom", renamedfrom_endpoint="https://fake_renamedfrom_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..2a3a04d4cd9 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from versioning.renamedfrom.aio import RenamedFromClient + + +class RenamedFromClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(RenamedFromClient, is_async=True) + return self.create_client_from_credential( + RenamedFromClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py +++ b/packages/typespec-python/test/azure/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py new file mode 100644 index 00000000000..2d8b087a478 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/conftest.py @@ -0,0 +1,45 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + returntypechangedfrom_subscription_id = os.environ.get( + "RETURNTYPECHANGEDFROM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + returntypechangedfrom_tenant_id = os.environ.get( + "RETURNTYPECHANGEDFROM_TENANT_ID", "00000000-0000-0000-0000-000000000000" + ) + returntypechangedfrom_client_id = os.environ.get( + "RETURNTYPECHANGEDFROM_CLIENT_ID", "00000000-0000-0000-0000-000000000000" + ) + returntypechangedfrom_client_secret = os.environ.get( + "RETURNTYPECHANGEDFROM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer( + regex=returntypechangedfrom_subscription_id, value="00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=returntypechangedfrom_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=returntypechangedfrom_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=returntypechangedfrom_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py new file mode 100644 index 00000000000..27b6ee78d2b --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import ReturnTypeChangedFromClientTestBase, ReturnTypeChangedFromPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestReturnTypeChangedFrom(ReturnTypeChangedFromClientTestBase): + @ReturnTypeChangedFromPreparer() + @recorded_by_proxy + def test_test(self, returntypechangedfrom_endpoint): + client = self.create_client(endpoint=returntypechangedfrom_endpoint) + response = client.test( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py new file mode 100644 index 00000000000..b75ea6fa0ab --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/test_return_type_changed_from_async.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import ReturnTypeChangedFromPreparer +from testpreparer_async import ReturnTypeChangedFromClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestReturnTypeChangedFromAsync(ReturnTypeChangedFromClientTestBaseAsync): + @ReturnTypeChangedFromPreparer() + @recorded_by_proxy_async + async def test_test(self, returntypechangedfrom_endpoint): + client = self.create_async_client(endpoint=returntypechangedfrom_endpoint) + response = await client.test( + body="str", + content_type="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py new file mode 100644 index 00000000000..3db19f8ddbe --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from versioning.returntypechangedfrom import ReturnTypeChangedFromClient + + +class ReturnTypeChangedFromClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(ReturnTypeChangedFromClient) + return self.create_client_from_credential( + ReturnTypeChangedFromClient, + credential=credential, + endpoint=endpoint, + ) + + +ReturnTypeChangedFromPreparer = functools.partial( + PowerShellPreparer, + "returntypechangedfrom", + returntypechangedfrom_endpoint="https://fake_returntypechangedfrom_endpoint.com", +) diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..9b673fad3f1 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from versioning.returntypechangedfrom.aio import ReturnTypeChangedFromClient + + +class ReturnTypeChangedFromClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(ReturnTypeChangedFromClient, is_async=True) + return self.create_client_from_credential( + ReturnTypeChangedFromClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py +++ b/packages/typespec-python/test/azure/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/apiview_mapping_python.json b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/apiview_mapping_python.json similarity index 100% rename from packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/apiview_mapping_python.json rename to packages/typespec-python/test/azure/generated/versioning-typechangedfrom/apiview_mapping_python.json diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py new file mode 100644 index 00000000000..9fcd84c05df --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/conftest.py @@ -0,0 +1,39 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import os +import pytest +from dotenv import load_dotenv +from devtools_testutils import ( + test_proxy, + add_general_regex_sanitizer, + add_body_key_sanitizer, + add_header_regex_sanitizer, +) + +load_dotenv() + + +# For security, please avoid record sensitive identity information in recordings +@pytest.fixture(scope="session", autouse=True) +def add_sanitizers(test_proxy): + typechangedfrom_subscription_id = os.environ.get( + "TYPECHANGEDFROM_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000" + ) + typechangedfrom_tenant_id = os.environ.get("TYPECHANGEDFROM_TENANT_ID", "00000000-0000-0000-0000-000000000000") + typechangedfrom_client_id = os.environ.get("TYPECHANGEDFROM_CLIENT_ID", "00000000-0000-0000-0000-000000000000") + typechangedfrom_client_secret = os.environ.get( + "TYPECHANGEDFROM_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000" + ) + add_general_regex_sanitizer(regex=typechangedfrom_subscription_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=typechangedfrom_tenant_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=typechangedfrom_client_id, value="00000000-0000-0000-0000-000000000000") + add_general_regex_sanitizer(regex=typechangedfrom_client_secret, value="00000000-0000-0000-0000-000000000000") + + add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]") + add_header_regex_sanitizer(key="Cookie", value="cookie;") + add_body_key_sanitizer(json_path="$..access_token", value="access_token") diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py new file mode 100644 index 00000000000..f923927e7d8 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from.py @@ -0,0 +1,25 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils import recorded_by_proxy +from testpreparer import TypeChangedFromClientTestBase, TypeChangedFromPreparer + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTypeChangedFrom(TypeChangedFromClientTestBase): + @TypeChangedFromPreparer() + @recorded_by_proxy + def test_test(self, typechangedfrom_endpoint): + client = self.create_client(endpoint=typechangedfrom_endpoint) + response = client.test( + body={"changedProp": "str", "prop": "str"}, + param="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py new file mode 100644 index 00000000000..cc7fe639551 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/test_type_changed_from_async.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +import pytest +from devtools_testutils.aio import recorded_by_proxy_async +from testpreparer import TypeChangedFromPreparer +from testpreparer_async import TypeChangedFromClientTestBaseAsync + + +@pytest.mark.skip("you may need to update the auto-generated test case before run it") +class TestTypeChangedFromAsync(TypeChangedFromClientTestBaseAsync): + @TypeChangedFromPreparer() + @recorded_by_proxy_async + async def test_test(self, typechangedfrom_endpoint): + client = self.create_async_client(endpoint=typechangedfrom_endpoint) + response = await client.test( + body={"changedProp": "str", "prop": "str"}, + param="str", + ) + + # please add some check logic here by yourself + # ... diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py new file mode 100644 index 00000000000..69d22259ad2 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer.py @@ -0,0 +1,26 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer +import functools +from versioning.typechangedfrom import TypeChangedFromClient + + +class TypeChangedFromClientTestBase(AzureRecordedTestCase): + + def create_client(self, endpoint): + credential = self.get_credential(TypeChangedFromClient) + return self.create_client_from_credential( + TypeChangedFromClient, + credential=credential, + endpoint=endpoint, + ) + + +TypeChangedFromPreparer = functools.partial( + PowerShellPreparer, "typechangedfrom", typechangedfrom_endpoint="https://fake_typechangedfrom_endpoint.com" +) diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py new file mode 100644 index 00000000000..cf09aeeeca5 --- /dev/null +++ b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/generated_tests/testpreparer_async.py @@ -0,0 +1,20 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) Python Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- +from devtools_testutils import AzureRecordedTestCase +from versioning.typechangedfrom.aio import TypeChangedFromClient + + +class TypeChangedFromClientTestBaseAsync(AzureRecordedTestCase): + + def create_async_client(self, endpoint): + credential = self.get_credential(TypeChangedFromClient, is_async=True) + return self.create_client_from_credential( + TypeChangedFromClient, + credential=credential, + endpoint=endpoint, + ) diff --git a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py index 6a6e1f38b17..7f73b97b23e 100644 --- a/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py +++ b/packages/typespec-python/test/azure/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_model_base.py b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_operations/_operations.py b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_operations/_operations.py index 5a42d431fb1..b471f950552 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_operations/_operations.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from corehttp.exceptions import ( ClientAuthenticationError, @@ -21,7 +21,7 @@ from corehttp.utils import case_insensitive_dict from .. import models as _models -from .._model_base import _deserialize +from .._model_base import _failsafe_deserialize from .._serialization import Serializer from .._vendor import ApiKeyClientMixinABC @@ -112,12 +112,6 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -142,7 +136,10 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py index 8f1da48b7dc..ecc0e05156c 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-api-key/authentication/apikey/aio/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from corehttp.exceptions import ( ClientAuthenticationError, @@ -20,7 +20,7 @@ from corehttp.runtime.pipeline import PipelineResponse from ... import models as _models -from ..._model_base import _deserialize +from ..._model_base import _failsafe_deserialize from ..._operations._operations import build_api_key_invalid_request, build_api_key_valid_request from .._vendor import ApiKeyClientMixinABC @@ -89,12 +89,6 @@ async def invalid(self, **kwargs: Any) -> None: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -121,7 +115,10 @@ async def invalid(self, **kwargs: Any) -> None: if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_model_base.py b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py index 84495af2fc1..9d307f997e9 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from corehttp.exceptions import ( ClientAuthenticationError, @@ -21,7 +21,7 @@ from corehttp.utils import case_insensitive_dict from .. import models as _models -from .._model_base import _deserialize +from .._model_base import _failsafe_deserialize from .._serialization import Serializer from .._vendor import CustomClientMixinABC @@ -112,12 +112,6 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -142,7 +136,10 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py index 22916e4a577..f51c6eb3445 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-http-custom/authentication/http/custom/aio/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from corehttp.exceptions import ( ClientAuthenticationError, @@ -20,7 +20,7 @@ from corehttp.runtime.pipeline import PipelineResponse from ... import models as _models -from ..._model_base import _deserialize +from ..._model_base import _failsafe_deserialize from ..._operations._operations import build_custom_invalid_request, build_custom_valid_request from .._vendor import CustomClientMixinABC @@ -89,12 +89,6 @@ async def invalid(self, **kwargs: Any) -> None: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -121,7 +115,10 @@ async def invalid(self, **kwargs: Any) -> None: if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_model_base.py b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py index 38aef5243db..84bddb75f32 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from corehttp.exceptions import ( ClientAuthenticationError, @@ -21,7 +21,7 @@ from corehttp.utils import case_insensitive_dict from .. import models as _models -from .._model_base import _deserialize +from .._model_base import _failsafe_deserialize from .._serialization import Serializer from .._vendor import OAuth2ClientMixinABC @@ -112,12 +112,6 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -142,7 +136,10 @@ def invalid(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py index 1d06ad9f0c9..dc6c81669a6 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-oauth2/authentication/oauth2/aio/_operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from corehttp.exceptions import ( ClientAuthenticationError, @@ -20,7 +20,7 @@ from corehttp.runtime.pipeline import PipelineResponse from ... import models as _models -from ..._model_base import _deserialize +from ..._model_base import _failsafe_deserialize from ..._operations._operations import build_oauth2_invalid_request, build_oauth2_valid_request from .._vendor import OAuth2ClientMixinABC @@ -89,12 +89,6 @@ async def invalid(self, **kwargs: Any) -> None: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 403: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=_deserialize(_models.InvalidAuth, response.json()) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -121,7 +115,10 @@ async def invalid(self, **kwargs: Any) -> None: if response.status_code not in [204]: map_error(status_code=response.status_code, response=response, error_map=error_map) - raise HttpResponseError(response=response) + error = None + if response.status_code == 403: + error = _failsafe_deserialize(_models.InvalidAuth, response.json()) + raise HttpResponseError(response=response, model=error) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/typespec-python/test/unbranded/generated/authentication-union/authentication/union/_model_base.py b/packages/typespec-python/test/unbranded/generated/authentication-union/authentication/union/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/authentication-union/authentication/union/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/authentication-union/authentication/union/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/client-naming/client/naming/_model_base.py b/packages/typespec-python/test/unbranded/generated/client-naming/client/naming/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/client-naming/client/naming/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/client-naming/client/naming/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/client-structure-default/client/structure/service/_model_base.py b/packages/typespec-python/test/unbranded/generated/client-structure-default/client/structure/service/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/client-structure-default/client/structure/service/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/client-structure-default/client/structure/service/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py b/packages/typespec-python/test/unbranded/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/client-structure-multiclient/client/structure/multiclient/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py b/packages/typespec-python/test/unbranded/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/client-structure-renamedoperation/client/structure/renamedoperation/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py b/packages/typespec-python/test/unbranded/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/client-structure-twooperationgroup/client/structure/twooperationgroup/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/encode-bytes/encode/bytes/_model_base.py b/packages/typespec-python/test/unbranded/generated/encode-bytes/encode/bytes/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/encode-bytes/encode/bytes/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/encode-bytes/encode/bytes/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/encode-datetime/encode/datetime/_model_base.py b/packages/typespec-python/test/unbranded/generated/encode-datetime/encode/datetime/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/encode-datetime/encode/datetime/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/encode-datetime/encode/datetime/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/encode-duration/encode/duration/_model_base.py b/packages/typespec-python/test/unbranded/generated/encode-duration/encode/duration/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/encode-duration/encode/duration/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/encode-duration/encode/duration/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/encode-numeric/encode/numeric/_model_base.py b/packages/typespec-python/test/unbranded/generated/encode-numeric/encode/numeric/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/encode-numeric/encode/numeric/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/encode-numeric/encode/numeric/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py b/packages/typespec-python/test/unbranded/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/headasbooleanfalse/headasbooleanfalse/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/headasbooleantrue/headasbooleantrue/_model_base.py b/packages/typespec-python/test/unbranded/generated/headasbooleantrue/headasbooleantrue/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/headasbooleantrue/headasbooleantrue/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/headasbooleantrue/headasbooleantrue/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/parameters-basic/parameters/basic/_model_base.py b/packages/typespec-python/test/unbranded/generated/parameters-basic/parameters/basic/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/parameters-basic/parameters/basic/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/parameters-basic/parameters/basic/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py b/packages/typespec-python/test/unbranded/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/parameters-body-optionality/parameters/bodyoptionality/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/parameters-collection-format/parameters/collectionformat/_model_base.py b/packages/typespec-python/test/unbranded/generated/parameters-collection-format/parameters/collectionformat/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/parameters-collection-format/parameters/collectionformat/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/parameters-collection-format/parameters/collectionformat/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/parameters-spread/parameters/spread/_model_base.py b/packages/typespec-python/test/unbranded/generated/parameters-spread/parameters/spread/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/parameters-spread/parameters/spread/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/parameters-spread/parameters/spread/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py b/packages/typespec-python/test/unbranded/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/payload-content-negotiation/payload/contentnegotiation/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py b/packages/typespec-python/test/unbranded/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/payload-json-merge-patch/payload/jsonmergepatch/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/payload-media-type/payload/mediatype/_model_base.py b/packages/typespec-python/test/unbranded/generated/payload-media-type/payload/mediatype/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/payload-media-type/payload/mediatype/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/payload-media-type/payload/mediatype/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/payload-multipart/payload/multipart/_model_base.py b/packages/typespec-python/test/unbranded/generated/payload-multipart/payload/multipart/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/payload-multipart/payload/multipart/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/payload-multipart/payload/multipart/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/payload-pageable/payload/pageable/_model_base.py b/packages/typespec-python/test/unbranded/generated/payload-pageable/payload/pageable/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/payload-pageable/payload/pageable/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/payload-pageable/payload/pageable/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/payload-xml/payload/xml/_model_base.py b/packages/typespec-python/test/unbranded/generated/payload-xml/payload/xml/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/payload-xml/payload/xml/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/payload-xml/payload/xml/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py b/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven1/resiliency/srv/driven1/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py b/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/resiliency-srv-driven2/resiliency/srv/driven2/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/routes/routes/_model_base.py b/packages/typespec-python/test/unbranded/generated/routes/routes/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/routes/routes/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/routes/routes/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py b/packages/typespec-python/test/unbranded/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/serialization-encoded-name-json/serialization/encodedname/json/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py b/packages/typespec-python/test/unbranded/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/server-endpoint-not-defined/server/endpoint/notdefined/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/server-path-multiple/server/path/multiple/_model_base.py b/packages/typespec-python/test/unbranded/generated/server-path-multiple/server/path/multiple/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/server-path-multiple/server/path/multiple/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/server-path-multiple/server/path/multiple/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/server-path-single/server/path/single/_model_base.py b/packages/typespec-python/test/unbranded/generated/server-path-single/server/path/single/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/server-path-single/server/path/single/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/server-path-single/server/path/single/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py b/packages/typespec-python/test/unbranded/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/server-versions-not-versioned/server/versions/notversioned/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/server-versions-versioned/server/versions/versioned/_model_base.py b/packages/typespec-python/test/unbranded/generated/server-versions-versioned/server/versions/versioned/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/server-versions-versioned/server/versions/versioned/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/server-versions-versioned/server/versions/versioned/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py b/packages/typespec-python/test/unbranded/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/special-headers-conditional-request/specialheaders/conditionalrequest/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py b/packages/typespec-python/test/unbranded/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/special-headers-repeatability/specialheaders/repeatability/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/special-words/specialwords/_model_base.py b/packages/typespec-python/test/unbranded/generated/special-words/specialwords/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/special-words/specialwords/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/special-words/specialwords/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-array/typetest/array/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-array/typetest/array/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-array/typetest/array/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-array/typetest/array/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-dictionary/typetest/dictionary/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-dictionary/typetest/dictionary/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-dictionary/typetest/dictionary/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-dictionary/typetest/dictionary/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-enum-extensible/typetest/enum/extensible/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-enum-fixed/typetest/enum/fixed/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-empty/typetest/model/empty/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-empty/typetest/model/empty/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-empty/typetest/model/empty/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-empty/typetest/model/empty/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-enumdiscriminator/typetest/model/enumdiscriminator/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-nesteddiscriminator/typetest/model/nesteddiscriminator/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-notdiscriminated/typetest/model/notdiscriminated/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-recursive/typetest/model/recursive/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-singlediscriminator/typetest/model/singlediscriminator/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-usage/typetest/model/usage/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-usage/typetest/model/usage/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-usage/typetest/model/usage/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-usage/typetest/model/usage/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-model-visibility/typetest/model/visibility/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-property-additionalproperties/typetest/property/additionalproperties/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-property-nullable/typetest/property/nullable/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-property-optional/typetest/property/optional/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-property-optional/typetest/property/optional/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-property-optional/typetest/property/optional/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-property-optional/typetest/property/optional/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-property-valuetypes/typetest/property/valuetypes/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-scalar/typetest/scalar/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-scalar/typetest/scalar/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-scalar/typetest/scalar/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-scalar/typetest/scalar/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/typetest-union/typetest/union/_model_base.py b/packages/typespec-python/test/unbranded/generated/typetest-union/typetest/union/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/typetest-union/typetest/union/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/typetest-union/typetest/union/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/versioning-added/versioning/added/_model_base.py b/packages/typespec-python/test/unbranded/generated/versioning-added/versioning/added/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/versioning-added/versioning/added/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/versioning-added/versioning/added/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py b/packages/typespec-python/test/unbranded/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/versioning-madeoptional/versioning/madeoptional/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/versioning-removed/versioning/removed/_model_base.py b/packages/typespec-python/test/unbranded/generated/versioning-removed/versioning/removed/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/versioning-removed/versioning/removed/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/versioning-removed/versioning/removed/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py b/packages/typespec-python/test/unbranded/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/versioning-renamedfrom/versioning/renamedfrom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py b/packages/typespec-python/test/unbranded/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/versioning-returntypechangedfrom/versioning/returntypechangedfrom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, diff --git a/packages/typespec-python/test/unbranded/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py b/packages/typespec-python/test/unbranded/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py index 93dadc3c4b1..ca57303fb2b 100644 --- a/packages/typespec-python/test/unbranded/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py +++ b/packages/typespec-python/test/unbranded/generated/versioning-typechangedfrom/versioning/typechangedfrom/_model_base.py @@ -894,6 +894,22 @@ def _deserialize( return _deserialize_with_callable(deserializer, value) +def _failsafe_deserialize( + deserializer: typing.Any, + value: typing.Any, + module: typing.Optional[str] = None, + rf: typing.Optional["_RestField"] = None, + format: typing.Optional[str] = None, +) -> typing.Any: + try: + return _deserialize(deserializer, value, module, rf, format) + except DeserializationError: + _LOGGER.warning( + "Ran into a deserialization error. Ignoring since this is failsafe deserialization", exc_info=True + ) + return None + + class _RestField: def __init__( self, From 8c0f31a4ff3f2dcca6c8db53a7c999a7e07e1e67 Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 11:49:02 +0000 Subject: [PATCH 6/8] Regenerate for autorest.python (2024-12-06 11:49:02) --- .../multiapi/sample/v1/models/_models_py3.py | 5 ++- .../multiapi/sample/v3/models/_models_py3.py | 5 ++- .../generated_samples/duration_get_null.py | 27 ++++++++++++ .../duration_put_positive_duration.py | 30 +++++++++++++ .../generated_samples/paths_get_empty.py | 28 +++++++++++++ .../custompollerpager/models/_models_py3.py | 5 ++- .../custombaseurlpaging/models/_models_py3.py | 5 ++- .../paging_get_pages_partial_url.py | 30 +++++++++++++ .../paging_get_pages_partial_url_operation.py | 30 +++++++++++++ .../generated_samples/http_success_head200.py | 37 ++++++++++++++++ .../generated_samples/http_success_head204.py | 37 ++++++++++++++++ .../generated_samples/http_success_head404.py | 37 ++++++++++++++++ .../head_exception_head200.py | 37 ++++++++++++++++ .../head_exception_head204.py | 37 ++++++++++++++++ .../generated_samples/http_success_head200.py | 36 ++++++++++++++++ .../generated_samples/http_success_head204.py | 36 ++++++++++++++++ .../generated_samples/http_success_head404.py | 36 ++++++++++++++++ .../Lro/lro/models/_models_py3.py | 5 ++- .../batch/paging/models/_models_py3.py | 5 ++- .../Paging/paging/models/_models_py3.py | 5 ++- .../storage/models/_models_py3.py | 5 ++- .../generated_samples/duration_get_null.py | 27 ++++++++++++ .../duration_put_positive_duration.py | 30 +++++++++++++ .../generated_samples/paths_get_empty.py | 28 +++++++++++++ .../paging_get_pages_partial_url.py | 30 +++++++++++++ .../paging_get_pages_partial_url_operation.py | 30 +++++++++++++ .../head_exception_head200.py | 37 ++++++++++++++++ .../head_exception_head204.py | 37 ++++++++++++++++ .../generated_samples/http_success_head200.py | 37 ++++++++++++++++ .../generated_samples/http_success_head204.py | 37 ++++++++++++++++ .../generated_samples/http_success_head404.py | 37 ++++++++++++++++ .../models/_models.py | 5 ++- .../multiapi/v0/models/_models_py3.py | 5 ++- .../multiapi/v1/models/_models_py3.py | 5 ++- .../multiapi/v3/models/_models_py3.py | 5 ++- .../v1/models/_models_py3.py | 5 ++- .../v3/models/_models_py3.py | 5 ++- .../v1/models/_models_py3.py | 5 ++- .../v3/models/_models_py3.py | 5 ++- .../v1/models/_models_py3.py | 5 ++- .../v3/models/_models_py3.py | 5 ++- .../multiapinoasync/v1/models/_models_py3.py | 5 ++- .../multiapinoasync/v3/models/_models_py3.py | 5 ++- .../multiapisecurity/v0/models/_models_py3.py | 5 ++- .../multiapisecurity/v1/models/_models_py3.py | 5 ++- .../submodule/v1/models/_models_py3.py | 5 ++- .../submodule/v3/models/_models_py3.py | 5 ++- .../multiapicombiner/models/_models.py | 5 ++- .../multiapicombiner/v1/models/_models_py3.py | 5 ++- .../multiapicombiner/v3/models/_models_py3.py | 5 ++- .../bodycomplex/models/_models_py3.py | 5 ++- .../models/_models_py3.py | 5 ++- .../Constants/constants/models/_models_py3.py | 5 ++- .../errorwithsecrets/models/_models_py3.py | 5 ++- .../models/_models_py3.py | 5 ++- .../_merge_patch_json_client_operations.py | 3 +- .../_merge_patch_json_client_operations.py | 3 +- .../modelflattening/models/_models_py3.py | 5 ++- .../_object_type_client_operations.py | 6 +-- .../_object_type_client_operations.py | 6 +-- .../requiredoptional/models/_models_py3.py | 5 ++- .../validation/models/_models_py3.py | 5 ++- .../Xml/xmlservice/models/_models_py3.py | 5 ++- .../aio/operations/_pet_operations.py | 42 ++++++++----------- .../operations/_pet_operations.py | 42 ++++++++----------- .../models/_models.py | 5 ++- .../aio/operations/_operations.py | 10 ++--- .../operations/_operations.py | 10 ++--- 68 files changed, 962 insertions(+), 113 deletions(-) create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py create mode 100644 packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py create mode 100644 packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py index 83aa365da8d..3748e0032bc 100644 --- a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py index 2f695839c35..0f284e05cf1 100644 --- a/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py +++ b/packages/autorest.python/samples/specification/multiapi/generated/azure/multiapi/sample/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py new file mode 100644 index 00000000000..045bab5d418 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_get_null.py @@ -0,0 +1,27 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from bodyduration import AutoRestDurationTestService + +""" +# PREREQUISITES + pip install autorestdurationtestservice +# USAGE + python duration_get_null.py +""" + + +def main(): + client = AutoRestDurationTestService() + + response = client.duration.get_null() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py new file mode 100644 index 00000000000..69e50626bbd --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/AzureBodyDuration/generated_samples/duration_put_positive_duration.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +import isodate + +from bodyduration import AutoRestDurationTestService + +""" +# PREREQUISITES + pip install autorestdurationtestservice +# USAGE + python duration_put_positive_duration.py +""" + + +def main(): + client = AutoRestDurationTestService() + + client.duration.put_positive_duration( + duration_body=isodate.parse_duration("P123DT22H14M12.011S"), + ) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py new file mode 100644 index 00000000000..c8c33fed4b6 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomBaseUri/generated_samples/paths_get_empty.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from custombaseurl import AutoRestParameterizedHostTestClient + +""" +# PREREQUISITES + pip install autorestparameterizedhosttestclient +# USAGE + python paths_get_empty.py +""" + + +def main(): + client = AutoRestParameterizedHostTestClient() + + client.paths.get_empty( + account_name="testaccount", + ) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py index 399fd76b242..e661461d066 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomPollerPager/custompollerpager/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, Union +from typing import Any, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class BodyParam(_serialization.Model): """BodyParam. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py index 177edc08ae3..fc88a8d66ac 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/custombaseurlpaging/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py new file mode 100644 index 00000000000..318995c0864 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from custombaseurlpaging import AutoRestParameterizedHostTestPagingClient + +""" +# PREREQUISITES + pip install autorestparameterizedhosttestpagingclient +# USAGE + python paging_get_pages_partial_url.py +""" + + +def main(): + client = AutoRestParameterizedHostTestPagingClient() + + response = client.paging.get_pages_partial_url( + account_name="testaccount", + ) + for item in response: + print(item) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py new file mode 100644 index 00000000000..37cd45add32 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/CustomUrlPaging/generated_samples/paging_get_pages_partial_url_operation.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from custombaseurlpaging import AutoRestParameterizedHostTestPagingClient + +""" +# PREREQUISITES + pip install autorestparameterizedhosttestpagingclient +# USAGE + python paging_get_pages_partial_url_operation.py +""" + + +def main(): + client = AutoRestParameterizedHostTestPagingClient() + + response = client.paging.get_pages_partial_url_operation( + account_name="testaccount", + ) + for item in response: + print(item) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py new file mode 100644 index 00000000000..be86f08131e --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head200.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from head import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadtestservice +# USAGE + python http_success_head200.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadTestService( + credential=DefaultAzureCredential(), + ) + + response = client.http_success.head200() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py new file mode 100644 index 00000000000..804edc5597d --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head204.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from head import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadtestservice +# USAGE + python http_success_head204.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadTestService( + credential=DefaultAzureCredential(), + ) + + response = client.http_success.head204() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py new file mode 100644 index 00000000000..4b50f67a8d9 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Head/generated_samples/http_success_head404.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from head import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadtestservice +# USAGE + python http_success_head404.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadTestService( + credential=DefaultAzureCredential(), + ) + + response = client.http_success.head404() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py new file mode 100644 index 00000000000..4285729aa20 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head200.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from headexceptions import AutoRestHeadExceptionTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadexceptiontestservice +# USAGE + python head_exception_head200.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadExceptionTestService( + credential=DefaultAzureCredential(), + ) + + response = client.head_exception.head200() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py new file mode 100644 index 00000000000..6ea218aa8e1 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadExceptions/generated_samples/head_exception_head204.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from headexceptions import AutoRestHeadExceptionTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadexceptiontestservice +# USAGE + python head_exception_head204.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadExceptionTestService( + credential=DefaultAzureCredential(), + ) + + response = client.head_exception.head204() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py new file mode 100644 index 00000000000..d0928b7aa33 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head200.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +import os + +from azure.core.credentials import AzureKeyCredential + +from headwithazurekeycredentialpolicy import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install autorestheadtestservice +# USAGE + python http_success_head200.py + + Before run the sample, please set environment variables AZURE_KEY with real value + which can access your service +""" + + +def main(): + client = AutoRestHeadTestService( + credential=AzureKeyCredential(key=os.getenv("AZURE_KEY")), + ) + + response = client.http_success.head200() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py new file mode 100644 index 00000000000..8dad096cb16 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head204.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +import os + +from azure.core.credentials import AzureKeyCredential + +from headwithazurekeycredentialpolicy import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install autorestheadtestservice +# USAGE + python http_success_head204.py + + Before run the sample, please set environment variables AZURE_KEY with real value + which can access your service +""" + + +def main(): + client = AutoRestHeadTestService( + credential=AzureKeyCredential(key=os.getenv("AZURE_KEY")), + ) + + response = client.http_success.head204() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py new file mode 100644 index 00000000000..b6c3e61f065 --- /dev/null +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/HeadWithAzureKeyCredentialPolicy/generated_samples/http_success_head404.py @@ -0,0 +1,36 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +import os + +from azure.core.credentials import AzureKeyCredential + +from headwithazurekeycredentialpolicy import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install autorestheadtestservice +# USAGE + python http_success_head404.py + + Before run the sample, please set environment variables AZURE_KEY with real value + which can access your service +""" + + +def main(): + client = AutoRestHeadTestService( + credential=AzureKeyCredential(key=os.getenv("AZURE_KEY")), + ) + + response = client.http_success.head404() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py index 5abc01aa699..68b6288aa8a 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Lro/lro/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Dict, Optional, Union +from typing import Any, Dict, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class OperationResult(_serialization.Model): """OperationResult. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py index 995c0f52b3c..e59e87a4ac9 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/PackageModeBatch/azure/packagemode/batch/paging/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, Union +from typing import Any, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class BodyParam(_serialization.Model): """BodyParam. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py index 0e95655c3bc..54d8bfc2674 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/Paging/paging/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, Union +from typing import Any, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class BodyParam(_serialization.Model): """BodyParam. diff --git a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py index 2e15afca307..406b6861daa 100644 --- a/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py +++ b/packages/autorest.python/test/azure/legacy/Expected/AcceptanceTests/StorageManagementClient/storage/models/_models_py3.py @@ -7,10 +7,13 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Bar(_serialization.Model): """The URIs that are used to perform a retrieval of a public blob, queue or table object. diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py new file mode 100644 index 00000000000..9197f6d6e6e --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_get_null.py @@ -0,0 +1,27 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from bodydurationversiontolerant import AutoRestDurationTestService + +""" +# PREREQUISITES + pip install autorestdurationtestservice +# USAGE + python duration_get_null.py +""" + + +def main(): + client = AutoRestDurationTestService() + + response = client.duration.get_null() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py new file mode 100644 index 00000000000..e3d92a29334 --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/AzureBodyDurationVersionTolerant/generated_samples/duration_put_positive_duration.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +import isodate + +from bodydurationversiontolerant import AutoRestDurationTestService + +""" +# PREREQUISITES + pip install autorestdurationtestservice +# USAGE + python duration_put_positive_duration.py +""" + + +def main(): + client = AutoRestDurationTestService() + + client.duration.put_positive_duration( + duration_body=isodate.parse_duration("P123DT22H14M12.011S"), + ) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py new file mode 100644 index 00000000000..cd2f88a83ff --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomBaseUriVersionTolerant/generated_samples/paths_get_empty.py @@ -0,0 +1,28 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from custombaseurlversiontolerant import AutoRestParameterizedHostTestClient + +""" +# PREREQUISITES + pip install autorestparameterizedhosttestclient +# USAGE + python paths_get_empty.py +""" + + +def main(): + client = AutoRestParameterizedHostTestClient() + + client.paths.get_empty( + account_name="testaccount", + ) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py new file mode 100644 index 00000000000..b9c7cfcfc2b --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from custombaseurlpagingversiontolerant import AutoRestParameterizedHostTestPagingClient + +""" +# PREREQUISITES + pip install autorestparameterizedhosttestpagingclient +# USAGE + python paging_get_pages_partial_url.py +""" + + +def main(): + client = AutoRestParameterizedHostTestPagingClient() + + response = client.paging.get_pages_partial_url( + account_name="testaccount", + ) + for item in response: + print(item) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py new file mode 100644 index 00000000000..9c315c6b84e --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/CustomUrlPagingVersionTolerant/generated_samples/paging_get_pages_partial_url_operation.py @@ -0,0 +1,30 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from custombaseurlpagingversiontolerant import AutoRestParameterizedHostTestPagingClient + +""" +# PREREQUISITES + pip install autorestparameterizedhosttestpagingclient +# USAGE + python paging_get_pages_partial_url_operation.py +""" + + +def main(): + client = AutoRestParameterizedHostTestPagingClient() + + response = client.paging.get_pages_partial_url_operation( + account_name="testaccount", + ) + for item in response: + print(item) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py new file mode 100644 index 00000000000..95892b1b19a --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head200.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from headexceptionsversiontolerant import AutoRestHeadExceptionTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadexceptiontestservice +# USAGE + python head_exception_head200.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadExceptionTestService( + credential=DefaultAzureCredential(), + ) + + response = client.head_exception.head200() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py new file mode 100644 index 00000000000..90ceb384219 --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadExceptionsVersionTolerant/generated_samples/head_exception_head204.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from headexceptionsversiontolerant import AutoRestHeadExceptionTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadexceptiontestservice +# USAGE + python head_exception_head204.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadExceptionTestService( + credential=DefaultAzureCredential(), + ) + + response = client.head_exception.head204() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py new file mode 100644 index 00000000000..e036e7e5142 --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head200.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from headversiontolerant import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadtestservice +# USAGE + python http_success_head200.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadTestService( + credential=DefaultAzureCredential(), + ) + + response = client.http_success.head200() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py new file mode 100644 index 00000000000..8637361f364 --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head204.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from headversiontolerant import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadtestservice +# USAGE + python http_success_head204.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadTestService( + credential=DefaultAzureCredential(), + ) + + response = client.http_success.head204() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py new file mode 100644 index 00000000000..ed7221131d0 --- /dev/null +++ b/packages/autorest.python/test/azure/version-tolerant/Expected/AcceptanceTests/HeadVersionTolerant/generated_samples/http_success_head404.py @@ -0,0 +1,37 @@ +# coding=utf-8 +# -------------------------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# Licensed under the MIT License. See License.txt in the project root for license information. +# Code generated by Microsoft (R) AutoRest Code Generator. +# Changes may cause incorrect behavior and will be lost if the code is regenerated. +# -------------------------------------------------------------------------- + +from azure.identity import DefaultAzureCredential + +from headversiontolerant import AutoRestHeadTestService + +""" +# PREREQUISITES + pip install azure-identity + pip install autorestheadtestservice +# USAGE + python http_success_head404.py + + Before run the sample, please set the values of the client ID, tenant ID and client secret + of the AAD application as environment variables: AZURE_CLIENT_ID, AZURE_TENANT_ID, + AZURE_CLIENT_SECRET. For more info about how to get the value, please see: + https://docs.microsoft.com/azure/active-directory/develop/howto-create-service-principal-portal +""" + + +def main(): + client = AutoRestHeadTestService( + credential=DefaultAzureCredential(), + ) + + response = client.http_success.head404() + print(response) + + +if __name__ == "__main__": + main() diff --git a/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py b/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py index 100da986cd1..2779069dfd1 100644 --- a/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py +++ b/packages/autorest.python/test/dpg/version-tolerant/Expected/AcceptanceTests/DPGTestModelsVersionTolerant/dpgtestmodelsversiontolerant/models/_models.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional, Union +from typing import Any, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Input(_serialization.Model): """Input. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py index 15d7bb640ad..8b98be42e59 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py index ed6f229313a..9c722d1c22b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py index a8292170171..b0579fca699 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py index ffbf90c6f18..93c4265bd8c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py index 2fa2f8f28e0..9c0180868ed 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py index 7c46380faec..762767d95bf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py index 0935a52d2d0..db26518e212 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py index 61d47a63696..d0e6dc50bed 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py index 32d705523ce..79116ab4352 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py index 24c341cd35b..c60e45da038 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py index 284706c8d1a..87fb36d250e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py index 133294256cc..29649508ccd 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py index f689de5518f..b992d5ee62c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py index 10facdeb65c..a9f3e5954c4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py index 5768703fec4..c2a687ce8c0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py index 7022dd2e9d2..560c1398930 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py index d47335f09b7..2188435a8fb 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py index 7db31c76bef..80fc2d5d141 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING from ... import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """Error. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py index b6f7560c84d..a0a1d0d6aa6 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyComplex/bodycomplex/models/_models_py3.py @@ -8,10 +8,13 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union import msrest.serialization +if TYPE_CHECKING: + from .. import models as _models + class ArrayWrapper(msrest.serialization.Model): """ArrayWrapper. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py index 5d43068368e..d684fe1f617 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/BodyFormUrlEncodedData/bodyformurlencodeddata/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Optional, Union +from typing import Any, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Paths14Hl8BdFormsdataurlencodedPetAddPetidPostRequestbodyContentApplicationXWwwFormUrlencodedSchema( _serialization.Model diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py index b0a67c1f5c8..3f916138cc1 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Constants/constants/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Literal, Optional, Union +from typing import Any, Literal, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class ModelAsStringNoRequiredOneValueDefault(_serialization.Model): """ModelAsStringNoRequiredOneValueDefault. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py index 8f5c8887e95..0721c41ba08 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ErrorWithSecrets/errorwithsecrets/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Error(_serialization.Model): """The error object. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py index 030adf044f8..34a5f6ea52b 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ExtensibleEnums/extensibleenumsswagger/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Optional, Union +from typing import Any, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class Pet(_serialization.Model): """Pet. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/aio/operations/_merge_patch_json_client_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/aio/operations/_merge_patch_json_client_operations.py index ab7b8ac82d9..fee2516ba7f 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/aio/operations/_merge_patch_json_client_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/aio/operations/_merge_patch_json_client_operations.py @@ -78,8 +78,7 @@ async def patch_single(self, body: JSON, **kwargs: Any) -> None: if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize("object", pipeline_response) - raise HttpResponseError(response=response, model=error) + raise HttpResponseError(response=response) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/operations/_merge_patch_json_client_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/operations/_merge_patch_json_client_operations.py index d3c3635842b..153f72810ea 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/operations/_merge_patch_json_client_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/MergePatchJson/mergepatchjson/operations/_merge_patch_json_client_operations.py @@ -98,8 +98,7 @@ def patch_single(self, body: JSON, **kwargs: Any) -> None: # pylint: disable=in if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize("object", pipeline_response) - raise HttpResponseError(response=response, model=error) + raise HttpResponseError(response=response) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py index 7ba62aeb64c..1473e8fb5fa 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ModelFlattening/modelflattening/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, Dict, List, Literal, Optional +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class BaseProduct(_serialization.Model): """The product documentation. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/aio/operations/_object_type_client_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/aio/operations/_object_type_client_operations.py index 3a0674713a1..bd723fa18c3 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/aio/operations/_object_type_client_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/aio/operations/_object_type_client_operations.py @@ -72,8 +72,7 @@ async def get(self, **kwargs: Any) -> JSON: if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize("object", pipeline_response) - raise HttpResponseError(response=response, model=error) + raise HttpResponseError(response=response) deserialized = self._deserialize("object", pipeline_response.http_response) @@ -127,8 +126,7 @@ async def put(self, put_object: JSON, **kwargs: Any) -> None: if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize("object", pipeline_response) - raise HttpResponseError(response=response, model=error) + raise HttpResponseError(response=response) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/operations/_object_type_client_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/operations/_object_type_client_operations.py index 14647cf9cfc..5fbec14d4a4 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/operations/_object_type_client_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/ObjectType/objecttype/operations/_object_type_client_operations.py @@ -106,8 +106,7 @@ def get(self, **kwargs: Any) -> JSON: if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize("object", pipeline_response) - raise HttpResponseError(response=response, model=error) + raise HttpResponseError(response=response) deserialized = self._deserialize("object", pipeline_response.http_response) @@ -161,8 +160,7 @@ def put(self, put_object: JSON, **kwargs: Any) -> None: # pylint: disable=incon if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize("object", pipeline_response) - raise HttpResponseError(response=response, model=error) + raise HttpResponseError(response=response) if cls: return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py index aa5163f7604..b5ef0ed3e28 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/RequiredOptional/requiredoptional/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Optional +from typing import Any, List, Optional, TYPE_CHECKING import msrest.serialization +if TYPE_CHECKING: + from .. import models as _models + class ArrayOptionalWrapper(msrest.serialization.Model): """ArrayOptionalWrapper. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py index ffbadc428e4..9722f402934 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Validation/validation/models/_models_py3.py @@ -6,10 +6,13 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- -from typing import Any, List, Literal, Optional +from typing import Any, List, Literal, Optional, TYPE_CHECKING import msrest.serialization +if TYPE_CHECKING: + from .. import models as _models + class ChildProduct(msrest.serialization.Model): """The product documentation. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py index 2e4a7cda59f..6b8562c0332 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/Xml/xmlservice/models/_models_py3.py @@ -8,10 +8,13 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class AccessPolicy(_serialization.Model): """An Access policy. diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py index 40f3f1cb254..497631f898c 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/aio/operations/_pet_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -68,14 +68,6 @@ async def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[_models.Pe 401: ClientAuthenticationError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 400: HttpResponseError, - 404: cast( - Type[HttpResponseError], - lambda response: ResourceNotFoundError( - response=response, model=self._deserialize(NotFoundErrorBase, response) - ), - ), - 501: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -99,8 +91,12 @@ async def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[_models.Pe response = pipeline_response.http_response if response.status_code not in [200, 202]: - map_error(status_code=response.status_code, response=response, error_map=error_map) # type: ignore - raise HttpResponseError(response=response) + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = None + if response.status_code == 404: + error = self._deserialize.failsafe_deserialize(_models.NotFoundErrorBase, pipeline_response) + raise ResourceNotFoundError(response=response, model=error) + raise HttpResponseError(response=response, model=error) deserialized = None if response.status_code == 200: @@ -126,12 +122,6 @@ async def do_something(self, what_action: str, **kwargs: Any) -> _models.PetActi 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=self._deserialize(PetActionError, response) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -156,7 +146,11 @@ async def do_something(self, what_action: str, **kwargs: Any) -> _models.PetActi if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + error = None + if response.status_code == 500: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + else: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) raise HttpResponseError(response=response, model=error) deserialized = self._deserialize("PetAction", pipeline_response.http_response) @@ -183,12 +177,6 @@ async def has_models_param(self, models: str = "value1", **kwargs: Any) -> None: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=self._deserialize(PetActionError, response) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -213,7 +201,11 @@ async def has_models_param(self, models: str = "value1", **kwargs: Any) -> None: if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + error = None + if response.status_code == 500: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + else: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) raise HttpResponseError(response=response, model=error) if cls: diff --git a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py index cbe689df1b7..b0bf565e9cf 100644 --- a/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py +++ b/packages/autorest.python/test/vanilla/legacy/Expected/AcceptanceTests/XmsErrorResponse/xmserrorresponse/operations/_pet_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar from azure.core.exceptions import ( ClientAuthenticationError, @@ -125,14 +125,6 @@ def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[_models.Pet]: 401: ClientAuthenticationError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 400: HttpResponseError, - 404: cast( - Type[HttpResponseError], - lambda response: ResourceNotFoundError( - response=response, model=self._deserialize(NotFoundErrorBase, response) - ), - ), - 501: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -156,8 +148,12 @@ def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[_models.Pet]: response = pipeline_response.http_response if response.status_code not in [200, 202]: - map_error(status_code=response.status_code, response=response, error_map=error_map) # type: ignore - raise HttpResponseError(response=response) + map_error(status_code=response.status_code, response=response, error_map=error_map) + error = None + if response.status_code == 404: + error = self._deserialize.failsafe_deserialize(_models.NotFoundErrorBase, pipeline_response) + raise ResourceNotFoundError(response=response, model=error) + raise HttpResponseError(response=response, model=error) deserialized = None if response.status_code == 200: @@ -183,12 +179,6 @@ def do_something(self, what_action: str, **kwargs: Any) -> _models.PetAction: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=self._deserialize(PetActionError, response) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -213,7 +203,11 @@ def do_something(self, what_action: str, **kwargs: Any) -> _models.PetAction: if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + error = None + if response.status_code == 500: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + else: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) raise HttpResponseError(response=response, model=error) deserialized = self._deserialize("PetAction", pipeline_response.http_response) @@ -242,12 +236,6 @@ def has_models_param( # pylint: disable=inconsistent-return-statements 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: cast( - Type[HttpResponseError], - lambda response: HttpResponseError( - response=response, model=self._deserialize(PetActionError, response) - ), - ), } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -272,7 +260,11 @@ def has_models_param( # pylint: disable=inconsistent-return-statements if response.status_code not in [200]: map_error(status_code=response.status_code, response=response, error_map=error_map) - error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + error = None + if response.status_code == 500: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) + else: + error = self._deserialize.failsafe_deserialize(_models.PetActionError, pipeline_response) raise HttpResponseError(response=response, model=error) if cls: diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py index b4fee3cdc46..5ddfca1b641 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmlVersionTolerant/xmlserviceversiontolerant/models/_models.py @@ -8,10 +8,13 @@ # -------------------------------------------------------------------------- import datetime -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union from .. import _serialization +if TYPE_CHECKING: + from .. import models as _models + class AccessPolicy(_serialization.Model): """An Access policy. diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/aio/operations/_operations.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/aio/operations/_operations.py index 2e8c6d3da1d..1df8ae62cab 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/aio/operations/_operations.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/aio/operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar, cast from azure.core.exceptions import ( ClientAuthenticationError, @@ -73,11 +73,9 @@ async def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[JSON]: """ error_map: MutableMapping = { 401: ClientAuthenticationError, + 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 400: HttpResponseError, - 404: cast(Type[HttpResponseError], lambda response: ResourceNotFoundError(response=response)), - 501: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -101,7 +99,7 @@ async def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[JSON]: response = pipeline_response.http_response if response.status_code not in [200, 202]: - map_error(status_code=response.status_code, response=response, error_map=error_map) # type: ignore + map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response) deserialized = None @@ -139,7 +137,6 @@ async def do_something(self, what_action: str, **kwargs: Any) -> JSON: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -193,7 +190,6 @@ async def has_models_param(self, *, models: str = "value1", **kwargs: Any) -> No 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) diff --git a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/operations/_operations.py b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/operations/_operations.py index 72fff0f9516..1db5a89a594 100644 --- a/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/operations/_operations.py +++ b/packages/autorest.python/test/vanilla/version-tolerant/Expected/AcceptanceTests/XmsErrorResponseVersionTolerant/xmserrorresponseversiontolerant/operations/_operations.py @@ -6,7 +6,7 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- import sys -from typing import Any, Callable, Dict, Optional, Type, TypeVar, cast +from typing import Any, Callable, Dict, Optional, TypeVar, cast from azure.core.exceptions import ( ClientAuthenticationError, @@ -130,11 +130,9 @@ def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[JSON]: """ error_map: MutableMapping = { 401: ClientAuthenticationError, + 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 400: HttpResponseError, - 404: cast(Type[HttpResponseError], lambda response: ResourceNotFoundError(response=response)), - 501: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -158,7 +156,7 @@ def get_pet_by_id(self, pet_id: str, **kwargs: Any) -> Optional[JSON]: response = pipeline_response.http_response if response.status_code not in [200, 202]: - map_error(status_code=response.status_code, response=response, error_map=error_map) # type: ignore + map_error(status_code=response.status_code, response=response, error_map=error_map) raise HttpResponseError(response=response) deserialized = None @@ -196,7 +194,6 @@ def do_something(self, what_action: str, **kwargs: Any) -> JSON: 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) @@ -252,7 +249,6 @@ def has_models_param( # pylint: disable=inconsistent-return-statements 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError, - 500: HttpResponseError, } error_map.update(kwargs.pop("error_map", {}) or {}) From cfc4dc46b920d6e3dd6c3028d977540be80b38ca Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 12:26:07 +0000 Subject: [PATCH 7/8] Update dependencies (2024-12-06 12:26:07) --- packages/autorest.python/package.json | 2 +- packages/typespec-python/package.json | 2 +- pnpm-lock.yaml | 14 +++++++------- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/packages/autorest.python/package.json b/packages/autorest.python/package.json index bf84e9fbe71..13b286e9406 100644 --- a/packages/autorest.python/package.json +++ b/packages/autorest.python/package.json @@ -29,7 +29,7 @@ }, "homepage": "https://github.com/Azure/autorest.python/blob/main/README.md", "dependencies": { - "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", + "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", "@autorest/system-requirements": "~1.0.2", "fs-extra": "~11.2.0", "tsx": "~4.19.1" diff --git a/packages/typespec-python/package.json b/packages/typespec-python/package.json index e43b2678aa9..ab59ae0262f 100644 --- a/packages/typespec-python/package.json +++ b/packages/typespec-python/package.json @@ -60,7 +60,7 @@ "js-yaml": "~4.1.0", "semver": "~7.6.2", "tsx": "~4.19.1", - "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", + "@typespec/http-client-python": "https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz", "fs-extra": "~11.2.0" }, "devDependencies": { diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml index cf4709e454b..c67391aa487 100644 --- a/pnpm-lock.yaml +++ b/pnpm-lock.yaml @@ -54,8 +54,8 @@ importers: specifier: ~1.0.2 version: 1.0.2 '@typespec/http-client-python': - specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz - version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) + specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz + version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) fs-extra: specifier: ~11.2.0 version: 11.2.0 @@ -76,8 +76,8 @@ importers: packages/typespec-python: dependencies: '@typespec/http-client-python': - specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz - version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) + specifier: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz + version: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi) fs-extra: specifier: ~11.2.0 version: 11.2.0 @@ -1233,8 +1233,8 @@ packages: engines: {node: '>=18.0.0'} hasBin: true - '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz': - resolution: {tarball: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz} + '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz': + resolution: {tarball: https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz} version: 0.3.12 engines: {node: '>=18.0.0'} peerDependencies: @@ -4919,7 +4919,7 @@ snapshots: yaml: 2.5.1 yargs: 17.7.2 - '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTI5Ny9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi)': + '@typespec/http-client-python@https://artprodcus3.artifacts.visualstudio.com/A0fb41ef4-5012-48a9-bf39-4ee3de03ee35/29ec6040-b234-4e31-b139-33dc4287b756/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2F6dXJlLXNkay9wcm9qZWN0SWQvMjllYzYwNDAtYjIzNC00ZTMxLWIxMzktMzNkYzQyODdiNzU2L2J1aWxkSWQvNDM5MTM4My9hcnRpZmFjdE5hbWUvYnVpbGRfYXJ0aWZhY3RzX3B5dGhvbg2/content?format=file&subPath=%2Fpackages%2Ftypespec-http-client-python-0.3.12.tgz(v3aklw2azvdrnx5j6agyixaimi)': dependencies: '@azure-tools/typespec-autorest': 0.48.0(2qbrnlizjwf7gxxublducwg43e) '@azure-tools/typespec-azure-core': 0.48.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))(@typespec/rest@0.62.0(@typespec/compiler@0.62.0)(@typespec/http@0.62.0(@typespec/compiler@0.62.0))) From a426d91327c7a9c14746523027ee5b41c00f2dd9 Mon Sep 17 00:00:00 2001 From: AutoPrFromHttpClientPython Date: Fri, 6 Dec 2024 12:45:24 +0000 Subject: [PATCH 8/8] Regenerate for autorest.python (2024-12-06 12:45:24) --- .../Multiapi/multiapi/v0/__init__.py | 5 +- .../Multiapi/multiapi/v0/_configuration.py | 43 +- .../multiapi/v0/_multiapi_service_client.py | 37 +- .../Multiapi/multiapi/v0/aio/__init__.py | 5 +- .../multiapi/v0/aio/_configuration.py | 43 +- .../v0/aio/_multiapi_service_client.py | 33 +- .../multiapi/v0/aio/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 46 +- .../Multiapi/multiapi/v0/models/__init__.py | 9 +- .../multiapi/v0/models/_models_py3.py | 24 +- .../multiapi/v0/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 66 +- .../Multiapi/multiapi/v1/__init__.py | 5 +- .../Multiapi/multiapi/v1/_configuration.py | 43 +- .../multiapi/v1/_multiapi_service_client.py | 37 +- .../Multiapi/multiapi/v1/_vendor.py | 7 +- .../Multiapi/multiapi/v1/aio/__init__.py | 5 +- .../multiapi/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 33 +- .../Multiapi/multiapi/v1/aio/_vendor.py | 7 +- .../multiapi/v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 +++---- .../_operation_group_one_operations.py | 46 +- .../Multiapi/multiapi/v1/models/__init__.py | 11 +- .../multiapi/v1/models/_models_py3.py | 33 +- .../multiapi/v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++------ .../_operation_group_one_operations.py | 66 +- .../Multiapi/multiapi/v2/__init__.py | 5 +- .../Multiapi/multiapi/v2/_configuration.py | 43 +- .../multiapi/v2/_multiapi_service_client.py | 39 +- .../Multiapi/multiapi/v2/_vendor.py | 7 +- .../Multiapi/multiapi/v2/aio/__init__.py | 5 +- .../multiapi/v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 35 +- .../Multiapi/multiapi/v2/aio/_vendor.py | 7 +- .../multiapi/v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 +-- .../_operation_group_two_operations.py | 47 +- .../Multiapi/multiapi/v2/models/__init__.py | 7 +- .../multiapi/v2/models/_models_py3.py | 18 +- .../multiapi/v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../Multiapi/multiapi/v3/__init__.py | 5 +- .../Multiapi/multiapi/v3/_configuration.py | 43 +- .../multiapi/v3/_multiapi_service_client.py | 39 +- .../Multiapi/multiapi/v3/_vendor.py | 7 +- .../Multiapi/multiapi/v3/aio/__init__.py | 5 +- .../multiapi/v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 35 +- .../Multiapi/multiapi/v3/aio/_vendor.py | 7 +- .../multiapi/v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 +-- .../_operation_group_one_operations.py | 115 ++- .../_operation_group_two_operations.py | 92 +-- .../Multiapi/multiapi/v3/models/__init__.py | 13 +- .../multiapi/v3/models/_models_py3.py | 34 +- .../models/_multiapi_service_client_enums.py | 3 +- .../multiapi/v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../v1/__init__.py | 5 +- .../v1/_configuration.py | 37 +- .../v1/_multiapi_service_client.py | 37 +- .../v1/_vendor.py | 7 +- .../v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 37 +- .../v1/aio/_multiapi_service_client.py | 35 +- .../v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 +++---- .../_operation_group_one_operations.py | 46 +- .../v1/models/__init__.py | 11 +- .../v1/models/_models_py3.py | 33 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++------ .../_operation_group_one_operations.py | 66 +- .../v2/__init__.py | 5 +- .../v2/_configuration.py | 37 +- .../v2/_multiapi_service_client.py | 39 +- .../v2/_vendor.py | 7 +- .../v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 37 +- .../v2/aio/_multiapi_service_client.py | 37 +- .../v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 +-- .../_operation_group_two_operations.py | 47 +- .../v2/models/__init__.py | 7 +- .../v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../v3/__init__.py | 5 +- .../v3/_configuration.py | 37 +- .../v3/_multiapi_service_client.py | 39 +- .../v3/_vendor.py | 7 +- .../v3/aio/__init__.py | 5 +- .../v3/aio/_configuration.py | 37 +- .../v3/aio/_multiapi_service_client.py | 37 +- .../v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 +-- .../_operation_group_one_operations.py | 115 ++- .../_operation_group_two_operations.py | 92 +-- .../v3/models/__init__.py | 13 +- .../v3/models/_models_py3.py | 34 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapicustombaseurl/v1/__init__.py | 5 +- .../v1/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 42 +- .../multiapicustombaseurl/v1/_vendor.py | 7 +- .../multiapicustombaseurl/v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 40 +- .../multiapicustombaseurl/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 44 +- .../v1/models/__init__.py | 5 +- .../v1/models/_models_py3.py | 9 +- .../v1/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 70 +- .../multiapicustombaseurl/v2/__init__.py | 5 +- .../v2/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 42 +- .../multiapicustombaseurl/v2/_vendor.py | 7 +- .../multiapicustombaseurl/v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 44 +- ...multiapi_custom_base_url_service_client.py | 40 +- .../multiapicustombaseurl/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 44 +- .../v2/models/__init__.py | 5 +- .../v2/models/_models_py3.py | 9 +- .../v2/operations/__init__.py | 5 +- ...stom_base_url_service_client_operations.py | 70 +- .../multiapidataplane/v1/__init__.py | 5 +- .../multiapidataplane/v1/_configuration.py | 43 +- .../v1/_multiapi_service_client.py | 36 +- .../multiapidataplane/v1/_vendor.py | 7 +- .../multiapidataplane/v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 32 +- .../multiapidataplane/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 +++---- .../_operation_group_one_operations.py | 46 +- .../multiapidataplane/v1/models/__init__.py | 11 +- .../v1/models/_models_py3.py | 33 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++------ .../_operation_group_one_operations.py | 66 +- .../multiapidataplane/v2/__init__.py | 5 +- .../multiapidataplane/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 38 +- .../multiapidataplane/v2/_vendor.py | 7 +- .../multiapidataplane/v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 34 +- .../multiapidataplane/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 +-- .../_operation_group_two_operations.py | 47 +- .../multiapidataplane/v2/models/__init__.py | 7 +- .../v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../multiapidataplane/v3/__init__.py | 5 +- .../multiapidataplane/v3/_configuration.py | 43 +- .../v3/_multiapi_service_client.py | 38 +- .../multiapidataplane/v3/_vendor.py | 7 +- .../multiapidataplane/v3/aio/__init__.py | 5 +- .../v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 34 +- .../multiapidataplane/v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 +-- .../_operation_group_one_operations.py | 115 ++- .../_operation_group_two_operations.py | 92 +-- .../multiapidataplane/v3/models/__init__.py | 13 +- .../v3/models/_models_py3.py | 34 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapikeywordonly/v1/__init__.py | 5 +- .../multiapikeywordonly/v1/_configuration.py | 43 +- .../v1/_multiapi_service_client.py | 36 +- .../multiapikeywordonly/v1/_vendor.py | 7 +- .../multiapikeywordonly/v1/aio/__init__.py | 5 +- .../v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 32 +- .../multiapikeywordonly/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 275 +++---- .../_operation_group_one_operations.py | 46 +- .../multiapikeywordonly/v1/models/__init__.py | 11 +- .../v1/models/_models_py3.py | 33 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 371 ++++------ .../_operation_group_one_operations.py | 66 +- .../multiapikeywordonly/v2/__init__.py | 5 +- .../multiapikeywordonly/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 38 +- .../multiapikeywordonly/v2/_vendor.py | 7 +- .../multiapikeywordonly/v2/aio/__init__.py | 5 +- .../v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 34 +- .../multiapikeywordonly/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 79 +- .../_operation_group_one_operations.py | 95 +-- .../_operation_group_two_operations.py | 48 +- .../multiapikeywordonly/v2/models/__init__.py | 7 +- .../v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 139 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 70 +- .../multiapikeywordonly/v3/__init__.py | 5 +- .../multiapikeywordonly/v3/_configuration.py | 43 +- .../v3/_multiapi_service_client.py | 38 +- .../multiapikeywordonly/v3/_vendor.py | 7 +- .../multiapikeywordonly/v3/aio/__init__.py | 5 +- .../v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 34 +- .../multiapikeywordonly/v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 +-- .../_operation_group_one_operations.py | 115 ++- .../_operation_group_two_operations.py | 92 +-- .../multiapikeywordonly/v3/models/__init__.py | 13 +- .../v3/models/_models_py3.py | 34 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapinoasync/v1/__init__.py | 5 +- .../multiapinoasync/v1/_configuration.py | 43 +- .../v1/_multiapi_service_client.py | 37 +- .../multiapinoasync/v1/_vendor.py | 7 +- .../multiapinoasync/v1/models/__init__.py | 11 +- .../multiapinoasync/v1/models/_models_py3.py | 33 +- .../multiapinoasync/v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++------ .../_operation_group_one_operations.py | 66 +- .../multiapinoasync/v2/__init__.py | 5 +- .../multiapinoasync/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 39 +- .../multiapinoasync/v2/_vendor.py | 7 +- .../multiapinoasync/v2/models/__init__.py | 7 +- .../multiapinoasync/v2/models/_models_py3.py | 18 +- .../multiapinoasync/v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../multiapinoasync/v3/__init__.py | 5 +- .../multiapinoasync/v3/_configuration.py | 43 +- .../v3/_multiapi_service_client.py | 39 +- .../multiapinoasync/v3/_vendor.py | 7 +- .../multiapinoasync/v3/models/__init__.py | 13 +- .../multiapinoasync/v3/models/_models_py3.py | 34 +- .../models/_multiapi_service_client_enums.py | 3 +- .../multiapinoasync/v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapisecurity/v0/__init__.py | 5 +- .../multiapisecurity/v0/_configuration.py | 43 +- .../v0/_multiapi_service_client.py | 36 +- .../multiapisecurity/v0/aio/__init__.py | 5 +- .../multiapisecurity/v0/aio/_configuration.py | 43 +- .../v0/aio/_multiapi_service_client.py | 32 +- .../v0/aio/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 46 +- .../multiapisecurity/v0/models/__init__.py | 9 +- .../multiapisecurity/v0/models/_models_py3.py | 24 +- .../v0/operations/__init__.py | 5 +- .../_operation_group_one_operations.py | 66 +- .../multiapisecurity/v1/__init__.py | 5 +- .../multiapisecurity/v1/_configuration.py | 43 +- .../v1/_multiapi_service_client.py | 36 +- .../multiapisecurity/v1/_vendor.py | 7 +- .../multiapisecurity/v1/aio/__init__.py | 5 +- .../multiapisecurity/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 32 +- .../multiapisecurity/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 +++---- .../_operation_group_one_operations.py | 46 +- .../multiapisecurity/v1/models/__init__.py | 11 +- .../multiapisecurity/v1/models/_models_py3.py | 33 +- .../v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++------ .../_operation_group_one_operations.py | 66 +- .../multiapiwithsubmodule/__init__.py | 2 +- .../submodule/v1/__init__.py | 5 +- .../submodule/v1/_configuration.py | 43 +- .../submodule/v1/_multiapi_service_client.py | 37 +- .../submodule/v1/_vendor.py | 7 +- .../submodule/v1/aio/__init__.py | 5 +- .../submodule/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 33 +- .../submodule/v1/aio/_vendor.py | 7 +- .../submodule/v1/aio/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 273 +++---- .../_operation_group_one_operations.py | 46 +- .../submodule/v1/models/__init__.py | 11 +- .../submodule/v1/models/_models_py3.py | 33 +- .../submodule/v1/operations/__init__.py | 7 +- .../_multiapi_service_client_operations.py | 369 ++++------ .../_operation_group_one_operations.py | 66 +- .../submodule/v2/__init__.py | 5 +- .../submodule/v2/_configuration.py | 43 +- .../submodule/v2/_multiapi_service_client.py | 39 +- .../submodule/v2/_vendor.py | 7 +- .../submodule/v2/aio/__init__.py | 5 +- .../submodule/v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 35 +- .../submodule/v2/aio/_vendor.py | 7 +- .../submodule/v2/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 77 +- .../_operation_group_one_operations.py | 95 +-- .../_operation_group_two_operations.py | 47 +- .../submodule/v2/models/__init__.py | 7 +- .../submodule/v2/models/_models_py3.py | 18 +- .../submodule/v2/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 137 ++-- .../_operation_group_one_operations.py | 139 ++-- .../_operation_group_two_operations.py | 71 +- .../submodule/v3/__init__.py | 5 +- .../submodule/v3/_configuration.py | 43 +- .../submodule/v3/_multiapi_service_client.py | 39 +- .../submodule/v3/_vendor.py | 7 +- .../submodule/v3/aio/__init__.py | 5 +- .../submodule/v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 35 +- .../submodule/v3/aio/_vendor.py | 7 +- .../submodule/v3/aio/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 91 +-- .../_operation_group_one_operations.py | 115 ++- .../_operation_group_two_operations.py | 92 +-- .../submodule/v3/models/__init__.py | 13 +- .../submodule/v3/models/_models_py3.py | 34 +- .../models/_multiapi_service_client_enums.py | 3 +- .../submodule/v3/operations/__init__.py | 9 +- .../_multiapi_service_client_operations.py | 131 ++-- .../_operation_group_one_operations.py | 153 ++-- .../_operation_group_two_operations.py | 134 ++-- .../multiapicombiner/__init__.py | 5 +- .../multiapicombiner/_configuration.py | 43 +- .../multiapicombiner/_vendor.py | 7 +- .../multiapicombiner/aio/__init__.py | 5 +- .../multiapicombiner/aio/_configuration.py | 43 +- .../multiapicombiner/aio/_vendor.py | 7 +- .../aio/operations/_operations.py | 462 ++++++------ .../multiapicombiner/models/_enums.py | 3 +- .../multiapicombiner/models/_models.py | 54 +- .../operations/_operations.py | 689 +++++++----------- .../multiapicombiner/v1/__init__.py | 5 +- .../multiapicombiner/v1/_configuration.py | 43 +- .../v1/_multiapi_service_client.py | 37 +- .../multiapicombiner/v1/_vendor.py | 7 +- .../multiapicombiner/v1/aio/__init__.py | 5 +- .../multiapicombiner/v1/aio/_configuration.py | 43 +- .../v1/aio/_multiapi_service_client.py | 33 +- .../multiapicombiner/v1/aio/_vendor.py | 7 +- .../v1/aio/operations/__init__.py | 7 +- .../v1/aio/operations/_operations.py | 308 ++++---- .../multiapicombiner/v1/models/__init__.py | 11 +- .../multiapicombiner/v1/models/_models_py3.py | 33 +- .../v1/operations/__init__.py | 7 +- .../v1/operations/_operations.py | 421 +++++------ .../multiapicombiner/v2/__init__.py | 5 +- .../multiapicombiner/v2/_configuration.py | 43 +- .../v2/_multiapi_service_client.py | 39 +- .../multiapicombiner/v2/_vendor.py | 7 +- .../multiapicombiner/v2/aio/__init__.py | 5 +- .../multiapicombiner/v2/aio/_configuration.py | 43 +- .../v2/aio/_multiapi_service_client.py | 35 +- .../multiapicombiner/v2/aio/_vendor.py | 7 +- .../v2/aio/operations/__init__.py | 9 +- .../v2/aio/operations/_operations.py | 202 ++--- .../multiapicombiner/v2/models/__init__.py | 7 +- .../multiapicombiner/v2/models/_models_py3.py | 18 +- .../v2/operations/__init__.py | 9 +- .../v2/operations/_operations.py | 318 +++----- .../multiapicombiner/v3/__init__.py | 5 +- .../multiapicombiner/v3/_configuration.py | 43 +- .../v3/_multiapi_service_client.py | 39 +- .../multiapicombiner/v3/_vendor.py | 7 +- .../multiapicombiner/v3/aio/__init__.py | 5 +- .../multiapicombiner/v3/aio/_configuration.py | 43 +- .../v3/aio/_multiapi_service_client.py | 35 +- .../multiapicombiner/v3/aio/_vendor.py | 7 +- .../v3/aio/operations/__init__.py | 9 +- .../v3/aio/operations/_operations.py | 269 +++---- .../multiapicombiner/v3/models/__init__.py | 13 +- .../multiapicombiner/v3/models/_models_py3.py | 34 +- .../models/_multiapi_service_client_enums.py | 3 +- .../v3/operations/__init__.py | 9 +- .../v3/operations/_operations.py | 390 ++++------ 417 files changed, 9337 insertions(+), 12257 deletions(-) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py index ff9fe140cda..5d72190ec6e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "0.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py index 02297b40119..9955ea175a5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient: """Service client for multiapi client testing. @@ -37,33 +38,37 @@ class MultiapiServiceClient: :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '0.0.0' + self._client, self._config, self._serialize, self._deserialize, "0.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py index 611e61dc65b..cc3420f9112 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "0.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py index f8e04726218..99f361877eb 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient: """Service client for multiapi client testing. @@ -38,31 +39,39 @@ class MultiapiServiceClient: """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '0.0.0' + self._client, self._config, self._serialize, self._deserialize, "0.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py index 9631441cd6a..f8443c0f5fe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'OperationGroupOneOperations', + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py index ca42dc5232a..e04784a2b44 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,14 +53,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -60,19 +62,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -81,10 +83,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -95,6 +95,4 @@ async def test_two( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py index 0b2869020ef..513632adef7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/__init__.py @@ -21,10 +21,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', + "Error", + "PagingResult", + "Product", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py index 8b98be42e59..89176e0e663 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py index 9631441cd6a..f8443c0f5fe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'OperationGroupOneOperations', + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py index a4756b1a8ca..3213f256983 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v0/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,40 +29,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "0.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "0.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,14 +75,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -90,19 +84,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -111,10 +105,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -125,6 +117,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py index 1c9a5b0ff2d..4d91e0a380e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py index fc1ba252cf5..3dc74cddca7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,33 +40,37 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py index 9e3856bf5cc..093bce4e8d9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py index 0c4d5171992..7cc3d40eb7d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,31 +41,39 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py index 96dfced3f55..ade2701e6d6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,7 +11,16 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -22,19 +31,23 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_lro_and_paging_request, + build_test_lro_request, + build_test_one_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -42,12 +55,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -59,19 +67,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -83,9 +93,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -96,27 +104,24 @@ async def test_one( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -125,7 +130,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -138,12 +143,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -160,19 +163,13 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload async def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -189,11 +186,7 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -208,12 +201,9 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -228,59 +218,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) async def _test_lro_and_paging_initial( self, @@ -289,16 +265,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -315,12 +292,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -336,12 +311,10 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -365,14 +338,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -380,7 +355,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -393,22 +368,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -416,9 +394,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -428,24 +404,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -453,38 +425,27 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace_async - async def test_different_calls( - self, - greeting_in_english: str, - **kwargs: Any - ) -> None: + async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -494,19 +455,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -517,9 +480,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -530,6 +491,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py index f1aa6183dd8..7b5ed061b53 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,14 +53,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -60,19 +62,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -81,10 +83,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -95,6 +95,4 @@ async def test_two( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py index 9c722d1c22b..996681e3e2c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py index 3848a9c8fdf..576f88f1b4e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_multiapi_service_client_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -28,133 +37,95 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_test_lro_and_paging_request( - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) -def build_test_different_calls_request( - *, - greeting_in_english: str, - **kwargs: Any -) -> HttpRequest: +def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -163,10 +134,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -179,19 +147,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -202,10 +172,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,27 +184,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -245,7 +210,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -258,12 +223,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -280,19 +243,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -308,11 +265,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -326,12 +279,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -345,59 +295,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -406,16 +342,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -432,12 +369,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -453,12 +388,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -482,14 +415,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -497,7 +432,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -510,32 +445,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -545,24 +481,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -570,37 +502,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - **kwargs: Any + self, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -611,19 +534,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -633,10 +558,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -647,6 +570,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py index d1aff3a2365..92c05501686 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v1/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,40 +29,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,14 +75,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -90,19 +84,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -111,10 +105,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -125,6 +117,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py index 0c09e0d6074..8a6a6365ac6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py index 73b1e3da5ac..267df04dacc 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py index c961d19c35b..10f716d8dd6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py index 7c58e8f62b4..3a081ccc216 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,34 +41,42 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py index 1dc47f74947..fb9447b4402 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,12 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -36,12 +42,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -53,19 +54,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -77,9 +80,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -89,24 +90,16 @@ async def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def test_different_calls( - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -119,19 +112,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -143,9 +138,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -156,6 +149,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py index 4d3b3f9ceb6..78943a9d4be 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,10 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -46,16 +54,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -71,11 +72,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -89,12 +86,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -106,18 +100,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -126,7 +121,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -141,10 +136,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -154,23 +147,15 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async - async def test_three( - self, - **kwargs: Any - ) -> None: + async def test_three(self, **kwargs: Any) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -178,19 +163,19 @@ async def test_three( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -199,10 +184,8 @@ async def test_three( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -213,6 +196,4 @@ async def test_three( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py index 055d7801caa..dd3455c83c3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/aio/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,15 +53,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_four( - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -63,19 +64,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -85,10 +86,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -99,6 +98,4 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py index bdef0772a67..ed8e322c54e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/__init__.py @@ -20,9 +20,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelTwo', + "Error", + "ModelTwo", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py index 14a84ef76da..46990252065 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -41,6 +36,7 @@ def __init__( self.status = status self.message = message + class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -53,7 +49,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -62,11 +58,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py index 5c2816d8007..b094ad6f29a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,81 +30,60 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_test_different_calls_request( - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -105,12 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -122,19 +103,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -145,10 +128,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -158,24 +139,16 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -188,19 +161,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -211,10 +186,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -225,6 +198,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py index 91fb46a415a..3cd1ed59cf0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_three_request( - **kwargs: Any -) -> HttpRequest: +def build_test_three_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -131,11 +116,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -149,12 +130,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -166,18 +144,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -186,7 +165,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -201,10 +180,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -214,23 +191,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -238,19 +207,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -259,10 +228,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -273,6 +240,4 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py index d58c381f306..8194728a647 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v2/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,43 +29,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - *, - parameter_one: bool, - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -78,15 +76,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_four( # pylint: disable=inconsistent-return-statements - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -96,19 +87,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -118,10 +109,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -132,6 +121,4 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py index f7b05316ad2..633f146ec60 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py index 24f2baa1933..119f8588d79 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py index b5996cca8ef..7a4058edefe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapi/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapi/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py index adaac305718..e41a3533676 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,34 +41,42 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py index 29467d39007..97c4d332d0c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,7 +10,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -19,19 +26,21 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_paging_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -39,10 +48,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -52,17 +58,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -72,22 +80,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -95,9 +106,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -107,11 +116,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @distributed_trace_async async def test_different_calls( @@ -134,19 +139,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -159,9 +166,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -172,6 +177,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py index 88c037aeaa6..117bb536374 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_one_operations.py @@ -11,7 +11,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -20,16 +27,20 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request +from ...operations._operation_group_one_operations import ( + build_test_operation_group_paging_request, + build_test_two_request, +) if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -49,14 +60,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -66,17 +71,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -86,32 +93,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -121,11 +129,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @overload async def test_two( @@ -149,11 +153,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -167,12 +167,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -184,18 +181,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -204,7 +202,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -219,10 +217,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -232,14 +228,9 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py index 80e0bbb37d5..c33382ea2c6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/aio/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,10 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -46,16 +54,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -71,11 +72,7 @@ async def test_four( @overload async def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -90,13 +87,8 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async - async def test_four( - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any - ) -> None: + async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -107,18 +99,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -126,7 +119,7 @@ async def test_four( _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -142,10 +135,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -156,15 +147,10 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def test_five( - self, - **kwargs: Any - ) -> None: + async def test_five(self, **kwargs: Any) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -172,19 +158,19 @@ async def test_five( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -193,10 +179,8 @@ async def test_five( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -207,6 +191,4 @@ async def test_five( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py index a21d2c11b05..63672cad01d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/__init__.py @@ -26,12 +26,13 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelThree', - 'PagingResult', - 'SourcePath', - 'ContentType', + "Error", + "ModelThree", + "PagingResult", + "SourcePath", + "ContentType", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py index b0579fca699..49a4432418f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -55,12 +51,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -68,6 +59,7 @@ def __init__( super().__init__(**kwargs) self.optional_property = optional_property + class PagingResult(_serialization.Model): """PagingResult. @@ -83,11 +75,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -99,6 +87,7 @@ def __init__( self.values = values self.next_link = next_link + class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -107,19 +96,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py index 853a409c981..7179ffb9c14 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/models/_multiapi_service_client_enums.py @@ -11,8 +11,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py index 5a4b0724f9b..d4e34c679e3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_multiapi_service_client_operations.py @@ -9,7 +9,14 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,32 +32,25 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request( - **kwargs: Any -) -> HttpRequest: +def build_test_paging_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) def build_test_different_calls_request( @@ -63,34 +63,27 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -98,10 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -111,17 +101,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -131,32 +123,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -166,11 +159,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -193,19 +182,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -217,10 +208,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -231,6 +220,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py index b715f292c50..4edc0e9ea1b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_one_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,64 +32,50 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class OperationGroupOneOperations: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -102,14 +95,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -119,17 +106,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -139,32 +128,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -174,11 +164,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @overload def test_two( @@ -202,11 +188,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -220,12 +202,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -237,18 +216,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -257,7 +237,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -272,10 +252,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -285,14 +263,9 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py index 22f547ac52a..eeb9c5e3872 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/Multiapi/multiapi/v3/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_five_request( - **kwargs: Any -) -> HttpRequest: +def build_test_five_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,11 +116,7 @@ def test_four( @overload def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -150,12 +131,9 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -167,18 +145,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -186,7 +165,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -202,10 +181,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,15 +193,10 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -232,19 +204,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -253,10 +225,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -267,6 +237,4 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py index 712154c9ff9..a8be991d4c3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_configuration.py @@ -14,7 +14,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -27,35 +28,27 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) + kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py index e85f4f96d9a..324fbdc58e0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_multiapi_service_client.py @@ -21,6 +21,7 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,33 +39,37 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: AzureKeyCredential, - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py index b1341bff761..c0bfc9b848e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_configuration.py @@ -14,7 +14,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -27,35 +28,27 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) + kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py index 4036c86c1af..7afa7077e11 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_multiapi_service_client.py @@ -21,6 +21,7 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,32 +39,38 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: AzureKeyCredential, - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py index 471b1ff46fb..97a2ca1cb9f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,7 +11,16 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -22,19 +31,23 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_lro_and_paging_request, + build_test_lro_request, + build_test_one_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -42,12 +55,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -59,19 +67,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -83,9 +93,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -96,27 +104,24 @@ async def test_one( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -125,7 +130,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -138,12 +143,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -160,19 +163,13 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload async def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -189,11 +186,7 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -208,12 +201,9 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -228,59 +218,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) async def _test_lro_and_paging_initial( self, @@ -289,16 +265,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -315,12 +292,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -336,12 +311,10 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -367,14 +340,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -382,7 +357,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -395,22 +370,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -418,9 +396,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -430,24 +406,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -455,38 +427,27 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace_async - async def test_different_calls( - self, - greeting_in_english: str, - **kwargs: Any - ) -> None: + async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -496,19 +457,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -519,9 +482,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -532,6 +493,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py index acfb0727b49..ce64ec4c546 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,14 +53,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -60,19 +62,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -81,10 +83,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -95,6 +95,4 @@ async def test_two( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py index 93c4265bd8c..93d38043e62 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py index b6bcc863bcf..ec3d941ed8d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_multiapi_service_client_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -28,133 +37,95 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_test_lro_and_paging_request( - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) -def build_test_different_calls_request( - *, - greeting_in_english: str, - **kwargs: Any -) -> HttpRequest: +def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -163,10 +134,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -179,19 +147,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -202,10 +172,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,27 +184,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -245,7 +210,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -258,12 +223,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -280,19 +243,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -308,11 +265,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -326,12 +279,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -345,59 +295,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -406,16 +342,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -432,12 +369,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -453,12 +388,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -483,14 +416,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -498,7 +433,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -511,32 +446,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -546,24 +482,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -571,37 +503,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - **kwargs: Any + self, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -612,19 +535,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -634,10 +559,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -648,6 +571,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py index 999c0681731..f6fee2b20d0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v1/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,40 +29,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,14 +75,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -90,19 +84,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -111,10 +105,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -125,6 +117,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py index fa0888e64c9..0912525614d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_configuration.py @@ -14,7 +14,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -27,35 +28,27 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) + kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py index 172f5da5b90..ab171b0b766 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_multiapi_service_client.py @@ -21,6 +21,7 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py index 45bee4fee54..c6dedcb071d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_configuration.py @@ -14,7 +14,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -27,35 +28,27 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) + kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py index 26f62b9439c..4d425a8b9d6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_multiapi_service_client.py @@ -21,6 +21,7 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,35 +40,41 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py index 9c81d309ff7..3bb6b8555bc 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,12 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -36,12 +42,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -53,19 +54,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -77,9 +80,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -89,24 +90,16 @@ async def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def test_different_calls( - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -119,19 +112,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -143,9 +138,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -156,6 +149,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py index a490264d924..015e383a8fa 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,10 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -46,16 +54,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -71,11 +72,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -89,12 +86,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -106,18 +100,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -126,7 +121,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -141,10 +136,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -154,23 +147,15 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async - async def test_three( - self, - **kwargs: Any - ) -> None: + async def test_three(self, **kwargs: Any) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -178,19 +163,19 @@ async def test_three( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -199,10 +184,8 @@ async def test_three( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -213,6 +196,4 @@ async def test_three( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py index 27e67f73930..a4da2d85b0a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/aio/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,15 +53,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_four( - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -63,19 +64,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -85,10 +86,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -99,6 +98,4 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py index bdef0772a67..ed8e322c54e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/__init__.py @@ -20,9 +20,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelTwo', + "Error", + "ModelTwo", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py index 14a84ef76da..46990252065 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -41,6 +36,7 @@ def __init__( self.status = status self.message = message + class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -53,7 +49,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -62,11 +58,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py index f42959dd244..be1a9b9e82e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,81 +30,60 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_test_different_calls_request( - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -105,12 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -122,19 +103,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -145,10 +128,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -158,24 +139,16 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -188,19 +161,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -211,10 +186,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -225,6 +198,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py index 503485608ba..7f8bf5f5700 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_three_request( - **kwargs: Any -) -> HttpRequest: +def build_test_three_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -131,11 +116,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -149,12 +130,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -166,18 +144,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -186,7 +165,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -201,10 +180,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -214,23 +191,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -238,19 +207,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -259,10 +228,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -273,6 +240,4 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py index 39b3c2c79f9..ecae6014d5d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v2/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,43 +29,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - *, - parameter_one: bool, - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -78,15 +76,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_four( # pylint: disable=inconsistent-return-statements - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -96,19 +87,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -118,10 +109,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -132,6 +121,4 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py index 0ad67c0eb34..eca4a1b0a56 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_configuration.py @@ -14,7 +14,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -27,35 +28,27 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) + kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py index 596b5c66218..e27caf2ecf8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_multiapi_service_client.py @@ -21,6 +21,7 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py index 8cc199d2b6a..1f5ce3c0e76 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_configuration.py @@ -14,7 +14,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -27,35 +28,27 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: AzureKeyCredential, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - kwargs.setdefault('sdk_moniker', 'multiapicredentialdefaultpolicy/{}'.format(VERSION)) + kwargs.setdefault("sdk_moniker", "multiapicredentialdefaultpolicy/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: self.authentication_policy = policies.AzureKeyCredentialPolicy(self.credential, "api-key", **kwargs) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py index 9b52d66d726..fd43b96f213 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_multiapi_service_client.py @@ -21,6 +21,7 @@ from ._configuration import MultiapiServiceClientConfiguration from .operations import MultiapiServiceClientOperationsMixin, OperationGroupOneOperations, OperationGroupTwoOperations + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,35 +40,41 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: AzureKeyCredential, - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: AzureKeyCredential, base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py index 6a987f06fcf..c1765973ab2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,7 +10,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -19,19 +26,21 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_paging_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -39,10 +48,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -53,17 +59,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -73,22 +81,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -96,9 +107,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -108,11 +117,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @distributed_trace_async async def test_different_calls( @@ -135,19 +140,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -160,9 +167,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -173,6 +178,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py index ae6fa327c9d..b1af64630fb 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_one_operations.py @@ -11,7 +11,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -20,16 +27,20 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request +from ...operations._operation_group_one_operations import ( + build_test_operation_group_paging_request, + build_test_two_request, +) if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -49,14 +60,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -67,17 +72,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -87,32 +94,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -122,11 +130,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @overload async def test_two( @@ -150,11 +154,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -168,12 +168,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -185,18 +182,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -205,7 +203,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -220,10 +218,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -233,14 +229,9 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py index 840a3b9d2fb..f447c159587 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/aio/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,10 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -46,16 +54,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -71,11 +72,7 @@ async def test_four( @overload async def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -90,13 +87,8 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async - async def test_four( - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any - ) -> None: + async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -107,18 +99,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -126,7 +119,7 @@ async def test_four( _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -142,10 +135,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -156,15 +147,10 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def test_five( - self, - **kwargs: Any - ) -> None: + async def test_five(self, **kwargs: Any) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -172,19 +158,19 @@ async def test_five( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -193,10 +179,8 @@ async def test_five( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -207,6 +191,4 @@ async def test_five( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py index a21d2c11b05..63672cad01d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/__init__.py @@ -26,12 +26,13 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelThree', - 'PagingResult', - 'SourcePath', - 'ContentType', + "Error", + "ModelThree", + "PagingResult", + "SourcePath", + "ContentType", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py index 9c0180868ed..79a8e667ab8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -55,12 +51,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -68,6 +59,7 @@ def __init__( super().__init__(**kwargs) self.optional_property = optional_property + class PagingResult(_serialization.Model): """PagingResult. @@ -83,11 +75,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -99,6 +87,7 @@ def __init__( self.values = values self.next_link = next_link + class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -107,19 +96,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py index 853a409c981..7179ffb9c14 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/models/_multiapi_service_client_enums.py @@ -11,8 +11,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py index f47bf6113ba..7225a31c9db 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_multiapi_service_client_operations.py @@ -9,7 +9,14 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,32 +32,25 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request( - **kwargs: Any -) -> HttpRequest: +def build_test_paging_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) def build_test_different_calls_request( @@ -63,34 +63,27 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -98,10 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -111,17 +101,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -131,32 +123,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -166,11 +159,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -193,19 +182,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -217,10 +208,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -231,6 +220,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py index b7ac24273fd..a0802150ed0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_one_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,64 +32,50 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class OperationGroupOneOperations: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -102,14 +95,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -119,17 +106,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -139,32 +128,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -174,11 +164,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @overload def test_two( @@ -202,11 +188,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -220,12 +202,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -237,18 +216,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -257,7 +237,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -272,10 +252,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -285,14 +263,9 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py index c07001a7d97..e37e6579d60 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCredentialDefaultPolicy/multiapicredentialdefaultpolicy/v3/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_five_request( - **kwargs: Any -) -> HttpRequest: +def build_test_five_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,11 +116,7 @@ def test_four( @overload def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -150,12 +131,9 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -167,18 +145,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -186,7 +165,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -202,10 +181,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,15 +193,10 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -232,19 +204,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -253,10 +225,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -267,6 +237,4 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py index 7652bc9a164..ccc133e71e1 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClient', + "MultiapiCustomBaseUrlServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py index 7f0f64c8eb7..e88c1c6d7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,13 +31,8 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too- :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -46,26 +42,24 @@ def __init__( self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py index 86f432fa78d..e961cb6fe23 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_multiapi_custom_base_url_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -34,31 +35,36 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v1' - self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) - _policies = kwargs.pop('policies', None) + def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: + _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v1" + self._config = MultiapiCustomBaseUrlServiceClientConfiguration( + credential=credential, endpoint=endpoint, **kwargs + ) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest @@ -78,7 +84,7 @@ def _send_request( request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py index e1066c871a7..856dac1195a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long - ABC -): + +class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py index 7652bc9a164..ccc133e71e1 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClient', + "MultiapiCustomBaseUrlServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py index 2fdd7034dba..9915f2c2e2e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,13 +31,8 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too- :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -46,26 +42,24 @@ def __init__( self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py index 179bc8893cf..37dbb09b777 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_multiapi_custom_base_url_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -34,30 +35,37 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v1' - self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) - _policies = kwargs.pop('policies', None) + def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: + _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v1" + self._config = MultiapiCustomBaseUrlServiceClientConfiguration( + credential=credential, endpoint=endpoint, **kwargs + ) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=_endpoint, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. @@ -78,7 +86,7 @@ def _send_request( request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py index 1ab725bccca..7ad36d04654 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long - ABC -): + +class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py index 172a9058373..40541c05116 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClientOperationsMixin', + "MultiapiCustomBaseUrlServiceClientOperationsMixin", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py index 18d2c97abdf..dc3cc527caa 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/aio/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long + +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -35,11 +43,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test( - self, - id: int, - **kwargs: Any - ) -> None: + async def test(self, id: int, **kwargs: Any) -> None: """Should be a mixin operation. Put in 1 for the required parameter and have the correct api version of 1.0.0 to pass. @@ -50,19 +54,19 @@ async def test( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_request( id=id, api_version=api_version, @@ -70,15 +74,13 @@ async def test( params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -89,6 +91,4 @@ async def test( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py index f02be92e17f..187235acc3a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/__init__.py @@ -19,8 +19,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', + "Error", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py index b58173f2878..4ccf63e55b5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py index 172a9058373..40541c05116 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClientOperationsMixin', + "MultiapiCustomBaseUrlServiceClientOperationsMixin", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py index ee05f3883c3..0a5f910a9f0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v1/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,43 +29,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_request( - *, - id: int, - **kwargs: Any -) -> HttpRequest: +def build_test_request(*, id: int, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/test") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["id"] = _SERIALIZER.query("id", id, "int") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long + +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -68,11 +66,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test( # pylint: disable=inconsistent-return-statements - self, - id: int, - **kwargs: Any - ) -> None: + def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """Should be a mixin operation. Put in 1 for the required parameter and have the correct api version of 1.0.0 to pass. @@ -83,19 +77,19 @@ def test( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_request( id=id, api_version=api_version, @@ -103,15 +97,13 @@ def test( # pylint: disable=inconsistent-return-statements params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -122,6 +114,4 @@ def test( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py index 7652bc9a164..ccc133e71e1 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClient', + "MultiapiCustomBaseUrlServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py index 9019197a1ba..5f2904887f8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,13 +31,8 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too- :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -46,26 +42,24 @@ def __init__( self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py index 7aac04a3271..ed15d88585e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_multiapi_custom_base_url_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -34,31 +35,36 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v2' - self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) - _policies = kwargs.pop('policies', None) + def __init__(self, credential: "TokenCredential", endpoint: str, **kwargs: Any) -> None: + _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v2" + self._config = MultiapiCustomBaseUrlServiceClientConfiguration( + credential=credential, endpoint=endpoint, **kwargs + ) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest @@ -78,7 +84,7 @@ def _send_request( request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py index e1066c871a7..856dac1195a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long - ABC -): + +class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py index 7652bc9a164..ccc133e71e1 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClient', + "MultiapiCustomBaseUrlServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py index 686cb16bcfd..0bdd1c9cfd0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long + +class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for MultiapiCustomBaseUrlServiceClient. Note that all parameters used to create this instance are saved as instance @@ -30,13 +31,8 @@ class MultiapiCustomBaseUrlServiceClientConfiguration: # pylint: disable=too- :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") @@ -46,26 +42,24 @@ def __init__( self.credential = credential self.endpoint = endpoint self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapicustombaseurl/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapicustombaseurl/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py index ab660ded68a..98cc9e0b5e0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_multiapi_custom_base_url_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOperationsMixin): """Service client for multiapi custom base url testing. @@ -34,30 +35,37 @@ class MultiapiCustomBaseUrlServiceClient(MultiapiCustomBaseUrlServiceClientOpera :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - endpoint: str, - **kwargs: Any - ) -> None: - _endpoint = '{Endpoint}/multiapiCustomBaseUrl/v2' - self._config = MultiapiCustomBaseUrlServiceClientConfiguration(credential=credential, endpoint=endpoint, **kwargs) - _policies = kwargs.pop('policies', None) + def __init__(self, credential: "AsyncTokenCredential", endpoint: str, **kwargs: Any) -> None: + _endpoint = "{Endpoint}/multiapiCustomBaseUrl/v2" + self._config = MultiapiCustomBaseUrlServiceClientConfiguration( + credential=credential, endpoint=endpoint, **kwargs + ) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=_endpoint, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. @@ -78,7 +86,7 @@ def _send_request( request_copy = deepcopy(request) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py index 1ab725bccca..7ad36d04654 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiCustomBaseUrlServiceClientMixinABC( # pylint: disable=name-too-long - ABC -): + +class MultiapiCustomBaseUrlServiceClientMixinABC(ABC): # pylint: disable=name-too-long """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiCustomBaseUrlServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py index 172a9058373..40541c05116 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClientOperationsMixin', + "MultiapiCustomBaseUrlServiceClientOperationsMixin", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py index 95c6edf3c1c..6bbcee4be50 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/aio/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long + +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -35,11 +43,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test( - self, - id: int, - **kwargs: Any - ) -> None: + async def test(self, id: int, **kwargs: Any) -> None: """Should be a mixin operation. Put in 2 for the required parameter and have the correct api version of 2.0.0 to pass. @@ -50,19 +54,19 @@ async def test( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_request( id=id, api_version=api_version, @@ -70,15 +74,13 @@ async def test( params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -89,6 +91,4 @@ async def test( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py index f02be92e17f..187235acc3a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/__init__.py @@ -19,8 +19,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', + "Error", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py index b58173f2878..4ccf63e55b5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py index 172a9058373..40541c05116 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiCustomBaseUrlServiceClientOperationsMixin', + "MultiapiCustomBaseUrlServiceClientOperationsMixin", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py index e03323f3a05..a7610c97bcb 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiCustomBaseUrl/multiapicustombaseurl/v2/operations/_multiapi_custom_base_url_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,43 +29,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_request( - *, - id: int, - **kwargs: Any -) -> HttpRequest: +def build_test_request(*, id: int, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/test") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["id"] = _SERIALIZER.query("id", id, "int") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long + +class MultiapiCustomBaseUrlServiceClientOperationsMixin( # pylint: disable=name-too-long MultiapiCustomBaseUrlServiceClientMixinABC ): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @@ -68,11 +66,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test( # pylint: disable=inconsistent-return-statements - self, - id: int, - **kwargs: Any - ) -> None: + def test(self, id: int, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """Should be a mixin operation. Put in 2 for the required parameter and have the correct api version of 2.0.0 to pass. @@ -83,19 +77,19 @@ def test( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version("test") or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_request( id=id, api_version=api_version, @@ -103,15 +97,13 @@ def test( # pylint: disable=inconsistent-return-statements params=_params, ) path_format_arguments = { - "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True), + "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True), } _request.url = self._client.format_url(_request.url, **path_format_arguments) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -122,6 +114,4 @@ def test( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py index 19124f3ce57..09e03ece143 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py index ec55857b82e..02e0b796d19 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,33 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py index bde90131f5d..3d6432fa758 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py index ea5937d3ed3..c5457654c34 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,31 +40,38 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py index 9cf4d485236..1592e7e3079 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,7 +11,16 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.polling.async_base_polling import AsyncLROBasePolling @@ -21,19 +30,23 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_lro_and_paging_request, + build_test_lro_request, + build_test_one_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -41,12 +54,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -58,19 +66,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -82,9 +92,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -95,27 +103,24 @@ async def test_one( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -124,7 +129,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -137,12 +142,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,19 +162,13 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload async def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -188,11 +185,7 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -207,12 +200,9 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -227,59 +217,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) async def _test_lro_and_paging_initial( self, @@ -288,16 +264,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -314,12 +291,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -335,12 +310,10 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -365,14 +338,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -380,7 +355,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -393,22 +368,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -416,9 +394,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -428,24 +404,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -453,38 +425,27 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace_async - async def test_different_calls( - self, - greeting_in_english: str, - **kwargs: Any - ) -> None: + async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -494,19 +455,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -517,9 +480,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -530,6 +491,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py index c822e39b56f..35141ff86c5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -21,10 +28,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -44,14 +52,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -59,19 +61,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -80,10 +82,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -94,6 +94,4 @@ async def test_two( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py index 762767d95bf..918a39761c2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py index 6a017091706..7deb4528d01 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_multiapi_service_client_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -27,133 +36,95 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_test_lro_and_paging_request( - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) -def build_test_different_calls_request( - *, - greeting_in_english: str, - **kwargs: Any -) -> HttpRequest: +def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -162,10 +133,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -178,19 +146,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -201,10 +171,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -215,27 +183,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -244,7 +209,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -257,12 +222,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -279,19 +242,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -307,11 +264,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -325,12 +278,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -344,59 +294,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -405,16 +341,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -431,12 +368,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -452,12 +387,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -481,14 +414,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -496,7 +431,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -509,32 +444,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -544,24 +480,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -569,37 +501,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - **kwargs: Any + self, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -610,19 +533,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -632,10 +557,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -646,6 +569,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py index 6114030a753..160c63eedd4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v1/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -21,40 +28,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,14 +74,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -89,19 +83,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -110,10 +104,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -124,6 +116,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py index 2fd8b4ff762..5e60bea3540 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py index 80b50b79d59..2f9a67a9e8c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,36 +39,39 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py index d45199c2a69..8d71e3f2f72 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py index 2ccfa08b81a..4809ee86266 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,34 +40,41 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py index bc6ed439f09..4b446c5f02c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,12 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -35,12 +41,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -52,19 +53,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -76,9 +79,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -88,24 +89,16 @@ async def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def test_different_calls( - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -118,19 +111,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -142,9 +137,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -155,6 +148,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py index 6e6e6e03d66..b6a71a2e585 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,16 +53,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -70,11 +71,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -88,12 +85,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -105,18 +99,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -125,7 +120,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -140,10 +135,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -153,23 +146,15 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async - async def test_three( - self, - **kwargs: Any - ) -> None: + async def test_three(self, **kwargs: Any) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -177,19 +162,19 @@ async def test_three( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -198,10 +183,8 @@ async def test_three( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -212,6 +195,4 @@ async def test_three( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py index 3e751d5baab..ded5facbd78 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/aio/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -21,10 +28,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -44,15 +52,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_four( - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -62,19 +63,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -84,10 +85,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -98,6 +97,4 @@ async def test_four( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py index bdef0772a67..ed8e322c54e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/__init__.py @@ -20,9 +20,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelTwo', + "Error", + "ModelTwo", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py index 14a84ef76da..46990252065 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -41,6 +36,7 @@ def __init__( self.status = status self.message = message + class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -53,7 +49,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -62,11 +58,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py index 60b659d5d6b..66ca2e3753e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,81 +29,60 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_test_different_calls_request( - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -104,12 +90,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -121,19 +102,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -144,10 +127,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -157,24 +138,16 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -187,19 +160,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -210,10 +185,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -224,6 +197,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py index 69dafaeb3e6..dfed4c01227 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,70 +29,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_three_request( - **kwargs: Any -) -> HttpRequest: +def build_test_three_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -105,16 +97,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -130,11 +115,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -148,12 +129,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -165,18 +143,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -185,7 +164,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -200,10 +179,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -213,23 +190,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -237,19 +206,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -258,10 +227,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -272,6 +239,4 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py index 449683618ce..7488db14ba2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v2/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -21,43 +28,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - *, - parameter_one: bool, - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -77,15 +75,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_four( # pylint: disable=inconsistent-return-statements - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -95,19 +86,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -117,10 +108,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -131,6 +120,4 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py index 8c1ed8b2ddf..9826b12b94f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py index c57b7afeb82..7b6fc625cf4 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,36 +39,39 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py index 827d3959bf9..0b30d2372cd 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapidataplane/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapidataplane/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py index d879d925712..56e1a56df02 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,34 +40,41 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py index fbbb099e1c0..9e5c6f71af5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,7 +10,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -18,19 +25,21 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_paging_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -38,10 +47,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -51,17 +57,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -71,22 +79,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -94,9 +105,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -106,11 +115,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @distributed_trace_async async def test_different_calls( @@ -133,19 +138,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -158,9 +165,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -171,6 +176,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py index deadb8ab34d..a6ead9ba8c6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_one_operations.py @@ -11,7 +11,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -19,16 +26,20 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request +from ...operations._operation_group_one_operations import ( + build_test_operation_group_paging_request, + build_test_two_request, +) if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -48,14 +59,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -65,17 +70,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -85,32 +92,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -120,11 +128,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @overload async def test_two( @@ -148,11 +152,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -166,12 +166,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -183,18 +180,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -203,7 +201,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -218,10 +216,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -231,14 +227,9 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py index 165d04b7fbc..19e03f22b23 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/aio/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,16 +53,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -70,11 +71,7 @@ async def test_four( @overload async def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -89,13 +86,8 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async - async def test_four( - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any - ) -> None: + async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -106,18 +98,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -125,7 +118,7 @@ async def test_four( _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -141,10 +134,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -155,15 +146,10 @@ async def test_four( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def test_five( - self, - **kwargs: Any - ) -> None: + async def test_five(self, **kwargs: Any) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -171,19 +157,19 @@ async def test_five( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -192,10 +178,8 @@ async def test_five( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -206,6 +190,4 @@ async def test_five( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py index a21d2c11b05..63672cad01d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/__init__.py @@ -26,12 +26,13 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelThree', - 'PagingResult', - 'SourcePath', - 'ContentType', + "Error", + "ModelThree", + "PagingResult", + "SourcePath", + "ContentType", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py index db26518e212..0b7fe77b89e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -55,12 +51,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -68,6 +59,7 @@ def __init__( super().__init__(**kwargs) self.optional_property = optional_property + class PagingResult(_serialization.Model): """PagingResult. @@ -83,11 +75,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -99,6 +87,7 @@ def __init__( self.values = values self.next_link = next_link + class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -107,19 +96,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py index 853a409c981..7179ffb9c14 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/models/_multiapi_service_client_enums.py @@ -11,8 +11,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py index 0e01b28839e..74b0a8bfab5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_multiapi_service_client_operations.py @@ -9,7 +9,14 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -24,32 +31,25 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request( - **kwargs: Any -) -> HttpRequest: +def build_test_paging_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) def build_test_different_calls_request( @@ -62,34 +62,27 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -97,10 +90,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -110,17 +100,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -130,32 +122,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -165,11 +158,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -192,19 +181,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -216,10 +207,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -230,6 +219,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py index 3483038be34..96c3dd80870 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_one_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -24,64 +31,50 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class OperationGroupOneOperations: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -101,14 +94,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -118,17 +105,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -138,32 +127,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -173,11 +163,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @overload def test_two( @@ -201,11 +187,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -219,12 +201,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -236,18 +215,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -256,7 +236,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -271,10 +251,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -284,14 +262,9 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py index 74799e52c56..7cb38df4f2c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiDataPlane/multiapidataplane/v3/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,70 +29,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_five_request( - **kwargs: Any -) -> HttpRequest: +def build_test_five_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -105,16 +97,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -130,11 +115,7 @@ def test_four( @overload def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -149,12 +130,9 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -166,18 +144,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -185,7 +164,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -201,10 +180,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -215,15 +192,10 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -231,19 +203,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -252,10 +224,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -266,6 +236,4 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py index c33bf8ae0ad..6810552fa19 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py index 1680db68ebe..4c91d87e39f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,33 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py index 01d21e467bc..e704da2b33e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py index 4eb5c4b4501..3d1a97d3380 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,31 +40,38 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py index 8f2974137d2..1a4887a3ccf 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,7 +11,16 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.polling.async_base_polling import AsyncLROBasePolling @@ -21,19 +30,23 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_lro_and_paging_request, + build_test_lro_request, + build_test_one_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -41,13 +54,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> None: """TestOne should be in an FirstVersionOperationsMixin. :keyword id: An int parameter. Required. @@ -59,19 +66,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -83,9 +92,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -96,27 +103,24 @@ async def test_one( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -125,7 +129,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -138,12 +142,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -160,19 +162,13 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload async def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -189,11 +185,7 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -208,12 +200,9 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -228,59 +217,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) async def _test_lro_and_paging_initial( self, @@ -290,16 +265,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -316,12 +292,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -337,12 +311,10 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -368,14 +340,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -383,7 +357,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -396,22 +370,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -419,9 +396,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -431,24 +406,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -456,39 +427,27 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace_async - async def test_different_calls( - self, - *, - greeting_in_english: str, - **kwargs: Any - ) -> None: + async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) -> None: """Has added parameters across the API versions. :keyword greeting_in_english: pass in 'hello' to pass test. Required. @@ -498,19 +457,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -521,9 +482,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -534,6 +493,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py index f7bf3c18dab..41b52864a73 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -21,10 +28,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -44,14 +52,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -59,19 +61,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -80,10 +82,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -94,6 +94,4 @@ async def test_two( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py index d0e6dc50bed..57b29f30605 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py index 588668551b4..b8fe62eda38 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_multiapi_service_client_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -27,133 +36,95 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_test_lro_and_paging_request( - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) -def build_test_different_calls_request( - *, - greeting_in_english: str, - **kwargs: Any -) -> HttpRequest: +def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -162,11 +133,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -179,19 +146,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -202,10 +171,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,27 +183,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -245,7 +209,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -258,12 +222,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -280,19 +242,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -308,11 +264,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -326,12 +278,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -345,59 +294,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -407,16 +342,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -433,12 +369,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -454,12 +388,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -484,14 +416,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -499,7 +433,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -512,32 +446,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -547,24 +482,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -572,38 +503,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - *, - greeting_in_english: str, - **kwargs: Any + self, *, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -614,19 +535,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -636,10 +559,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -650,6 +571,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py index cef549b274d..1db6fa5ca6b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v1/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -21,40 +28,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,14 +74,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -89,19 +83,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -110,10 +104,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -124,6 +116,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py index 6c56eb09de7..74eee4f7f7d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py index 9c30b7b7883..30bdc5c6c06 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,36 +39,39 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py index 340e36ee459..2faf9fb852f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py index e8488e1306d..0dc415a7c7e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,34 +40,41 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py index 400c5bfdd1c..b6feefbb231 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,12 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -35,13 +41,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -53,19 +53,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -77,9 +79,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -89,25 +89,16 @@ async def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def test_different_calls( - self, - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -120,19 +111,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -144,9 +137,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -157,6 +148,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py index 138759d999d..f6f4410b013 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,16 +53,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -70,11 +71,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -88,12 +85,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -105,18 +99,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -125,7 +120,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -140,10 +135,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -153,23 +146,15 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async - async def test_three( - self, - **kwargs: Any - ) -> None: + async def test_three(self, **kwargs: Any) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -177,19 +162,19 @@ async def test_three( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -198,10 +183,8 @@ async def test_three( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -212,6 +195,4 @@ async def test_three( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py index cb666b70c10..58c8d1ca210 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/aio/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -21,10 +28,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -44,16 +52,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_four( - self, - *, - parameter_one: bool, - **kwargs: Any - ) -> None: + async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :keyword parameter_one: A boolean parameter. Required. @@ -63,19 +63,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -85,10 +85,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -99,6 +97,4 @@ async def test_four( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py index bdef0772a67..ed8e322c54e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/__init__.py @@ -20,9 +20,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelTwo', + "Error", + "ModelTwo", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py index 14a84ef76da..46990252065 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -41,6 +36,7 @@ def __init__( self.status = status self.message = message + class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -53,7 +49,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -62,11 +58,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py index b12b08dc834..a5233c33e63 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,81 +29,60 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_test_different_calls_request( - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -104,13 +90,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -122,19 +102,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -145,10 +127,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -158,25 +138,16 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -189,19 +160,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -212,10 +185,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -226,6 +197,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py index 86f64e9544f..9c4025eee8f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,70 +29,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_three_request( - **kwargs: Any -) -> HttpRequest: +def build_test_three_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -105,16 +97,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -130,11 +115,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -148,12 +129,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -165,18 +143,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -185,7 +164,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -200,10 +179,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -213,23 +190,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -237,19 +206,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -258,10 +227,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -272,6 +239,4 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py index a635aa4185c..9aa1d1c87ce 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v2/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -21,43 +28,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - *, - parameter_one: bool, - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -77,15 +75,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - *, - parameter_one: bool, - **kwargs: Any + self, *, parameter_one: bool, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -96,19 +88,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -118,10 +110,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -132,6 +122,4 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py index fb0f812979a..525afec9c0b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py index 02a60977c75..1e86a7b095e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,36 +39,39 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py index 97bbef80a2b..fa0ebc27201 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,38 +29,32 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', []) - kwargs.setdefault('sdk_moniker', 'multiapikeywordonly/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", []) + kwargs.setdefault("sdk_moniker", "multiapikeywordonly/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if not self.credential_scopes and not self.authentication_policy: raise ValueError("You must provide either credential_scopes or authentication_policy as kwargs") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py index 63f6cce4446..519ff60a57e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,34 +40,41 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py index 37956f27972..e75413f0e1e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,7 +10,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -18,19 +25,21 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_paging_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -38,10 +47,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -51,17 +57,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -71,22 +79,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -94,9 +105,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -106,11 +115,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @distributed_trace_async async def test_different_calls( @@ -134,19 +139,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -159,9 +166,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -172,6 +177,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py index 6dcfb2235fe..a8632ecebad 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_one_operations.py @@ -11,7 +11,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -19,16 +26,20 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request +from ...operations._operation_group_one_operations import ( + build_test_operation_group_paging_request, + build_test_two_request, +) if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -48,14 +59,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -65,17 +70,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -85,32 +92,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -120,11 +128,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @overload async def test_two( @@ -148,11 +152,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -166,12 +166,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -183,18 +180,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -203,7 +201,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -218,10 +216,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -231,14 +227,9 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py index e4f271fee50..d159601c24a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/aio/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,16 +53,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -70,11 +71,7 @@ async def test_four( @overload async def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -89,13 +86,8 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async - async def test_four( - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any - ) -> None: + async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -106,18 +98,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -125,7 +118,7 @@ async def test_four( _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -141,10 +134,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -155,15 +146,10 @@ async def test_four( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def test_five( - self, - **kwargs: Any - ) -> None: + async def test_five(self, **kwargs: Any) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -171,19 +157,19 @@ async def test_five( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -192,10 +178,8 @@ async def test_five( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -206,6 +190,4 @@ async def test_five( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py index a21d2c11b05..63672cad01d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/__init__.py @@ -26,12 +26,13 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelThree', - 'PagingResult', - 'SourcePath', - 'ContentType', + "Error", + "ModelThree", + "PagingResult", + "SourcePath", + "ContentType", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py index 79116ab4352..558a0c7c0ca 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -55,12 +51,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -68,6 +59,7 @@ def __init__( super().__init__(**kwargs) self.optional_property = optional_property + class PagingResult(_serialization.Model): """PagingResult. @@ -83,11 +75,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -99,6 +87,7 @@ def __init__( self.values = values self.next_link = next_link + class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -107,19 +96,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py index 853a409c981..7179ffb9c14 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/models/_multiapi_service_client_enums.py @@ -11,8 +11,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py index 90018679649..6f2a65b7f58 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_multiapi_service_client_operations.py @@ -9,7 +9,14 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -24,32 +31,25 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request( - **kwargs: Any -) -> HttpRequest: +def build_test_paging_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) def build_test_different_calls_request( @@ -62,34 +62,27 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -97,10 +90,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -110,17 +100,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -130,32 +122,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -165,11 +158,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -193,19 +182,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -217,10 +208,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -231,6 +220,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py index 62ef65c8a3f..f45c29d6e24 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_one_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -24,64 +31,50 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class OperationGroupOneOperations: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -101,14 +94,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -118,17 +105,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -138,32 +127,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -173,11 +163,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @overload def test_two( @@ -201,11 +187,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -219,12 +201,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -236,18 +215,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -256,7 +236,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -271,10 +251,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -284,14 +262,9 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py index 0baf6a1a677..65c837d0fe7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiKeywordOnly/multiapikeywordonly/v3/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,70 +29,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_five_request( - **kwargs: Any -) -> HttpRequest: +def build_test_five_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -105,16 +97,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -130,11 +115,7 @@ def test_four( @overload def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -149,12 +130,9 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -166,18 +144,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -185,7 +164,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -201,10 +180,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -215,15 +192,10 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -231,19 +203,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -252,10 +224,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -266,6 +236,4 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py index e500e64f9d9..c4224d94b4c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapinoasync/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapinoasync/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py index 0e354639f45..6c7eda4e245 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,33 +40,37 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py index c60e45da038..3379281e03e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py index 8a730caa052..f9a3405e005 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_multiapi_service_client_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -28,133 +37,95 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_test_lro_and_paging_request( - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) -def build_test_different_calls_request( - *, - greeting_in_english: str, - **kwargs: Any -) -> HttpRequest: +def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -163,10 +134,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -179,19 +147,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -202,10 +172,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,27 +184,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -245,7 +210,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -258,12 +223,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -280,19 +243,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -308,11 +265,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -326,12 +279,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -345,59 +295,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -406,16 +342,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -432,12 +369,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -453,12 +388,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -482,14 +415,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -497,7 +432,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -510,32 +445,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -545,24 +481,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -570,37 +502,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - **kwargs: Any + self, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -611,19 +534,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -633,10 +558,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -647,6 +570,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py index dcc682277c3..fa640fca422 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v1/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,40 +29,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,14 +75,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -90,19 +84,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -111,10 +105,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -125,6 +117,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py index dc493e12998..91e7712fba7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapinoasync/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapinoasync/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py index 374aa10b974..b42aa5f432b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py index bdef0772a67..ed8e322c54e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/__init__.py @@ -20,9 +20,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelTwo', + "Error", + "ModelTwo", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py index 14a84ef76da..46990252065 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -41,6 +36,7 @@ def __init__( self.status = status self.message = message + class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -53,7 +49,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -62,11 +58,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py index f733b815f65..9cd9fc89b1d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,81 +30,60 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_test_different_calls_request( - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -105,12 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -122,19 +103,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -145,10 +128,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -158,24 +139,16 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -188,19 +161,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -211,10 +186,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -225,6 +198,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py index b9bf5a48d4d..cd290cb1235 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_three_request( - **kwargs: Any -) -> HttpRequest: +def build_test_three_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -131,11 +116,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -149,12 +130,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -166,18 +144,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -186,7 +165,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -201,10 +180,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -214,23 +191,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -238,19 +207,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -259,10 +228,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -273,6 +240,4 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py index d65087d810f..240741ec538 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v2/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,43 +29,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - *, - parameter_one: bool, - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -78,15 +76,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_four( # pylint: disable=inconsistent-return-statements - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -96,19 +87,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -118,10 +109,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -132,6 +121,4 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py index feb2223ffe9..74845e1428b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapinoasync/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapinoasync/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py index 5043f2ec322..089af5a6d19 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py index a21d2c11b05..63672cad01d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/__init__.py @@ -26,12 +26,13 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelThree', - 'PagingResult', - 'SourcePath', - 'ContentType', + "Error", + "ModelThree", + "PagingResult", + "SourcePath", + "ContentType", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py index 87fb36d250e..04f6ab18536 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -55,12 +51,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -68,6 +59,7 @@ def __init__( super().__init__(**kwargs) self.optional_property = optional_property + class PagingResult(_serialization.Model): """PagingResult. @@ -83,11 +75,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -99,6 +87,7 @@ def __init__( self.values = values self.next_link = next_link + class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -107,19 +96,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py index 853a409c981..7179ffb9c14 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/models/_multiapi_service_client_enums.py @@ -11,8 +11,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py index ae49a51f0e9..b23e72f4dea 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_multiapi_service_client_operations.py @@ -9,7 +9,14 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,32 +32,25 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request( - **kwargs: Any -) -> HttpRequest: +def build_test_paging_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) def build_test_different_calls_request( @@ -63,34 +63,27 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -98,10 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -111,17 +101,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -131,32 +123,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -166,11 +159,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -193,19 +182,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -217,10 +208,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -231,6 +220,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py index 3b6f2f326a1..d230ae6ed31 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_one_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,64 +32,50 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class OperationGroupOneOperations: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -102,14 +95,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -119,17 +106,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -139,32 +128,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -174,11 +164,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @overload def test_two( @@ -202,11 +188,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -220,12 +202,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -237,18 +216,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -257,7 +237,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -272,10 +252,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -285,14 +263,9 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py index 7d5f71cbbe8..238f6b6713a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiNoAsync/multiapinoasync/v3/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_five_request( - **kwargs: Any -) -> HttpRequest: +def build_test_five_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,11 +116,7 @@ def test_four( @overload def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -150,12 +131,9 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -167,18 +145,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -186,7 +165,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -202,10 +181,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,15 +193,10 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -232,19 +204,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -253,10 +225,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -267,6 +237,4 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py index 113c6839f8b..9851b5dfa42 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,36 +29,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "0.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py index 4cb81c54e62..fda5b5367df 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient: """Service client for multiapi client testing. @@ -36,33 +37,36 @@ class MultiapiServiceClient: :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '0.0.0' + self._client, self._config, self._serialize, self._deserialize, "0.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py index c6876a549cf..74a93dfdd47 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,36 +29,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "0.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "0.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py index 76477c40daf..0010438f296 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient: """Service client for multiapi client testing. @@ -37,31 +38,38 @@ class MultiapiServiceClient: """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '0.0.0' + self._client, self._config, self._serialize, self._deserialize, "0.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py index 9631441cd6a..f8443c0f5fe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'OperationGroupOneOperations', + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py index da36d153d4c..e5a310e7549 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -21,10 +28,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -44,14 +52,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -59,19 +61,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -80,10 +82,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -94,6 +94,4 @@ async def test_two( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py index 0b2869020ef..513632adef7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/__init__.py @@ -21,10 +21,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', + "Error", + "PagingResult", + "Product", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py index 29649508ccd..9797a5a8c77 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py index 9631441cd6a..f8443c0f5fe 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/__init__.py @@ -17,8 +17,9 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'OperationGroupOneOperations', + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py index ed85ed89e58..41c1c3b31d9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v0/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -21,40 +28,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "0.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "0.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,14 +74,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -89,19 +83,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "0.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "0.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -110,10 +104,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -124,6 +116,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py index 091c7f939c4..ab8adb477f6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,36 +29,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.BearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py index 9a3cdde8c50..e26d21c5737 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -38,33 +39,36 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: PipelineClient = PipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py index e7cd00bea2d..cc1b4d0032a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_configuration.py @@ -15,7 +15,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -28,36 +29,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapisecurity/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapisecurity/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = policies.AsyncBearerTokenCredentialPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py index 4529fbd189a..7e9382d9988 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_multiapi_service_client.py @@ -22,6 +22,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,31 +40,38 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncPipelineClient = AsyncPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py index 5cebd8b5abd..3dc7f3d0999 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,7 +11,16 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.polling.async_base_polling import AsyncLROBasePolling @@ -21,19 +30,23 @@ from azure.core.utils import case_insensitive_dict from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_lro_and_paging_request, + build_test_lro_request, + build_test_one_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -41,12 +54,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -58,19 +66,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -82,9 +92,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -95,27 +103,24 @@ async def test_one( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -124,7 +129,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -137,12 +142,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,19 +162,13 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload async def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -188,11 +185,7 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -207,12 +200,9 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -227,59 +217,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) async def _test_lro_and_paging_initial( self, @@ -288,16 +264,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -314,12 +291,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -335,12 +310,10 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -365,14 +338,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -380,7 +355,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -393,22 +368,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -416,9 +394,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -428,24 +404,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -453,38 +425,27 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncLROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace_async - async def test_different_calls( - self, - greeting_in_english: str, - **kwargs: Any - ) -> None: + async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -494,19 +455,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -517,9 +480,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -530,6 +491,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py index aea40996aea..1a3584ea4a0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -21,10 +28,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -44,14 +52,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -59,19 +61,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -80,10 +82,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -94,6 +94,4 @@ async def test_two( raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py index b992d5ee62c..c9ddcd7fa74 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py index 0813100f380..4efe481b2f2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_multiapi_service_client_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -27,133 +36,95 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_test_lro_and_paging_request( - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) -def build_test_different_calls_request( - *, - greeting_in_english: str, - **kwargs: Any -) -> HttpRequest: +def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -162,10 +133,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -178,19 +146,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -201,10 +171,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -215,27 +183,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -244,7 +209,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -257,12 +222,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -279,19 +242,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -307,11 +264,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -325,12 +278,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -344,59 +294,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -405,16 +341,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -431,12 +368,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -452,12 +387,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -481,14 +414,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -496,7 +431,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -509,32 +444,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -544,24 +480,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -569,37 +501,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, LROBasePolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, LROBasePolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - **kwargs: Any + self, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -610,19 +533,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -632,10 +557,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -646,6 +569,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py index e31c7f58bac..883e25b35e0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiSecurity/multiapisecurity/v1/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -21,40 +28,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -74,14 +74,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -89,19 +83,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -110,10 +104,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -124,6 +116,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py index 5960c353a89..d55ccad1f57 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/__init__.py @@ -1 +1 @@ -__path__ = __import__('pkgutil').extend_path(__path__, __name__) # type: ignore \ No newline at end of file +__path__ = __import__("pkgutil").extend_path(__path__, __name__) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py index 8aed01eef43..3dd2e667247 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py index 64207a9ba7a..d804d78d30d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,33 +41,37 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py index fac9f3e204e..d9524f4ef39 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py index c3749bdf9c3..4bc4f474464 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,31 +42,39 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py index 39ddb101c6a..7f47fddb36c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_multiapi_service_client_operations.py @@ -11,7 +11,16 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -22,19 +31,23 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_lro_and_paging_request, build_test_lro_request, build_test_one_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_lro_and_paging_request, + build_test_lro_request, + build_test_one_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -42,12 +55,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> None: """TestOne should be in an FirstVersionOperationsMixin. :param id: An int parameter. Required. @@ -59,19 +67,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -83,9 +93,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -96,27 +104,24 @@ async def test_one( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -125,7 +130,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -138,12 +143,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -160,19 +163,13 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload async def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -189,11 +186,7 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -208,12 +201,9 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -228,59 +218,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) async def _test_lro_and_paging_initial( self, @@ -289,16 +265,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -315,12 +292,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -336,12 +311,10 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -367,14 +340,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -382,7 +357,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -395,22 +370,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -418,9 +396,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -430,24 +406,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -455,38 +427,27 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace_async - async def test_different_calls( - self, - greeting_in_english: str, - **kwargs: Any - ) -> None: + async def test_different_calls(self, greeting_in_english: str, **kwargs: Any) -> None: """Has added parameters across the API versions. :param greeting_in_english: pass in 'hello' to pass test. Required. @@ -496,19 +457,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -519,9 +482,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -532,6 +493,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py index 28e6b513ae6..f76a78464d8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/aio/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,14 +53,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -60,19 +62,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -81,10 +83,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -95,6 +95,4 @@ async def test_two( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py index a9f3e5954c4..cbc5fbd265b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py index b928092d59f..dc6be7ed447 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py index a422dcf113e..3a3cd22d345 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_multiapi_service_client_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -28,133 +37,95 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_test_lro_and_paging_request( - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) -def build_test_different_calls_request( - *, - greeting_in_english: str, - **kwargs: Any -) -> HttpRequest: +def build_test_different_calls_request(*, greeting_in_english: str, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -163,10 +134,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -179,19 +147,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -202,10 +172,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,27 +184,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -245,7 +210,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -258,12 +223,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -280,19 +243,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -308,11 +265,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -326,12 +279,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -345,59 +295,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -406,16 +342,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -432,12 +369,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -453,12 +388,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -483,14 +416,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -498,7 +433,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -511,32 +446,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -546,24 +482,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( client_request_id=client_request_id, test_lro_and_paging_options=test_lro_and_paging_options, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -571,37 +503,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - **kwargs: Any + self, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -612,19 +535,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -634,10 +559,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -648,6 +571,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py index 01e61d3cfb0..f9fd0e6235a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v1/operations/_operation_group_one_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,40 +29,33 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -75,14 +75,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -90,19 +84,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_two_request( api_version=api_version, headers=_headers, @@ -111,10 +105,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -125,6 +117,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py index 9725fcb3b82..ccf9343999a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py index 1d59320acea..d4e7e77885b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,36 +42,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py index 58f33c99bff..3ef47ecd0f2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py index 2055ea2d2f8..dba17194fa3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -42,34 +43,42 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py index 41e0dbe6cda..3dcd40687ba 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,12 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -36,12 +42,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + async def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -53,19 +54,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -77,9 +80,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -89,24 +90,16 @@ async def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def test_different_calls( - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -119,19 +112,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -143,9 +138,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -156,6 +149,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py index 763490fe36b..07f2b00ced9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,10 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -46,16 +54,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -71,11 +72,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -89,12 +86,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -106,18 +100,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -126,7 +121,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -141,10 +136,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -154,23 +147,15 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async - async def test_three( - self, - **kwargs: Any - ) -> None: + async def test_three(self, **kwargs: Any) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -178,19 +163,19 @@ async def test_three( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -199,10 +184,8 @@ async def test_three( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -213,6 +196,4 @@ async def test_three( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py index a4e5b03665a..9f90431c490 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/aio/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -22,10 +29,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -45,15 +53,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_four( - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + async def test_four(self, parameter_one: bool, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -63,19 +64,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -85,10 +86,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -99,6 +98,4 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py index bdef0772a67..ed8e322c54e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/__init__.py @@ -20,9 +20,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelTwo', + "Error", + "ModelTwo", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py index 14a84ef76da..46990252065 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -41,6 +36,7 @@ def __init__( self.status = status self.message = message + class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -53,7 +49,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -62,11 +58,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py index babbe906b47..f4643b30912 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_multiapi_service_client_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,81 +30,60 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_test_different_calls_request( - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -105,12 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one( - self, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :param id: An int parameter. Required. @@ -122,19 +103,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_test_one_request( id=id, message=message, @@ -145,10 +128,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -158,24 +139,16 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -188,19 +161,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -211,10 +186,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -225,6 +198,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py index 2ecb72bfc04..7348de6757f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_one_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_three_request( - **kwargs: Any -) -> HttpRequest: +def build_test_three_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -131,11 +116,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -149,12 +130,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -166,18 +144,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -186,7 +165,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -201,10 +180,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -214,23 +191,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -238,19 +207,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_three_request( api_version=api_version, headers=_headers, @@ -259,10 +228,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -273,6 +240,4 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py index 02afdc02e66..63bc7ccc0d0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v2/operations/_operation_group_two_operations.py @@ -8,7 +8,14 @@ import sys from typing import Any, Callable, Dict, Optional, TypeVar -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -22,43 +29,34 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - *, - parameter_one: bool, - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(*, parameter_one: bool, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -78,15 +76,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_four( # pylint: disable=inconsistent-return-statements - self, - parameter_one: bool, - **kwargs: Any - ) -> None: + def test_four(self, parameter_one: bool, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFour should be in OperationGroupTwoOperations. :param parameter_one: A boolean parameter. Required. @@ -96,19 +87,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -118,10 +109,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -132,6 +121,4 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py index 8f0d64c37df..f39362fec82 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py index 9b29d6aa0e6..b507e04b23a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -41,36 +42,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py index 303ce374204..6da45f95f65 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapiwithsubmodule/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapiwithsubmodule/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py index cf20d39419c..abaf7900ef2 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -42,34 +43,42 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py index 8bd8701687c..b1d0c672ba7 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_multiapi_service_client_operations.py @@ -10,7 +10,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -19,19 +26,21 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._multiapi_service_client_operations import build_test_different_calls_request, build_test_paging_request +from ...operations._multiapi_service_client_operations import ( + build_test_different_calls_request, + build_test_paging_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -39,10 +48,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -53,17 +59,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -73,22 +81,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -96,9 +107,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -108,11 +117,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @distributed_trace_async async def test_different_calls( @@ -135,19 +140,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -160,9 +167,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -173,6 +178,4 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py index b9c451da297..87e66f8c331 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_one_operations.py @@ -11,7 +11,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -20,16 +27,20 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operation_group_one_operations import build_test_operation_group_paging_request, build_test_two_request +from ...operations._operation_group_one_operations import ( + build_test_operation_group_paging_request, + build_test_two_request, +) if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupOneOperations: + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -49,14 +60,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -67,17 +72,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -87,32 +94,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -122,11 +130,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @overload async def test_two( @@ -150,11 +154,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -168,12 +168,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -185,18 +182,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -205,7 +203,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -220,10 +218,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -233,14 +229,9 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py index ccc32916176..05151725dd8 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/aio/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -23,10 +30,11 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -46,16 +54,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -71,11 +72,7 @@ async def test_four( @overload async def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -90,13 +87,8 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async - async def test_four( - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any - ) -> None: + async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -107,18 +99,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -126,7 +119,7 @@ async def test_four( _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -142,10 +135,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -156,15 +147,10 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def test_five( - self, - **kwargs: Any - ) -> None: + async def test_five(self, **kwargs: Any) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -172,19 +158,19 @@ async def test_five( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -193,10 +179,8 @@ async def test_five( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -207,6 +191,4 @@ async def test_five( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py index a21d2c11b05..63672cad01d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/__init__.py @@ -26,12 +26,13 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelThree', - 'PagingResult', - 'SourcePath', - 'ContentType', + "Error", + "ModelThree", + "PagingResult", + "SourcePath", + "ContentType", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py index c2a687ce8c0..a7d63282c19 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -55,12 +51,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -68,6 +59,7 @@ def __init__( super().__init__(**kwargs) self.optional_property = optional_property + class PagingResult(_serialization.Model): """PagingResult. @@ -83,11 +75,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -99,6 +87,7 @@ def __init__( self.values = values self.next_link = next_link + class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -107,19 +96,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py index 853a409c981..7179ffb9c14 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/models/_multiapi_service_client_enums.py @@ -11,8 +11,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py index be08986800c..63452e61cf6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py index 376c4d761fd..95b84481ec3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_multiapi_service_client_operations.py @@ -9,7 +9,14 @@ from typing import Any, Callable, Dict, Iterable, Optional, TypeVar import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,32 +32,25 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_paging_request( - **kwargs: Any -) -> HttpRequest: +def build_test_paging_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) def build_test_different_calls_request( @@ -63,34 +63,27 @@ def build_test_different_calls_request( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -98,10 +91,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -111,17 +101,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_paging_request( headers=_headers, params=_params, @@ -131,32 +123,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -166,11 +159,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -193,19 +182,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -217,10 +208,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -231,6 +220,4 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py index 058189a157d..fb261b27c7a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_one_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -25,64 +32,50 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_test_operation_group_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) -def build_test_two_request( - **kwargs: Any -) -> HttpRequest: +def build_test_two_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class OperationGroupOneOperations: + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -102,14 +95,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -119,17 +106,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_test_operation_group_paging_request( headers=_headers, params=_params, @@ -139,32 +128,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -174,11 +164,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @overload def test_two( @@ -202,11 +188,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -220,12 +202,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -237,18 +216,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -257,7 +237,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -272,10 +252,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -285,14 +263,9 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py index f835e9c2e6f..efedc6f0ca9 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/MultiapiWithSubmodule/multiapiwithsubmodule/submodule/v3/operations/_operation_group_two_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -23,70 +30,55 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_test_four_request( - **kwargs: Any -) -> HttpRequest: +def build_test_four_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_test_five_request( - **kwargs: Any -) -> HttpRequest: +def build_test_five_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -class OperationGroupTwoOperations: + +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -106,16 +98,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -131,11 +116,7 @@ def test_four( @overload def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -150,12 +131,9 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -167,18 +145,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -186,7 +165,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -202,10 +181,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -216,15 +193,10 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -232,19 +204,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_test_five_request( api_version=api_version, headers=_headers, @@ -253,10 +225,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -267,6 +237,4 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py index 54a3a3b8fda..5213f6b72da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py index 01d4dc77f92..f26bfbe150b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py index 54a3a3b8fda..5213f6b72da 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py index cfe7c788bc7..924dbdb2511 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py index 598f872188b..c3992e19425 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/aio/operations/_operations.py @@ -11,7 +11,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -20,19 +27,28 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_paging_request, build_operation_group_one_test_operation_group_paging_request, build_operation_group_one_test_two_request, build_operation_group_two_test_five_request, build_operation_group_two_test_four_request +from ...operations._operations import ( + build_multiapi_service_test_different_calls_request, + build_multiapi_service_test_paging_request, + build_operation_group_one_test_operation_group_paging_request, + build_operation_group_one_test_two_request, + build_operation_group_two_test_five_request, + build_operation_group_two_test_four_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] + from typing import Any, AsyncIterable, AsyncIterator, Callable, Dict, IO, Optional, TypeVar, Union, cast, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import StreamClosedError +from azure.core.exceptions import StreamConsumedError from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload @@ -69,9 +85,7 @@ def __init__(self, *args, **kwargs): ) @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -83,18 +97,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -103,7 +118,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -118,10 +133,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -131,13 +144,10 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -146,10 +156,7 @@ async def test_two( method_valid_on=['2.0.0'], ) @distributed_trace_async - async def test_three( - self, - **kwargs: Any - ) -> None: + async def test_three(self, **kwargs: Any) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -157,19 +164,19 @@ async def test_three( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -178,10 +185,8 @@ async def test_three( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -192,17 +197,14 @@ async def test_three( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -212,17 +214,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -232,32 +236,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -267,10 +272,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) + return AsyncItemPaged(get_next, extract_data) class OperationGroupTwoOperations: @@ -300,11 +302,7 @@ def __init__(self, *args, **kwargs): } ) @distributed_trace_async - async def test_four( - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any - ) -> None: + async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -315,18 +313,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -334,7 +333,7 @@ async def test_four( _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -350,10 +349,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -364,17 +361,14 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace_async - async def test_five( - self, - **kwargs: Any - ) -> None: + async def test_five(self, **kwargs: Any) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -382,19 +376,19 @@ async def test_five( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -403,10 +397,8 @@ async def test_five( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -417,7 +409,7 @@ async def test_five( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): @@ -432,16 +424,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -458,12 +451,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -479,28 +470,27 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -509,7 +499,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -522,12 +512,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -544,7 +532,7 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -554,9 +542,7 @@ async def _test_lro_initial( ) @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -571,57 +557,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -652,14 +626,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -667,7 +643,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -680,22 +656,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -703,9 +682,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -715,24 +692,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -740,29 +713,24 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -793,19 +761,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -818,9 +788,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -831,20 +799,14 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['1.0.0', '2.0.0'], ) @distributed_trace_async - async def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -856,19 +818,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -880,9 +844,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -892,13 +854,10 @@ async def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -907,10 +866,7 @@ async def test_one( method_valid_on=['3.0.0'], ) @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -920,17 +876,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -940,22 +898,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -963,9 +924,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -975,9 +934,6 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) + return AsyncItemPaged(get_next, extract_data) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py index 454cdc8be88..8f82806ad3c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_enums.py @@ -13,8 +13,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py index 560c1398930..e572bc9fc4c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/models/_models.py @@ -15,6 +15,7 @@ + class Error(_serialization.Model): """Error. @@ -29,13 +30,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -61,11 +56,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -88,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -116,13 +102,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int @@ -146,7 +126,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -155,11 +135,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. @@ -182,12 +158,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -203,19 +174,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py index f0771525fbe..f0a399d0445 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/operations/_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -26,7 +33,7 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() @@ -35,86 +42,66 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import StreamClosedError +from azure.core.exceptions import StreamConsumedError from azure.core.polling import LROPoller, NoPolling, PollingMethod from azure.mgmt.core.polling.arm_polling import ARMPolling from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload from .._validation import api_version_validation -def build_operation_group_one_test_two_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") -def build_operation_group_one_test_three_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + +def build_operation_group_one_test_three_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_operation_group_one_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any + **kwargs: Any, ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) class OperationGroupOneOperations: """ @@ -143,9 +130,7 @@ def __init__(self, *args, **kwargs): ) @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -157,18 +142,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -177,7 +163,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -192,10 +178,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -205,13 +189,10 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -220,10 +201,7 @@ def test_two( method_valid_on=['2.0.0'], ) @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -231,19 +209,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -252,10 +230,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -266,17 +242,14 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -286,17 +259,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -306,32 +281,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -341,66 +317,47 @@ def get_next(next_link=None): return pipeline_response + return ItemPaged(get_next, extract_data) - return ItemPaged( - get_next, extract_data - ) - -def build_operation_group_two_test_four_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_two_test_four_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") -def build_operation_group_two_test_five_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + +def build_operation_group_two_test_five_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) class OperationGroupTwoOperations: """ @@ -430,9 +387,7 @@ def __init__(self, *args, **kwargs): ) @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -444,18 +399,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -463,7 +419,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -479,10 +435,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -493,17 +447,14 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['3.0.0'], ) @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -511,19 +462,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -532,10 +483,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -546,148 +495,108 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore def build_multiapi_service_test_lro_and_paging_request( # pylint: disable=name-too-long - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") -def build_multiapi_service_test_lro_request( - **kwargs: Any -) -> HttpRequest: + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) + +def build_multiapi_service_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any + **kwargs: Any, ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") -def build_multiapi_service_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + +def build_multiapi_service_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") -def build_multiapi_service_test_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + +def build_multiapi_service_test_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: @@ -701,16 +610,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -727,12 +637,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -748,28 +656,27 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -778,7 +685,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -791,12 +698,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -813,7 +718,7 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -823,9 +728,7 @@ def _test_lro_initial( ) @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -839,57 +742,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -919,14 +810,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -934,7 +827,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -947,32 +840,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -982,24 +876,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -1007,29 +897,24 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) + ) @api_version_validation( @@ -1045,7 +930,7 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any + **kwargs: Any, ) -> None: """Has added parameters across the API versions. @@ -1060,19 +945,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -1084,10 +971,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1098,20 +983,14 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore @api_version_validation( method_valid_on=['1.0.0', '2.0.0'], ) @distributed_trace - def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -1123,19 +1002,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -1146,10 +1027,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1159,13 +1038,10 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore @@ -1174,10 +1050,7 @@ def test_one( method_valid_on=['3.0.0'], ) @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -1187,17 +1060,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -1207,32 +1082,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1242,9 +1118,6 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) + return ItemPaged(get_next, extract_data) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py index 91bd75b1569..79e32d87d94 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py index 0aabb0fc6b3..24b18be3e61 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,33 +40,37 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): Retry-After header is present. """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py index f63da44aaf6..f706e232a1e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "1.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "1.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py index 4db51ca16f9..b402ddec33b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,31 +41,39 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '1.0.0' + self._client, self._config, self._serialize, self._deserialize, "1.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py index 8eb92c30ca1..01939b64a84 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py index e658f770445..e915b1c289f 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/aio/operations/_operations.py @@ -11,7 +11,16 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod from azure.core.rest import AsyncHttpResponse, HttpRequest @@ -22,19 +31,24 @@ from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling from ... import models as _models -from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_lro_and_paging_request, build_multiapi_service_test_lro_request, build_multiapi_service_test_one_request, build_operation_group_one_test_two_request +from ...operations._operations import ( + build_multiapi_service_test_different_calls_request, + build_multiapi_service_test_lro_and_paging_request, + build_multiapi_service_test_lro_request, + build_multiapi_service_test_one_request, + build_operation_group_one_test_two_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -42,13 +56,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> None: """TestOne should be in an FirstVersionOperationsMixin. :keyword id: An int parameter. Required. @@ -60,19 +68,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -84,9 +94,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -97,27 +105,24 @@ async def test_one( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore async def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -126,7 +131,7 @@ async def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -139,12 +144,10 @@ async def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -161,19 +164,13 @@ async def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload async def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -190,11 +187,7 @@ async def begin_test_lro( @overload async def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -209,12 +202,9 @@ async def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -229,59 +219,45 @@ async def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) async def _test_lro_and_paging_initial( self, @@ -291,16 +267,17 @@ async def _test_lro_and_paging_initial( **kwargs: Any ) -> AsyncIterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[AsyncIterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -317,12 +294,10 @@ async def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -338,12 +313,10 @@ async def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def begin_test_lro_and_paging( self, @@ -369,14 +342,16 @@ async def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -384,7 +359,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -397,22 +372,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -420,9 +398,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -432,24 +408,20 @@ async def get_next(next_link=None): return pipeline_response - - polling: Union[bool, AsyncPollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = await self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - await raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + await raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): async def internal_get_next(next_link=None): @@ -457,39 +429,27 @@ async def internal_get_next(next_link=None): return pipeline_response return await get_next(next_link) - return AsyncItemPaged( - internal_get_next, extract_data - ) + return AsyncItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) - else: polling_method = polling + polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(AsyncPollingMethod, AsyncNoPolling()) + else: + polling_method = polling if cont_token: return AsyncLROPoller[AsyncIterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return AsyncLROPoller[AsyncIterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace_async - async def test_different_calls( - self, - *, - greeting_in_english: str, - **kwargs: Any - ) -> None: + async def test_different_calls(self, *, greeting_in_english: str, **kwargs: Any) -> None: """Has added parameters across the API versions. :keyword greeting_in_english: pass in 'hello' to pass test. Required. @@ -499,19 +459,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -522,9 +484,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -535,10 +495,10 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -558,14 +518,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_two( - self, - **kwargs: Any - ) -> None: + async def test_two(self, **kwargs: Any) -> None: """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -573,19 +527,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_one_test_two_request( api_version=api_version, headers=_headers, @@ -594,10 +548,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -608,6 +560,4 @@ async def test_two( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py index 34986ba29ae..e389a34d387 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/__init__.py @@ -22,11 +22,12 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'PagingResult', - 'Product', - 'TestLroAndPagingOptions', + "Error", + "PagingResult", + "Product", + "TestLroAndPagingOptions", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py index 2188435a8fb..d1cf00f88d6 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class PagingResult(_serialization.Model): """PagingResult. @@ -59,11 +55,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.Product"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.Product"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -75,6 +67,7 @@ def __init__( self.values = values self.next_link = next_link + class Product(_serialization.Model): """Product. @@ -86,12 +79,7 @@ class Product(_serialization.Model): "id": {"key": "id", "type": "int"}, } - def __init__( - self, - *, - id: Optional[int] = None, # pylint: disable=redefined-builtin - **kwargs: Any - ) -> None: + def __init__(self, *, id: Optional[int] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: :paramtype id: int @@ -99,6 +87,7 @@ def __init__( super().__init__(**kwargs) self.id = id + class TestLroAndPagingOptions(_serialization.Model): """Parameter group. @@ -114,13 +103,7 @@ class TestLroAndPagingOptions(_serialization.Model): "timeout": {"key": "timeout", "type": "int"}, } - def __init__( - self, - *, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any - ) -> None: + def __init__(self, *, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any) -> None: """ :keyword maxresults: Sets the maximum number of items to return in the response. :paramtype maxresults: int diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py index 8eb92c30ca1..01939b64a84 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/__init__.py @@ -18,9 +18,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py index e4dd39c9826..a7e0aeec963 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v1/operations/_operations.py @@ -10,7 +10,16 @@ from typing import Any, Callable, Dict, IO, Iterable, Iterator, Optional, TypeVar, Union, cast, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, StreamClosedError, StreamConsumedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + StreamClosedError, + StreamConsumedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.polling import LROPoller, NoPolling, PollingMethod @@ -28,160 +37,116 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_multiapi_service_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_multiapi_service_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_multiapi_service_test_lro_request( - **kwargs: Any -) -> HttpRequest: +def build_multiapi_service_test_lro_request(**kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lro") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, headers=_headers, **kwargs) def build_multiapi_service_test_lro_and_paging_request( # pylint: disable=name-too-long - *, - client_request_id: Optional[str] = None, - maxresults: Optional[int] = None, - timeout: int = 30, - **kwargs: Any + *, client_request_id: Optional[str] = None, maxresults: Optional[int] = None, timeout: int = 30, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/lroAndPaging") # Construct headers if client_request_id is not None: - _headers['client-request-id'] = _SERIALIZER.header("client_request_id", client_request_id, 'str') + _headers["client-request-id"] = _SERIALIZER.header("client_request_id", client_request_id, "str") if maxresults is not None: - _headers['maxresults'] = _SERIALIZER.header("maxresults", maxresults, 'int') + _headers["maxresults"] = _SERIALIZER.header("maxresults", maxresults, "int") if timeout is not None: - _headers['timeout'] = _SERIALIZER.header("timeout", timeout, 'int') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["timeout"] = _SERIALIZER.header("timeout", timeout, "int") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long - *, - greeting_in_english: str, - **kwargs: Any + *, greeting_in_english: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_operation_group_one_test_two_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "1.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "1.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -190,11 +155,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument @distributed_trace def test_one( # pylint: disable=inconsistent-return-statements - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any ) -> None: """TestOne should be in an FirstVersionOperationsMixin. @@ -207,19 +168,21 @@ def test_one( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -230,10 +193,8 @@ def test_one( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -244,27 +205,24 @@ def test_one( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore def _test_lro_initial( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -273,7 +231,7 @@ def _test_lro_initial( _content = product else: if product is not None: - _json = self._serialize.body(product, 'Product') + _json = self._serialize.body(product, "Product") else: _json = None @@ -286,12 +244,10 @@ def _test_lro_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -308,19 +264,13 @@ def _test_lro_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @overload def begin_test_lro( - self, - product: Optional[_models.Product] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[_models.Product] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -336,11 +286,7 @@ def begin_test_lro( @overload def begin_test_lro( - self, - product: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, product: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -354,12 +300,9 @@ def begin_test_lro( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def begin_test_lro( - self, - product: Optional[Union[_models.Product, IO[bytes]]] = None, - **kwargs: Any + self, product: Optional[Union[_models.Product, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.Product]: """Put in whatever shape of Product you want, will return a Product with id equal to 100. @@ -373,59 +316,45 @@ def begin_test_lro( _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = kwargs.pop("params", {}) or {} - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.Product] = kwargs.pop( - 'cls', None - ) - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.Product] = kwargs.pop("cls", None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_initial( product=product, content_type=content_type, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): - deserialized = self._deserialize( - 'Product', - pipeline_response.http_response - ) + deserialized = self._deserialize("Product", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized - if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[_models.Product].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[_models.Product]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) def _test_lro_and_paging_initial( self, @@ -435,16 +364,17 @@ def _test_lro_and_paging_initial( **kwargs: Any ) -> Iterator[bytes]: error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[Iterator[bytes]] = kwargs.pop( - 'cls', None - ) + cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) _maxresults = None _timeout = None @@ -461,12 +391,10 @@ def _test_lro_and_paging_initial( ) _request.url = self._client.format_url(_request.url) - _decompress = kwargs.pop('decompress', True) + _decompress = kwargs.pop("decompress", True) _stream = True - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -482,12 +410,10 @@ def _test_lro_and_paging_initial( deserialized = response.stream_download(self._client._pipeline, decompress=_decompress) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def begin_test_lro_and_paging( self, @@ -512,14 +438,16 @@ def begin_test_lro_and_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: _maxresults = None @@ -527,7 +455,7 @@ def prepare_request(next_link=None): if test_lro_and_paging_options is not None: _maxresults = test_lro_and_paging_options.maxresults _timeout = test_lro_and_paging_options.timeout - + _request = build_multiapi_service_test_lro_and_paging_request( client_request_id=client_request_id, maxresults=_maxresults, @@ -540,32 +468,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response.http_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response.http_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -575,24 +504,20 @@ def get_next(next_link=None): return pipeline_response - - polling: Union[bool, PollingMethod] = kwargs.pop('polling', True) - lro_delay = kwargs.pop( - 'polling_interval', - self._config.polling_interval - ) - cont_token: Optional[str] = kwargs.pop('continuation_token', None) + polling: Union[bool, PollingMethod] = kwargs.pop("polling", True) + lro_delay = kwargs.pop("polling_interval", self._config.polling_interval) + cont_token: Optional[str] = kwargs.pop("continuation_token", None) if cont_token is None: raw_result = self._test_lro_and_paging_initial( test_lro_and_paging_options=test_lro_and_paging_options, client_request_id=client_request_id, - cls=lambda x,y,z: x, + cls=lambda x, y, z: x, headers=_headers, params=_params, **kwargs ) - raw_result.http_response.read() # type: ignore - kwargs.pop('error_map', None) + raw_result.http_response.read() # type: ignore + kwargs.pop("error_map", None) def get_long_running_output(pipeline_response): def internal_get_next(next_link=None): @@ -600,38 +525,28 @@ def internal_get_next(next_link=None): return pipeline_response return get_next(next_link) - return ItemPaged( - internal_get_next, extract_data - ) + return ItemPaged(internal_get_next, extract_data) if polling is True: - polling_method: PollingMethod = cast(PollingMethod, ARMPolling( - lro_delay, - - - **kwargs - )) - elif polling is False: polling_method = cast(PollingMethod, NoPolling()) - else: polling_method = polling + polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) + elif polling is False: + polling_method = cast(PollingMethod, NoPolling()) + else: + polling_method = polling if cont_token: return LROPoller[Iterable["_models.Product"]].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, - deserialization_callback=get_long_running_output + deserialization_callback=get_long_running_output, ) return LROPoller[Iterable["_models.Product"]]( self._client, raw_result, get_long_running_output, polling_method # type: ignore - ) - - + ) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - *, - greeting_in_english: str, - **kwargs: Any + self, *, greeting_in_english: str, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -642,19 +557,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "1.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, api_version=api_version, @@ -664,10 +581,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -678,10 +593,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -701,14 +616,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_two( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_two(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestTwo should be in OperationGroupOneOperations. :return: None or the result of cls(response) @@ -716,19 +625,19 @@ def test_two( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "1.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "1.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_one_test_two_request( api_version=api_version, headers=_headers, @@ -737,10 +646,8 @@ def test_two( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -751,6 +658,4 @@ def test_two( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py index abaaaa33886..f9444070b03 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py index 5deb24da5c6..5cef307ba5e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py index 926aea303a9..2156d35f402 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "2.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "2.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py index d4cfa27e1e5..258353eff7a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,34 +41,42 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '2.0.0' + self._client, self._config, self._serialize, self._deserialize, "2.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py index 6aaa304b8a7..33560f2557a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py index 6ad79f6b9d2..88972f2a779 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/aio/operations/_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator_async import distributed_trace_async @@ -17,19 +24,24 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_one_request, build_operation_group_one_test_three_request, build_operation_group_one_test_two_request, build_operation_group_two_test_four_request +from ...operations._operations import ( + build_multiapi_service_test_different_calls_request, + build_multiapi_service_test_one_request, + build_operation_group_one_test_three_request, + build_operation_group_one_test_two_request, + build_operation_group_two_test_four_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -37,13 +49,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace_async - async def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + async def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -55,19 +61,21 @@ async def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -79,9 +87,7 @@ async def test_one( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -91,25 +97,16 @@ async def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async async def test_different_calls( - self, - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -122,19 +119,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -146,9 +145,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +156,10 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -182,16 +179,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -207,11 +197,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -225,12 +211,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -242,18 +225,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -262,7 +246,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -277,10 +261,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -290,23 +272,15 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace_async - async def test_three( - self, - **kwargs: Any - ) -> None: + async def test_three(self, **kwargs: Any) -> None: """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -314,19 +288,19 @@ async def test_three( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -335,10 +309,8 @@ async def test_three( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -349,10 +321,10 @@ async def test_three( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -372,16 +344,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace_async - async def test_four( - self, - *, - parameter_one: bool, - **kwargs: Any - ) -> None: + async def test_four(self, *, parameter_one: bool, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :keyword parameter_one: A boolean parameter. Required. @@ -391,19 +355,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_two_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -413,10 +377,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -427,6 +389,4 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py index bdef0772a67..ed8e322c54e 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/__init__.py @@ -20,9 +20,10 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelTwo', + "Error", + "ModelTwo", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py index 14a84ef76da..46990252065 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/models/_models_py3.py @@ -10,6 +10,7 @@ from ... import _serialization + class Error(_serialization.Model): """Error. @@ -24,13 +25,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -41,6 +36,7 @@ def __init__( self.status = status self.message = message + class ModelTwo(_serialization.Model): """Only exists in api version 2.0.0. @@ -53,7 +49,7 @@ class ModelTwo(_serialization.Model): """ _validation = { - 'id': {'required': True}, + "id": {"required": True}, } _attribute_map = { @@ -62,11 +58,7 @@ class ModelTwo(_serialization.Model): } def __init__( - self, - *, - id: int, # pylint: disable=redefined-builtin - message: Optional[str] = None, - **kwargs: Any + self, *, id: int, message: Optional[str] = None, **kwargs: Any # pylint: disable=redefined-builtin ) -> None: """ :keyword id: Required. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py index 6aaa304b8a7..33560f2557a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py index afc59e1e43f..b15213994b5 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v2/operations/_operations.py @@ -9,7 +9,14 @@ import sys from typing import Any, Callable, Dict, IO, Optional, TypeVar, Union, overload -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse from azure.core.tracing.decorator import distributed_trace @@ -24,168 +31,123 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_multiapi_service_test_one_request( - *, - id: int, - message: Optional[str] = None, - **kwargs: Any -) -> HttpRequest: +def build_multiapi_service_test_one_request(*, id: int, message: Optional[str] = None, **kwargs: Any) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testOneEndpoint") # Construct parameters - _params['id'] = _SERIALIZER.query("id", id, 'int') + _params["id"] = _SERIALIZER.query("id", id, "int") if message is not None: - _params['message'] = _SERIALIZER.query("message", message, 'str') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["message"] = _SERIALIZER.query("message", message, "str") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_operation_group_one_test_two_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_operation_group_one_test_three_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_one_test_three_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testThreeEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) def build_operation_group_two_test_four_request( # pylint: disable=name-too-long - *, - parameter_one: bool, - **kwargs: Any + *, parameter_one: bool, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "2.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "2.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['parameterOne'] = _SERIALIZER.query("parameter_one", parameter_one, 'bool') - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["parameterOne"] = _SERIALIZER.query("parameter_one", parameter_one, "bool") + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -193,13 +155,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_one( - self, - *, - id: int, - message: Optional[str] = None, - **kwargs: Any - ) -> _models.ModelTwo: + def test_one(self, *, id: int, message: Optional[str] = None, **kwargs: Any) -> _models.ModelTwo: """TestOne should be in an SecondVersionOperationsMixin. Returns ModelTwo. :keyword id: An int parameter. Required. @@ -211,19 +167,21 @@ def test_one( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_one') or "2.0.0")) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_one") or "2.0.0") ) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_one_request( id=id, message=message, @@ -234,10 +192,8 @@ def test_one( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -247,25 +203,16 @@ def test_one( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements - self, - *, - greeting_in_english: str, - greeting_in_chinese: Optional[str] = None, - **kwargs: Any + self, *, greeting_in_english: str, greeting_in_chinese: Optional[str] = None, **kwargs: Any ) -> None: """Has added parameters across the API versions. @@ -278,19 +225,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "2.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -301,10 +250,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -315,10 +262,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -338,16 +285,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_two( - self, - parameter_one: Optional[_models.ModelTwo] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[_models.ModelTwo] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -363,11 +303,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -381,12 +317,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelTwo, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelTwo: """TestTwo should be in OperationGroupOneOperations. Takes in ModelTwo and ouputs ModelTwo. @@ -398,18 +331,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelTwo] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelTwo] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -418,7 +352,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelTwo') + _json = self._serialize.body(parameter_one, "ModelTwo") else: _json = None @@ -433,10 +367,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -446,23 +378,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelTwo', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelTwo", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore - - @distributed_trace - def test_three( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_three(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestThree should be in OperationGroupOneOperations. Takes in ModelTwo. :return: None or the result of cls(response) @@ -470,19 +394,19 @@ def test_three( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_one_test_three_request( api_version=api_version, headers=_headers, @@ -491,10 +415,8 @@ def test_three( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -505,10 +427,10 @@ def test_three( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -528,15 +450,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - *, - parameter_one: bool, - **kwargs: Any + self, *, parameter_one: bool, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -547,19 +463,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "2.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_two_test_four_request( parameter_one=parameter_one, api_version=api_version, @@ -569,10 +485,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -583,6 +497,4 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py index 01d4dc77f92..f26bfbe150b 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "TokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = ARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py index 7650ad6847b..ad63699a83c 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials import TokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -39,36 +40,40 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): :paramtype api_version: str """ - def __init__( - self, - credential: "TokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any - ) -> None: + def __init__(self, credential: "TokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),ARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + ARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any - ) -> HttpResponse: + def _send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse: """Runs the network request through the client's chained policies. >>> from azure.core.rest import HttpRequest diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py index 614290bca5a..70517e0b7af 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/_vendor.py @@ -15,12 +15,11 @@ from .._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "PipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py index 8f5084e3ef5..30c0d2f93d3 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/__init__.py @@ -20,9 +20,10 @@ except ImportError: _patch_all = [] from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClient', + "MultiapiServiceClient", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py index cfe7c788bc7..924dbdb2511 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_configuration.py @@ -16,7 +16,8 @@ VERSION = "unknown" -class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes + +class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance-attributes """Configuration for MultiapiServiceClient. Note that all parameters used to create this instance are saved as instance @@ -29,36 +30,30 @@ class MultiapiServiceClientConfiguration: # pylint: disable=too-many-instance :paramtype api_version: str """ - def __init__( - self, - credential: "AsyncTokenCredential", - **kwargs: Any - ) -> None: - api_version: str = kwargs.pop('api_version', "3.0.0") + def __init__(self, credential: "AsyncTokenCredential", **kwargs: Any) -> None: + api_version: str = kwargs.pop("api_version", "3.0.0") if credential is None: raise ValueError("Parameter 'credential' must not be None.") self.credential = credential self.api_version = api_version - self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) - kwargs.setdefault('sdk_moniker', 'multiapicombiner/{}'.format(VERSION)) + self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) + kwargs.setdefault("sdk_moniker", "multiapicombiner/{}".format(VERSION)) self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) - - def _configure( - self, - **kwargs: Any - ) -> None: - self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs) - self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs) - self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs) - self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs) - self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs) - self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs) - self.redirect_policy = kwargs.get('redirect_policy') or policies.AsyncRedirectPolicy(**kwargs) - self.retry_policy = kwargs.get('retry_policy') or policies.AsyncRetryPolicy(**kwargs) - self.authentication_policy = kwargs.get('authentication_policy') + def _configure(self, **kwargs: Any) -> None: + self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs) + self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs) + self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs) + self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs) + self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs) + self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs) + self.redirect_policy = kwargs.get("redirect_policy") or policies.AsyncRedirectPolicy(**kwargs) + self.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**kwargs) + self.authentication_policy = kwargs.get("authentication_policy") if self.credential and not self.authentication_policy: - self.authentication_policy = AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs) + self.authentication_policy = AsyncARMChallengeAuthenticationPolicy( + self.credential, *self.credential_scopes, **kwargs + ) diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py index d7f9a304bc0..f840ea392b0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_multiapi_service_client.py @@ -23,6 +23,7 @@ if TYPE_CHECKING: from azure.core.credentials_async import AsyncTokenCredential + class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """Service client for multiapi client testing. @@ -40,34 +41,42 @@ class MultiapiServiceClient(MultiapiServiceClientOperationsMixin): """ def __init__( - self, - credential: "AsyncTokenCredential", - base_url: str = "http://localhost:3000", - **kwargs: Any + self, credential: "AsyncTokenCredential", base_url: str = "http://localhost:3000", **kwargs: Any ) -> None: self._config = MultiapiServiceClientConfiguration(credential=credential, **kwargs) - _policies = kwargs.pop('policies', None) + _policies = kwargs.pop("policies", None) if _policies is None: - _policies = [policies.RequestIdPolicy(**kwargs),self._config.headers_policy,self._config.user_agent_policy,self._config.proxy_policy,policies.ContentDecodePolicy(**kwargs),AsyncARMAutoResourceProviderRegistrationPolicy(),self._config.redirect_policy,self._config.retry_policy,self._config.authentication_policy,self._config.custom_hook_policy,self._config.logging_policy,policies.DistributedTracingPolicy(**kwargs),policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,self._config.http_logging_policy] + _policies = [ + policies.RequestIdPolicy(**kwargs), + self._config.headers_policy, + self._config.user_agent_policy, + self._config.proxy_policy, + policies.ContentDecodePolicy(**kwargs), + AsyncARMAutoResourceProviderRegistrationPolicy(), + self._config.redirect_policy, + self._config.retry_policy, + self._config.authentication_policy, + self._config.custom_hook_policy, + self._config.logging_policy, + policies.DistributedTracingPolicy(**kwargs), + policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None, + self._config.http_logging_policy, + ] self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, policies=_policies, **kwargs) - client_models = {k: v for k, v in _models.__dict__.items() if isinstance(v, type)} self._serialize = Serializer(client_models) self._deserialize = Deserializer(client_models) self._serialize.client_side_validation = False self.operation_group_one = OperationGroupOneOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) self.operation_group_two = OperationGroupTwoOperations( - self._client, self._config, self._serialize, self._deserialize, '3.0.0' + self._client, self._config, self._serialize, self._deserialize, "3.0.0" ) - def _send_request( - self, - request: HttpRequest, *, stream: bool = False, - **kwargs: Any + self, request: HttpRequest, *, stream: bool = False, **kwargs: Any ) -> Awaitable[AsyncHttpResponse]: """Runs the network request through the client's chained policies. diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py index d0b3400218b..9c6fd05bb21 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/_vendor.py @@ -15,12 +15,11 @@ from ..._serialization import Deserializer, Serializer -class MultiapiServiceClientMixinABC( - ABC -): + +class MultiapiServiceClientMixinABC(ABC): """DO NOT use this class. It is for internal typing use only.""" + _client: "AsyncPipelineClient" _config: MultiapiServiceClientConfiguration _serialize: "Serializer" _deserialize: "Deserializer" - diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py index 6aaa304b8a7..33560f2557a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py index 98c53aed427..c33a78c1d80 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/aio/operations/_operations.py @@ -11,7 +11,14 @@ import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.pipeline import PipelineResponse from azure.core.rest import AsyncHttpResponse, HttpRequest from azure.core.tracing.decorator import distributed_trace @@ -20,19 +27,25 @@ from azure.mgmt.core.exceptions import ARMErrorFormat from ... import models as _models -from ...operations._operations import build_multiapi_service_test_different_calls_request, build_multiapi_service_test_paging_request, build_operation_group_one_test_operation_group_paging_request, build_operation_group_one_test_two_request, build_operation_group_two_test_five_request, build_operation_group_two_test_four_request +from ...operations._operations import ( + build_multiapi_service_test_different_calls_request, + build_multiapi_service_test_paging_request, + build_operation_group_one_test_operation_group_paging_request, + build_operation_group_one_test_two_request, + build_operation_group_two_test_five_request, + build_operation_group_two_test_four_request, +) from .._vendor import MultiapiServiceClientMixinABC if sys.version_info >= (3, 9): from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -40,10 +53,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -53,17 +63,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -73,22 +85,25 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): @@ -96,9 +111,7 @@ async def get_next(next_link=None): _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -108,11 +121,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @distributed_trace_async async def test_different_calls( @@ -136,19 +145,21 @@ async def test_different_calls( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -161,9 +172,7 @@ async def test_different_calls( _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # type: ignore # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -174,10 +183,10 @@ async def test_different_calls( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -197,14 +206,8 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> AsyncIterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> AsyncIterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -214,17 +217,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -234,32 +239,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request async def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -269,11 +275,7 @@ async def get_next(next_link=None): return pipeline_response - - return AsyncItemPaged( - get_next, extract_data - ) - + return AsyncItemPaged(get_next, extract_data) @overload async def test_two( @@ -297,11 +299,7 @@ async def test_two( @overload async def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -315,12 +313,9 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async async def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -332,18 +327,19 @@ async def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -352,7 +348,7 @@ async def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -367,10 +363,8 @@ async def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -380,18 +374,15 @@ async def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -411,16 +402,9 @@ def __init__(self, *args, **kwargs) -> None: self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload async def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -436,11 +420,7 @@ async def test_four( @overload async def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -455,13 +435,8 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace_async - async def test_four( - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any - ) -> None: + async def test_four(self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any) -> None: """TestFour should be in OperationGroupTwoOperations. :param input: Input parameter. Is either a SourcePath type or a IO[bytes] type. Default value @@ -472,18 +447,19 @@ async def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -491,7 +467,7 @@ async def test_four( _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -507,10 +483,8 @@ async def test_four( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -521,15 +495,10 @@ async def test_four( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def test_five( - self, - **kwargs: Any - ) -> None: + async def test_five(self, **kwargs: Any) -> None: """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -537,19 +506,19 @@ async def test_five( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -558,10 +527,8 @@ async def test_five( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -572,6 +539,4 @@ async def test_five( raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py index a21d2c11b05..63672cad01d 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/__init__.py @@ -26,12 +26,13 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'Error', - 'ModelThree', - 'PagingResult', - 'SourcePath', - 'ContentType', + "Error", + "ModelThree", + "PagingResult", + "SourcePath", + "ContentType", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py index 80fc2d5d141..9305829f310 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_models_py3.py @@ -13,6 +13,7 @@ if TYPE_CHECKING: from .. import models as _models + class Error(_serialization.Model): """Error. @@ -27,13 +28,7 @@ class Error(_serialization.Model): "message": {"key": "message", "type": "str"}, } - def __init__( - self, - *, - status: Optional[int] = None, - message: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, status: Optional[int] = None, message: Optional[str] = None, **kwargs: Any) -> None: """ :keyword status: :paramtype status: int @@ -44,6 +39,7 @@ def __init__( self.status = status self.message = message + class ModelThree(_serialization.Model): """Only exists in api version 3.0.0. @@ -55,12 +51,7 @@ class ModelThree(_serialization.Model): "optional_property": {"key": "optionalProperty", "type": "str"}, } - def __init__( - self, - *, - optional_property: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, optional_property: Optional[str] = None, **kwargs: Any) -> None: """ :keyword optional_property: :paramtype optional_property: str @@ -68,6 +59,7 @@ def __init__( super().__init__(**kwargs) self.optional_property = optional_property + class PagingResult(_serialization.Model): """PagingResult. @@ -83,11 +75,7 @@ class PagingResult(_serialization.Model): } def __init__( - self, - *, - values: Optional[List["_models.ModelThree"]] = None, - next_link: Optional[str] = None, - **kwargs: Any + self, *, values: Optional[List["_models.ModelThree"]] = None, next_link: Optional[str] = None, **kwargs: Any ) -> None: """ :keyword values: @@ -99,6 +87,7 @@ def __init__( self.values = values self.next_link = next_link + class SourcePath(_serialization.Model): """Uri or local path to source data. @@ -107,19 +96,14 @@ class SourcePath(_serialization.Model): """ _validation = { - 'source': {'max_length': 2048}, + "source": {"max_length": 2048}, } _attribute_map = { "source": {"key": "source", "type": "str"}, } - def __init__( - self, - *, - source: Optional[str] = None, - **kwargs: Any - ) -> None: + def __init__(self, *, source: Optional[str] = None, **kwargs: Any) -> None: """ :keyword source: File source path. :paramtype source: str diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py index 853a409c981..7179ffb9c14 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/models/_multiapi_service_client_enums.py @@ -11,8 +11,7 @@ class ContentType(str, Enum, metaclass=CaseInsensitiveEnumMeta): - """Content type for upload. - """ + """Content type for upload.""" APPLICATION_PDF = "application/pdf" """Content Type 'application/pdf'""" diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py index 6aaa304b8a7..33560f2557a 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/__init__.py @@ -19,10 +19,11 @@ from ._patch import __all__ as _patch_all from ._patch import * from ._patch import patch_sdk as _patch_sdk + __all__ = [ - 'MultiapiServiceClientOperationsMixin', - 'OperationGroupOneOperations', - 'OperationGroupTwoOperations', + "MultiapiServiceClientOperationsMixin", + "OperationGroupOneOperations", + "OperationGroupTwoOperations", ] -__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore +__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore _patch_sdk() diff --git a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py index 45d5174de3a..624e1d0b4b0 100644 --- a/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py +++ b/packages/autorest.python/test/multiapi/Expected/AcceptanceTests/multiapicombiner/multiapicombiner/v3/operations/_operations.py @@ -10,7 +10,14 @@ from typing import Any, Callable, Dict, IO, Iterable, Optional, TypeVar, Union, overload import urllib.parse -from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, ResourceNotModifiedError, map_error +from azure.core.exceptions import ( + ClientAuthenticationError, + HttpResponseError, + ResourceExistsError, + ResourceNotFoundError, + ResourceNotModifiedError, + map_error, +) from azure.core.paging import ItemPaged from azure.core.pipeline import PipelineResponse from azure.core.rest import HttpRequest, HttpResponse @@ -26,32 +33,25 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore -T = TypeVar('T') +T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] _SERIALIZER = Serializer() _SERIALIZER.client_side_validation = False -def build_multiapi_service_test_paging_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_multiapi_service_test_paging_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) def build_multiapi_service_test_different_calls_request( # pylint: disable=name-too-long @@ -59,147 +59,111 @@ def build_multiapi_service_test_different_calls_request( # pylint: disable=name greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any + **kwargs: Any, ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/testDifferentCalls") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['greetingInEnglish'] = _SERIALIZER.header("greeting_in_english", greeting_in_english, 'str') + _headers["greetingInEnglish"] = _SERIALIZER.header("greeting_in_english", greeting_in_english, "str") if greeting_in_chinese is not None: - _headers['greetingInChinese'] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, 'str') + _headers["greetingInChinese"] = _SERIALIZER.header("greeting_in_chinese", greeting_in_chinese, "str") if greeting_in_french is not None: - _headers['greetingInFrench'] = _SERIALIZER.header("greeting_in_french", greeting_in_french, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["greetingInFrench"] = _SERIALIZER.header("greeting_in_french", greeting_in_french, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) def build_operation_group_one_test_operation_group_paging_request( # pylint: disable=name-too-long - **kwargs: Any + **kwargs: Any, ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) - accept = _headers.pop('Accept', "application/json") + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/paging/1") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs) -def build_operation_group_one_test_two_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_one_test_two_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/one/testTwoEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="GET", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_operation_group_two_test_four_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_two_test_four_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFourEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers if content_type is not None: - _headers['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str') - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') + _headers["Content-Type"] = _SERIALIZER.header("content_type", content_type, "str") + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") - return HttpRequest( - method="POST", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) + return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_operation_group_two_test_five_request( # pylint: disable=name-too-long - **kwargs: Any -) -> HttpRequest: +def build_operation_group_two_test_five_request(**kwargs: Any) -> HttpRequest: # pylint: disable=name-too-long _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', "3.0.0")) - accept = _headers.pop('Accept', "application/json") + api_version: str = kwargs.pop("api_version", _params.pop("api-version", "3.0.0")) + accept = _headers.pop("Accept", "application/json") # Construct URL _url = kwargs.pop("template_url", "/multiapi/two/testFiveEndpoint") # Construct parameters - _params['api-version'] = _SERIALIZER.query("api_version", api_version, 'str') + _params["api-version"] = _SERIALIZER.query("api_version", api_version, "str") # Construct headers - _headers['Accept'] = _SERIALIZER.header("accept", accept, 'str') - - return HttpRequest( - method="PUT", - url=_url, - params=_params, - headers=_headers, - **kwargs - ) - -class MultiapiServiceClientOperationsMixin( - MultiapiServiceClientMixinABC -): + _headers["Accept"] = _SERIALIZER.header("accept", accept, "str") + + return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) + + +class MultiapiServiceClientOperationsMixin(MultiapiServiceClientMixinABC): def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument try: return self._config.api_version @@ -207,10 +171,7 @@ def _api_version(self, op_name: str) -> str: # pylint: disable=unused-argument return "" @distributed_trace - def test_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -220,17 +181,19 @@ def test_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_multiapi_service_test_paging_request( headers=_headers, params=_params, @@ -240,32 +203,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._config.api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -275,11 +239,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @distributed_trace def test_different_calls( # pylint: disable=inconsistent-return-statements @@ -288,7 +248,7 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements greeting_in_english: str, greeting_in_chinese: Optional[str] = None, greeting_in_french: Optional[str] = None, - **kwargs: Any + **kwargs: Any, ) -> None: """Has added parameters across the API versions. @@ -303,19 +263,21 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version('test_different_calls') or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None + api_version: str = kwargs.pop( + "api_version", _params.pop("api-version", self._api_version("test_different_calls") or "3.0.0") ) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_multiapi_service_test_different_calls_request( greeting_in_english=greeting_in_english, greeting_in_chinese=greeting_in_chinese, @@ -327,10 +289,8 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -341,10 +301,10 @@ def test_different_calls( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore + return cls(pipeline_response, None, {}) # type: ignore -class OperationGroupOneOperations: +class OperationGroupOneOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -364,14 +324,8 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @distributed_trace - def test_operation_group_paging( - self, - **kwargs: Any - ) -> Iterable["_models.ModelThree"]: + def test_operation_group_paging(self, **kwargs: Any) -> Iterable["_models.ModelThree"]: """Returns ModelThree with optionalProperty 'paged'. :return: An iterator like instance of either ModelThree or the result of cls(response) @@ -381,17 +335,19 @@ def test_operation_group_paging( _headers = kwargs.pop("headers", {}) or {} _params = kwargs.pop("params", {}) or {} - cls: ClsType[_models.PagingResult] = kwargs.pop( - 'cls', None - ) + cls: ClsType[_models.PagingResult] = kwargs.pop("cls", None) error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) + def prepare_request(next_link=None): if not next_link: - + _request = build_operation_group_one_test_operation_group_paging_request( headers=_headers, params=_params, @@ -401,32 +357,33 @@ def prepare_request(next_link=None): else: # make call to next link with the client's api-version _parsed_next_link = urllib.parse.urlparse(next_link) - _next_request_params = case_insensitive_dict({ - key: [urllib.parse.quote(v) for v in value] for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items()}) + _next_request_params = case_insensitive_dict( + { + key: [urllib.parse.quote(v) for v in value] + for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() + } + ) _next_request_params["api-version"] = self._api_version - _request = HttpRequest("GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params) + _request = HttpRequest( + "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params + ) _request.url = self._client.format_url(_request.url) _request.method = "GET" return _request def extract_data(pipeline_response): - deserialized = self._deserialize( - "PagingResult", - pipeline_response - ) + deserialized = self._deserialize("PagingResult", pipeline_response) list_of_elem = deserialized.values if cls: - list_of_elem = cls(list_of_elem) # type: ignore + list_of_elem = cls(list_of_elem) # type: ignore return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): _request = prepare_request(next_link) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -436,11 +393,7 @@ def get_next(next_link=None): return pipeline_response - - return ItemPaged( - get_next, extract_data - ) - + return ItemPaged(get_next, extract_data) @overload def test_two( @@ -448,7 +401,7 @@ def test_two( parameter_one: Optional[_models.ModelThree] = None, *, content_type: str = "application/json", - **kwargs: Any + **kwargs: Any, ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -464,11 +417,7 @@ def test_two( @overload def test_two( - self, - parameter_one: Optional[IO[bytes]] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, parameter_one: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -482,12 +431,9 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_two( - self, - parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, - **kwargs: Any + self, parameter_one: Optional[Union[_models.ModelThree, IO[bytes]]] = None, **kwargs: Any ) -> _models.ModelThree: """TestTwo should be in OperationGroupOneOperations. Takes in ModelThree and ouputs ModelThree. @@ -499,18 +445,19 @@ def test_two( :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[_models.ModelThree] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[_models.ModelThree] = kwargs.pop("cls", None) content_type = content_type or "application/json" _json = None @@ -519,7 +466,7 @@ def test_two( _content = parameter_one else: if parameter_one is not None: - _json = self._serialize.body(parameter_one, 'ModelThree') + _json = self._serialize.body(parameter_one, "ModelThree") else: _json = None @@ -534,10 +481,8 @@ def test_two( _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -547,18 +492,15 @@ def test_two( error = self._deserialize.failsafe_deserialize(_models.Error, pipeline_response) raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) - deserialized = self._deserialize( - 'ModelThree', - pipeline_response.http_response - ) + deserialized = self._deserialize("ModelThree", pipeline_response.http_response) if cls: - return cls(pipeline_response, deserialized, {}) # type: ignore + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized # type: ignore -class OperationGroupTwoOperations: +class OperationGroupTwoOperations: """ .. warning:: **DO NOT** instantiate this class directly. @@ -578,16 +520,9 @@ def __init__(self, *args, **kwargs): self._deserialize = input_args.pop(0) if input_args else kwargs.pop("deserializer") self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") - - - @overload def test_four( - self, - input: Optional[_models.SourcePath] = None, - *, - content_type: str = "application/json", - **kwargs: Any + self, input: Optional[_models.SourcePath] = None, *, content_type: str = "application/json", **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -603,11 +538,7 @@ def test_four( @overload def test_four( - self, - input: Optional[IO[bytes]] = None, - *, - content_type: Optional[str] = None, - **kwargs: Any + self, input: Optional[IO[bytes]] = None, *, content_type: Optional[str] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -622,12 +553,9 @@ def test_four( :raises ~azure.core.exceptions.HttpResponseError: """ - @distributed_trace def test_four( # pylint: disable=inconsistent-return-statements - self, - input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, - **kwargs: Any + self, input: Optional[Union[_models.SourcePath, IO[bytes]]] = None, **kwargs: Any ) -> None: """TestFour should be in OperationGroupTwoOperations. @@ -639,18 +567,19 @@ def test_four( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - content_type: Optional[str] = kwargs.pop('content_type', _headers.pop('Content-Type', None)) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None)) + cls: ClsType[None] = kwargs.pop("cls", None) _json = None _content = None @@ -658,7 +587,7 @@ def test_four( # pylint: disable=inconsistent-return-statements _content = input else: if input is not None: - _json = self._serialize.body(input, 'SourcePath') + _json = self._serialize.body(input, "SourcePath") else: _json = None content_type = content_type or "application/json" @@ -674,10 +603,8 @@ def test_four( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -688,15 +615,10 @@ def test_four( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def test_five( # pylint: disable=inconsistent-return-statements - self, - **kwargs: Any - ) -> None: + def test_five(self, **kwargs: Any) -> None: # pylint: disable=inconsistent-return-statements """TestFive should be in OperationGroupTwoOperations. :return: None or the result of cls(response) @@ -704,19 +626,19 @@ def test_five( # pylint: disable=inconsistent-return-statements :raises ~azure.core.exceptions.HttpResponseError: """ error_map: MutableMapping = { - 401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError, 304: ResourceNotModifiedError + 401: ClientAuthenticationError, + 404: ResourceNotFoundError, + 409: ResourceExistsError, + 304: ResourceNotModifiedError, } - error_map.update(kwargs.pop('error_map', {}) or {}) + error_map.update(kwargs.pop("error_map", {}) or {}) _headers = kwargs.pop("headers", {}) or {} _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) - api_version: str = kwargs.pop('api_version', _params.pop('api-version', self._api_version or "3.0.0")) - cls: ClsType[None] = kwargs.pop( - 'cls', None - ) + api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "3.0.0")) + cls: ClsType[None] = kwargs.pop("cls", None) - _request = build_operation_group_two_test_five_request( api_version=api_version, headers=_headers, @@ -725,10 +647,8 @@ def test_five( # pylint: disable=inconsistent-return-statements _request.url = self._client.format_url(_request.url) _stream = False - pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - _request, - stream=_stream, - **kwargs + pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -739,6 +659,4 @@ def test_five( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) # type: ignore - - + return cls(pipeline_response, None, {}) # type: ignore