← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_version_info.py

(jelmer) Use the absolute_import feature everywhere in bzrlib,
 and add a source test to make sure it's used everywhere. (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_version_info.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005-2011 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
23
23
 
24
24
from bzrlib import (
25
25
    errors,
26
 
    symbol_versioning,
 
26
    registry,
27
27
    tests,
28
28
    version_info_formats,
29
29
    )
53
53
 
54
54
        return wt
55
55
 
 
56
    def create_tree_with_dotted_revno(self):
 
57
        wt = self.make_branch_and_tree('branch')
 
58
        self.build_tree(['branch/a'])
 
59
        wt.add('a')
 
60
        wt.commit('a', rev_id='r1')
 
61
 
 
62
        other = wt.bzrdir.sprout('other').open_workingtree()
 
63
        self.build_tree(['other/b.a'])
 
64
        other.add(['b.a'])
 
65
        other.commit('b.a', rev_id='o2')
 
66
 
 
67
        os.chdir('branch')
 
68
        self.run_bzr('merge ../other')
 
69
        wt.commit('merge', rev_id='merge')
 
70
 
 
71
        wt.update(revision='o2')
 
72
 
 
73
        return wt
 
74
 
56
75
    def test_rio_null(self):
57
76
        wt = self.make_branch_and_tree('branch')
58
77
 
63
82
        self.assertContainsRe(val, 'build-date:')
64
83
        self.assertContainsRe(val, 'revno: 0')
65
84
 
 
85
    def test_rio_dotted_revno(self):
 
86
        wt = self.create_tree_with_dotted_revno()
 
87
 
 
88
        sio = StringIO()
 
89
        builder = RioVersionInfoBuilder(wt.branch, working_tree=wt)
 
90
        builder.generate(sio)
 
91
        val = sio.getvalue()
 
92
        self.assertContainsRe(val, 'revno: 1.1.1')
 
93
 
66
94
    def test_rio_version_text(self):
67
95
        wt = self.create_branch()
68
96
 
136
164
            return new_stanzas[0]
137
165
 
138
166
        stanza = regen()
139
 
        self.failUnless('date' in stanza)
140
 
        self.failUnless('build-date' in stanza)
 
167
        self.assertTrue('date' in stanza)
 
168
        self.assertTrue('build-date' in stanza)
141
169
        self.assertEqual(['3'], stanza.get_all('revno'))
142
170
        self.assertEqual(['r3'], stanza.get_all('revision-id'))
143
171
 
191
219
        builder.generate(sio)
192
220
        val = sio.getvalue()
193
221
        self.assertContainsRe(val, "'revision_id': None")
194
 
        self.assertContainsRe(val, "'revno': 0")
 
222
        self.assertContainsRe(val, "'revno': '0'")
 
223
        self.assertNotContainsString(val, '\n\n\n\n')
 
224
 
 
225
    def test_python_dotted_revno(self):
 
226
        wt = self.create_tree_with_dotted_revno()
 
227
 
 
228
        sio = StringIO()
 
229
        builder = PythonVersionInfoBuilder(wt.branch, working_tree=wt)
 
230
        builder.generate(sio)
 
231
        val = sio.getvalue()
 
232
        self.assertContainsRe(val, "'revno': '1.1.1'")
195
233
 
196
234
    def test_python_version(self):
197
235
        wt = self.create_branch()
222
260
            return tvi
223
261
 
224
262
        tvi = regen()
225
 
        self.assertEqual(3, tvi.version_info['revno'])
 
263
        self.assertEqual('3', tvi.version_info['revno'])
226
264
        self.assertEqual('r3', tvi.version_info['revision_id'])
227
 
        self.failUnless(tvi.version_info.has_key('date'))
 
265
        self.assertTrue(tvi.version_info.has_key('date'))
228
266
        self.assertEqual(None, tvi.version_info['clean'])
229
267
 
230
268
        tvi = regen(check_for_clean=True)
282
320
        self.assertRaises(errors.MissingTemplateVariable, 
283
321
            builder.generate, sio)
284
322
 
 
323
    def test_custom_dotted_revno(self):
 
324
        sio = StringIO()
 
325
        wt = self.create_tree_with_dotted_revno()
 
326
        builder = CustomVersionInfoBuilder(wt.branch, working_tree=wt, 
 
327
            template='{revno} revid: {revision_id}')
 
328
        builder.generate(sio)
 
329
        self.assertEquals("1.1.1 revid: o2", sio.getvalue())
 
330
 
285
331
    def test_custom_version_text(self):
286
332
        wt = self.create_branch()
287
333
 
325
371
 
326
372
    def setUp(self):
327
373
        super(TestVersionInfoFormatRegistry, self).setUp()
328
 
        registry = version_info_formats.format_registry
329
 
        self._default_key = registry._default_key
330
 
        self._dict = registry._dict.copy()
331
 
        self._help_dict = registry._help_dict.copy()
332
 
        self._info_dict = registry._info_dict.copy()
333
 
        self.addCleanup(self._cleanup)
334
 
 
335
 
    def _cleanup(self):
336
 
        # Restore the registry to pristine state after the test runs
337
 
        registry = version_info_formats.format_registry
338
 
        registry._default_key = self._default_key
339
 
        registry._dict = self._dict
340
 
        registry._help_dict = self._help_dict
341
 
        registry._info_dict = self._info_dict
 
374
        self.overrideAttr(version_info_formats,
 
375
                          'format_registry', registry.Registry())
342
376
 
343
377
    def test_register_remove(self):
344
378
        registry = version_info_formats.format_registry