1
# Copyright (C) 2006 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Tests for lock-breaking user interface"""
26
from bzrlib.branch import Branch
27
from bzrlib.bzrdir import BzrDir
28
from bzrlib.tests.blackbox import ExternalBase
31
class TestBreakLock(ExternalBase):
33
# General principal for break-lock: All the elements that might be locked
34
# by a bzr operation on PATH, are candidates that break-lock may unlock.
35
# so pathologically if we have a lightweight checkout A, of branch B, which
36
# is bound to location C, the following things should be checked for locks
40
# wt.branch.repository
41
# wt.branch.get_master_branch()
42
# wt.branch.get_master_branch().repository
43
# so for smoke tests all we need is a bound branch with a checkout of that
44
# and we can then use different urls to test individual cases, for as much
45
# granularity as needed.
48
super(TestBreakLock, self).setUp()
51
'master-repo/master-branch/',
55
bzrlib.bzrdir.BzrDir.create('master-repo').create_repository()
56
self.master_branch = bzrlib.bzrdir.BzrDir.create_branch_convenience(
57
'master-repo/master-branch')
58
bzrlib.bzrdir.BzrDir.create('repo').create_repository()
59
local_branch = bzrlib.bzrdir.BzrDir.create_branch_convenience('repo/branch')
60
local_branch.bind(self.master_branch)
61
checkoutdir = bzrlib.bzrdir.BzrDir.create('checkout')
62
bzrlib.branch.BranchReferenceFormat().initialize(
63
checkoutdir, local_branch)
64
self.wt = checkoutdir.create_workingtree()
66
def test_break_lock_help(self):
67
out, err = self.run_bzr('break-lock --help')
68
# shouldn't fail and should not produce error output
69
self.assertEqual('', err)
71
def test_break_lock_everything_locked(self):
72
### if everything is locked, we should be able to unlock the lot.
73
# however, we dont test breaking the working tree because we
74
# cannot accurately do so right now: the dirstate lock is held
75
# by an os lock, and we need to spawn a separate process to lock it
78
# lock most of the dir:
79
self.wt.branch.lock_write()
80
self.master_branch.lock_write()
82
# we need 5 yes's - wt, branch, repo, bound branch, bound repo.
83
self.run_bzr('break-lock checkout', stdin="y\ny\ny\ny\n")
84
# a new tree instance should be lockable
85
branch = bzrlib.branch.Branch.open('checkout')
88
# and a new instance of the master branch
89
mb = branch.get_master_branch()
92
self.assertRaises(errors.LockBroken, self.wt.unlock)
93
self.assertRaises(errors.LockBroken, self.master_branch.unlock)
96
class TestBreakLockOldBranch(ExternalBase):
98
def test_break_lock_format_5_bzrdir(self):
99
# break lock on a format 5 bzrdir should just return
100
self.make_branch_and_tree('foo', format=bzrlib.bzrdir.BzrDirFormat5())
101
out, err = self.run_bzr('break-lock foo')
102
self.assertEqual('', out)
103
self.assertEqual('', err)