1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
# Copyright (C) 2006 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
"""Tests for lock-breaking user interface"""
import os
import bzrlib
import bzrlib.errors as errors
from bzrlib.branch import Branch
from bzrlib.bzrdir import BzrDir
from bzrlib.tests.blackbox import ExternalBase
class TestBreakLock(ExternalBase):
# General principal for break-lock: All the elements that might be locked
# by a bzr operation on PATH, are candidates that break-lock may unlock.
# so pathologically if we have a lightweight checkout A, of branch B, which
# is bound to location C, the following things should be checked for locks
# to break:
# wt = WorkingTree(A)
# wt.branch
# wt.branch.repository
# wt.branch.get_master_branch()
# wt.branch.get_master_branch().repository
# so for smoke tests all we need is a bound branch with a checkout of that
# and we can then use different urls to test individual cases, for as much
# granularity as needed.
def setUp(self):
super(TestBreakLock, self).setUp()
self.build_tree(
['master-repo/',
'master-repo/master-branch/',
'repo/',
'repo/branch/',
'checkout/'])
bzrlib.bzrdir.BzrDir.create('master-repo').create_repository()
self.master_branch = bzrlib.bzrdir.BzrDir.create_branch_convenience(
'master-repo/master-branch')
bzrlib.bzrdir.BzrDir.create('repo').create_repository()
local_branch = bzrlib.bzrdir.BzrDir.create_branch_convenience('repo/branch')
local_branch.bind(self.master_branch)
checkoutdir = bzrlib.bzrdir.BzrDir.create('checkout')
bzrlib.branch.BranchReferenceFormat().initialize(
checkoutdir, local_branch)
self.wt = checkoutdir.create_workingtree()
def test_break_lock_help(self):
out, err = self.run_bzr('break-lock', '--help')
# shouldn't fail and should not produce error output
self.assertEqual('', err)
def test_break_lock_everything_locked(self):
### if everything is locked, we should be able to unlock the lot.
# sketch of test:
# lock the lot:
self.wt.lock_write()
self.master_branch.lock_write()
# run the break-lock
# we need 5 yes's - wt, branch, repo, bound branch, bound repo.
self.run_bzr('break-lock', 'checkout', stdin="y\ny\ny\ny\ny\n")
# a new tree instance should be lockable
wt = bzrlib.workingtree.WorkingTree.open('checkout')
wt.lock_write()
wt.unlock()
# and a new instance of the master branch
mb = wt.branch.get_master_branch()
mb.lock_write()
mb.unlock()
self.assertRaises(errors.LockBroken, self.wt.unlock)
self.assertRaises(errors.LockBroken, self.master_branch.unlock)
def test_saying_no_leaves_it_locked(self):
### if 'no' is answered, objects should remain locked.
self.wt.lock_write()
self.master_branch.lock_write()
# run the break-lock
# we need 5 yes's - wt, branch, repo, bound branch, bound repo.
self.run_bzr('break-lock', 'checkout', stdin="n\nn\nn\nn\nn\n")
# a new tree instance should not be lockable
wt = bzrlib.workingtree.WorkingTree.open('checkout')
self.assertRaises(errors.LockContention, wt.lock_write)
# and a new instance of the master branch
mb = wt.branch.get_master_branch()
self.assertRaises(errors.LockContention, mb.lock_write)
# unlock our branches normally.
self.wt.unlock()
self.master_branch.unlock()
class TestBreakLockOldBranch(ExternalBase):
def test_break_lock_format_5_bzrdir(self):
# break lock on a format 5 bzrdir should just return
self.make_branch_and_tree('foo', format=bzrlib.bzrdir.BzrDirFormat5())
out, err = self.run_bzr('break-lock', 'foo')
self.assertEqual('', out)
self.assertEqual('', err)
|