@typespec/http-client-python 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (334) hide show
  1. package/README.md +118 -0
  2. package/dist/emitter/code-model.d.ts +4 -0
  3. package/dist/emitter/code-model.d.ts.map +1 -0
  4. package/dist/emitter/code-model.js +195 -0
  5. package/dist/emitter/code-model.js.map +1 -0
  6. package/dist/emitter/emitter.d.ts +6 -0
  7. package/dist/emitter/emitter.d.ts.map +1 -0
  8. package/dist/emitter/emitter.js +104 -0
  9. package/dist/emitter/emitter.js.map +1 -0
  10. package/dist/emitter/external-process.d.ts +20 -0
  11. package/dist/emitter/external-process.d.ts.map +1 -0
  12. package/dist/emitter/external-process.js +44 -0
  13. package/dist/emitter/external-process.js.map +1 -0
  14. package/dist/emitter/http.d.ts +7 -0
  15. package/dist/emitter/http.d.ts.map +1 -0
  16. package/dist/emitter/http.js +268 -0
  17. package/dist/emitter/http.js.map +1 -0
  18. package/dist/emitter/index.d.ts +3 -0
  19. package/dist/emitter/index.d.ts.map +1 -0
  20. package/dist/emitter/index.js +3 -0
  21. package/dist/emitter/index.js.map +1 -0
  22. package/dist/emitter/lib.d.ts +30 -0
  23. package/dist/emitter/lib.d.ts.map +1 -0
  24. package/dist/emitter/lib.js +33 -0
  25. package/dist/emitter/lib.js.map +1 -0
  26. package/dist/emitter/types.d.ts +36 -0
  27. package/dist/emitter/types.d.ts.map +1 -0
  28. package/dist/emitter/types.js +491 -0
  29. package/dist/emitter/types.js.map +1 -0
  30. package/dist/emitter/utils.d.ts +26 -0
  31. package/dist/emitter/utils.d.ts.map +1 -0
  32. package/dist/emitter/utils.js +155 -0
  33. package/dist/emitter/utils.js.map +1 -0
  34. package/emitter/src/code-model.ts +272 -0
  35. package/emitter/src/emitter.ts +127 -0
  36. package/emitter/src/external-process.ts +52 -0
  37. package/emitter/src/http.ts +382 -0
  38. package/emitter/src/index.ts +2 -0
  39. package/emitter/src/lib.ts +59 -0
  40. package/emitter/src/types.ts +573 -0
  41. package/emitter/src/utils.ts +215 -0
  42. package/emitter/temp/tsconfig.tsbuildinfo +1 -0
  43. package/emitter/test/utils.test.ts +22 -0
  44. package/emitter/tsconfig.build.json +11 -0
  45. package/emitter/tsconfig.json +7 -0
  46. package/emitter/vitest.config.ts +4 -0
  47. package/eng/scripts/Build-Packages.ps1 +86 -0
  48. package/eng/scripts/Check-GitChanges.ps1 +22 -0
  49. package/eng/scripts/Functions.ps1 +26 -0
  50. package/eng/scripts/Generate.ps1 +11 -0
  51. package/eng/scripts/Generation.psm1 +22 -0
  52. package/eng/scripts/Initialize-Repository.ps1 +40 -0
  53. package/eng/scripts/Test-Packages.ps1 +65 -0
  54. package/eng/scripts/ci/format.ts +3 -0
  55. package/eng/scripts/ci/lint.ts +39 -0
  56. package/eng/scripts/ci/mypy.ini +38 -0
  57. package/eng/scripts/ci/pylintrc +59 -0
  58. package/eng/scripts/ci/pyproject.toml +18 -0
  59. package/eng/scripts/ci/pyrightconfig.json +6 -0
  60. package/eng/scripts/ci/regenerate.ts +299 -0
  61. package/eng/scripts/ci/run-ci.ts +88 -0
  62. package/eng/scripts/ci/run_apiview.py +40 -0
  63. package/eng/scripts/ci/run_mypy.py +49 -0
  64. package/eng/scripts/ci/run_pylint.py +50 -0
  65. package/eng/scripts/ci/run_pyright.py +58 -0
  66. package/eng/scripts/ci/util.py +72 -0
  67. package/eng/scripts/ci/utils.ts +48 -0
  68. package/eng/scripts/setup/__pycache__/venvtools.cpython-38.pyc +0 -0
  69. package/eng/scripts/setup/install.py +53 -0
  70. package/eng/scripts/setup/prepare.py +42 -0
  71. package/eng/scripts/setup/run-python3.ts +25 -0
  72. package/eng/scripts/setup/run_tsp.py +42 -0
  73. package/eng/scripts/setup/system-requirements.ts +261 -0
  74. package/eng/scripts/setup/venvtools.py +87 -0
  75. package/generator/LICENSE +21 -0
  76. package/generator/README.md +1 -0
  77. package/generator/dev_requirements.txt +13 -0
  78. package/generator/pygen/__init__.py +107 -0
  79. package/generator/pygen/_version.py +7 -0
  80. package/generator/pygen/black.py +71 -0
  81. package/generator/pygen/codegen/__init__.py +338 -0
  82. package/generator/pygen/codegen/_utils.py +17 -0
  83. package/generator/pygen/codegen/models/__init__.py +204 -0
  84. package/generator/pygen/codegen/models/base.py +186 -0
  85. package/generator/pygen/codegen/models/base_builder.py +118 -0
  86. package/generator/pygen/codegen/models/client.py +433 -0
  87. package/generator/pygen/codegen/models/code_model.py +237 -0
  88. package/generator/pygen/codegen/models/combined_type.py +149 -0
  89. package/generator/pygen/codegen/models/constant_type.py +129 -0
  90. package/generator/pygen/codegen/models/credential_types.py +214 -0
  91. package/generator/pygen/codegen/models/dictionary_type.py +127 -0
  92. package/generator/pygen/codegen/models/enum_type.py +238 -0
  93. package/generator/pygen/codegen/models/imports.py +291 -0
  94. package/generator/pygen/codegen/models/list_type.py +143 -0
  95. package/generator/pygen/codegen/models/lro_operation.py +142 -0
  96. package/generator/pygen/codegen/models/lro_paging_operation.py +32 -0
  97. package/generator/pygen/codegen/models/model_type.py +359 -0
  98. package/generator/pygen/codegen/models/operation.py +530 -0
  99. package/generator/pygen/codegen/models/operation_group.py +184 -0
  100. package/generator/pygen/codegen/models/paging_operation.py +155 -0
  101. package/generator/pygen/codegen/models/parameter.py +412 -0
  102. package/generator/pygen/codegen/models/parameter_list.py +387 -0
  103. package/generator/pygen/codegen/models/primitive_types.py +659 -0
  104. package/generator/pygen/codegen/models/property.py +170 -0
  105. package/generator/pygen/codegen/models/request_builder.py +189 -0
  106. package/generator/pygen/codegen/models/request_builder_parameter.py +115 -0
  107. package/generator/pygen/codegen/models/response.py +348 -0
  108. package/generator/pygen/codegen/models/utils.py +21 -0
  109. package/generator/pygen/codegen/serializers/__init__.py +574 -0
  110. package/generator/pygen/codegen/serializers/base_serializer.py +21 -0
  111. package/generator/pygen/codegen/serializers/builder_serializer.py +1507 -0
  112. package/generator/pygen/codegen/serializers/client_serializer.py +294 -0
  113. package/generator/pygen/codegen/serializers/enum_serializer.py +15 -0
  114. package/generator/pygen/codegen/serializers/general_serializer.py +213 -0
  115. package/generator/pygen/codegen/serializers/import_serializer.py +126 -0
  116. package/generator/pygen/codegen/serializers/metadata_serializer.py +198 -0
  117. package/generator/pygen/codegen/serializers/model_init_serializer.py +33 -0
  118. package/generator/pygen/codegen/serializers/model_serializer.py +317 -0
  119. package/generator/pygen/codegen/serializers/operation_groups_serializer.py +89 -0
  120. package/generator/pygen/codegen/serializers/operations_init_serializer.py +44 -0
  121. package/generator/pygen/codegen/serializers/parameter_serializer.py +221 -0
  122. package/generator/pygen/codegen/serializers/patch_serializer.py +19 -0
  123. package/generator/pygen/codegen/serializers/request_builders_serializer.py +52 -0
  124. package/generator/pygen/codegen/serializers/sample_serializer.py +168 -0
  125. package/generator/pygen/codegen/serializers/test_serializer.py +292 -0
  126. package/generator/pygen/codegen/serializers/types_serializer.py +31 -0
  127. package/generator/pygen/codegen/serializers/utils.py +68 -0
  128. package/generator/pygen/codegen/templates/client.py.jinja2 +37 -0
  129. package/generator/pygen/codegen/templates/client_container.py.jinja2 +12 -0
  130. package/generator/pygen/codegen/templates/config.py.jinja2 +73 -0
  131. package/generator/pygen/codegen/templates/config_container.py.jinja2 +16 -0
  132. package/generator/pygen/codegen/templates/conftest.py.jinja2 +28 -0
  133. package/generator/pygen/codegen/templates/enum.py.jinja2 +13 -0
  134. package/generator/pygen/codegen/templates/enum_container.py.jinja2 +10 -0
  135. package/generator/pygen/codegen/templates/init.py.jinja2 +24 -0
  136. package/generator/pygen/codegen/templates/keywords.jinja2 +19 -0
  137. package/generator/pygen/codegen/templates/lro_operation.py.jinja2 +16 -0
  138. package/generator/pygen/codegen/templates/lro_paging_operation.py.jinja2 +18 -0
  139. package/generator/pygen/codegen/templates/macros.jinja2 +12 -0
  140. package/generator/pygen/codegen/templates/metadata.json.jinja2 +167 -0
  141. package/generator/pygen/codegen/templates/model_base.py.jinja2 +1157 -0
  142. package/generator/pygen/codegen/templates/model_container.py.jinja2 +12 -0
  143. package/generator/pygen/codegen/templates/model_dpg.py.jinja2 +97 -0
  144. package/generator/pygen/codegen/templates/model_init.py.jinja2 +28 -0
  145. package/generator/pygen/codegen/templates/model_msrest.py.jinja2 +92 -0
  146. package/generator/pygen/codegen/templates/operation.py.jinja2 +21 -0
  147. package/generator/pygen/codegen/templates/operation_group.py.jinja2 +75 -0
  148. package/generator/pygen/codegen/templates/operation_groups_container.py.jinja2 +19 -0
  149. package/generator/pygen/codegen/templates/operation_tools.jinja2 +81 -0
  150. package/generator/pygen/codegen/templates/operations_folder_init.py.jinja2 +17 -0
  151. package/generator/pygen/codegen/templates/packaging_templates/CHANGELOG.md.jinja2 +6 -0
  152. package/generator/pygen/codegen/templates/packaging_templates/LICENSE.jinja2 +21 -0
  153. package/generator/pygen/codegen/templates/packaging_templates/MANIFEST.in.jinja2 +8 -0
  154. package/generator/pygen/codegen/templates/packaging_templates/README.md.jinja2 +107 -0
  155. package/generator/pygen/codegen/templates/packaging_templates/dev_requirements.txt.jinja2 +9 -0
  156. package/generator/pygen/codegen/templates/packaging_templates/setup.py.jinja2 +108 -0
  157. package/generator/pygen/codegen/templates/paging_operation.py.jinja2 +21 -0
  158. package/generator/pygen/codegen/templates/patch.py.jinja2 +19 -0
  159. package/generator/pygen/codegen/templates/pkgutil_init.py.jinja2 +1 -0
  160. package/generator/pygen/codegen/templates/request_builder.py.jinja2 +28 -0
  161. package/generator/pygen/codegen/templates/request_builders.py.jinja2 +10 -0
  162. package/generator/pygen/codegen/templates/rest_init.py.jinja2 +12 -0
  163. package/generator/pygen/codegen/templates/sample.py.jinja2 +44 -0
  164. package/generator/pygen/codegen/templates/serialization.py.jinja2 +2114 -0
  165. package/generator/pygen/codegen/templates/test.py.jinja2 +50 -0
  166. package/generator/pygen/codegen/templates/testpreparer.py.jinja2 +26 -0
  167. package/generator/pygen/codegen/templates/types.py.jinja2 +7 -0
  168. package/generator/pygen/codegen/templates/validation.py.jinja2 +38 -0
  169. package/generator/pygen/codegen/templates/vendor.py.jinja2 +95 -0
  170. package/generator/pygen/codegen/templates/version.py.jinja2 +4 -0
  171. package/generator/pygen/m2r.py +65 -0
  172. package/generator/pygen/postprocess/__init__.py +183 -0
  173. package/generator/pygen/postprocess/get_all.py +19 -0
  174. package/generator/pygen/postprocess/venvtools.py +75 -0
  175. package/generator/pygen/preprocess/__init__.py +515 -0
  176. package/generator/pygen/preprocess/helpers.py +27 -0
  177. package/generator/pygen/preprocess/python_mappings.py +224 -0
  178. package/generator/pygen/utils.py +163 -0
  179. package/generator/pygen.egg-info/PKG-INFO +25 -0
  180. package/generator/pygen.egg-info/SOURCES.txt +66 -0
  181. package/generator/pygen.egg-info/dependency_links.txt +1 -0
  182. package/generator/pygen.egg-info/requires.txt +4 -0
  183. package/generator/pygen.egg-info/top_level.txt +1 -0
  184. package/generator/requirements.txt +12 -0
  185. package/generator/setup.py +55 -0
  186. package/generator/test/azure/mock_api_tests/asynctests/test_azure_arm_models_common_types_managed_identity_async.py +63 -0
  187. package/generator/test/azure/mock_api_tests/asynctests/test_azure_arm_models_resource_async.py +284 -0
  188. package/generator/test/azure/mock_api_tests/asynctests/test_azure_client_generator_core_access_async.py +101 -0
  189. package/generator/test/azure/mock_api_tests/asynctests/test_azure_client_generator_core_flatten_async.py +93 -0
  190. package/generator/test/azure/mock_api_tests/asynctests/test_azure_client_generator_core_usage_async.py +31 -0
  191. package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_basic_async.py +76 -0
  192. package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_lro_rpc_async.py +22 -0
  193. package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_lro_standard_async.py +39 -0
  194. package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_model_async.py +33 -0
  195. package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_page_async.py +58 -0
  196. package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_scalar_async.py +41 -0
  197. package/generator/test/azure/mock_api_tests/asynctests/test_azure_core_traits_async.py +87 -0
  198. package/generator/test/azure/mock_api_tests/asynctests/test_azure_example_basic_async.py +30 -0
  199. package/generator/test/azure/mock_api_tests/asynctests/test_azure_special_headers_client_request_id_async.py +30 -0
  200. package/generator/test/azure/mock_api_tests/conftest.py +150 -0
  201. package/generator/test/azure/mock_api_tests/test_azure_arm_models_common_types_managed_identity.py +60 -0
  202. package/generator/test/azure/mock_api_tests/test_azure_arm_models_resource.py +254 -0
  203. package/generator/test/azure/mock_api_tests/test_azure_client_generator_core_access.py +92 -0
  204. package/generator/test/azure/mock_api_tests/test_azure_client_generator_core_flatten.py +84 -0
  205. package/generator/test/azure/mock_api_tests/test_azure_client_generator_core_usage.py +28 -0
  206. package/generator/test/azure/mock_api_tests/test_azure_core_basic.py +70 -0
  207. package/generator/test/azure/mock_api_tests/test_azure_core_lro_rpc.py +20 -0
  208. package/generator/test/azure/mock_api_tests/test_azure_core_lro_standard.py +32 -0
  209. package/generator/test/azure/mock_api_tests/test_azure_core_model.py +30 -0
  210. package/generator/test/azure/mock_api_tests/test_azure_core_page.py +51 -0
  211. package/generator/test/azure/mock_api_tests/test_azure_core_scalar.py +35 -0
  212. package/generator/test/azure/mock_api_tests/test_azure_core_traits.py +85 -0
  213. package/generator/test/azure/mock_api_tests/test_azure_example_basic.py +29 -0
  214. package/generator/test/azure/mock_api_tests/test_azure_special_headers_client_request_id.py +29 -0
  215. package/generator/test/azure/requirements.txt +89 -0
  216. package/generator/test/azure/tox.ini +56 -0
  217. package/generator/test/generic_mock_api_tests/asynctests/test_authentication_async.py +121 -0
  218. package/generator/test/generic_mock_api_tests/asynctests/test_client_naming_async.py +69 -0
  219. package/generator/test/generic_mock_api_tests/asynctests/test_client_structure_async.py +62 -0
  220. package/generator/test/generic_mock_api_tests/asynctests/test_encode_bytes_async.py +133 -0
  221. package/generator/test/generic_mock_api_tests/asynctests/test_encode_datetime_async.py +127 -0
  222. package/generator/test/generic_mock_api_tests/asynctests/test_encode_duration_async.py +63 -0
  223. package/generator/test/generic_mock_api_tests/asynctests/test_encode_numeric_async.py +35 -0
  224. package/generator/test/generic_mock_api_tests/asynctests/test_headasboolean_async.py +35 -0
  225. package/generator/test/generic_mock_api_tests/asynctests/test_parameters_basic_async.py +24 -0
  226. package/generator/test/generic_mock_api_tests/asynctests/test_parameters_body_optionality_async.py +30 -0
  227. package/generator/test/generic_mock_api_tests/asynctests/test_parameters_collection_format_async.py +44 -0
  228. package/generator/test/generic_mock_api_tests/asynctests/test_parameters_spread_async.py +76 -0
  229. package/generator/test/generic_mock_api_tests/asynctests/test_payload_content_negotiation_async.py +37 -0
  230. package/generator/test/generic_mock_api_tests/asynctests/test_payload_json_merge_patch_async.py +98 -0
  231. package/generator/test/generic_mock_api_tests/asynctests/test_payload_media_type_async.py +27 -0
  232. package/generator/test/generic_mock_api_tests/asynctests/test_payload_multipart_async.py +153 -0
  233. package/generator/test/generic_mock_api_tests/asynctests/test_payload_pageable_async.py +19 -0
  234. package/generator/test/generic_mock_api_tests/asynctests/test_payload_xml_async.py +103 -0
  235. package/generator/test/generic_mock_api_tests/asynctests/test_resiliency_srv_driven_async.py +128 -0
  236. package/generator/test/generic_mock_api_tests/asynctests/test_routes_async.py +331 -0
  237. package/generator/test/generic_mock_api_tests/asynctests/test_serialization_encoded_name_json_async.py +24 -0
  238. package/generator/test/generic_mock_api_tests/asynctests/test_server_endpoint_not_defined_async.py +18 -0
  239. package/generator/test/generic_mock_api_tests/asynctests/test_server_path_multiple_async.py +25 -0
  240. package/generator/test/generic_mock_api_tests/asynctests/test_server_path_single_async.py +18 -0
  241. package/generator/test/generic_mock_api_tests/asynctests/test_server_versions_not_versioned_async.py +28 -0
  242. package/generator/test/generic_mock_api_tests/asynctests/test_server_versions_versioned_async.py +34 -0
  243. package/generator/test/generic_mock_api_tests/asynctests/test_special_headers_conditional_request_async.py +38 -0
  244. package/generator/test/generic_mock_api_tests/asynctests/test_special_headers_repeatability_async.py +19 -0
  245. package/generator/test/generic_mock_api_tests/asynctests/test_special_words_async.py +42 -0
  246. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_array_async.py +118 -0
  247. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_dictionary_async.py +98 -0
  248. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_enum_extensible_async.py +25 -0
  249. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_enum_fixed_async.py +27 -0
  250. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_empty_async.py +32 -0
  251. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_inheritance_enum_discriminator_async.py +70 -0
  252. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_inheritance_nested_discriminator_async.py +85 -0
  253. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_inheritance_not_discriminated_async.py +34 -0
  254. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_inheritance_recursive_async.py +34 -0
  255. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_inheritance_single_discriminator_async.py +67 -0
  256. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_usage_async.py +32 -0
  257. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_model_visibility_async.py +47 -0
  258. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_property_additionalproperties_async.py +352 -0
  259. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_property_nullable_async.py +110 -0
  260. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_property_optional_async.py +197 -0
  261. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_property_valuetypes_async.py +315 -0
  262. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_scalar_async.py +60 -0
  263. package/generator/test/generic_mock_api_tests/asynctests/test_typetest_union_async.py +90 -0
  264. package/generator/test/generic_mock_api_tests/asynctests/test_versioning_added_async.py +36 -0
  265. package/generator/test/generic_mock_api_tests/asynctests/test_versioning_made_optional_async.py +21 -0
  266. package/generator/test/generic_mock_api_tests/asynctests/test_versioning_removed_async.py +21 -0
  267. package/generator/test/generic_mock_api_tests/asynctests/test_versioning_renamed_from_async.py +29 -0
  268. package/generator/test/generic_mock_api_tests/asynctests/test_versioning_return_type_changed_from_async.py +18 -0
  269. package/generator/test/generic_mock_api_tests/asynctests/test_versioning_type_changed_from_async.py +22 -0
  270. package/generator/test/generic_mock_api_tests/conftest.py +113 -0
  271. package/generator/test/generic_mock_api_tests/data/image.jpg +0 -0
  272. package/generator/test/generic_mock_api_tests/data/image.png +0 -0
  273. package/generator/test/generic_mock_api_tests/test_authentication.py +113 -0
  274. package/generator/test/generic_mock_api_tests/test_client_naming.py +57 -0
  275. package/generator/test/generic_mock_api_tests/test_client_structure.py +57 -0
  276. package/generator/test/generic_mock_api_tests/test_encode_bytes.py +128 -0
  277. package/generator/test/generic_mock_api_tests/test_encode_datetime.py +123 -0
  278. package/generator/test/generic_mock_api_tests/test_encode_duration.py +60 -0
  279. package/generator/test/generic_mock_api_tests/test_encode_numeric.py +31 -0
  280. package/generator/test/generic_mock_api_tests/test_headasboolean.py +33 -0
  281. package/generator/test/generic_mock_api_tests/test_parameters_basic.py +22 -0
  282. package/generator/test/generic_mock_api_tests/test_parameters_body_optionality.py +27 -0
  283. package/generator/test/generic_mock_api_tests/test_parameters_collection_format.py +37 -0
  284. package/generator/test/generic_mock_api_tests/test_parameters_spread.py +66 -0
  285. package/generator/test/generic_mock_api_tests/test_payload_content_negotiation.py +33 -0
  286. package/generator/test/generic_mock_api_tests/test_payload_json_merge_patch.py +93 -0
  287. package/generator/test/generic_mock_api_tests/test_payload_media_type.py +25 -0
  288. package/generator/test/generic_mock_api_tests/test_payload_multipart.py +140 -0
  289. package/generator/test/generic_mock_api_tests/test_payload_pageable.py +18 -0
  290. package/generator/test/generic_mock_api_tests/test_payload_xml.py +93 -0
  291. package/generator/test/generic_mock_api_tests/test_resiliency_srv_driven.py +122 -0
  292. package/generator/test/generic_mock_api_tests/test_routes.py +285 -0
  293. package/generator/test/generic_mock_api_tests/test_serialization_encoded_name_json.py +21 -0
  294. package/generator/test/generic_mock_api_tests/test_server_endpoint_not_defined.py +17 -0
  295. package/generator/test/generic_mock_api_tests/test_server_path_multiple.py +21 -0
  296. package/generator/test/generic_mock_api_tests/test_server_path_single.py +17 -0
  297. package/generator/test/generic_mock_api_tests/test_server_versions_not_versioned.py +25 -0
  298. package/generator/test/generic_mock_api_tests/test_server_versions_versioned.py +30 -0
  299. package/generator/test/generic_mock_api_tests/test_special_headers_conditional_request.py +34 -0
  300. package/generator/test/generic_mock_api_tests/test_special_headers_repeatability.py +18 -0
  301. package/generator/test/generic_mock_api_tests/test_special_words.py +37 -0
  302. package/generator/test/generic_mock_api_tests/test_typetest_array.py +103 -0
  303. package/generator/test/generic_mock_api_tests/test_typetest_dictionary.py +86 -0
  304. package/generator/test/generic_mock_api_tests/test_typetest_enum_extensible.py +23 -0
  305. package/generator/test/generic_mock_api_tests/test_typetest_enum_fixed.py +25 -0
  306. package/generator/test/generic_mock_api_tests/test_typetest_model_empty.py +29 -0
  307. package/generator/test/generic_mock_api_tests/test_typetest_model_inheritance_enum_discriminator.py +58 -0
  308. package/generator/test/generic_mock_api_tests/test_typetest_model_inheritance_nested_discriminator.py +79 -0
  309. package/generator/test/generic_mock_api_tests/test_typetest_model_inheritance_not_discriminated.py +31 -0
  310. package/generator/test/generic_mock_api_tests/test_typetest_model_inheritance_recursive.py +32 -0
  311. package/generator/test/generic_mock_api_tests/test_typetest_model_inheritance_single_discriminator.py +60 -0
  312. package/generator/test/generic_mock_api_tests/test_typetest_model_usage.py +28 -0
  313. package/generator/test/generic_mock_api_tests/test_typetest_model_visibility.py +40 -0
  314. package/generator/test/generic_mock_api_tests/test_typetest_property_additionalproperties.py +313 -0
  315. package/generator/test/generic_mock_api_tests/test_typetest_property_nullable.py +102 -0
  316. package/generator/test/generic_mock_api_tests/test_typetest_property_optional.py +174 -0
  317. package/generator/test/generic_mock_api_tests/test_typetest_property_valuetypes.py +286 -0
  318. package/generator/test/generic_mock_api_tests/test_typetest_scalar.py +53 -0
  319. package/generator/test/generic_mock_api_tests/test_typetest_union.py +80 -0
  320. package/generator/test/generic_mock_api_tests/test_versioning_added.py +33 -0
  321. package/generator/test/generic_mock_api_tests/test_versioning_made_optional.py +20 -0
  322. package/generator/test/generic_mock_api_tests/test_versioning_removed.py +20 -0
  323. package/generator/test/generic_mock_api_tests/test_versioning_renamed_from.py +27 -0
  324. package/generator/test/generic_mock_api_tests/test_versioning_return_type_changed_from.py +17 -0
  325. package/generator/test/generic_mock_api_tests/test_versioning_type_changed_from.py +21 -0
  326. package/generator/test/generic_mock_api_tests/unittests/test_model_base_serialization.py +4067 -0
  327. package/generator/test/generic_mock_api_tests/unittests/test_model_base_xml_serialization.py +1001 -0
  328. package/generator/test/unbranded/mock_api_tests/asynctests/test_unbranded_async.py +24 -0
  329. package/generator/test/unbranded/mock_api_tests/cadl-ranch-config.yaml +27 -0
  330. package/generator/test/unbranded/mock_api_tests/conftest.py +35 -0
  331. package/generator/test/unbranded/mock_api_tests/test_unbranded.py +57 -0
  332. package/generator/test/unbranded/requirements.txt +72 -0
  333. package/generator/test/unbranded/tox.ini +56 -0
  334. package/package.json +93 -0
@@ -0,0 +1,1001 @@
1
+ # ------------------------------------
2
+ # Copyright (c) Microsoft Corporation.
3
+ # Licensed under the MIT License.
4
+ # ------------------------------------
5
+ import xml.etree.ElementTree as ET
6
+
7
+ from typing import (
8
+ List,
9
+ Literal,
10
+ Dict,
11
+ )
12
+
13
+ from specialwords._model_base import (
14
+ _get_element,
15
+ Model,
16
+ rest_field,
17
+ rest_discriminator,
18
+ _deserialize_xml,
19
+ )
20
+
21
+
22
+ def assert_xml_equals(x1, x2):
23
+ ET.dump(x1)
24
+ ET.dump(x2)
25
+
26
+ assert x1.tag == x2.tag
27
+ assert (x1.text or "").strip() == (x2.text or "").strip()
28
+ # assert x1.tail == x2.tail # Swagger does not change tail
29
+ assert x1.attrib == x2.attrib
30
+ assert len(x1) == len(x2)
31
+ for c1, c2 in zip(x1, x2):
32
+ assert_xml_equals(c1, c2)
33
+
34
+
35
+ class TestXmlDeserialization:
36
+ def test_basic(self):
37
+ """Test an ultra basic XML."""
38
+ basic_xml = """<?xml version="1.0"?>
39
+ <Data country="france">
40
+ <Int>12</Int>
41
+ <EmptyInt/>
42
+ <Float>12.34</Float>
43
+ <EmptyFloat/>
44
+ <Bool>true</Bool>
45
+ <EmptyBool/>
46
+ <String>test</String>
47
+ <EmptyString/>
48
+ </Data>"""
49
+
50
+ class XmlModel(Model):
51
+ int_field: int = rest_field(name="int", xml={"name": "Int"})
52
+ empty_int: int = rest_field(name="empty_int", xml={"name": "EmptyInt"})
53
+ float_field: float = rest_field(name="float", xml={"name": "Float"})
54
+ empty_float: float = rest_field(name="empty_float", xml={"name": "EmptyFloat"})
55
+ bool_field: bool = rest_field(name="bool", xml={"name": "Bool"})
56
+ empty_bool: bool = rest_field(name="empty_bool", xml={"name": "EmptyBool"})
57
+ string: str = rest_field(name="string", xml={"name": "String"})
58
+ empty_string: str = rest_field(name="empty_string", xml={"name": "EmptyString"})
59
+ not_set: str = rest_field(name="not_set", xml={"name": "NotSet"})
60
+ country: str = rest_field(name="country", xml={"name": "country", "attribute": True})
61
+
62
+ def __init__(self, *args, **kwargs):
63
+ super().__init__(*args, **kwargs)
64
+
65
+ _xml = {"name": "Data"}
66
+
67
+ result = _deserialize_xml(XmlModel, basic_xml)
68
+
69
+ assert result.int_field == 12
70
+ assert result.empty_int is None
71
+ assert result.float_field == 12.34
72
+ assert result.empty_float is None
73
+ assert result.bool_field is True
74
+ assert result.empty_bool is None
75
+ assert result.string == "test"
76
+ assert result.country == "france"
77
+ assert result.empty_string == ""
78
+ assert result.not_set is None
79
+
80
+ def test_basic_unicode(self):
81
+ """Test a XML with unicode."""
82
+ basic_xml = """<?xml version="1.0" encoding="utf-8"?>
83
+ <Data language="français"/>"""
84
+
85
+ class XmlModel(Model):
86
+ language: str = rest_field(name="language", xml={"name": "language", "attribute": True})
87
+
88
+ def __init__(self, *args, **kwargs):
89
+ super().__init__(*args, **kwargs)
90
+
91
+ _xml = {"name": "Data"}
92
+
93
+ result = _deserialize_xml(XmlModel, basic_xml)
94
+
95
+ assert result.language == "français"
96
+
97
+ def test_basic_text(self):
98
+ """Test a XML with unicode."""
99
+ basic_xml = """<?xml version="1.0" encoding="utf-8"?>
100
+ <Data language="english">I am text</Data>"""
101
+
102
+ class XmlModel(Model):
103
+ language: str = rest_field(name="language", xml={"name": "language", "attribute": True})
104
+ content: str = rest_field(name="content", xml={"name": "content", "text": True})
105
+
106
+ def __init__(self, *args, **kwargs):
107
+ super().__init__(*args, **kwargs)
108
+
109
+ _xml = {"name": "Data"}
110
+
111
+ result = _deserialize_xml(XmlModel, basic_xml)
112
+
113
+ assert result.language == "english"
114
+ assert result.content == "I am text"
115
+
116
+ def test_dict_type(self):
117
+ """Test dict type."""
118
+ basic_xml = """<?xml version="1.0"?>
119
+ <Data>
120
+ <Metadata>
121
+ <Key1>value1</Key1>
122
+ <Key2>value2</Key2>
123
+ </Metadata>
124
+ </Data>"""
125
+
126
+ class XmlModel(Model):
127
+ metadata: Dict[str, str] = rest_field(name="Metadata", xml={"name": "Metadata"})
128
+
129
+ def __init__(self, *args, **kwargs):
130
+ super().__init__(*args, **kwargs)
131
+
132
+ _xml = {"name": "Data"}
133
+
134
+ result = _deserialize_xml(XmlModel, basic_xml)
135
+
136
+ assert len(result.metadata) == 2
137
+ assert result.metadata["Key1"] == "value1"
138
+ assert result.metadata["Key2"] == "value2"
139
+
140
+ def test_basic_empty_list(self):
141
+ """Test an basic XML with an empty node."""
142
+ basic_xml = """<?xml version="1.0"?>
143
+ <Data>
144
+ <Age/>
145
+ </Data>"""
146
+
147
+ class XmlModel(Model):
148
+ age: List[str] = rest_field(name="age", xml={"name": "Age"})
149
+
150
+ def __init__(self, *args, **kwargs):
151
+ super().__init__(*args, **kwargs)
152
+
153
+ _xml = {"name": "Data"}
154
+
155
+ result = _deserialize_xml(XmlModel, basic_xml)
156
+ assert result.age == []
157
+
158
+ def test_basic_empty_list_unwrapped(self):
159
+ """Test an basic XML with an empty node."""
160
+ basic_xml = """<?xml version="1.0"?>
161
+ <Data/>"""
162
+
163
+ class XmlModel(Model):
164
+ age: List[str] = rest_field(name="age", xml={"name": "Age", "unwrapped": True})
165
+
166
+ def __init__(self, *args, **kwargs):
167
+ super().__init__(*args, **kwargs)
168
+
169
+ _xml = {"name": "Data"}
170
+
171
+ result = _deserialize_xml(XmlModel, basic_xml)
172
+ assert result.age is None
173
+
174
+ def test_list_wrapped_items_name_basic_types(self):
175
+ """Test XML list and wrap, items is basic type and there is itemsName."""
176
+
177
+ basic_xml = """<?xml version="1.0"?>
178
+ <AppleBarrel>
179
+ <GoodApples>
180
+ <Apple>granny</Apple>
181
+ <Apple>fuji</Apple>
182
+ </GoodApples>
183
+ </AppleBarrel>"""
184
+
185
+ class AppleBarrel(Model):
186
+ good_apples: List[str] = rest_field(name="GoodApples", xml={"name": "GoodApples", "itemsName": "Apple"})
187
+
188
+ def __init__(self, *args, **kwargs):
189
+ super().__init__(*args, **kwargs)
190
+
191
+ _xml = {"name": "AppleBarrel"}
192
+
193
+ result = _deserialize_xml(AppleBarrel, basic_xml)
194
+ assert result.good_apples == ["granny", "fuji"]
195
+
196
+ def test_list_not_wrapped_items_name_basic_types(self):
197
+ """Test XML list and no wrap, items is basic type and there is itemsName."""
198
+
199
+ basic_xml = """<?xml version="1.0"?>
200
+ <AppleBarrel>
201
+ <Apple>granny</Apple>
202
+ <Apple>fuji</Apple>
203
+ </AppleBarrel>"""
204
+
205
+ class AppleBarrel(Model):
206
+ good_apples: List[str] = rest_field(
207
+ name="GoodApples",
208
+ xml={"name": "GoodApples", "unwrapped": True, "itemsName": "Apple"},
209
+ )
210
+
211
+ def __init__(self, *args, **kwargs):
212
+ super().__init__(*args, **kwargs)
213
+
214
+ _xml = {"name": "AppleBarrel"}
215
+
216
+ result = _deserialize_xml(AppleBarrel, basic_xml)
217
+ assert result.good_apples == ["granny", "fuji"]
218
+
219
+ def test_list_wrapped_items_name_complex_types(self):
220
+ """Test XML list and wrap, items is ref and there is itemsName."""
221
+
222
+ basic_xml = """<?xml version="1.0"?>
223
+ <AppleBarrel>
224
+ <GoodApples>
225
+ <Apple name="granny"/>
226
+ <Apple name="fuji"/>
227
+ </GoodApples>
228
+ </AppleBarrel>"""
229
+
230
+ class Apple(Model):
231
+ name: str = rest_field(name="name", xml={"name": "name", "attribute": True})
232
+
233
+ def __init__(self, *args, **kwargs):
234
+ super().__init__(*args, **kwargs)
235
+
236
+ _xml = {"name": "Apple"}
237
+
238
+ class AppleBarrel(Model):
239
+ good_apples: List[Apple] = rest_field(name="GoodApples", xml={"name": "GoodApples", "itemsName": "Apple"})
240
+
241
+ def __init__(self, *args, **kwargs):
242
+ super().__init__(*args, **kwargs)
243
+
244
+ _xml = {"name": "AppleBarrel"}
245
+
246
+ result = _deserialize_xml(AppleBarrel, basic_xml)
247
+ assert [apple.name for apple in result.good_apples] == ["granny", "fuji"]
248
+
249
+ def test_list_not_wrapped_items_name_complex_types(self):
250
+ """Test XML list and wrap, items is ref and there is itemsName."""
251
+
252
+ basic_xml = """<?xml version="1.0"?>
253
+ <AppleBarrel>
254
+ <Apple name="granny"/>
255
+ <Apple name="fuji"/>
256
+ </AppleBarrel>"""
257
+
258
+ class Apple(Model):
259
+ name: str = rest_field(name="name", xml={"name": "name", "attribute": True})
260
+
261
+ def __init__(self, *args, **kwargs):
262
+ super().__init__(*args, **kwargs)
263
+
264
+ _xml = {"name": "Apple"}
265
+
266
+ class AppleBarrel(Model):
267
+ good_apples: List[Apple] = rest_field(
268
+ name="GoodApples",
269
+ xml={"name": "GoodApples", "unwrapped": True, "itemsName": "Apple"},
270
+ )
271
+
272
+ def __init__(self, *args, **kwargs):
273
+ super().__init__(*args, **kwargs)
274
+
275
+ _xml = {"name": "AppleBarrel"}
276
+
277
+ result = _deserialize_xml(AppleBarrel, basic_xml)
278
+ assert [apple.name for apple in result.good_apples] == ["granny", "fuji"]
279
+
280
+ def test_list_not_wrapped_items_name_complex_types(self):
281
+ """Test XML list and wrap, items is ref and there is itemsName."""
282
+
283
+ basic_xml = """<?xml version="1.0"?>
284
+ <AppleBarrel>
285
+ <Apple name="granny"/>
286
+ <Apple name="fuji"/>
287
+ </AppleBarrel>"""
288
+
289
+ class Apple(Model):
290
+ name: str = rest_field(name="name", xml={"name": "name", "attribute": True})
291
+
292
+ def __init__(self, *args, **kwargs):
293
+ super().__init__(*args, **kwargs)
294
+
295
+ _xml = {"name": "Apple"}
296
+
297
+ class AppleBarrel(Model):
298
+ good_apples: List[Apple] = rest_field(
299
+ name="GoodApples",
300
+ xml={"name": "GoodApples", "unwrapped": True, "itemsName": "Apple"},
301
+ )
302
+
303
+ def __init__(self, *args, **kwargs):
304
+ super().__init__(*args, **kwargs)
305
+
306
+ _xml = {"name": "AppleBarrel"}
307
+
308
+ result = _deserialize_xml(AppleBarrel, basic_xml)
309
+ assert [apple.name for apple in result.good_apples] == ["granny", "fuji"]
310
+
311
+ def test_basic_additional_properties(self):
312
+ """Test additional properties."""
313
+ basic_xml = """<?xml version="1.0"?>
314
+ <Data>
315
+ <add1>text</add1>
316
+ <add2>
317
+ <add2>a</add2>
318
+ <add2>b</add2>
319
+ <add2>c</add2>
320
+ </add2>
321
+ <add3>
322
+ <a>a</a>
323
+ <b>b</b>
324
+ </add3>
325
+ </Data>"""
326
+
327
+ class XmlModel(Model):
328
+ name: str = rest_field(name="name", xml={"name": "Name"})
329
+
330
+ def __init__(self, *args, **kwargs):
331
+ super().__init__(*args, **kwargs)
332
+
333
+ _xml = {"name": "Data"}
334
+
335
+ result = _deserialize_xml(XmlModel, basic_xml)
336
+
337
+ assert result.name is None
338
+ assert result["add1"] == "text"
339
+ assert result["add2"] == ["a", "b", "c"]
340
+ assert result["add3"] == {"a": "a", "b": "b"}
341
+
342
+ def test_basic_namespace(self):
343
+ """Test an ultra basic XML."""
344
+ basic_xml = """<?xml version="1.0"?>
345
+ <Data xmlns:fictional="http://characters.example.com">
346
+ <fictional:Age>37</fictional:Age>
347
+ </Data>"""
348
+
349
+ class XmlModel(Model):
350
+ age: int = rest_field(
351
+ name="age",
352
+ xml={
353
+ "name": "Age",
354
+ "prefix": "fictional",
355
+ "ns": "http://characters.example.com",
356
+ },
357
+ )
358
+
359
+ def __init__(self, *args, **kwargs):
360
+ super().__init__(*args, **kwargs)
361
+
362
+ _xml = {"name": "Data"}
363
+
364
+ result = _deserialize_xml(XmlModel, basic_xml)
365
+ assert result.age == 37
366
+
367
+ def test_complex_namespace(self):
368
+ """Test recursive namespace."""
369
+ basic_xml = """<?xml version="1.0"?>
370
+ <entry xmlns="http://www.w3.org/2005/Atom" xmlns:i="http://www.w3.org/2001/XMLSchema-instance">
371
+ <author>
372
+ <name>lmazuel</name>
373
+ </author>
374
+ <AuthorizationRules xmlns="http://schemas.microsoft.com/netservices/2010/10/servicebus/connect">
375
+ <AuthorizationRule i:type="SharedAccessAuthorizationRule">
376
+ <KeyName>testpolicy</KeyName>
377
+ </AuthorizationRule>
378
+ </AuthorizationRules>
379
+ <MessageCountDetails xmlns="http://schemas.microsoft.com/netservices/2010/10/servicebus/connect">
380
+ <d2p1:ActiveMessageCount xmlns:d2p1="http://schemas.microsoft.com/netservices/2011/06/servicebus">12</d2p1:ActiveMessageCount>
381
+ </MessageCountDetails>
382
+ </entry>"""
383
+
384
+ class QueueDescriptionResponseAuthor(Model):
385
+ name: str = rest_field(name="name", xml={"ns": "http://www.w3.org/2005/Atom"})
386
+
387
+ def __init__(self, *args, **kwargs):
388
+ super().__init__(*args, **kwargs)
389
+
390
+ _xml = {"ns": "http://www.w3.org/2005/Atom"}
391
+
392
+ class AuthorizationRule(Model):
393
+ type: str = rest_field(
394
+ name="type",
395
+ xml={
396
+ "attribute": True,
397
+ "prefix": "i",
398
+ "ns": "http://www.w3.org/2001/XMLSchema-instance",
399
+ },
400
+ )
401
+ key_name: str = rest_field(
402
+ name="KeyName",
403
+ xml={"ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect"},
404
+ )
405
+
406
+ def __init__(self, *args, **kwargs):
407
+ super().__init__(*args, **kwargs)
408
+
409
+ _xml = {"ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect"}
410
+
411
+ class MessageCountDetails(Model):
412
+ active_message_count: int = rest_field(
413
+ name="ActiveMessageCount",
414
+ xml={
415
+ "prefix": "d2p1",
416
+ "ns": "http://schemas.microsoft.com/netservices/2011/06/servicebus",
417
+ },
418
+ )
419
+
420
+ def __init__(self, *args, **kwargs):
421
+ super().__init__(*args, **kwargs)
422
+
423
+ _xml = {
424
+ "name": "CountDetails",
425
+ "ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect",
426
+ }
427
+
428
+ class XmlRoot(Model):
429
+ author: QueueDescriptionResponseAuthor = rest_field(name="author")
430
+ authorization_rules: List[AuthorizationRule] = rest_field(
431
+ name="AuthorizationRules",
432
+ xml={
433
+ "ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect",
434
+ },
435
+ )
436
+ message_count_details: MessageCountDetails = rest_field(
437
+ name="MessageCountDetails",
438
+ xml={
439
+ "ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect",
440
+ },
441
+ )
442
+
443
+ def __init__(self, *args, **kwargs):
444
+ super().__init__(*args, **kwargs)
445
+
446
+ _xml = {"name": "entry", "ns": "http://www.w3.org/2005/Atom"}
447
+
448
+ result = _deserialize_xml(XmlRoot, basic_xml)
449
+
450
+ assert result.author.name == "lmazuel"
451
+ assert result.authorization_rules[0].key_name == "testpolicy"
452
+ assert result.authorization_rules[0].type == "SharedAccessAuthorizationRule"
453
+ assert result.message_count_details.active_message_count == 12
454
+
455
+ def test_polymorphic_deserialization(self):
456
+ basic_xml = """<?xml version="1.0"?>
457
+ <entry xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
458
+ <Filter xsi:type="CorrelationFilter">
459
+ <CorrelationId>12</CorrelationId>
460
+ </Filter>
461
+ </entry>"""
462
+
463
+ class RuleFilter(Model):
464
+ __mapping__: Dict[str, Model] = {}
465
+ type: Literal[None] = rest_discriminator(
466
+ name="type",
467
+ xml={
468
+ "attribute": True,
469
+ "prefix": "xsi",
470
+ "ns": "http://www.w3.org/2001/XMLSchema-instance",
471
+ },
472
+ )
473
+
474
+ def __init__(self, *args, **kwargs):
475
+ super().__init__(*args, **kwargs)
476
+ self.type: Literal[None] = None
477
+
478
+ _xml = {"name": "Filter"}
479
+
480
+ class CorrelationFilter(RuleFilter):
481
+ type: Literal["CorrelationFilter"] = rest_discriminator(
482
+ name="type",
483
+ xml={
484
+ "attribute": True,
485
+ "prefix": "xsi",
486
+ "ns": "http://www.w3.org/2001/XMLSchema-instance",
487
+ },
488
+ )
489
+ correlation_id: int = rest_field(name="CorrelationId")
490
+
491
+ def __init__(self, *args, **kwargs):
492
+ super().__init__(*args, **kwargs)
493
+ self.type: Literal["CorrelationFilter"] = "CorrelationFilter"
494
+
495
+ class SqlFilter(RuleFilter):
496
+ type: Literal["SqlFilter"] = rest_discriminator(
497
+ name="type",
498
+ xml={
499
+ "attribute": True,
500
+ "prefix": "xsi",
501
+ "ns": "http://www.w3.org/2001/XMLSchema-instance",
502
+ },
503
+ )
504
+
505
+ def __init__(self, *args, **kwargs):
506
+ super().__init__(*args, **kwargs)
507
+ self.type: Literal["SqlFilter"] = "SqlFilter"
508
+
509
+ class XmlRoot(Model):
510
+ filter: RuleFilter = rest_field(name="Filter")
511
+
512
+ def __init__(self, *args, **kwargs):
513
+ super().__init__(*args, **kwargs)
514
+
515
+ _xml = {"name": "entry"}
516
+
517
+ result = _deserialize_xml(XmlRoot, basic_xml)
518
+
519
+ assert isinstance(result.filter, CorrelationFilter)
520
+ assert result.filter.correlation_id == 12
521
+
522
+
523
+ class TestXmlSerialization:
524
+ def test_basic(self):
525
+ """Test an ultra basic XML."""
526
+ basic_xml = ET.fromstring(
527
+ """<?xml version="1.0"?>
528
+ <Data country="france">
529
+ <Age>37</Age>
530
+ </Data>"""
531
+ )
532
+
533
+ class XmlModel(Model):
534
+ age: int = rest_field(xml={"name": "Age"})
535
+ country: str = rest_field(xml={"name": "country", "attribute": True})
536
+
537
+ def __init__(self, *args, **kwargs):
538
+ super().__init__(*args, **kwargs)
539
+
540
+ _xml = {"name": "Data"}
541
+
542
+ xml_model = XmlModel(age=37, country="france")
543
+ assert_xml_equals(_get_element(xml_model), basic_xml)
544
+
545
+ def test_basic_unicode(self):
546
+ """Test a XML with unicode."""
547
+ basic_xml = ET.fromstring(
548
+ """<?xml version="1.0" encoding="utf-8"?>
549
+ <Data language="français"/>""".encode(
550
+ "utf-8"
551
+ )
552
+ )
553
+
554
+ class XmlModel(Model):
555
+ language: str = rest_field(xml={"name": "language", "attribute": True})
556
+
557
+ def __init__(self, *args, **kwargs):
558
+ super().__init__(*args, **kwargs)
559
+
560
+ _xml = {"name": "Data"}
561
+
562
+ xml_model = XmlModel(language="français")
563
+ assert_xml_equals(_get_element(xml_model), basic_xml)
564
+
565
+ def test_nested_unicode(self):
566
+ class XmlModel(Model):
567
+ message_text: str = rest_field(name="MessageText", xml={"name": "MessageText"})
568
+
569
+ def __init__(self, *args, **kwargs):
570
+ super().__init__(*args, **kwargs)
571
+
572
+ _xml = {"name": "Message"}
573
+
574
+ xml_model_no_unicode = XmlModel(message_text="message1")
575
+ xml_element = _get_element(xml_model_no_unicode)
576
+ xml_content = ET.tostring(xml_element, encoding="utf8")
577
+ assert (
578
+ xml_content
579
+ == b"<?xml version='1.0' encoding='utf8'?>\n<Message><MessageText>message1</MessageText></Message>"
580
+ )
581
+
582
+ xml_model_with_unicode = XmlModel(message_text="message1㚈")
583
+ xml_element = _get_element(xml_model_with_unicode)
584
+ xml_content = ET.tostring(xml_element, encoding="utf8")
585
+ assert (
586
+ xml_content
587
+ == b"<?xml version='1.0' encoding='utf8'?>\n<Message><MessageText>message1\xe3\x9a\x88</MessageText></Message>"
588
+ )
589
+
590
+ def test_type_basic(self):
591
+ """Test basic types."""
592
+ basic_xml = ET.fromstring(
593
+ """<?xml version="1.0"?>
594
+ <Data>
595
+ <Age>37</Age>
596
+ <Enabled>true</Enabled>
597
+ </Data>"""
598
+ )
599
+
600
+ class XmlModel(Model):
601
+ age: int = rest_field(name="age", xml={"name": "Age"})
602
+ enabled: bool = rest_field(name="enabled", xml={"name": "Enabled"})
603
+
604
+ def __init__(self, *args, **kwargs):
605
+ super().__init__(*args, **kwargs)
606
+
607
+ _xml = {"name": "Data"}
608
+
609
+ xml_model = XmlModel(age=37, enabled=True)
610
+ assert_xml_equals(_get_element(xml_model), basic_xml)
611
+
612
+ def test_basic_text(self):
613
+ """Test a XML with unicode."""
614
+ basic_xml = ET.fromstring(
615
+ """<?xml version="1.0" encoding="utf-8"?>
616
+ <Data language="english">I am text</Data>"""
617
+ )
618
+
619
+ class XmlModel(Model):
620
+ language: str = rest_field(name="language", xml={"name": "language", "attribute": True})
621
+ content: str = rest_field(name="content", xml={"text": True})
622
+
623
+ def __init__(self, *args, **kwargs):
624
+ super().__init__(*args, **kwargs)
625
+
626
+ _xml = {"name": "Data"}
627
+
628
+ xml_model = XmlModel(language="english", content="I am text")
629
+ assert_xml_equals(_get_element(xml_model), basic_xml)
630
+
631
+ def test_dict_type(self):
632
+ """Test dict type."""
633
+ basic_xml = ET.fromstring(
634
+ """<?xml version="1.0"?>
635
+ <Data>
636
+ <Metadata>
637
+ <Key1>value1</Key1>
638
+ <Key2>value2</Key2>
639
+ </Metadata>
640
+ </Data>"""
641
+ )
642
+
643
+ class XmlModel(Model):
644
+ metadata: Dict[str, str] = rest_field(name="Metadata", xml={"name": "Metadata"})
645
+
646
+ def __init__(self, *args, **kwargs):
647
+ super().__init__(*args, **kwargs)
648
+
649
+ _xml = {"name": "Data"}
650
+
651
+ xml_model = XmlModel(
652
+ metadata={
653
+ "Key1": "value1",
654
+ "Key2": "value2",
655
+ }
656
+ )
657
+ assert_xml_equals(_get_element(xml_model), basic_xml)
658
+
659
+ def test_additional_properties(self):
660
+ """Test additional properties."""
661
+ basic_xml = ET.fromstring(
662
+ """<?xml version="1.0"?>
663
+ <Data>
664
+ <Name>test</Name>
665
+ <add1>text</add1>
666
+ <add2>
667
+ <add2>a</add2>
668
+ <add2>b</add2>
669
+ <add2>c</add2>
670
+ </add2>
671
+ <add3>
672
+ <a>a</a>
673
+ <b>b</b>
674
+ </add3>
675
+ </Data>"""
676
+ )
677
+
678
+ class XmlModel(Model):
679
+ name: str = rest_field(name="name", xml={"name": "Name"})
680
+
681
+ def __init__(self, *args, **kwargs):
682
+ super().__init__(*args, **kwargs)
683
+
684
+ _xml = {"name": "Data"}
685
+
686
+ xml_model = XmlModel(
687
+ {
688
+ "name": "test",
689
+ "add1": "text",
690
+ "add2": ["a", "b", "c"],
691
+ "add3": {"a": "a", "b": "b"},
692
+ }
693
+ )
694
+ assert_xml_equals(_get_element(xml_model), basic_xml)
695
+
696
+ def test_list_wrapped_basic_types(self):
697
+ """Test XML list and wrap, items is basic type and there is no itemsName."""
698
+
699
+ basic_xml = ET.fromstring(
700
+ """<?xml version="1.0"?>
701
+ <AppleBarrel>
702
+ <GoodApples>
703
+ <GoodApples>granny</GoodApples>
704
+ <GoodApples>fuji</GoodApples>
705
+ </GoodApples>
706
+ </AppleBarrel>"""
707
+ )
708
+
709
+ class AppleBarrel(Model):
710
+ good_apples: List[str] = rest_field(name="GoodApples", xml={"name": "GoodApples"})
711
+
712
+ def __init__(self, *args, **kwargs):
713
+ super().__init__(*args, **kwargs)
714
+
715
+ _xml = {"name": "AppleBarrel"}
716
+
717
+ xml_model = AppleBarrel(good_apples=["granny", "fuji"])
718
+ assert_xml_equals(_get_element(xml_model), basic_xml)
719
+
720
+ def test_list_not_wrapped_basic_types(self):
721
+ """Test XML list and no wrap, items is basic type and there is no itemsName."""
722
+
723
+ basic_xml = ET.fromstring(
724
+ """<?xml version="1.0"?>
725
+ <AppleBarrel>
726
+ <GoodApples>granny</GoodApples>
727
+ <GoodApples>fuji</GoodApples>
728
+ </AppleBarrel>"""
729
+ )
730
+
731
+ class AppleBarrel(Model):
732
+ good_apples: List[str] = rest_field(name="GoodApples", xml={"name": "GoodApples", "unwrapped": True})
733
+
734
+ def __init__(self, *args, **kwargs):
735
+ super().__init__(*args, **kwargs)
736
+
737
+ _xml = {"name": "AppleBarrel"}
738
+
739
+ xml_model = AppleBarrel(good_apples=["granny", "fuji"])
740
+ assert_xml_equals(_get_element(xml_model), basic_xml)
741
+
742
+ def test_list_wrapped_basic_types_items_name(self):
743
+ """Test XML list and wrap, items is basic type and itemsName."""
744
+
745
+ basic_xml = ET.fromstring(
746
+ """<?xml version="1.0"?>
747
+ <AppleBarrel>
748
+ <GoodApples>
749
+ <string>granny</string>
750
+ <string>fuji</string>
751
+ </GoodApples>
752
+ </AppleBarrel>"""
753
+ )
754
+
755
+ class AppleBarrel(Model):
756
+ good_apples: List[str] = rest_field(name="GoodApples", xml={"name": "GoodApples", "itemsName": "string"})
757
+
758
+ def __init__(self, *args, **kwargs):
759
+ super().__init__(*args, **kwargs)
760
+
761
+ _xml = {"name": "AppleBarrel"}
762
+
763
+ xml_model = AppleBarrel(good_apples=["granny", "fuji"])
764
+ assert_xml_equals(_get_element(xml_model), basic_xml)
765
+
766
+ def test_list_not_wrapped_basic_types_items_name(self):
767
+ """Test XML list and no wrap, items is basic type and itemsName."""
768
+
769
+ basic_xml = ET.fromstring(
770
+ """<?xml version="1.0"?>
771
+ <AppleBarrel>
772
+ <string>granny</string>
773
+ <string>fuji</string>
774
+ </AppleBarrel>"""
775
+ )
776
+
777
+ class AppleBarrel(Model):
778
+ good_apples: List[str] = rest_field(
779
+ name="GoodApples",
780
+ xml={"name": "GoodApples", "unwrapped": True, "itemsName": "string"},
781
+ )
782
+
783
+ def __init__(self, *args, **kwargs):
784
+ super().__init__(*args, **kwargs)
785
+
786
+ _xml = {"name": "AppleBarrel"}
787
+
788
+ xml_model = AppleBarrel(good_apples=["granny", "fuji"])
789
+ assert_xml_equals(_get_element(xml_model), basic_xml)
790
+
791
+ def test_list_wrapped_complex_types(self):
792
+ """Test XML list and wrap, items is ref."""
793
+
794
+ basic_xml = ET.fromstring(
795
+ """<?xml version="1.0"?>
796
+ <AppleBarrel>
797
+ <GoodApples>
798
+ <Apple name="granny"/>
799
+ <Apple name="fuji"/>
800
+ </GoodApples>
801
+ </AppleBarrel>"""
802
+ )
803
+
804
+ class Apple(Model):
805
+ name: str = rest_field(name="name", xml={"name": "name", "attribute": True})
806
+
807
+ def __init__(self, *args, **kwargs):
808
+ super().__init__(*args, **kwargs)
809
+
810
+ _xml = {"name": "Apple"}
811
+
812
+ class AppleBarrel(Model):
813
+ good_apples: List[Apple] = rest_field(name="GoodApples", xml={"name": "GoodApples"})
814
+
815
+ def __init__(self, *args, **kwargs):
816
+ super().__init__(*args, **kwargs)
817
+
818
+ _xml = {"name": "AppleBarrel"}
819
+
820
+ test = AppleBarrel({"GoodApples": [{"name": "granny"}, {"name": "fuji"}]})
821
+ xml_model = AppleBarrel(good_apples=[Apple(name="granny"), Apple(name="fuji")])
822
+ assert_xml_equals(_get_element(xml_model), basic_xml)
823
+
824
+ def test_list_not_wrapped_complex_types(self):
825
+ """Test XML list and wrap, items is ref."""
826
+
827
+ basic_xml = ET.fromstring(
828
+ """<?xml version="1.0"?>
829
+ <AppleBarrel>
830
+ <Apple name="granny"/>
831
+ <Apple name="fuji"/>
832
+ </AppleBarrel>"""
833
+ )
834
+
835
+ class Apple(Model):
836
+ name: str = rest_field(name="name", xml={"name": "name", "attribute": True})
837
+
838
+ def __init__(self, *args, **kwargs):
839
+ super().__init__(*args, **kwargs)
840
+
841
+ _xml = {"name": "Apple"}
842
+
843
+ class AppleBarrel(Model):
844
+ good_apples: List[Apple] = rest_field(name="GoodApples", xml={"name": "GoodApples", "unwrapped": True})
845
+
846
+ def __init__(self, *args, **kwargs):
847
+ super().__init__(*args, **kwargs)
848
+
849
+ xml_model = AppleBarrel(good_apples=[Apple(name="granny"), Apple(name="fuji")])
850
+ assert_xml_equals(_get_element(xml_model), basic_xml)
851
+
852
+ def test_two_complex_same_type(self):
853
+ """Two different attribute are same type"""
854
+
855
+ basic_xml = ET.fromstring(
856
+ """<?xml version="1.0"?>
857
+ <AppleBarrel>
858
+ <EuropeanApple name="granny"/>
859
+ <USAApple name="fuji"/>
860
+ </AppleBarrel>"""
861
+ )
862
+
863
+ class Apple(Model):
864
+ name: str = rest_field(name="name", xml={"name": "name", "attribute": True})
865
+
866
+ def __init__(self, *args, **kwargs):
867
+ super().__init__(*args, **kwargs)
868
+
869
+ _xml = {"name": "Apple"}
870
+
871
+ class AppleBarrel(Model):
872
+ eu_apple: Apple = rest_field(name="EuropeanApple", xml={"name": "EuropeanApple"})
873
+ us_apple: Apple = rest_field(name="USAApple", xml={"name": "USAApple"})
874
+
875
+ def __init__(self, *args, **kwargs):
876
+ super().__init__(*args, **kwargs)
877
+
878
+ xml_model = AppleBarrel(
879
+ eu_apple=Apple(name="granny"),
880
+ us_apple=Apple(name="fuji"),
881
+ )
882
+ assert_xml_equals(_get_element(xml_model), basic_xml)
883
+
884
+ def test_basic_namespace(self):
885
+ """Test an ultra basic XML."""
886
+ basic_xml = ET.fromstring(
887
+ """<?xml version="1.0"?>
888
+ <Data xmlns:fictional="http://characters.example.com">
889
+ <fictional:Age>37</fictional:Age>
890
+ </Data>"""
891
+ )
892
+
893
+ class XmlModel(Model):
894
+ age: int = rest_field(
895
+ name="age",
896
+ xml={
897
+ "name": "Age",
898
+ "prefix": "fictional",
899
+ "ns": "http://characters.example.com",
900
+ },
901
+ )
902
+
903
+ def __init__(self, *args, **kwargs):
904
+ super().__init__(*args, **kwargs)
905
+
906
+ _xml = {"name": "Data"}
907
+
908
+ xml_model = XmlModel(
909
+ age=37,
910
+ )
911
+ assert_xml_equals(_get_element(xml_model), basic_xml)
912
+
913
+ def test_complex_namespace(self):
914
+ """Test recursive namespace."""
915
+ basic_xml = ET.fromstring(
916
+ """<?xml version="1.0"?>
917
+ <entry xmlns="http://www.w3.org/2005/Atom" xmlns:i="http://www.w3.org/2001/XMLSchema-instance">
918
+ <author>
919
+ <name>lmazuel</name>
920
+ </author>
921
+ <AuthorizationRules xmlns="http://schemas.microsoft.com/netservices/2010/10/servicebus/connect">
922
+ <AuthorizationRule i:type="SharedAccessAuthorizationRule">
923
+ <KeyName>testpolicy</KeyName>
924
+ </AuthorizationRule>
925
+ </AuthorizationRules>
926
+ <MessageCountDetails xmlns="http://schemas.microsoft.com/netservices/2010/10/servicebus/connect">
927
+ <d2p1:ActiveMessageCount xmlns:d2p1="http://schemas.microsoft.com/netservices/2011/06/servicebus">12</d2p1:ActiveMessageCount>
928
+ </MessageCountDetails>
929
+ </entry>"""
930
+ )
931
+
932
+ class QueueDescriptionResponseAuthor(Model):
933
+ name: str = rest_field(name="name", xml={"ns": "http://www.w3.org/2005/Atom"})
934
+
935
+ def __init__(self, *args, **kwargs):
936
+ super().__init__(*args, **kwargs)
937
+
938
+ _xml = {"ns": "http://www.w3.org/2005/Atom"}
939
+
940
+ class AuthorizationRule(Model):
941
+ type: str = rest_field(
942
+ name="type",
943
+ xml={
944
+ "attribute": True,
945
+ "prefix": "i",
946
+ "ns": "http://www.w3.org/2001/XMLSchema-instance",
947
+ },
948
+ )
949
+ key_name: str = rest_field(
950
+ name="KeyName",
951
+ xml={"ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect"},
952
+ )
953
+
954
+ def __init__(self, *args, **kwargs):
955
+ super().__init__(*args, **kwargs)
956
+
957
+ _xml = {"ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect"}
958
+
959
+ class MessageCountDetails(Model):
960
+ active_message_count: int = rest_field(
961
+ name="ActiveMessageCount",
962
+ xml={
963
+ "prefix": "d2p1",
964
+ "ns": "http://schemas.microsoft.com/netservices/2011/06/servicebus",
965
+ },
966
+ )
967
+
968
+ def __init__(self, *args, **kwargs):
969
+ super().__init__(*args, **kwargs)
970
+
971
+ _xml = {
972
+ "name": "CountDetails",
973
+ "ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect",
974
+ }
975
+
976
+ class XmlRoot(Model):
977
+ author: QueueDescriptionResponseAuthor = rest_field(name="author")
978
+ authorization_rules: List[AuthorizationRule] = rest_field(
979
+ name="AuthorizationRules",
980
+ xml={
981
+ "ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect",
982
+ },
983
+ )
984
+ message_count_details: MessageCountDetails = rest_field(
985
+ name="MessageCountDetails",
986
+ xml={
987
+ "ns": "http://schemas.microsoft.com/netservices/2010/10/servicebus/connect",
988
+ },
989
+ )
990
+
991
+ def __init__(self, *args, **kwargs):
992
+ super().__init__(*args, **kwargs)
993
+
994
+ _xml = {"name": "entry", "ns": "http://www.w3.org/2005/Atom"}
995
+
996
+ xml_model = XmlRoot(
997
+ author=QueueDescriptionResponseAuthor(name="lmazuel"),
998
+ authorization_rules=[AuthorizationRule(type="SharedAccessAuthorizationRule", key_name="testpolicy")],
999
+ message_count_details=MessageCountDetails(active_message_count=12),
1000
+ )
1001
+ assert_xml_equals(_get_element(xml_model), basic_xml)