1# Copyright (C) 2007-2012, 2016 Canonical Ltd
2#
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.
7#
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.
12#
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
16
17from breezy import (
18    controldir,
19    errors,
20    tests,
21    workingtree,
22    )
23from breezy.tests.script import TestCaseWithTransportAndScript
24
25
26class TestReconfigure(TestCaseWithTransportAndScript):
27
28    def test_no_type(self):
29        branch = self.make_branch('branch')
30        self.run_bzr_error(['No target configuration specified'],
31                           'reconfigure branch')
32
33    def test_branch_to_tree(self):
34        branch = self.make_branch('branch')
35        self.run_bzr('reconfigure --tree branch')
36        tree = workingtree.WorkingTree.open('branch')
37
38    def test_tree_to_branch(self):
39        tree = self.make_branch_and_tree('tree')
40        self.run_bzr('reconfigure --branch tree')
41        self.assertRaises(errors.NoWorkingTree,
42                          workingtree.WorkingTree.open, 'tree')
43
44    def test_branch_to_specified_checkout(self):
45        branch = self.make_branch('branch')
46        parent = self.make_branch('parent')
47        self.run_bzr('reconfigure branch --checkout --bind-to parent')
48
49    def test_force(self):
50        tree = self.make_branch_and_tree('tree')
51        self.build_tree(['tree/file'])
52        tree.add('file')
53        self.run_bzr_error(['Working tree ".*" has uncommitted changes'],
54                           'reconfigure --branch tree')
55        self.run_bzr('reconfigure --force --branch tree')
56
57    def test_lightweight_checkout_to_checkout(self):
58        branch = self.make_branch('branch')
59        checkout = branch.create_checkout('checkout', lightweight=True)
60        self.run_bzr('reconfigure --checkout checkout')
61
62    def test_lightweight_checkout_to_tree(self):
63        branch = self.make_branch('branch')
64        checkout = branch.create_checkout('checkout', lightweight=True)
65        self.run_bzr('reconfigure --tree checkout')
66
67    def test_no_args(self):
68        branch = self.make_branch('branch')
69        self.run_bzr_error(['No target configuration specified'],
70                           'reconfigure', working_dir='branch')
71
72    def test_checkout_to_lightweight_checkout(self):
73        branch = self.make_branch('branch')
74        checkout = branch.create_checkout('checkout')
75        self.run_bzr('reconfigure --lightweight-checkout checkout')
76
77    def test_standalone_to_use_shared(self):
78        self.build_tree(['repo/'])
79        tree = self.make_branch_and_tree('repo/tree')
80        repo = self.make_repository('repo', shared=True)
81        self.run_bzr('reconfigure --use-shared', working_dir='repo/tree')
82        tree = workingtree.WorkingTree.open('repo/tree')
83        self.assertNotEqual(tree.controldir.root_transport.base,
84                            tree.branch.repository.controldir.root_transport.base)
85
86    def test_use_shared_to_standalone(self):
87        repo = self.make_repository('repo', shared=True)
88        branch = controldir.ControlDir.create_branch_convenience('repo/tree')
89        self.assertNotEqual(branch.controldir.root_transport.base,
90                            branch.repository.controldir.root_transport.base)
91        self.run_bzr('reconfigure --standalone', working_dir='repo/tree')
92        tree = workingtree.WorkingTree.open('repo/tree')
93        self.assertEqual(tree.controldir.root_transport.base,
94                         tree.branch.repository.controldir.root_transport.base)
95
96    def test_make_with_trees(self):
97        repo = self.make_repository('repo', shared=True)
98        repo.set_make_working_trees(False)
99        self.run_bzr('reconfigure --with-trees', working_dir='repo')
100        self.assertIs(True, repo.make_working_trees())
101
102    def test_make_with_trees_already_trees(self):
103        repo = self.make_repository('repo', shared=True)
104        repo.set_make_working_trees(True)
105        self.run_bzr_error([" already creates working trees"],
106                           'reconfigure --with-trees repo')
107
108    def test_make_without_trees(self):
109        repo = self.make_repository('repo', shared=True)
110        repo.set_make_working_trees(True)
111        self.run_bzr('reconfigure --with-no-trees', working_dir='repo')
112        self.assertIs(False, repo.make_working_trees())
113
114    def test_make_without_trees_already_no_trees(self):
115        repo = self.make_repository('repo', shared=True)
116        repo.set_make_working_trees(False)
117        self.run_bzr_error([" already doesn't create working trees"],
118                           'reconfigure --with-no-trees repo')
119
120    def test_make_with_trees_nonshared_repo(self):
121        branch = self.make_branch('branch')
122        self.run_bzr_error(
123            ["Requested reconfiguration of '.*' is not supported"],
124            'reconfigure --with-trees branch')
125
126    def test_make_without_trees_leaves_tree_alone(self):
127        repo = self.make_repository('repo', shared=True)
128        branch = controldir.ControlDir.create_branch_convenience('repo/branch')
129        tree = workingtree.WorkingTree.open('repo/branch')
130        self.build_tree(['repo/branch/foo'])
131        tree.add('foo')
132        self.run_bzr('reconfigure --with-no-trees --force',
133                     working_dir='repo/branch')
134        self.assertPathExists('repo/branch/foo')
135        tree = workingtree.WorkingTree.open('repo/branch')
136
137    def test_shared_format_to_standalone(self, format=None):
138        repo = self.make_repository('repo', shared=True, format=format)
139        branch = controldir.ControlDir.create_branch_convenience('repo/tree')
140        self.assertNotEqual(branch.controldir.root_transport.base,
141                            branch.repository.controldir.root_transport.base)
142        tree = workingtree.WorkingTree.open('repo/tree')
143        self.build_tree_contents([('repo/tree/file', b'foo\n')])
144        tree.add(['file'])
145        tree.commit('added file')
146        self.run_bzr('reconfigure --standalone', working_dir='repo/tree')
147        tree = workingtree.WorkingTree.open('repo/tree')
148        self.build_tree_contents([('repo/tree/file', b'bar\n')])
149        self.check_file_contents('repo/tree/file', b'bar\n')
150        self.run_bzr('revert', working_dir='repo/tree')
151        self.check_file_contents('repo/tree/file', b'foo\n')
152        self.assertEqual(tree.controldir.root_transport.base,
153                         tree.branch.repository.controldir.root_transport.base)
154
155    def test_shared_knit_to_standalone(self):
156        self.test_shared_format_to_standalone('knit')
157
158    def test_shared_pack092_to_standalone(self):
159        self.test_shared_format_to_standalone('pack-0.92')
160
161    def test_shared_rich_root_pack_to_standalone(self):
162        self.test_shared_format_to_standalone('rich-root-pack')
163
164    def test_lightweight_format_checkout_to_tree(self, format=None):
165        branch = self.make_branch('branch', format=format)
166        checkout = branch.create_checkout('checkout', lightweight=True)
167        tree = workingtree.WorkingTree.open('checkout')
168        self.build_tree_contents([('checkout/file', b'foo\n')])
169        tree.add(['file'])
170        tree.commit('added file')
171        self.run_bzr('reconfigure --tree', working_dir='checkout')
172        tree = workingtree.WorkingTree.open('checkout')
173        self.build_tree_contents([('checkout/file', b'bar\n')])
174        self.check_file_contents('checkout/file', b'bar\n')
175        self.run_bzr('revert', working_dir='checkout')
176        self.check_file_contents('checkout/file', b'foo\n')
177
178    def test_lightweight_knit_checkout_to_tree(self):
179        self.test_lightweight_format_checkout_to_tree('knit')
180
181    def test_lightweight_pack092_checkout_to_tree(self):
182        self.test_lightweight_format_checkout_to_tree('pack-0.92')
183
184    def test_lightweight_rich_root_pack_checkout_to_tree(self):
185        self.test_lightweight_format_checkout_to_tree('rich-root-pack')
186
187    def test_branch_and_use_shared(self):
188        self.run_script("""\
189$ brz init -q branch
190$ echo foo > branch/foo
191$ brz add -q branch/foo
192$ brz commit -q -m msg branch
193$ brz init-shared-repo -q .
194$ brz reconfigure --branch --use-shared branch
195$ brz info branch
196Repository branch (format: ...)
197Location:
198  shared repository: .
199  repository branch: branch
200""")
201
202    def test_use_shared_and_branch(self):
203        self.run_script("""\
204$ brz init -q branch
205$ echo foo > branch/foo
206$ brz add -q branch/foo
207$ brz commit -q -m msg branch
208$ brz init-shared-repo -q .
209$ brz reconfigure --use-shared --branch branch
210$ brz info branch
211Repository branch (format: ...)
212Location:
213  shared repository: .
214  repository branch: branch
215""")
216
217
218class TestReconfigureStacking(tests.TestCaseWithTransport):
219
220    def test_reconfigure_stacking(self):
221        """Test a fairly realistic scenario for stacking:
222
223         * make a branch with some history
224         * branch it
225         * make the second branch stacked on the first
226         * commit in the second
227         * then make the second unstacked, so it has to fill in history from
228           the original fallback lying underneath its original content
229
230        See discussion in <https://bugs.launchpad.net/bzr/+bug/391411>
231        """
232        # there are also per_branch tests that exercise remote operation etc
233        tree_1 = self.make_branch_and_tree('b1', format='2a')
234        self.build_tree(['b1/foo'])
235        tree_1.add(['foo'])
236        tree_1.commit('add foo')
237        branch_1 = tree_1.branch
238        # now branch and commit again
239        bzrdir_2 = tree_1.controldir.sprout('b2')
240        tree_2 = bzrdir_2.open_workingtree()
241        branch_2 = tree_2.branch
242        # now reconfigure to be stacked
243        out, err = self.run_bzr('reconfigure --stacked-on b1 b2')
244        self.assertContainsRe(out, '^.*/b2/ is now stacked on ../b1\n$')
245        self.assertEqual('', err)
246        # can also give the absolute URL of the branch, and it gets stored
247        # as a relative path if possible
248        out, err = self.run_bzr('reconfigure --stacked-on %s b2'
249                                % (self.get_url('b1'),))
250        self.assertContainsRe(out, '^.*/b2/ is now stacked on ../b1\n$')
251        self.assertEqual('', err)
252        # Refresh the branch as 'reconfigure' modified it
253        branch_2 = branch_2.controldir.open_branch()
254        # It should be given a relative URL to the destination, if possible,
255        # because that's most likely to work across different transports
256        self.assertEqual('../b1', branch_2.get_stacked_on_url())
257        # commit, and it should be stored into b2's repo
258        self.build_tree_contents([('foo', b'new foo')])
259        tree_2.commit('update foo')
260        # Now turn it off again
261        out, err = self.run_bzr('reconfigure --unstacked b2')
262        self.assertContainsRe(out,
263                              '^.*/b2/ is now not stacked\n$')
264        self.assertEqual('', err)
265        # Refresh the branch as 'reconfigure' modified it
266        branch_2 = branch_2.controldir.open_branch()
267        self.assertRaises(errors.NotStacked, branch_2.get_stacked_on_url)
268
269    # XXX: Needs a test for reconfiguring stacking and shape at the same time;
270    # no branch at location; stacked-on is not a branch; quiet mode.
271    # -- mbp 20090706
272