1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
from unittest import TestCase
from tests.fixtures.books import BookForm, Books
from tests.fixtures.books.fixtures import books
from tests.fixtures.models import Parent
from xsdata.formats.dataclass.serializers import PycodeSerializer
from xsdata.models.enums import Namespace
class PycodeSerializerTests(TestCase):
def setUp(self) -> None:
self.serializer = PycodeSerializer()
def test_render(self):
result = self.serializer.render(books, var_name="books")
expected = (
"from tests.fixtures.books.books import BookForm\n"
"from tests.fixtures.books.books import Books\n"
"from xsdata.models.datatype import XmlDate\n"
"\n"
"\n"
"books = Books(\n"
" book=[\n"
" BookForm(\n"
" author='Hightower, Kim',\n"
" title='The First Book',\n"
" genre='Fiction',\n"
" price=44.95,\n"
" pub_date=XmlDate(2000, 10, 1),\n"
" review='An amazing story of nothing.',\n"
" id='bk001'\n"
" ),\n"
" BookForm(\n"
" author='Nagata, Suanne',\n"
" title='Becoming Somebody',\n"
" genre='Biography',\n"
" price=33.95,\n"
" pub_date=XmlDate(2001, 1, 10),\n"
" review='A masterpiece of the fine art of gossiping.',\n"
" id='bk002'\n"
" ),\n"
" ]\n"
")\n"
)
self.assertEqual(expected, result)
def test_write_class_with_default_values(self):
books = Books(book=[BookForm(author="me")])
result = self.serializer.render(books, var_name="books")
expected = (
"from tests.fixtures.books.books import BookForm\n"
"from tests.fixtures.books.books import Books\n"
"\n"
"\n"
"books = Books(\n"
" book=[\n"
" BookForm(\n"
" author='me'\n"
" ),\n"
" ]\n"
")\n"
)
self.assertEqual(expected, result)
def test_write_string_with_unicode_characters(self):
books = Books(book=[BookForm(author="Backslashes \\ One Two \x12 Three")])
result = self.serializer.render(books, var_name="books")
expected = (
"from tests.fixtures.books.books import BookForm\n"
"from tests.fixtures.books.books import Books\n"
"\n"
"\n"
"books = Books(\n"
" book=[\n"
" BookForm(\n"
" author='Backslashes \\\\ One Two \\x12 Three'\n"
" ),\n"
" ]\n"
")\n"
)
self.assertEqual(expected, result)
def test_write_object_with_empty_array(self):
iterator = self.serializer.write_object([], 0, set())
self.assertEqual("[]", "".join(iterator))
iterator = self.serializer.write_object((), 0, set())
self.assertEqual("()", "".join(iterator))
iterator = self.serializer.write_object(set(), 0, set())
self.assertEqual("set()", "".join(iterator))
def test_write_object_with_mapping(self):
iterator = self.serializer.write_object({}, 0, set())
self.assertEqual("{}", "".join(iterator))
iterator = self.serializer.write_object({"foo": "bar"}, 0, set())
self.assertEqual("{\n 'foo': 'bar',\n}", "".join(iterator))
def test_write_object_with_enum(self):
iterator = self.serializer.write_object(Namespace.SOAP11, 0, set())
self.assertEqual("Namespace.SOAP11", "".join(iterator))
def test_build_imports_with_nested_types(self):
expected = "from tests.fixtures.models import Parent\n"
actual = self.serializer.build_imports({Parent.Inner})
self.assertEqual(expected, actual)
|
