← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_rio.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-03-29 08:06:16 UTC
  • mfrom: (5120.1.1 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100329080616-84azimjwafaukcey
(igc) Fix py2exe packaging of sqlite3

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_rio.py
1
 
# Copyright (C) 2005, 2006 by Canonical Ltd
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Tests for rio serialization
18
18
 
24
24
 
25
25
import cStringIO
26
26
import os
 
27
import re
27
28
import sys
28
29
from tempfile import TemporaryFile
29
30
 
 
31
from bzrlib import (
 
32
    rio,
 
33
    )
30
34
from bzrlib.tests import TestCaseInTempDir, TestCase
31
35
from bzrlib.rio import (RioWriter, Stanza, read_stanza, read_stanzas, rio_file,
32
36
                        RioReader)
50
54
        # these aren't enforced at construction time
51
55
        ## self.assertRaises(ValueError,
52
56
        ##        Stanza, complex=42 + 3j)
53
 
        ## self.assertRaises(ValueError, 
 
57
        ## self.assertRaises(ValueError,
54
58
        ##        Stanza, several=range(10))
55
59
 
56
60
    def test_empty_value(self):
119
123
    def test_repeated_field(self):
120
124
        """Repeated field in rio"""
121
125
        s = Stanza()
122
 
        for k, v in [('a', '10'), ('b', '20'), ('a', '100'), ('b', '200'), 
 
126
        for k, v in [('a', '10'), ('b', '20'), ('a', '100'), ('b', '200'),
123
127
                     ('a', '1000'), ('b', '2000')]:
124
128
            s.add(k, v)
125
129
        s2 = read_stanza(s.to_lines())
137
141
    def test_blank_line(self):
138
142
        s = Stanza(none='', one='\n', two='\n\n')
139
143
        self.assertEqualDiff(s.to_string(), """\
140
 
none: 
141
 
one: 
 
144
none:\x20
 
145
one:\x20
142
146
\t
143
 
two: 
 
147
two:\x20
144
148
\t
145
149
\t
146
150
""")
150
154
    def test_whitespace_value(self):
151
155
        s = Stanza(space=' ', tabs='\t\t\t', combo='\n\t\t\n')
152
156
        self.assertEqualDiff(s.to_string(), """\
153
 
combo: 
 
157
combo:\x20
154
158
\t\t\t
155
159
\t
156
 
space:  
 
160
space:\x20\x20
157
161
tabs: \t\t\t
158
162
""")
159
163
        s2 = read_stanza(s.to_lines())
162
166
 
163
167
    def test_quoted(self):
164
168
        """rio quoted string cases"""
165
 
        s = Stanza(q1='"hello"', 
166
 
                   q2=' "for', 
 
169
        s = Stanza(q1='"hello"',
 
170
                   q2=' "for',
167
171
                   q3='\n\n"for"\n',
168
172
                   q4='for\n"\nfor',
169
173
                   q5='\n',
170
 
                   q6='"', 
 
174
                   q6='"',
171
175
                   q7='""',
172
176
                   q8='\\',
173
177
                   q9='\\"\\"',
183
187
        s = read_stanza([])
184
188
        self.assertEqual(s, None)
185
189
        self.assertTrue(s is None)
186
 
        
 
190
 
187
191
    def test_read_iter(self):
188
192
        """Read several stanzas from file"""
189
193
        tmpf = TemporaryFile()
200
204
        reader = read_stanzas(tmpf)
201
205
        read_iter = iter(reader)
202
206
        stuff = list(reader)
203
 
        self.assertEqual(stuff, 
 
207
        self.assertEqual(stuff,
204
208
                [ Stanza(version_header='1'),
205
209
                  Stanza(name="foo", val='123'),
206
210
                  Stanza(name="bar", val='129319'), ])
255
259
        tmpf.write('''\
256
260
s: "one"
257
261
 
258
 
s: 
 
262
s:\x20
259
263
\t"one"
260
264
\t
261
265
 
265
269
 
266
270
s: """
267
271
 
268
 
s: 
 
272
s:\x20
269
273
\t
270
274
 
271
275
s: \\
272
276
 
273
 
s: 
 
277
s:\x20
274
278
\t\\
275
279
\t\\\\
276
280
\t
320
324
        self.assertRaises(TypeError, s.add, 10, {})
321
325
 
322
326
    def test_rio_unicode(self):
323
 
        # intentionally use cStringIO which doesn't accomodate unencoded unicode objects
324
 
        sio = cStringIO.StringIO()
325
327
        uni_data = u'\N{KATAKANA LETTER O}'
326
328
        s = Stanza(foo=uni_data)
327
329
        self.assertEquals(s.get('foo'), uni_data)
331
333
        new_s = read_stanza(raw_lines)
332
334
        self.assertEquals(new_s.get('foo'), uni_data)
333
335
 
 
336
    def test_rio_to_unicode(self):
 
337
        uni_data = u'\N{KATAKANA LETTER O}'
 
338
        s = Stanza(foo=uni_data)
 
339
        unicode_str = s.to_unicode()
 
340
        self.assertEqual(u'foo: %s\n' % (uni_data,), unicode_str)
 
341
        new_s = rio.read_stanza_unicode(unicode_str.splitlines(True))
 
342
        self.assertEqual(uni_data, new_s.get('foo'))
 
343
 
 
344
    def test_nested_rio_unicode(self):
 
345
        uni_data = u'\N{KATAKANA LETTER O}'
 
346
        s = Stanza(foo=uni_data)
 
347
        parent_stanza = Stanza(child=s.to_unicode())
 
348
        raw_lines = parent_stanza.to_lines()
 
349
        self.assertEqual(['child: foo: ' + uni_data.encode('utf-8') + '\n',
 
350
                          '\t\n',
 
351
                         ], raw_lines)
 
352
        new_parent = read_stanza(raw_lines)
 
353
        child_text = new_parent.get('child')
 
354
        self.assertEqual(u'foo: %s\n' % uni_data, child_text)
 
355
        new_child = rio.read_stanza_unicode(child_text.splitlines(True))
 
356
        self.assertEqual(uni_data, new_child.get('foo'))
 
357
 
 
358
    def mail_munge(self, lines, dos_nl=True):
 
359
        new_lines = []
 
360
        for line in lines:
 
361
            line = re.sub(' *\n', '\n', line)
 
362
            if dos_nl:
 
363
                line = re.sub('([^\r])\n', '\\1\r\n', line)
 
364
            new_lines.append(line)
 
365
        return new_lines
 
366
 
 
367
    def test_patch_rio(self):
 
368
        stanza = Stanza(data='#\n\r\\r ', space=' ' * 255, hash='#' * 255)
 
369
        lines = rio.to_patch_lines(stanza)
 
370
        for line in lines:
 
371
            self.assertContainsRe(line, '^# ')
 
372
            self.assertTrue(72 >= len(line))
 
373
        for line in rio.to_patch_lines(stanza, max_width=12):
 
374
            self.assertTrue(12 >= len(line))
 
375
        new_stanza = rio.read_patch_stanza(self.mail_munge(lines,
 
376
                                                           dos_nl=False))
 
377
        lines = self.mail_munge(lines)
 
378
        new_stanza = rio.read_patch_stanza(lines)
 
379
        self.assertEqual('#\n\r\\r ', new_stanza.get('data'))
 
380
        self.assertEqual(' '* 255, new_stanza.get('space'))
 
381
        self.assertEqual('#'* 255, new_stanza.get('hash'))
 
382
 
 
383
    def test_patch_rio_linebreaks(self):
 
384
        stanza = Stanza(breaktest='linebreak -/'*30)
 
385
        self.assertContainsRe(rio.to_patch_lines(stanza, 71)[0],
 
386
                              'linebreak\\\\\n')
 
387
        stanza = Stanza(breaktest='linebreak-/'*30)
 
388
        self.assertContainsRe(rio.to_patch_lines(stanza, 70)[0],
 
389
                              'linebreak-\\\\\n')
 
390
        stanza = Stanza(breaktest='linebreak/'*30)
 
391
        self.assertContainsRe(rio.to_patch_lines(stanza, 70)[0],
 
392
                              'linebreak\\\\\n')