~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/lock_helpers.py

  • Committer: Jelmer Vernooij
  • Date: 2009-04-10 15:58:09 UTC
  • mto: This revision was merged to the branch mainline in revision 4284.
  • Revision ID: jelmer@samba.org-20090410155809-kdibzcjvp7pdb83f
Fix missing import.

Show diffs side-by-side

added added

removed removed

Lines of Context:
17
17
"""Helper functions/classes for testing locking"""
18
18
 
19
19
from bzrlib import errors
20
 
from bzrlib.decorators import only_raises
21
20
 
22
21
 
23
22
class TestPreventLocking(errors.LockError):
45
44
        self.__dict__['_allow_read'] = True
46
45
        self.__dict__['_allow_unlock'] = True
47
46
 
48
 
    def __eq__(self, other):
49
 
        # Branch objects look for controlfiles == repo.controlfiles.
50
 
        if type(other) is LockWrapper:
51
 
            return self._other == other._other
52
 
        return False
53
 
 
54
47
    def __getattr__(self, attr):
55
48
        return getattr(self._other, attr)
56
49
 
69
62
            return self._other.lock_write()
70
63
        raise TestPreventLocking('lock_write disabled')
71
64
 
72
 
    @only_raises(errors.LockNotHeld, errors.LockBroken)
73
65
    def unlock(self):
74
66
        self._sequence.append((self._other_id, 'ul', self._allow_unlock))
75
67
        if self._allow_unlock: