/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_switch.py

  • Committer: Robert Collins
  • Date: 2008-02-06 04:06:42 UTC
  • mfrom: (3216 +trunk)
  • mto: This revision was merged to the branch mainline in revision 3217.
  • Revision ID: robertc@robertcollins.net-20080206040642-2efx3l4iv5f95lxp
Merge up with bzr.dev.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_switch.py
24
24
 
25
25
class TestSwitch(tests.TestCaseWithTransport):
26
26
 
 
27
    def setUp(self):
 
28
        super(TestSwitch, self).setUp()
 
29
        self.lightweight = True
 
30
 
27
31
    def _setup_tree(self):
28
32
        tree = self.make_branch_and_tree('branch-1')
29
33
        self.build_tree(['branch-1/file-1'])
39
43
        tree.add('file-2')
40
44
        tree.remove('file-1')
41
45
        tree.commit('rev2')
42
 
        checkout = tree.branch.create_checkout('checkout', lightweight=True)
 
46
        checkout = tree.branch.create_checkout('checkout',
 
47
            lightweight=self.lightweight)
43
48
        self.build_tree(['checkout/file-3'])
44
49
        checkout.add('file-3')
45
50
        self.failIfExists('checkout/file-1')
52
57
    def test_switch_after_branch_moved(self):
53
58
        """Test switch after the branch is moved."""
54
59
        tree = self._setup_tree()
55
 
        checkout = tree.branch.create_checkout('checkout', lightweight=True)
 
60
        checkout = tree.branch.create_checkout('checkout',
 
61
            lightweight=self.lightweight)
56
62
        self.build_tree(['branch-1/file-2'])
57
63
        tree.add('file-2')
58
64
        tree.remove('file-1')
59
65
        tree.commit('rev2')
 
66
        self.build_tree(['checkout/file-3'])
 
67
        checkout.add('file-3')
 
68
        checkout_dir = checkout.bzrdir
 
69
        # rename the branch on disk, the checkout object is now invalid.
60
70
        os.rename('branch-1', 'branch-2')
61
71
        to_branch = branch.Branch.open('branch-2')
62
 
        self.build_tree(['checkout/file-3'])
63
 
        checkout.add('file-3')
64
 
        switch.switch(checkout.bzrdir, to_branch)
 
72
        # Check fails without --force
 
73
        err = self.assertRaises((errors.NotBranchError,
 
74
            errors.BoundBranchConnectionFailure),
 
75
            switch.switch, checkout.bzrdir, to_branch)
 
76
        switch.switch(checkout.bzrdir, to_branch, force=True)
65
77
        self.failIfExists('checkout/file-1')
66
78
        self.failUnlessExists('checkout/file-2')
67
79
        self.failUnlessExists('checkout/file-3')
68
80
 
69
 
    def test_switch_on_heavy_checkout(self):
70
 
        """Test graceful failure on heavyweight checkouts."""
71
 
        tree = self._setup_tree()
72
 
        checkout = tree.branch.create_checkout('checkout-1', lightweight=False)
73
 
        branch2 = self.make_branch('branch-2')
74
 
        err = self.assertRaises(errors.BzrCommandError,
75
 
            switch.switch, checkout.bzrdir, branch2)
76
 
        self.assertContainsRe(str(err),
77
 
            "The switch command can only be used on a lightweight checkout")
78
 
 
79
81
    def test_switch_when_pending_merges(self):
80
82
        """Test graceful failure if pending merges are outstanding."""
81
83
        # Create 2 branches and a checkout
82
84
        tree = self._setup_tree()
83
85
        tree2 = tree.bzrdir.sprout('branch-2').open_workingtree()
84
 
        checkout = tree.branch.create_checkout('checkout', lightweight=True)
 
86
        checkout = tree.branch.create_checkout('checkout',
 
87
            lightweight=self.lightweight)
85
88
        # Change tree2 and merge it into the checkout without committing
86
89
        self.build_tree(['branch-2/file-2'])
87
90
        tree2.add('file-2')
92
95
            switch.switch, checkout.bzrdir, tree2.branch)
93
96
        self.assertContainsRe(str(err),
94
97
            "Pending merges must be committed or reverted before using switch")
 
98
 
 
99
 
 
100
class TestSwitchHeavyweight(TestSwitch):
 
101
 
 
102
    def setUp(self):
 
103
        super(TestSwitchHeavyweight, self).setUp()
 
104
        self.lightweight = False
 
105
 
 
106
    def test_switch_with_local_commits(self):
 
107
        """Test switch complains about local commits unless --force given."""
 
108
        tree = self._setup_tree()
 
109
        to_branch = tree.bzrdir.sprout('branch-2').open_branch()
 
110
        self.build_tree(['branch-1/file-2'])
 
111
        tree.add('file-2')
 
112
        tree.remove('file-1')
 
113
        tree.commit('rev2')
 
114
        checkout = tree.branch.create_checkout('checkout')
 
115
        self.build_tree(['checkout/file-3'])
 
116
        checkout.add('file-3')
 
117
        checkout.commit(message='local only commit', local=True)
 
118
        self.build_tree(['checkout/file-4'])
 
119
        # Check the error reporting is as expected
 
120
        err = self.assertRaises(errors.BzrCommandError,
 
121
            switch.switch, checkout.bzrdir, to_branch)
 
122
        self.assertContainsRe(str(err),
 
123
            'Cannot switch as local commits found in the checkout.')
 
124
        # Check all is ok when force is given
 
125
        self.failIfExists('checkout/file-1')
 
126
        self.failUnlessExists('checkout/file-2')
 
127
        switch.switch(checkout.bzrdir, to_branch, force=True)
 
128
        self.failUnlessExists('checkout/file-1')
 
129
        self.failIfExists('checkout/file-2')
 
130
        self.failIfExists('checkout/file-3')
 
131
        self.failUnlessExists('checkout/file-4')
 
132
        # Check that the checkout is a true mirror of the bound branch
 
133
        missing_in_checkout = checkout.branch.missing_revisions(to_branch)
 
134
        self.assertEqual([], missing_in_checkout)
 
135
        missing_in_remote = to_branch.missing_revisions(checkout.branch)
 
136
        self.assertEqual([], missing_in_remote)