def test_rio_to_unicode(self):
uni_data = u'\N{KATAKANA LETTER O}'
s = Stanza(foo=uni_data)
unicode_str = s.to_unicode()
self.assertEqual(u'foo: %s\n' % (uni_data, ), unicode_str)
new_s = rio.read_stanza_unicode(unicode_str.splitlines(True))
self.assertEqual(uni_data, new_s.get('foo'))
def test_read_several(self):
"""Read several stanzas from file"""
tmpf = TemporaryFile()
tmpf.write(b"""\
version_header: 1
name: foo
val: 123
name: quoted
address: "Willowglen"
\t 42 Wallaby Way
\t Sydney
name: bar
val: 129319
""")
tmpf.seek(0)
s = read_stanza(tmpf)
self.assertEqual(s, Stanza(version_header='1'))
s = read_stanza(tmpf)
self.assertEqual(s, Stanza(name="foo", val='123'))
s = read_stanza(tmpf)
self.assertEqual(s.get('name'), 'quoted')
self.assertEqual(s.get('address'),
' "Willowglen"\n 42 Wallaby Way\n Sydney')
s = read_stanza(tmpf)
self.assertEqual(s, Stanza(name="bar", val='129319'))
s = read_stanza(tmpf)
self.assertEqual(s, None)
self.check_rio_file(tmpf)
def test_rio_unicode(self):
uni_data = u'\N{KATAKANA LETTER O}'
s = Stanza(foo=uni_data)
self.assertEqual(s.get('foo'), uni_data)
raw_lines = s.to_lines()
self.assertEqual(raw_lines,
[b'foo: ' + uni_data.encode('utf-8') + b'\n'])
new_s = read_stanza(raw_lines)
self.assertEqual(new_s.get('foo'), uni_data)
def test_stanza(self):
"""Construct rio stanza in memory"""
s = Stanza(number='42', name="fred")
self.assertTrue('number' in s)
self.assertFalse('color' in s)
self.assertFalse('42' in s)
self.assertEqual(list(s.iter_pairs()), [('name', 'fred'),
('number', '42')])
self.assertEqual(s.get('number'), '42')
self.assertEqual(s.get('name'), 'fred')
def test_patch_rio_linebreaks(self):
stanza = Stanza(breaktest='linebreak -/' * 30)
self.assertContainsRe(
rio.to_patch_lines(stanza, 71)[0], b'linebreak\\\\\n')
stanza = Stanza(breaktest='linebreak-/' * 30)
self.assertContainsRe(
rio.to_patch_lines(stanza, 70)[0], b'linebreak-\\\\\n')
stanza = Stanza(breaktest='linebreak/' * 30)
self.assertContainsRe(
rio.to_patch_lines(stanza, 70)[0], b'linebreak\\\\\n')
def test_to_file(self):
"""Write rio to file"""
tmpf = TemporaryFile()
s = Stanza(a_thing='something with "quotes like \\"this\\""',
number='42', name='fred')
s.write(tmpf)
tmpf.seek(0)
self.assertEqual(tmpf.read(), b'''\
a_thing: something with "quotes like \\"this\\""
name: fred
number: 42
''')
def test_whitespace_value(self):
s = Stanza(space=' ', tabs='\t\t\t', combo='\n\t\t\n')
self.assertEqual(s.to_string(), b"""\
combo:\x20
\t\t\t
\t
space:\x20\x20
tabs: \t\t\t
""")
s2 = read_stanza(s.to_lines())
self.assertEqual(s, s2)
self.rio_file_stanzas([s])
def test_blank_line(self):
s = Stanza(none='', one='\n', two='\n\n')
self.assertEqual(s.to_string(), b"""\
none:\x20
one:\x20
\t
two:\x20
\t
\t
""")
s2 = read_stanza(s.to_lines())
self.assertEqual(s, s2)
def test_rio_surrogateescape(self):
raw_bytes = b'\xcb'
self.assertRaises(UnicodeDecodeError, raw_bytes.decode, 'utf-8')
try:
uni_data = raw_bytes.decode('utf-8', 'surrogateescape')
except LookupError:
self.skipTest('surrogateescape is not available on Python < 3')
s = Stanza(foo=uni_data)
self.assertEqual(s.get('foo'), uni_data)
raw_lines = s.to_lines()
self.assertEqual(raw_lines,
[b'foo: ' + uni_data.encode('utf-8', 'surrogateescape') + b'\n'])
new_s = read_stanza(raw_lines)
self.assertEqual(new_s.get('foo'), uni_data)
def test_multiline_string(self):
tmpf = TemporaryFile()
s = Stanza(
motto="war is peace\nfreedom is slavery\nignorance is strength")
s.write(tmpf)
tmpf.seek(0)
self.assertEqual(tmpf.read(), b'''\
motto: war is peace
\tfreedom is slavery
\tignorance is strength
''')
tmpf.seek(0)
s2 = read_stanza(tmpf)
self.assertEqual(s, s2)
def test_quoted(self):
"""rio quoted string cases"""
s = Stanza(q1='"hello"',
q2=' "for',
q3='\n\n"for"\n',
q4='for\n"\nfor',
q5='\n',
q6='"',
q7='""',
q8='\\',
q9='\\"\\"',
)
s2 = read_stanza(s.to_lines())
self.assertEqual(s, s2)
def test_read_iter(self):
"""Read several stanzas from file"""
tmpf = TemporaryFile()
tmpf.write(b"""\
version_header: 1
name: foo
val: 123
name: bar
val: 129319
""")
tmpf.seek(0)
reader = read_stanzas(tmpf)
read_iter = iter(reader)
stuff = list(reader)
self.assertEqual(stuff,
[Stanza(version_header='1'),
Stanza(name="foo", val='123'),
Stanza(name="bar", val='129319'), ])
def test_patch_rio(self):
stanza = Stanza(data='#\n\r\\r ', space=' ' * 255, hash='#' * 255)
lines = rio.to_patch_lines(stanza)
for line in lines:
self.assertContainsRe(line, b'^# ')
self.assertTrue(72 >= len(line))
for line in rio.to_patch_lines(stanza, max_width=12):
self.assertTrue(12 >= len(line))
new_stanza = rio.read_patch_stanza(self.mail_munge(lines,
dos_nl=False))
lines = self.mail_munge(lines)
new_stanza = rio.read_patch_stanza(lines)
self.assertEqual('#\n\r\\r ', new_stanza.get('data'))
self.assertEqual(' ' * 255, new_stanza.get('space'))
self.assertEqual('#' * 255, new_stanza.get('hash'))
def test_repeated_field(self):
"""Repeated field in rio"""
s = Stanza()
for k, v in [('a', '10'), ('b', '20'), ('a', '100'), ('b', '200'),
('a', '1000'), ('b', '2000')]:
s.add(k, v)
s2 = read_stanza(s.to_lines())
self.assertEqual(s, s2)
self.assertEqual(s.get_all('a'), ['10', '100', '1000'])
self.assertEqual(s.get_all('b'), ['20', '200', '2000'])
def test_nested_rio_unicode(self):
uni_data = u'\N{KATAKANA LETTER O}'
s = Stanza(foo=uni_data)
parent_stanza = Stanza(child=s.to_unicode())
raw_lines = parent_stanza.to_lines()
self.assertEqual([
b'child: foo: ' + uni_data.encode('utf-8') + b'\n',
b'\t\n',
], raw_lines)
new_parent = read_stanza(raw_lines)
child_text = new_parent.get('child')
self.assertEqual(u'foo: %s\n' % uni_data, child_text)
new_child = rio.read_stanza_unicode(child_text.splitlines(True))
self.assertEqual(uni_data, new_child.get('foo'))
def test_empty_value(self):
"""Serialize stanza with empty field"""
s = Stanza(empty='')
self.assertEquals(s.to_string(), b"empty: \n")
def generate(self, to_file):
info = Stanza()
revision_id = self._get_revision_id()
if revision_id != NULL_REVISION:
info.add('revision-id', revision_id)
rev = self._branch.repository.get_revision(revision_id)
info.add('date', create_date_str(rev.timestamp, rev.timezone))
try:
revno = self._get_revno_str(revision_id)
except errors.GhostRevisionsHaveNoRevno:
revno = None
for hook in RioVersionInfoBuilder.hooks['revision']:
hook(rev, info)
else:
revno = '0'
info.add('build-date', create_date_str())
if revno is not None:
info.add('revno', revno)
if self._branch.nick is not None:
info.add('branch-nick', self._branch.nick)
if self._check or self._include_file_revs:
self._extract_file_revisions()
if self._check:
if self._clean:
info.add('clean', 'True')
else:
info.add('clean', 'False')
if self._include_history:
log = Stanza()
for (revision_id, message, timestamp,
timezone) in self._iter_revision_history():
log.add('id', revision_id)
log.add('message', message)
log.add('date', create_date_str(timestamp, timezone))
info.add('revisions', log.to_unicode())
if self._include_file_revs:
files = Stanza()
for path in sorted(self._file_revisions.keys()):
files.add('path', path)
files.add('revision', self._file_revisions[path])
info.add('file-revisions', files.to_unicode())
to_file.write(info.to_unicode())
def test_rio_raises_type_error_key(self):
"""TypeError on adding invalid type to Stanza"""
s = Stanza()
self.assertRaises(TypeError, s.add, 10, {})
def test_to_lines(self):
"""Write simple rio stanza to string"""
s = Stanza(number='42', name='fred')
self.assertEqual(list(s.to_lines()),
[b'name: fred\n', b'number: 42\n'])
def test_backslash(self):
s = Stanza(q='\\')
t = s.to_string()
self.assertEqual(t, b'q: \\\n')
s2 = read_stanza(s.to_lines())
self.assertEqual(s, s2)
def test_as_dict(self):
"""Convert rio Stanza to dictionary"""
s = Stanza(number='42', name='fred')
sd = s.as_dict()
self.assertEqual(sd, dict(number='42', name='fred'))
def test_write_empty_stanza(self):
"""Write empty stanza"""
l = list(Stanza().to_lines())
self.assertEqual(l, [])