|
1
|
|
|
# |
|
2
|
|
|
# Copyright (C) 2012 - 2017 Satoru SATOH <ssato @ redhat.com> |
|
3
|
|
|
# License: MIT |
|
4
|
|
|
# |
|
5
|
|
|
# pylint: disable=missing-docstring, invalid-name, protected-access |
|
6
|
|
|
from __future__ import absolute_import |
|
7
|
|
|
import os.path |
|
8
|
|
|
import unittest |
|
9
|
|
|
|
|
10
|
|
|
import anyconfig.backend.xml as TT |
|
11
|
|
|
import anyconfig.backend.tests.ini |
|
12
|
|
|
import anyconfig.tests.common |
|
13
|
|
|
import anyconfig.compat |
|
14
|
|
|
|
|
15
|
|
|
from anyconfig.tests.common import dicts_equal, to_bytes |
|
16
|
|
|
|
|
17
|
|
|
|
|
18
|
|
|
XML_W_NS_S = """ |
|
19
|
|
|
<a xmlns="http://example.com/ns/config" |
|
20
|
|
|
xmlns:val="http://example.com/ns/config/val"> |
|
21
|
|
|
<b>1</b> |
|
22
|
|
|
<val:c>C</val:c> |
|
23
|
|
|
</a> |
|
24
|
|
|
""" |
|
25
|
|
|
|
|
26
|
|
|
CNF_0 = {'config': {'@attrs': {'name': 'foo'}, |
|
27
|
|
|
'a': '0', |
|
28
|
|
|
'b': {'@attrs': {'id': 'b0'}, '@text': 'bbb'}, |
|
29
|
|
|
'c': None, |
|
30
|
|
|
'sect0': {'d': 'x, y, z'}, |
|
31
|
|
|
'list1': [{'item': '0'}, {'item': '1'}, {'item': '2'}], |
|
32
|
|
|
'list2': {'@attrs': {'id': 'list2'}, |
|
33
|
|
|
'@children': [{'item': 'i'}, |
|
34
|
|
|
{'item': 'j'}]}}} |
|
35
|
|
|
CNF_0_S = """\ |
|
36
|
|
|
<?xml version="1.0" encoding="UTF-8"?> |
|
37
|
|
|
<config name='foo'> |
|
38
|
|
|
<a>0</a> |
|
39
|
|
|
<b id="b0">bbb</b> |
|
40
|
|
|
<c/> |
|
41
|
|
|
<sect0> |
|
42
|
|
|
<d>x, y, z</d> |
|
43
|
|
|
</sect0> |
|
44
|
|
|
<list1> |
|
45
|
|
|
<item>0</item> |
|
46
|
|
|
<item>1</item> |
|
47
|
|
|
<item>2</item> |
|
48
|
|
|
</list1> |
|
49
|
|
|
<list2 id="list2"> |
|
50
|
|
|
<item>i</item> |
|
51
|
|
|
<item>j</item> |
|
52
|
|
|
</list2> |
|
53
|
|
|
</config> |
|
54
|
|
|
""" |
|
55
|
|
|
|
|
56
|
|
|
|
|
57
|
|
|
class Test_00(unittest.TestCase): |
|
58
|
|
|
|
|
59
|
|
|
def test_10__namespaces_from_file(self): |
|
60
|
|
|
ref = {"http://example.com/ns/config": '', |
|
61
|
|
|
"http://example.com/ns/config/val": "val"} |
|
62
|
|
|
xmlfile = anyconfig.compat.StringIO(XML_W_NS_S) |
|
63
|
|
|
self.assertTrue(dicts_equal(TT._namespaces_from_file(xmlfile), ref)) |
|
64
|
|
|
|
|
65
|
|
|
def test_20__process_elem_text__whitespaces(self): |
|
66
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a> </a>"), {}, {}) |
|
67
|
|
|
TT._process_elem_text(elem, dic, subdic) |
|
68
|
|
|
self.assertTrue(not dic) |
|
69
|
|
|
self.assertTrue(not subdic) |
|
70
|
|
|
|
|
71
|
|
|
def test_22__process_elem_text__wo_attrs_and_children(self): |
|
72
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a>A</a>"), {}, {}) |
|
73
|
|
|
TT._process_elem_text(elem, dic, subdic, text="#text") |
|
74
|
|
|
self.assertTrue(dicts_equal(dic, {"a": 'A'})) |
|
75
|
|
|
self.assertTrue(not subdic) |
|
76
|
|
|
|
|
77
|
|
|
def test_22__process_elem_text__wo_attrs_and_children_parse(self): |
|
78
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a>A</a>"), {}, {}) |
|
79
|
|
|
TT._process_elem_text(elem, dic, subdic, text="#text", |
|
80
|
|
|
ac_parse_value=True) |
|
81
|
|
|
self.assertTrue(dicts_equal(dic, {"a": 'A'})) |
|
82
|
|
|
self.assertTrue(not subdic) |
|
83
|
|
|
|
|
84
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a>1</a>"), {}, {}) |
|
85
|
|
|
TT._process_elem_text(elem, dic, subdic, text="#text", |
|
86
|
|
|
ac_parse_value=True) |
|
87
|
|
|
self.assertTrue(dicts_equal(dic, {"a": 1})) |
|
88
|
|
|
self.assertTrue(not subdic) |
|
89
|
|
|
|
|
90
|
|
|
def test_24__process_elem_text__w_attrs(self): |
|
91
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a id='1'>A</a>"), {}, {}) |
|
92
|
|
|
TT._process_elem_text(elem, dic, subdic, text="#text") |
|
93
|
|
|
self.assertTrue(not dic) |
|
94
|
|
|
self.assertTrue(dicts_equal(subdic, {"#text": 'A'})) |
|
95
|
|
|
|
|
96
|
|
|
def test_24__process_elem_text__w_children(self): |
|
97
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a>A<b/></a>"), {}, {}) |
|
98
|
|
|
TT._process_elem_text(elem, dic, subdic, text="#text") |
|
99
|
|
|
self.assertTrue(not dic) |
|
100
|
|
|
self.assertTrue(dicts_equal(subdic, {"#text": 'A'})) |
|
101
|
|
|
|
|
102
|
|
|
def test_30__process_elem_attrs__wo_text_and_children(self): |
|
103
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a id='A'/>"), {}, {}) |
|
104
|
|
|
TT._process_elem_attrs(elem, dic, subdic) |
|
105
|
|
|
self.assertTrue(not dic) |
|
106
|
|
|
self.assertTrue(dicts_equal(subdic, {"@attrs": {"id": 'A'}})) |
|
107
|
|
|
|
|
108
|
|
|
def test_32__process_elem_attrs__w_text(self): |
|
109
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a id='A'>AAA</a>"), {}, {}) |
|
110
|
|
|
TT._process_elem_attrs(elem, dic, subdic) |
|
111
|
|
|
self.assertTrue(not dic) |
|
112
|
|
|
self.assertTrue(dicts_equal(subdic, {"@attrs": {"id": 'A'}})) |
|
113
|
|
|
|
|
114
|
|
|
def test_34__process_elem_attrs__merge_attrs(self): |
|
115
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a id='A'/>"), {}, {}) |
|
116
|
|
|
TT._process_elem_attrs(elem, dic, subdic, merge_attrs=True) |
|
117
|
|
|
self.assertTrue(dicts_equal(dic, {"a": {"id": 'A'}})) |
|
118
|
|
|
self.assertTrue(not subdic) |
|
119
|
|
|
|
|
120
|
|
|
def test_36__process_elem_attrs__wo_text_and_children_parse(self): |
|
121
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a id='1'/>"), {}, {}) |
|
122
|
|
|
TT._process_elem_attrs(elem, dic, subdic, ac_parse_value=True) |
|
123
|
|
|
self.assertTrue(not dic) |
|
124
|
|
|
self.assertTrue(dicts_equal(subdic, {"@attrs": {"id": 1}})) |
|
125
|
|
|
|
|
126
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a id='A'/>"), {}, {}) |
|
127
|
|
|
TT._process_elem_attrs(elem, dic, subdic, ac_parse_value=True) |
|
128
|
|
|
self.assertTrue(not dic) |
|
129
|
|
|
self.assertTrue(dicts_equal(subdic, {"@attrs": {"id": 'A'}})) |
|
130
|
|
|
|
|
131
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a id='true'/>"), {}, {}) |
|
132
|
|
|
TT._process_elem_attrs(elem, dic, subdic, ac_parse_value=True) |
|
133
|
|
|
self.assertTrue(not dic) |
|
134
|
|
|
self.assertTrue(dicts_equal(subdic, {"@attrs": {"id": True}})) |
|
135
|
|
|
|
|
136
|
|
|
def test_40__process_children_elems__root(self): |
|
137
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<list><i>A</i><i>B</i></list>"), {}, |
|
138
|
|
|
{}) |
|
139
|
|
|
TT._process_children_elems(elem, dic, subdic) |
|
140
|
|
|
self.assertTrue(dicts_equal(dic, {"list": [{"i": "A"}, {"i": "B"}]})) |
|
141
|
|
|
self.assertTrue(not subdic) |
|
142
|
|
|
|
|
143
|
|
|
def test_42__process_children_elems__w_attr(self): |
|
144
|
|
|
(elem, dic) = (TT.ET.XML("<list id='xyz'><i>A</i><i>B</i></list>"), {}) |
|
145
|
|
|
subdic = {"id": "xyz"} |
|
146
|
|
|
ref = subdic.copy() |
|
147
|
|
|
ref.update({"#children": [{"i": "A"}, {"i": "B"}]}) |
|
148
|
|
|
|
|
149
|
|
|
TT._process_children_elems(elem, dic, subdic, children="#children") |
|
150
|
|
|
self.assertTrue(not dic) |
|
151
|
|
|
self.assertTrue(dicts_equal(subdic, ref), subdic) |
|
152
|
|
|
|
|
153
|
|
|
def test_44__process_children_elems__w_children_have_unique_keys(self): |
|
154
|
|
|
(elem, dic, subdic) = (TT.ET.XML("<a><x>X</x><y>Y</y></a>"), {}, {}) |
|
155
|
|
|
TT._process_children_elems(elem, dic, subdic) |
|
156
|
|
|
self.assertTrue(dicts_equal(dic, {"a": {"x": "X", "y": "Y"}})) |
|
157
|
|
|
self.assertTrue(not subdic) |
|
158
|
|
|
|
|
159
|
|
|
def test_46__process_children_elems__w_merge_attrs(self): |
|
160
|
|
|
elem = TT.ET.XML("<a z='Z'><x>X</x><y>Y</y></a>") |
|
161
|
|
|
dic = {"a": {"@attrs": {"z": "Z"}}} |
|
162
|
|
|
subdic = dic["a"]["@attrs"] |
|
163
|
|
|
TT._process_children_elems(elem, dic, subdic, merge_attrs=True) |
|
164
|
|
|
self.assertTrue(dicts_equal(dic, {"a": {"x": "X", "y": "Y", |
|
165
|
|
|
"z": "Z"}}), dic) |
|
166
|
|
|
|
|
167
|
|
|
|
|
168
|
|
|
def _xml_to_container(snippet, **opts): |
|
169
|
|
|
return TT.elem_to_container(TT.ET.XML(snippet), to_container=dict, **opts) |
|
170
|
|
|
|
|
171
|
|
|
|
|
172
|
|
|
class Test_00_1(unittest.TestCase): |
|
173
|
|
|
|
|
174
|
|
|
def test_10_elem_to_container__None(self): |
|
175
|
|
|
self.assertEqual(TT.elem_to_container(None), dict()) |
|
176
|
|
|
|
|
177
|
|
|
def test_10_root_to_container__None(self): |
|
178
|
|
|
self.assertEqual(TT.root_to_container(None), dict()) |
|
179
|
|
|
|
|
180
|
|
|
def test_12_elem_to_container__empty(self): |
|
181
|
|
|
self.assertEqual(_xml_to_container("<a/>"), dict(a=None)) |
|
182
|
|
|
|
|
183
|
|
|
def test_20_elem_to_container__attrs(self): |
|
184
|
|
|
ref = dict(a={"@attrs": dict(x='1', y='y')}) |
|
185
|
|
|
self.assertEqual(_xml_to_container("<a x='1' y='y'/>"), ref) |
|
186
|
|
|
|
|
187
|
|
|
def test_30_elem_to_container__child(self): |
|
188
|
|
|
ref = dict(a=dict(b="b")) |
|
189
|
|
|
self.assertEqual(_xml_to_container("<a><b>b</b></a>"), ref) |
|
190
|
|
|
|
|
191
|
|
|
def test_32_elem_to_container__children__same_keys(self): |
|
192
|
|
|
ref = {'a': [{'b': '1'}, {'b': '2'}]} |
|
193
|
|
|
self.assertEqual(_xml_to_container("<a><b>1</b><b>2</b></a>"), ref) |
|
194
|
|
|
|
|
195
|
|
|
def test_34_elem_to_container__children(self): |
|
196
|
|
|
ref = {'a': {'b': 'b', 'c': 'c'}} |
|
197
|
|
|
self.assertEqual(_xml_to_container("<a><b>b</b><c>c</c></a>"), ref) |
|
198
|
|
|
|
|
199
|
|
|
def test_36_elem_to_container__children__same_keys_w_text(self): |
|
200
|
|
|
ref = {'a': {'@text': 'aaa', '@children': [{'b': '1'}, {'b': '2'}]}} |
|
201
|
|
|
self.assertEqual(_xml_to_container("<a>aaa<b>1</b><b>2</b></a>"), ref) |
|
202
|
|
|
|
|
203
|
|
|
def test_40_elem_to_container__text(self): |
|
204
|
|
|
self.assertEqual(_xml_to_container("<a>A</a>"), {'a': 'A'}) |
|
205
|
|
|
|
|
206
|
|
|
def test_42_elem_to_container__text_attrs(self): |
|
207
|
|
|
ref = dict(a={"@attrs": {'x': 'X'}, "@text": "A"}) |
|
208
|
|
|
self.assertEqual(_xml_to_container("<a x='X'>A</a>"), ref) |
|
209
|
|
|
|
|
210
|
|
|
def test_50_root_to_container__text_attrs_tags(self): |
|
211
|
|
|
ref = dict(a={"_attrs": {'x': 'X'}, "_text": "A"}) |
|
212
|
|
|
tags = dict(attrs="_attrs", text="_text") |
|
213
|
|
|
self.assertEqual(TT.root_to_container(TT.ET.XML("<a x='X'>A</a>"), |
|
214
|
|
|
dict, {}, tags=tags), |
|
215
|
|
|
ref) |
|
216
|
|
|
|
|
217
|
|
|
|
|
218
|
|
|
def tree_to_string(tree): |
|
219
|
|
|
return TT.ET.tostring(tree.getroot()) |
|
220
|
|
|
|
|
221
|
|
|
|
|
222
|
|
|
class Test_00_2(unittest.TestCase): |
|
223
|
|
|
|
|
224
|
|
|
def test_00_container_to_etree__None(self): |
|
225
|
|
|
self.assertTrue(TT.container_to_etree(None) is None) |
|
226
|
|
|
|
|
227
|
|
|
def test_10_container_to_etree__text_attrs(self): |
|
228
|
|
|
ref = to_bytes('<a x="X" y="Y">A</a>') |
|
229
|
|
|
obj = dict(a={"@attrs": {'x': 'X', 'y': 'Y'}, "@text": "A"}) |
|
230
|
|
|
res = TT.container_to_etree(obj) |
|
231
|
|
|
self.assertEqual(tree_to_string(res), ref) |
|
232
|
|
|
|
|
233
|
|
|
def test_12_container_to_etree__text_attrs_tags(self): |
|
234
|
|
|
ref = to_bytes('<a x="X" y="Y">A</a>') |
|
235
|
|
|
obj = dict(a={"_attrs": {'x': 'X', 'y': 'Y'}, "_text": "A"}) |
|
236
|
|
|
tags = dict(attrs="_attrs", text="_text") |
|
237
|
|
|
res = TT.container_to_etree(obj, tags=tags) |
|
238
|
|
|
self.assertEqual(tree_to_string(res), ref) |
|
239
|
|
|
|
|
240
|
|
|
def test_20_container_to_etree__child(self): |
|
241
|
|
|
ref = to_bytes("<a><b>b</b></a>") |
|
242
|
|
|
obj = dict(a=dict(b="b")) |
|
243
|
|
|
res = TT.container_to_etree(obj) |
|
244
|
|
|
self.assertEqual(tree_to_string(res), ref) |
|
245
|
|
|
|
|
246
|
|
|
def test_22_container_to_etree__children(self): |
|
247
|
|
|
ref = to_bytes("<a><b>b</b><c>c</c></a>") |
|
248
|
|
|
obj = {'a': {'@children': [{'b': 'b'}, {'c': 'c'}]}} |
|
249
|
|
|
res = TT.container_to_etree(obj) |
|
250
|
|
|
self.assertEqual(tree_to_string(res), ref) |
|
251
|
|
|
|
|
252
|
|
|
|
|
253
|
|
|
class Test10(anyconfig.backend.tests.ini.Test10): |
|
254
|
|
|
|
|
255
|
|
|
cnf = CNF_0 |
|
256
|
|
|
cnf_s = CNF_0_S.encode("utf-8") |
|
257
|
|
|
load_options = dump_options = dict(ac_parse_value=False) |
|
258
|
|
|
|
|
259
|
|
|
def setUp(self): |
|
260
|
|
|
self.psr = TT.Parser() |
|
261
|
|
|
|
|
262
|
|
|
|
|
263
|
|
|
class Test20(anyconfig.backend.tests.ini.Test20): |
|
264
|
|
|
|
|
265
|
|
|
psr_cls = TT.Parser |
|
266
|
|
|
cnf = CNF_0 |
|
267
|
|
|
cnf_s = CNF_0_S |
|
268
|
|
|
cnf_fn = "conf0.xml" |
|
269
|
|
|
|
|
270
|
|
|
def setUp(self): |
|
271
|
|
|
self.psr = self.psr_cls() |
|
272
|
|
|
self.workdir = anyconfig.tests.common.setup_workdir() |
|
273
|
|
|
self.cpath = os.path.join(self.workdir, self.cnf_fn) |
|
274
|
|
|
with self.psr.wopen(self.cpath) as out: |
|
275
|
|
|
if anyconfig.compat.IS_PYTHON_3: |
|
276
|
|
|
out.write(self.cnf_s.encode("utf-8")) |
|
277
|
|
|
else: |
|
278
|
|
|
out.write(self.cnf_s) |
|
279
|
|
|
|
|
280
|
|
|
def test_12_load__w_options(self): |
|
281
|
|
|
cnf = self.psr.load(self.cpath, ac_parse_value=False) |
|
282
|
|
|
self._assert_dicts_equal(cnf) |
|
283
|
|
|
|
|
284
|
|
|
def test_22_dump__w_special_option(self): |
|
285
|
|
|
self.psr.dump(self.cnf, self.cpath, ac_parse_value=False) |
|
286
|
|
|
cnf = self.psr.load(self.cpath) |
|
287
|
|
|
self._assert_dicts_equal(cnf) |
|
288
|
|
|
|
|
289
|
|
|
# vim:sw=4:ts=4:et: |
|
290
|
|
|
|