← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to zap.py

  • Committer: Robert Collins
  • Date: 2006-04-24 01:45:23 UTC
  • mto: (364.1.3 bzrtools)
  • mto: This revision was merged to the branch mainline in revision 366.
  • Revision ID: robertc@robertcollins.net-20060424014523-ef18f22034f17ec2
Backout debugging tweak to fai.py

Show diffs side-by-side

added added

removed removed

Lines of Context:
zap.py
 
1
from shutil import rmtree
 
2
 
 
3
from bzrlib.errors import NoWorkingTree, NotLocalUrl, NotBranchError
 
4
from bzrlib.delta import compare_trees
 
5
from bzrlib.workingtree import WorkingTree
 
6
 
 
7
from errors import NotCheckout, UncommittedCheckout
 
8
 
 
9
 
 
10
def zap(path, remove_branch=False):
 
11
    try:
 
12
        wt = WorkingTree.open(path)
 
13
    except (NoWorkingTree, NotBranchError):
 
14
        raise NotCheckout(path)
 
15
    tree_base = wt.bzrdir.transport.base
 
16
    branch = wt.branch
 
17
    branch_base = branch.bzrdir.transport.base
 
18
    if tree_base == branch_base:
 
19
        raise NotCheckout(path)
 
20
    delta = compare_trees(wt.basis_tree(), wt, want_unchanged=False)
 
21
    if delta.has_changed():
 
22
        raise UncommittedCheckout()
 
23
    rmtree(path)
 
24
    if remove_branch:
 
25
        t = branch.bzrdir.transport
 
26
        while t.base != branch_base:
 
27
            t = t.clone('..')
 
28
        t = t.clone('..')
 
29
        t.delete_tree('')
 
30
 
 
31
 
 
32
def test_suite():
 
33
    import os
 
34
    from unittest import makeSuite
 
35
    
 
36
    from bzrlib.bzrdir import BzrDir, BzrDirMetaFormat1
 
37
    from bzrlib.branch import BranchReferenceFormat
 
38
    from bzrlib.tests import TestCaseInTempDir
 
39
 
 
40
    class TestZap(TestCaseInTempDir):
 
41
 
 
42
        def make_checkout(self):
 
43
            wt = BzrDir.create_standalone_workingtree('source')
 
44
            os.mkdir('checkout')
 
45
            checkout = BzrDirMetaFormat1().initialize('checkout')
 
46
            BranchReferenceFormat().initialize(checkout, wt.branch)
 
47
            return checkout.create_workingtree()
 
48
 
 
49
        def test_is_checkout(self):
 
50
            self.assertRaises(NotCheckout, zap, '.')
 
51
            wt = BzrDir.create_standalone_workingtree('.')
 
52
            self.assertRaises(NotCheckout, zap, '.')
 
53
 
 
54
        def test_zap_works(self):
 
55
            self.make_checkout()
 
56
            self.assertIs(True, os.path.exists('checkout'))
 
57
            zap('checkout')
 
58
            self.assertIs(False, os.path.exists('checkout'))
 
59
            self.assertIs(True, os.path.exists('source'))
 
60
 
 
61
        def test_zap_branch(self):
 
62
            self.make_checkout()
 
63
            base = WorkingTree.open('checkout').branch.base
 
64
            self.assertIs(True, os.path.exists('checkout'))
 
65
            zap('checkout', remove_branch=True)
 
66
            self.assertIs(False, os.path.exists('checkout'))
 
67
            self.assertIs(False, os.path.exists('source'))
 
68
 
 
69
        def test_checks_modified(self):
 
70
            checkout = self.make_checkout()
 
71
            os.mkdir('checkout/foo')
 
72
            checkout.add('foo')
 
73
            self.assertRaises(UncommittedCheckout, zap, 'checkout')
 
74
            checkout.commit('commit changes to branch')
 
75
            zap('checkout')
 
76
 
 
77
    return makeSuite(TestZap)