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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
|
# coding: utf-8
from typing import Any
import pytest # type: ignore # NOQA
from roundtrip import YAML, round_trip, round_trip_load # type: ignore
def register_xxx(**kw: Any) -> None:
import ruyaml as yaml
class XXX(yaml.comments.CommentedMap):
@staticmethod
def yaml_dump(dumper: Any, data: Any) -> Any:
return dumper.represent_mapping('!xxx', data)
@classmethod
def yaml_load(cls, constructor: Any, node: Any) -> Any:
data = cls()
yield data
constructor.construct_mapping(node, data)
yaml.add_constructor('!xxx', XXX.yaml_load, constructor=yaml.RoundTripConstructor)
yaml.add_representer(XXX, XXX.yaml_dump, representer=yaml.RoundTripRepresenter)
class TestIndentFailures:
def test_tag(self) -> None:
round_trip(
"""\
!!python/object:__main__.Developer
name: Anthon
location: Germany
language: python
""",
)
def test_full_tag(self) -> None:
round_trip(
"""\
!!tag:yaml.org,2002:python/object:__main__.Developer
name: Anthon
location: Germany
language: python
""",
)
def test_standard_tag(self) -> None:
round_trip(
"""\
!!tag:yaml.org,2002:python/object:map
name: Anthon
location: Germany
language: python
""",
)
def test_Y1(self) -> None:
round_trip(
"""\
!yyy
name: Anthon
location: Germany
language: python
""",
)
def test_Y2(self) -> None:
round_trip(
"""\
!!yyy
name: Anthon
location: Germany
language: python
""",
)
# @pytest.mark.xfail(strict=True) # type: ignore
def test_spec_6_26_tag_shorthands(self) -> None:
from io import StringIO
from textwrap import dedent
from ruyaml import YAML
inp = dedent(
"""\
%TAG !e! tag:example.com,2000:app/
---
- !local foo
- !!str bar
- !e!tag%21 baz
""",
)
yaml = YAML()
data = yaml.load(inp)
buf = StringIO()
yaml.dump(data, buf)
print('buf:\n', buf.getvalue(), sep='')
assert buf.getvalue() == inp
class TestTagGeneral:
def test_unknow_handle(self) -> None:
from ruyaml.parser import ParserError
with pytest.raises(ParserError):
round_trip(
"""\
%TAG !x! tag:example.com,2000:app/
---
- !y!tag%21 baz
""",
)
class TestRoundTripCustom:
def test_X1(self) -> None:
register_xxx()
round_trip(
"""\
!xxx
name: Anthon
location: Germany
language: python
""",
)
@pytest.mark.xfail(strict=True) # type: ignore
def test_X_pre_tag_comment(self) -> None:
register_xxx()
round_trip(
"""\
-
# hello
!xxx
name: Anthon
location: Germany
language: python
""",
)
@pytest.mark.xfail(strict=True) # type: ignore
def test_X_post_tag_comment(self) -> None:
register_xxx()
round_trip(
"""\
- !xxx
# hello
name: Anthon
location: Germany
language: python
""",
)
def test_scalar_00(self) -> None:
# https://stackoverflow.com/a/45967047/1307905
round_trip(
"""\
Outputs:
Vpc:
Value: !Ref: vpc # first tag
Export:
Name: !Sub "${AWS::StackName}-Vpc" # second tag
""",
)
class TestIssue201:
def test_encoded_unicode_tag(self) -> None:
round_trip_load(
"""
s: !!python/%75nicode 'abc'
""",
)
class TestImplicitTaggedNodes:
def test_scalar(self) -> None:
data = round_trip(
"""\
- !SString abcdefg
- !SFloat 1.0
- !SInt 1961
- !SBool true
- !SLit |
glitter in the dark near the Tanhäuser gate
""",
)
# tagged scalers have string or string types as value
assert data[0].count('d') == 1
assert data[1].count('1') == 1
assert data[2].count('1') == 2
assert data[3].count('u') == 1
assert data[4].count('a') == 4
def test_mapping(self) -> None:
round_trip(
"""\
- !Mapping {a: 1, b: 2}
""",
)
def test_sequence(self) -> None:
yaml = YAML()
yaml.brace_single_entry_mapping_in_flow_sequence = True
yaml.mapping_value_align = True
yaml.round_trip(
"""
- !Sequence [a, {b: 1}, {c: {d: 3}}]
""",
)
def test_sequence2(self) -> None:
yaml = YAML()
yaml.mapping_value_align = True
yaml.round_trip(
"""
- !Sequence [a, b: 1, c: {d: 3}]
""",
)
|
