17
17
"""Tests for _rio_*."""
25
def load_tests(standard_tests, module, loader):
25
def load_tests(loader, standard_tests, pattern):
26
26
suite, _ = tests.permute_tests_for_extension(standard_tests, loader,
27
'bzrlib._rio_py', 'bzrlib._rio_pyx')
27
'breezy._rio_py', 'breezy._rio_pyx')
31
31
class TestValidTag(tests.TestCase):
33
module = None # Filled in by test parameterization
33
module = None # Filled in by test parameterization
36
36
self.assertTrue(self.module._valid_tag("foo"))
51
51
self.assertFalse(self.module._valid_tag(""))
53
53
def test_unicode(self):
54
self.assertRaises(TypeError, self.module._valid_tag, u"foo")
54
# When str is a unicode type, it is valid for a tag
55
self.assertTrue(self.module._valid_tag(u"foo"))
56
57
def test_non_ascii_char(self):
57
58
self.assertFalse(self.module._valid_tag("\xb5"))
60
61
class TestReadUTF8Stanza(tests.TestCase):
62
module = None # Filled in by test parameterization
63
module = None # Filled in by test parameterization
64
65
def assertReadStanza(self, result, line_iter):
65
66
s = self.module._read_stanza_utf8(line_iter)
66
self.assertEquals(result, s)
67
self.assertEqual(result, s)
68
69
for tag, value in s.iter_pairs():
69
70
self.assertIsInstance(tag, str)
70
self.assertIsInstance(value, unicode)
71
self.assertIsInstance(value, str)
72
73
def assertReadStanzaRaises(self, exception, line_iter):
73
74
self.assertRaises(exception, self.module._read_stanza_utf8, line_iter)
79
80
self.assertReadStanza(None, [])
81
82
def test_none(self):
82
self.assertReadStanza(None, [""])
83
self.assertReadStanza(None, [b""])
84
85
def test_simple(self):
85
self.assertReadStanza(rio.Stanza(foo="bar"), ["foo: bar\n", ""])
86
self.assertReadStanza(rio.Stanza(foo="bar"), [b"foo: bar\n", b""])
87
88
def test_multi_line(self):
88
self.assertReadStanza(rio.Stanza(foo="bar\nbla"),
89
["foo: bar\n", "\tbla\n"])
89
self.assertReadStanza(
90
rio.Stanza(foo="bar\nbla"), [b"foo: bar\n", b"\tbla\n"])
91
92
def test_repeated(self):
93
94
s.add("foo", "bar")
94
95
s.add("foo", "foo")
95
self.assertReadStanza(s, ["foo: bar\n", "foo: foo\n"])
96
self.assertReadStanza(s, [b"foo: bar\n", b"foo: foo\n"])
97
98
def test_invalid_early_colon(self):
98
self.assertReadStanzaRaises(ValueError, ["f:oo: bar\n"])
99
self.assertReadStanzaRaises(ValueError, [b"f:oo: bar\n"])
100
101
def test_invalid_tag(self):
101
self.assertReadStanzaRaises(ValueError, ["f%oo: bar\n"])
102
self.assertReadStanzaRaises(ValueError, [b"f%oo: bar\n"])
103
104
def test_continuation_too_early(self):
104
self.assertReadStanzaRaises(ValueError, ["\tbar\n"])
105
self.assertReadStanzaRaises(ValueError, [b"\tbar\n"])
106
107
def test_large(self):
108
value = b"bla" * 9000
108
109
self.assertReadStanza(rio.Stanza(foo=value),
109
["foo: %s\n" % value])
110
[b"foo: %s\n" % value])
111
112
def test_non_ascii_char(self):
112
113
self.assertReadStanza(rio.Stanza(foo=u"n\xe5me"),
113
[u"foo: n\xe5me\n".encode("utf-8")])
114
[u"foo: n\xe5me\n".encode("utf-8")])
116
117
class TestReadUnicodeStanza(tests.TestCase):
118
module = None # Filled in by test parameterization
119
module = None # Filled in by test parameterization
120
121
def assertReadStanza(self, result, line_iter):
121
122
s = self.module._read_stanza_unicode(line_iter)
122
self.assertEquals(result, s)
123
self.assertEqual(result, s)
123
124
if s is not None:
124
125
for tag, value in s.iter_pairs():
125
126
self.assertIsInstance(tag, str)
126
self.assertIsInstance(value, unicode)
127
self.assertIsInstance(value, str)
128
129
def assertReadStanzaRaises(self, exception, line_iter):
129
130
self.assertRaises(exception, self.module._read_stanza_unicode,
142
143
self.assertReadStanza(rio.Stanza(foo="bar"), [u"foo: bar\n", u""])
144
145
def test_multi_line(self):
145
self.assertReadStanza(rio.Stanza(foo="bar\nbla"),
146
[u"foo: bar\n", u"\tbla\n"])
146
self.assertReadStanza(rio.Stanza(foo="bar\nbla"),
147
[u"foo: bar\n", u"\tbla\n"])
148
149
def test_repeated(self):