1
from shutil import rmtree
3
from bzrlib.branch import Branch
4
from bzrlib.errors import NoWorkingTree, NotLocalUrl, NotBranchError
5
from bzrlib.workingtree import WorkingTree
7
from errors import (NotCheckout, UncommittedCheckout, ParentMissingRevisions,
11
def zap(path, remove_branch=False):
13
wt = WorkingTree.open(path)
14
except (NoWorkingTree, NotBranchError):
15
raise NotCheckout(path)
16
tree_base = wt.bzrdir.transport.base
18
branch_base = branch.bzrdir.transport.base
19
if tree_base == branch_base:
20
raise NotCheckout(path)
21
delta = wt.changes_from(wt.basis_tree(), want_unchanged=False)
22
if delta.has_changed():
23
raise UncommittedCheckout()
25
parent_loc = branch.get_parent()
26
if parent_loc is None:
28
parent = Branch.open(parent_loc)
29
p_ancestry = parent.repository.get_ancestry(parent.last_revision())
30
if branch.last_revision() not in p_ancestry:
31
raise ParentMissingRevisions(branch.get_parent())
34
t = branch.bzrdir.transport
35
while t.base != branch_base:
43
from unittest import makeSuite
45
from bzrlib.bzrdir import BzrDir, BzrDirMetaFormat1
46
from bzrlib.branch import BranchReferenceFormat
47
from bzrlib.tests import TestCaseInTempDir
49
class TestZap(TestCaseInTempDir):
51
def make_checkout(self):
52
wt = BzrDir.create_standalone_workingtree('source')
54
checkout = BzrDirMetaFormat1().initialize('checkout')
55
BranchReferenceFormat().initialize(checkout, wt.branch)
56
return checkout.create_workingtree()
58
def make_checkout2(self):
59
wt = self.make_checkout()
60
wt2 = wt.branch.bzrdir.sprout('source2').open_workingtree()
62
checkout = BzrDirMetaFormat1().initialize('checkout2')
63
BranchReferenceFormat().initialize(checkout, wt2.branch)
64
return checkout.create_workingtree()
66
def test_is_checkout(self):
67
self.assertRaises(NotCheckout, zap, '.')
68
wt = BzrDir.create_standalone_workingtree('.')
69
self.assertRaises(NotCheckout, zap, '.')
71
def test_zap_works(self):
73
self.assertIs(True, os.path.exists('checkout'))
75
self.assertIs(False, os.path.exists('checkout'))
76
self.assertIs(True, os.path.exists('source'))
78
def test_zap_branch(self):
80
base = WorkingTree.open('checkout').branch.base
81
self.assertIs(True, os.path.exists('checkout'))
82
self.assertRaises(NoParent, zap, 'checkout', remove_branch=True)
83
self.assertIs(True, os.path.exists('checkout'))
84
self.assertIs(True, os.path.exists('source'))
85
zap('checkout2', remove_branch=True)
86
self.assertIs(False, os.path.exists('checkout2'))
87
self.assertIs(False, os.path.exists('source2'))
89
def test_checks_modified(self):
90
checkout = self.make_checkout()
91
os.mkdir('checkout/foo')
93
self.assertRaises(UncommittedCheckout, zap, 'checkout')
94
checkout.commit('commit changes to branch')
97
return makeSuite(TestZap)
added
removed
zap.py
