~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_workingtree.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2006-07-06 03:15:29 UTC
  • mfrom: (1711.2.78 jam-integration)
  • Revision ID: pqm@pqm.ubuntu.com-20060706031529-e189d8c3f42076be
(jam) allow plugins to include benchmarks

Show diffs side-by-side

added added

removed removed

Lines of Context:
18
18
from cStringIO import StringIO
19
19
import os
20
20
 
21
 
from bzrlib import (
22
 
    bzrdir,
23
 
    conflicts,
24
 
    errors,
25
 
    workingtree,
26
 
    )
 
21
import bzrlib
27
22
from bzrlib.branch import Branch
 
23
from bzrlib import bzrdir, conflicts, errors, workingtree
28
24
from bzrlib.bzrdir import BzrDir
 
25
from bzrlib.errors import NotBranchError, NotVersionedError
29
26
from bzrlib.lockdir import LockDir
30
 
from bzrlib.mutabletree import needs_tree_write_lock
31
 
from bzrlib.symbol_versioning import zero_thirteen
32
 
from bzrlib.tests import TestCase, TestCaseWithTransport, TestSkipped
 
27
from bzrlib.osutils import pathjoin, getcwd, has_symlinks
 
28
from bzrlib.tests import TestCaseWithTransport, TestSkipped
 
29
from bzrlib.trace import mutter
33
30
from bzrlib.transport import get_transport
34
 
from bzrlib.workingtree import (
35
 
    TreeEntry,
36
 
    TreeDirectory,
37
 
    TreeFile,
38
 
    TreeLink,
39
 
    )
40
 
 
 
31
from bzrlib.workingtree import (TreeEntry, TreeDirectory, TreeFile, TreeLink,
 
32
                                WorkingTree)
41
33
 
42
34
class TestTreeDirectory(TestCaseWithTransport):
43
35
 
97
89
    def initialize(self, a_bzrdir, revision_id=None):
98
90
        """Sample branches cannot be created."""
99
91
        t = a_bzrdir.get_workingtree_transport(self)
100
 
        t.put_bytes('format', self.get_format_string())
 
92
        t.put('format', StringIO(self.get_format_string()))
101
93
        return 'A tree'
102
94
 
103
95
    def is_supported(self):
174
166
        t = control.get_workingtree_transport(None)
175
167
        self.assertEqualDiff('Bazaar-NG Working Tree format 3',
176
168
                             t.get('format').read())
177
 
        self.assertEqualDiff(t.get('inventory').read(), 
178
 
                              '<inventory format="5">\n'
179
 
                              '</inventory>\n',
180
 
                             )
 
169
        self.assertEqualDiff('<inventory format="5">\n'
 
170
                             '</inventory>\n',
 
171
                             t.get('inventory').read())
181
172
        self.assertEqualDiff('### bzr hashcache v5\n',
182
173
                             t.get('stat-cache').read())
183
174
        self.assertFalse(t.has('inventory.basis'))
217
208
        control.create_branch()
218
209
        tree = workingtree.WorkingTreeFormat3().initialize(control)
219
210
        tree._control_files._transport.delete("pending-merges")
220
 
        self.assertEqual([], tree.get_parent_ids())
 
211
        self.assertEqual([], tree.pending_merges())
221
212
 
222
213
 
223
214
class TestFormat2WorkingTree(TestCaseWithTransport):
225
216
 
226
217
    def create_format2_tree(self, url):
227
218
        return self.make_branch_and_tree(
228
 
            url, format=bzrdir.BzrDirFormat6())
 
219
            url, format=bzrlib.bzrdir.BzrDirFormat6())
229
220
 
230
221
    def test_conflicts(self):
231
222
        # test backwards compatability
253
244
class TestNonFormatSpecificCode(TestCaseWithTransport):
254
245
    """This class contains tests of workingtree that are not format specific."""
255
246
 
 
247
    
256
248
    def test_gen_file_id(self):
257
 
        file_id = self.applyDeprecated(zero_thirteen, workingtree.gen_file_id,
258
 
                                      'filename')
259
 
        self.assertStartsWith(file_id, 'filename-')
260
 
 
261
 
    def test_gen_root_id(self):
262
 
        file_id = self.applyDeprecated(zero_thirteen, workingtree.gen_root_id)
263
 
        self.assertStartsWith(file_id, 'tree_root-')
264
 
        
265
 
 
266
 
class InstrumentedTree(object):
267
 
    """A instrumented tree to check the needs_tree_write_lock decorator."""
268
 
 
269
 
    def __init__(self):
270
 
        self._locks = []
271
 
 
272
 
    def lock_tree_write(self):
273
 
        self._locks.append('t')
274
 
 
275
 
    @needs_tree_write_lock
276
 
    def method_with_tree_write_lock(self, *args, **kwargs):
277
 
        """A lock_tree_write decorated method that returns its arguments."""
278
 
        return args, kwargs
279
 
 
280
 
    @needs_tree_write_lock
281
 
    def method_that_raises(self):
282
 
        """This method causes an exception when called with parameters.
283
 
        
284
 
        This allows the decorator code to be checked - it should still call
285
 
        unlock.
286
 
        """
287
 
 
288
 
    def unlock(self):
289
 
        self._locks.append('u')
290
 
 
291
 
 
292
 
class TestInstrumentedTree(TestCase):
293
 
 
294
 
    def test_needs_tree_write_lock(self):
295
 
        """@needs_tree_write_lock should be semantically transparent."""
296
 
        tree = InstrumentedTree()
297
 
        self.assertEqual(
298
 
            'method_with_tree_write_lock',
299
 
            tree.method_with_tree_write_lock.__name__)
300
 
        self.assertEqual(
301
 
            "A lock_tree_write decorated method that returns its arguments.",
302
 
            tree.method_with_tree_write_lock.__doc__)
303
 
        args = (1, 2, 3)
304
 
        kwargs = {'a':'b'}
305
 
        result = tree.method_with_tree_write_lock(1,2,3, a='b')
306
 
        self.assertEqual((args, kwargs), result)
307
 
        self.assertEqual(['t', 'u'], tree._locks)
308
 
        self.assertRaises(TypeError, tree.method_that_raises, 'foo')
309
 
        self.assertEqual(['t', 'u', 't', 'u'], tree._locks)
310
 
 
311
 
 
312
 
class TestAutoResolve(TestCaseWithTransport):
313
 
 
314
 
    def test_auto_resolve(self):
315
 
        base = self.make_branch_and_tree('base')
316
 
        self.build_tree_contents([('base/hello', 'Hello')])
317
 
        base.add('hello', 'hello_id')
318
 
        base.commit('Hello')
319
 
        other = base.bzrdir.sprout('other').open_workingtree()
320
 
        self.build_tree_contents([('other/hello', 'hELLO')])
321
 
        other.commit('Case switch')
322
 
        this = base.bzrdir.sprout('this').open_workingtree()
323
 
        self.failUnlessExists('this/hello')
324
 
        self.build_tree_contents([('this/hello', 'Hello World')])
325
 
        this.commit('Add World')
326
 
        this.merge_from_branch(other.branch)
327
 
        self.assertEqual([conflicts.TextConflict('hello', None, 'hello_id')],
328
 
                         this.conflicts())
329
 
        this.auto_resolve()
330
 
        self.assertEqual([conflicts.TextConflict('hello', None, 'hello_id')],
331
 
                         this.conflicts())
332
 
        self.build_tree_contents([('this/hello', '<<<<<<<')])
333
 
        this.auto_resolve()
334
 
        self.assertEqual([conflicts.TextConflict('hello', None, 'hello_id')],
335
 
                         this.conflicts())
336
 
        self.build_tree_contents([('this/hello', '=======')])
337
 
        this.auto_resolve()
338
 
        self.assertEqual([conflicts.TextConflict('hello', None, 'hello_id')],
339
 
                         this.conflicts())
340
 
        self.build_tree_contents([('this/hello', '\n>>>>>>>')])
341
 
        remaining, resolved = this.auto_resolve()
342
 
        self.assertEqual([conflicts.TextConflict('hello', None, 'hello_id')],
343
 
                         this.conflicts())
344
 
        self.assertEqual([], resolved)
345
 
        self.build_tree_contents([('this/hello', 'hELLO wORLD')])
346
 
        remaining, resolved = this.auto_resolve()
347
 
        self.assertEqual([], this.conflicts())
348
 
        self.assertEqual([conflicts.TextConflict('hello', None, 'hello_id')],
349
 
                         resolved)
350
 
        self.failIfExists('this/hello.BASE')
351
 
 
352
 
    def test_auto_resolve_dir(self):
353
 
        tree = self.make_branch_and_tree('tree')
354
 
        self.build_tree(['tree/hello/'])
355
 
        tree.add('hello', 'hello-id')
356
 
        file_conflict = conflicts.TextConflict('file', None, 'hello-id')
357
 
        tree.set_conflicts(conflicts.ConflictList([file_conflict]))
358
 
        tree.auto_resolve()
 
249
        self.assertStartsWith(bzrlib.workingtree.gen_file_id('bar'), 'bar-')
 
250
        self.assertStartsWith(bzrlib.workingtree.gen_file_id('Mwoo oof\t m'), 'Mwoooofm-')
 
251
        self.assertStartsWith(bzrlib.workingtree.gen_file_id('..gam.py'), 'gam.py-')
 
252
        self.assertStartsWith(bzrlib.workingtree.gen_file_id('..Mwoo oof\t m'), 'Mwoooofm-')
 
253
 
 
254
    def test_next_id_suffix(self):
 
255
        bzrlib.workingtree._gen_id_suffix = None
 
256
        bzrlib.workingtree._next_id_suffix()
 
257
        self.assertNotEqual(None, bzrlib.workingtree._gen_id_suffix)
 
258
        bzrlib.workingtree._gen_id_suffix = "foo-"
 
259
        bzrlib.workingtree._gen_id_serial = 1
 
260
        self.assertEqual("foo-2", bzrlib.workingtree._next_id_suffix())
 
261
        self.assertEqual("foo-3", bzrlib.workingtree._next_id_suffix())
 
262
        self.assertEqual("foo-4", bzrlib.workingtree._next_id_suffix())
 
263
        self.assertEqual("foo-5", bzrlib.workingtree._next_id_suffix())
 
264
        self.assertEqual("foo-6", bzrlib.workingtree._next_id_suffix())
 
265
        self.assertEqual("foo-7", bzrlib.workingtree._next_id_suffix())
 
266
        self.assertEqual("foo-8", bzrlib.workingtree._next_id_suffix())
 
267
        self.assertEqual("foo-9", bzrlib.workingtree._next_id_suffix())
 
268
        self.assertEqual("foo-10", bzrlib.workingtree._next_id_suffix())
 
269
 
 
270
    def test__translate_ignore_rule(self):
 
271
        tree = self.make_branch_and_tree('.')
 
272
        # translation should return the regex, the number of groups in it,
 
273
        # and the original rule in a tuple.
 
274
        # there are three sorts of ignore rules:
 
275
        # root only - regex is the rule itself without the leading ./
 
276
        self.assertEqual(
 
277
            "(rootdirrule$)", 
 
278
            tree._translate_ignore_rule("./rootdirrule"))
 
279
        # full path - regex is the rule itself
 
280
        self.assertEqual(
 
281
            "(path\\/to\\/file$)",
 
282
            tree._translate_ignore_rule("path/to/file"))
 
283
        # basename only rule - regex is a rule that ignores everything up
 
284
        # to the last / in the filename
 
285
        self.assertEqual(
 
286
            "((?:.*/)?(?!.*/)basenamerule$)",
 
287
            tree._translate_ignore_rule("basenamerule"))
 
288
 
 
289
    def test__combine_ignore_rules(self):
 
290
        tree = self.make_branch_and_tree('.')
 
291
        # the combined ignore regexs need the outer group indices
 
292
        # placed in a dictionary with the rules that were combined.
 
293
        # an empty set of rules
 
294
        # this is returned as a list of combined regex,rule sets, because
 
295
        # python has a limit of 100 combined regexes.
 
296
        compiled_rules = tree._combine_ignore_rules([])
 
297
        self.assertEqual([], compiled_rules)
 
298
        # one of each type of rule.
 
299
        compiled_rules = tree._combine_ignore_rules(
 
300
            ["rule1", "rule/two", "./three"])[0]
 
301
        # what type *is* the compiled regex to do an isinstance of ?
 
302
        self.assertEqual(3, compiled_rules[0].groups)
 
303
        self.assertEqual(
 
304
            {0:"rule1",1:"rule/two",2:"./three"},
 
305
            compiled_rules[1])
 
306
 
 
307
    def test__combine_ignore_rules_grouping(self):
 
308
        tree = self.make_branch_and_tree('.')
 
309
        # when there are too many rules, the output is split into groups of 100
 
310
        rules = []
 
311
        for index in range(198):
 
312
            rules.append('foo')
 
313
        self.assertEqual(2, len(tree._combine_ignore_rules(rules)))
 
314
 
 
315
    def test__get_ignore_rules_as_regex(self):
 
316
        tree = self.make_branch_and_tree('.')
 
317
        self.build_tree_contents([('.bzrignore', 'CVS\n.hg\n')])
 
318
        reference_output = tree._combine_ignore_rules(['CVS', '.hg'])[0]
 
319
        regex_rules = tree._get_ignore_rules_as_regex()[0]
 
320
        self.assertEqual(len(reference_output[1]), regex_rules[0].groups)
 
321
        self.assertEqual(reference_output[1], regex_rules[1])