loafle/openapi-generator-original
3
0
Fork 1
mirror of https://github.com/OpenAPITools/openapi-generator.git synced 2025-12-21 19:07:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Python: Correctly serialize enum with its value (#18327) (#18328)

Browse Source
This commit is contained in:
BaptisteSaves
2024-04-10 15:12:37 +02:00
committed by GitHub
parent 9351217048
commit 6bcc28d06e
19 changed files with 233 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
samples/openapi3/client/petstore/python/tests/test_api_client.py
View File

@@ -9,11 +9,9 @@ $ cd OpenAPIetstore-python
$ pytest
"""
import os
import time
import atexit
import weakref
import unittest
from enum import Enum
from dateutil.parser import parse
import petstore_api
@@ -155,6 +153,16 @@ class ApiClientTests(unittest.TestCase):
result = self.api_client.sanitize_for_serialization(data)
self.assertEqual(result, "1997-07-16T19:20:30.450000+01:00")
def test_sanitize_for_serialization_list_enum(self):
class EnumSerialization(int, Enum):
NUMBER_0 = 0
NUMBER_1 = 1
data = [EnumSerialization.NUMBER_1]
result = self.api_client.sanitize_for_serialization(data)
self.assertEqual(result, [1])
self.assertNotIsInstance(result[0], EnumSerialization)
def test_sanitize_for_serialization_list(self):
data = [1]
result = self.api_client.sanitize_for_serialization(data)
@@ -203,10 +211,11 @@ class ApiClientTests(unittest.TestCase):
"category2": "example2"
}
result = self.api_client.parameters_to_url_query([('value', dictionary)], {})
self.assertEqual(result, "value=%7B%22category%22%3A%20%22example%22%2C%20%22category2%22%3A%20%22example2%22%7D")
self.assertEqual(result,
"value=%7B%22category%22%3A%20%22example%22%2C%20%22category2%22%3A%20%22example2%22%7D")
def test_parameters_to_url_query_complex_values(self):
data='value={"number": 1, "string": "str", "bool": true, "dict": {"number": 1, "string": "str", "bool": true}}'
data = 'value={"number": 1, "string": "str", "bool": true, "dict": {"number": 1, "string": "str", "bool": true}}'
dictionary = {
"number": 1,
"string": "str",
@@ -218,10 +227,11 @@ class ApiClientTests(unittest.TestCase):
}
}
result = self.api_client.parameters_to_url_query([('value', dictionary)], {})
self.assertEqual(result, 'value=%7B%22number%22%3A%201%2C%20%22string%22%3A%20%22str%22%2C%20%22bool%22%3A%20true%2C%20%22dict%22%3A%20%7B%22number%22%3A%201%2C%20%22string%22%3A%20%22str%22%2C%20%22bool%22%3A%20true%7D%7D')
self.assertEqual(result,
'value=%7B%22number%22%3A%201%2C%20%22string%22%3A%20%22str%22%2C%20%22bool%22%3A%20true%2C%20%22dict%22%3A%20%7B%22number%22%3A%201%2C%20%22string%22%3A%20%22str%22%2C%20%22bool%22%3A%20true%7D%7D')
def test_parameters_to_url_query_dict_values(self):
data='value={"strValues": ["one", "two", "three"], "dictValues": [{"name": "value1", "age": 14}, {"name": "value2", "age": 12}]}'
data = 'value={"strValues": ["one", "two", "three"], "dictValues": [{"name": "value1", "age": 14}, {"name": "value2", "age": 12}]}'
dictionary = {
"strValues": [
"one",
@@ -240,12 +250,14 @@ class ApiClientTests(unittest.TestCase):
]
}
result = self.api_client.parameters_to_url_query([('value', dictionary)], {})
self.assertEqual(result, 'value=%7B%22strValues%22%3A%20%5B%22one%22%2C%20%22two%22%2C%20%22three%22%5D%2C%20%22dictValues%22%3A%20%5B%7B%22name%22%3A%20%22value1%22%2C%20%22age%22%3A%2014%7D%2C%20%7B%22name%22%3A%20%22value2%22%2C%20%22age%22%3A%2012%7D%5D%7D')
self.assertEqual(result,
'value=%7B%22strValues%22%3A%20%5B%22one%22%2C%20%22two%22%2C%20%22three%22%5D%2C%20%22dictValues%22%3A%20%5B%7B%22name%22%3A%20%22value1%22%2C%20%22age%22%3A%2014%7D%2C%20%7B%22name%22%3A%20%22value2%22%2C%20%22age%22%3A%2012%7D%5D%7D')
def test_parameters_to_url_query_boolean_value(self):
result = self.api_client.parameters_to_url_query([('boolean', True)], {})
self.assertEqual(result, "boolean=true")
def test_parameters_to_url_query_list_value(self):
params = self.api_client.parameters_to_url_query(params=[('list', [1, 2, 3])], collection_formats={'list': 'multi'})
params = self.api_client.parameters_to_url_query(params=[('list', [1, 2, 3])],
collection_formats={'list': 'multi'})
self.assertEqual(params, "list=1&list=2&list=3")

25
samples/openapi3/client/petstore/python/tests/test_fake_api.py
View File

@@ -15,6 +15,8 @@ import unittest
import unittest
from unittest.mock import patch, Mock
import petstore_api
from petstore_api import OuterObjectWithEnumProperty, OuterEnumInteger
class TestFakeApi(unittest.TestCase):
"""StrLikeJson unit test stubs"""
@@ -141,4 +143,25 @@ class TestFakeApi(unittest.TestCase):
"petstore_api.api_client.ApiClient.call_api", return_value=mock_resp
):
returned = self.fake_api.fake_return_byte_like_json()
self.assertEqual(b'{"a": "a"}', returned)
self.assertEqual(b'{"a": "a"}', returned)
def testIntEnumReturnsValue(self):
"""
Fixes #18327 (https://github.com/OpenAPITools/openapi-generator/issues/18327)
The enum value should be used in the param and not the enum name
"""
mock_resp = Mock()
mock_resp.status = 200
mock_resp.data = b'{"value": "0"}'
mock_resp.getheaders.return_value = {}
mock_resp.getheader = (
lambda name: "text/plain" if name == "content-type" else Mock()
)
with patch(
"petstore_api.api_client.ApiClient.call_api", return_value=mock_resp
) as call_api_mock:
self.fake_api.fake_property_enum_integer_serialize(
outer_object_with_enum_property=OuterObjectWithEnumProperty(value=OuterEnumInteger.NUMBER_0),
param=[OuterEnumInteger.NUMBER_0])
self.assertEqual(call_api_mock.call_args[0][1],
'http://petstore.swagger.io:80/v2/fake/property/enum-int?param=0')