24
24
from StringIO import StringIO
25
25
from bzrlib.tests import TestCase, TestCaseInTempDir
26
26
from bzrlib.branch import Branch
27
from bzrlib.rio import read_stanzas
28
29
# TODO: jam 20051228 When part of bzrlib, this should become
29
30
# from bzrlib.generate_version_info import foo
146
147
self.assertContainsRe(val, 'id: r3')
147
148
self.assertContainsRe(val, 'message: a2')
150
def test_rio_version(self):
151
b, wt = self.create_branch()
155
generate_rio_version(b, to_file=sio, **kwargs)
157
stanzas = list(read_stanzas(sio))
158
self.assertEqual(1, len(stanzas))
162
self.failUnless('date' in stanza)
163
self.assertEqual(['3'], stanza.get_all('revno'))
164
self.assertEqual(['r3'], stanza.get_all('revision_id'))
166
stanza = regen(check_for_clean=True)
167
self.assertEqual(['True'], stanza.get_all('clean'))
169
open('branch/c', 'wb').write('now unclean\n')
170
stanza = regen(check_for_clean=True)
171
self.assertEqual(['False'], stanza.get_all('clean'))
172
os.remove('branch/c')
174
stanza = regen(include_revision_history=True)
175
txt = stanza['revisions']
176
revision_stanza = list(read_stanzas(StringIO(txt)))
177
self.assertEqual(1, len(revision_stanza))
178
revision_stanza = revision_stanza[0]
179
self.assertEqual(['r1', 'r2', 'r3'], revision_stanza.get_all('id'))
180
self.assertEqual(['a', 'b', 'a2'], revision_stanza.get_all('message'))
149
182
def test_python_version(self):
150
183
b, wt = self.create_branch()