← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_lockable_files.py

  • Committer: Robert Collins
  • Date: 2006-05-04 10:36:13 UTC
  • mto: (1697.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 1701.
  • Revision ID: robertc@robertcollins.net-20060504103613-bf4d13f925aa8520
Extend LockableFiles to support break_lock() calls.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_lockable_files.py
16
16
 
17
17
from StringIO import StringIO
18
18
 
 
19
import bzrlib
19
20
from bzrlib.branch import Branch
 
21
import bzrlib.errors as errors
20
22
from bzrlib.errors import BzrBadParameterNotString, NoSuchFile, ReadOnlyError
21
23
from bzrlib.lockable_files import LockableFiles, TransportLock
22
24
from bzrlib.lockdir import LockDir
95
97
    def test__escape_empty(self):
96
98
        self.assertEqual('', self.lockable._escape(''))
97
99
 
 
100
    def test_break_lock(self):
 
101
        # some locks are not breakable
 
102
        self.lockable.lock_write()
 
103
        try:
 
104
            self.assertRaises(AssertionError, self.lockable.break_lock)
 
105
        except NotImplementedError:
 
106
            # this lock cannot be broken
 
107
            self.lockable.unlock()
 
108
            return
 
109
        l2 = self.get_lockable()
 
110
        orig_factory = bzrlib.ui.ui_factory
 
111
        # silent ui - no need for stdout
 
112
        bzrlib.ui.ui_factory = bzrlib.ui.SilentUIFactory()
 
113
        bzrlib.ui.ui_factory.stdin = StringIO("y\n")
 
114
        try:
 
115
            l2.break_lock()
 
116
        finally:
 
117
            bzrlib.ui.ui_factory = orig_factory
 
118
        try:
 
119
            l2.lock_write()
 
120
            l2.unlock()
 
121
        finally:
 
122
            self.assertRaises(errors.LockBroken, self.lockable.unlock)
 
123
            self.assertFalse(self.lockable.is_locked())
 
124
 
98
125
 
99
126
# This method of adapting tests to parameters is different to 
100
127
# the TestProviderAdapters used elsewhere, but seems simpler for this 
106
133
        super(TestLockableFiles_TransportLock, self).setUp()
107
134
        transport = get_transport('.')
108
135
        transport.mkdir('.bzr')
109
 
        sub_transport = transport.clone('.bzr')
110
 
        self.lockable = LockableFiles(sub_transport, 'my-lock', TransportLock)
 
136
        self.sub_transport = transport.clone('.bzr')
 
137
        self.lockable = self.get_lockable()
111
138
        self.lockable.create_lock()
 
139
 
 
140
    def tearDown(self):
 
141
        super(TestLockableFiles_TransportLock, self).tearDown()
 
142
        del self.sub_transport
 
143
 
 
144
    def get_lockable(self):
 
145
        return LockableFiles(self.sub_transport, 'my-lock', TransportLock)
112
146
        
113
147
 
114
148
class TestLockableFiles_LockDir(TestCaseInTempDir,
118
152
    def setUp(self):
119
153
        super(TestLockableFiles_LockDir, self).setUp()
120
154
        self.transport = get_transport('.')
121
 
        self.lockable = LockableFiles(self.transport, 'my-lock', LockDir)
 
155
        self.lockable = self.get_lockable()
122
156
        # the lock creation here sets mode - test_permissions on branch 
123
157
        # tests that implicitly, but it might be a good idea to factor 
124
158
        # out the mode checking logic and have it applied to loackable files
125
159
        # directly. RBC 20060418
126
160
        self.lockable.create_lock()
127
161
 
128
 
    def test_lock_is_lockdir(self):
129
 
        """Created instance should use a LockDir.
130
 
        
131
 
        This primarily tests the mixin adapter works properly.
132
 
        """
133
 
        ## self.assertIsInstance(self.lockable, LockableFiles)
134
 
        ## self.assertIsInstance(self.lockable._lock_strategy,
135
 
                              ## LockDirStrategy)
 
162
    def get_lockable(self):
 
163
        return LockableFiles(self.transport, 'my-lock', LockDir)
136
164
 
137
165
    def test_lock_created(self):
138
166
        self.assertTrue(self.transport.has('my-lock'))