← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to tests/blackbox.py

  • Committer: Aaron Bentley
  • Date: 2011-06-28 07:59:00 UTC
  • Revision ID: aaron@aaronbentley.com-20110628075900-9vg08zbpwvl52hat
Fix docstring.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/blackbox.py
1
 
from bzrlib.tests.blackbox import ExternalBase
 
1
import os
 
2
import os.path
2
3
from unittest import makeSuite
3
 
import os.path
4
 
class TestBzrTools(ExternalBase):
 
4
 
 
5
from bzrlib import branch, osutils, workingtree
 
6
from bzrlib.config import LocationConfig
 
7
from bzrlib.transport import get_transport
 
8
from bzrlib.tests import (
 
9
    HardlinkFeature,
 
10
    ModuleAvailableFeature,
 
11
    TestCaseWithTransport,
 
12
    )
 
13
from bzrlib.plugins.bzrtools import command
 
14
 
 
15
 
 
16
LzmaFeature = ModuleAvailableFeature("lzma")
 
17
 
 
18
 
 
19
class TestBzrTools(TestCaseWithTransport):
 
20
 
 
21
    def setUp(self):
 
22
        TestCaseWithTransport.setUp(self)
 
23
        command._testing = True
 
24
        self.addCleanup(command._stop_testing)
 
25
 
5
26
    @staticmethod
6
27
    def touch(filename):
7
28
        file(filename, 'wb').write('')
8
29
 
9
 
    def test_clean_tree(self):
10
 
        self.runbzr('init')
11
 
        self.touch('name')
12
 
        self.touch('name~')
13
 
        assert os.path.lexists('name~')
14
 
        self.touch('name.pyc')
15
 
        self.runbzr('clean-tree')
16
 
        assert os.path.lexists('name~')
17
 
        assert not os.path.lexists('name')
18
 
        self.runbzr('clean-tree --detritus')
19
 
        assert not os.path.lexists('name~')
20
 
        assert os.path.lexists('name.pyc')
21
 
        self.runbzr('clean-tree --ignored')
22
 
        assert not os.path.lexists('name.pyc')
23
 
 
24
30
    def test_shelve(self):
25
 
        self.runbzr('init')
26
 
        self.runbzr('commit -m uc --unchanged')
27
 
        self.runbzr('shelve -r 1 -m foo', retcode=0)
 
31
        self.run_bzr('init')
 
32
        self.run_bzr('commit -m uc --unchanged')
 
33
        self.run_bzr('shelve1 -r 1 -m foo --all', retcode=3)
 
34
        file('foo', 'wb').write('foo')
 
35
        self.run_bzr('add foo')
 
36
        self.run_bzr('commit -m foo')
 
37
        self.run_bzr('shelve1 -r 1 -m foo --all', retcode=0)
28
38
 
29
39
    def test_fetch_ghosts(self):
30
 
        self.runbzr('init')
31
 
        self.runbzr('fetch-ghosts .')
 
40
        self.run_bzr('init')
 
41
        self.run_bzr('fetch-ghosts .')
 
42
 
 
43
    def test_fetch_ghosts_with_saved(self):
 
44
        wt = self.make_branch_and_tree('.')
 
45
        wt.branch.set_parent('.')
 
46
        self.run_bzr('fetch-ghosts')
32
47
 
33
48
    def test_patch(self):
34
 
        self.runbzr('init')
 
49
        self.run_bzr('init')
35
50
        file('myfile', 'wb').write('hello')
36
 
        self.runbzr('add')
37
 
        self.runbzr('commit -m hello')
 
51
        self.run_bzr('add')
 
52
        self.run_bzr('commit -m hello')
38
53
        file('myfile', 'wb').write('goodbye')
39
 
        file('mypatch', 'wb').write(self.runbzr('diff', retcode=1, backtick=1))
40
 
        self.runbzr('revert')
 
54
        file('mypatch', 'wb').write(self.run_bzr('diff', retcode=1)[0])
 
55
        self.run_bzr('revert')
41
56
        assert file('myfile', 'rb').read() == 'hello'
42
 
        self.runbzr('patch mypatch')
 
57
        self.run_bzr('patch --silent mypatch')
43
58
        assert file('myfile', 'rb').read() == 'goodbye'
44
59
 
45
60
    def test_branch_history(self):
46
 
        self.runbzr('init')
 
61
        self.run_bzr('init')
47
62
        file('myfile', 'wb').write('hello')
48
 
        self.runbzr('add')
49
 
        self.runbzr('commit -m hello')
50
 
        self.runbzr('branch-history')
 
63
        self.run_bzr('add')
 
64
        self.run_bzr('commit -m hello')
 
65
        self.run_bzr('branch-history')
51
66
 
52
67
    def test_branch_history(self):
53
 
        self.runbzr('init')
54
 
        file('myfile', 'wb').write('hello')
55
 
        self.runbzr('add')
56
 
        self.runbzr('commit -m hello')
57
 
        self.runbzr('graph-ancestry . graph.dot')
58
 
        self.runbzr('branch . my_branch')
59
 
        self.runbzr('graph-ancestry . graph.dot --merge-branch my_branch')
 
68
        self.run_bzr('init')
 
69
        file('myfile', 'wb').write('hello')
 
70
        self.run_bzr('add')
 
71
        self.run_bzr('commit -m hello')
 
72
        self.run_bzr('graph-ancestry . graph.dot')
 
73
        self.run_bzr('branch . my_branch')
 
74
        self.run_bzr('graph-ancestry . graph.dot --merge-branch my_branch')
 
75
 
 
76
    def test_fetch_ghosts(self):
 
77
        self.run_bzr('init')
 
78
        file('myfile', 'wb').write('hello')
 
79
        self.run_bzr('add')
 
80
        self.run_bzr('commit -m hello')
 
81
        self.run_bzr('branch . my_branch')
 
82
        self.run_bzr('fetch-ghosts my_branch')
 
83
 
 
84
    def test_zap(self):
 
85
        self.run_bzr('init source')
 
86
        self.run_bzr('checkout --lightweight source checkout')
 
87
        self.run_bzr('zap checkout')
 
88
        self.assertIs(False, os.path.exists('checkout'))
 
89
        self.assertIs(True, os.path.exists('source'))
 
90
 
 
91
    def test_zap_modified(self):
 
92
        tree = self.make_branch_and_tree('source')
 
93
        checkout = tree.branch.create_checkout('checkout', lightweight=True)
 
94
        self.build_tree(['checkout/file'])
 
95
        checkout.add('file')
 
96
        self.run_bzr_error(('This checkout has uncommitted changes',),
 
97
                           'zap checkout')
 
98
        self.assertPathExists('checkout')
 
99
        self.run_bzr('zap checkout --force')
 
100
        self.assertPathDoesNotExist('checkout')
 
101
        self.assertPathExists('source')
 
102
 
 
103
    def test_zap_branch(self):
 
104
        self.run_bzr('init source')
 
105
        self.run_bzr('checkout --lightweight source checkout')
 
106
        self.run_bzr('zap --branch checkout', retcode=3)
 
107
        self.assertIs(True, os.path.exists('checkout'))
 
108
        self.assertIs(True, os.path.exists('source'))
 
109
        self.run_bzr('branch source source2')
 
110
        self.run_bzr('checkout --lightweight source2 checkout2')
 
111
        self.run_bzr('zap --branch checkout2')
 
112
        self.assertIs(False, os.path.exists('checkout2'))
 
113
        self.assertIs(False, os.path.exists('source2'))
 
114
 
 
115
    def test_branches(self):
 
116
        self.run_bzr('init source')
 
117
        self.run_bzr('init source/subsource')
 
118
        self.run_bzr('checkout --lightweight source checkout')
 
119
        self.run_bzr('init checkout/subcheckout')
 
120
        self.run_bzr('init checkout/.bzr/subcheckout')
 
121
        out = self.run_bzr('branches')[0]
 
122
        lines = out.split('\n')
 
123
        self.assertIs(True, 'source' in lines)
 
124
        self.assertIs(True, 'source/subsource' in lines)
 
125
        self.assertIs(True, 'checkout/subcheckout' in lines)
 
126
        self.assertIs(True, 'checkout' not in lines)
 
127
 
 
128
    def test_import_upstream(self):
 
129
        self.run_bzr('init source')
 
130
        os.mkdir('source/src')
 
131
        f = file('source/src/myfile', 'wb')
 
132
        f.write('hello?')
 
133
        f.close()
 
134
        os.chdir('source')
 
135
        self.run_bzr('add')
 
136
        self.run_bzr('commit -m hello')
 
137
        self.run_bzr('export ../source-0.1.tar.gz')
 
138
        self.run_bzr('export ../source-0.1.tar.bz2')
 
139
        self.run_bzr('export ../source-0.1')
 
140
        self.run_bzr('init ../import')
 
141
        os.chdir('../import')
 
142
        self.run_bzr('import ../source-0.1.tar.gz')
 
143
        self.assertPathExists('src/myfile')
 
144
        result = self.run_bzr('import ../source-0.1.tar.gz', retcode=3)[1]
 
145
        self.assertContainsRe(result, 'Working tree has uncommitted changes')
 
146
        self.run_bzr('commit -m commit')
 
147
        self.run_bzr('import ../source-0.1.tar.gz')
 
148
        os.chdir('..')
 
149
        self.run_bzr('init import2')
 
150
        self.run_bzr('import source-0.1.tar.gz import2')
 
151
        self.assertPathExists('import2/src/myfile')
 
152
        self.run_bzr('import source-0.1.tar.gz import3')
 
153
        self.assertPathExists('import3/src/myfile')
 
154
        self.run_bzr('import source-0.1.tar.bz2 import4')
 
155
        self.assertPathExists('import4/src/myfile')
 
156
        self.run_bzr('import source-0.1 import5')
 
157
        self.assertPathExists('import5/src/myfile')
 
158
 
 
159
    def test_import_upstream_lzma(self):
 
160
        self.requireFeature(LzmaFeature)
 
161
        self.run_bzr('init source')
 
162
        os.mkdir('source/src')
 
163
        f = file('source/src/myfile', 'wb')
 
164
        f.write('hello?')
 
165
        f.close()
 
166
        os.chdir('source')
 
167
        self.run_bzr('add')
 
168
        self.run_bzr('commit -m hello')
 
169
        self.run_bzr('export ../source-0.1.tar.lzma')
 
170
        self.run_bzr('export ../source-0.1.tar.xz')
 
171
        os.chdir('..')
 
172
        self.run_bzr('import source-0.1.tar.lzma import1')
 
173
        self.assertPathExists('import1/src/myfile')
 
174
        self.run_bzr('import source-0.1.tar.xz import2')
 
175
        self.assertPathExists('import2/src/myfile')
 
176
 
 
177
    def test_cbranch(self):
 
178
        source = self.make_branch_and_tree('source')
 
179
        config = LocationConfig(osutils.abspath('target'))
 
180
        config.set_user_option('cbranch_target', 'target_branch')
 
181
        self.run_bzr('cbranch source target')
 
182
        checkout = workingtree.WorkingTree.open('target')
 
183
        self.assertEqual(checkout.branch.base,
 
184
                         get_transport('target').base)
 
185
        self.assertEqual(checkout.branch.get_master_branch().base,
 
186
                         get_transport('target_branch').base)
 
187
        self.assertEqual(checkout.branch.get_master_branch().get_parent(),
 
188
                         get_transport('source').base)
 
189
 
 
190
    def test_cbranch_hardlink(self):
 
191
        self.requireFeature(HardlinkFeature)
 
192
        # Later formats don't support hardlinks.  Boo!
 
193
        source = self.make_branch_and_tree('source', format='1.9')
 
194
        self.build_tree(['source/file'])
 
195
        source.add('file')
 
196
        source.commit('added file')
 
197
        config = LocationConfig(osutils.abspath('target'))
 
198
        config.set_user_option('cbranch_target', 'target_branch')
 
199
        self.run_bzr('cbranch source target --lightweight')
 
200
        checkout = workingtree.WorkingTree.open('target')
 
201
        self.assertNotEqual(os.lstat(checkout.abspath('file')).st_ino,
 
202
                            os.lstat(source.abspath('file')).st_ino)
 
203
        config = LocationConfig(osutils.abspath('target2'))
 
204
        config.set_user_option('cbranch_target', 'target_branch2')
 
205
        self.run_bzr('cbranch source target2 --lightweight --hardlink')
 
206
        checkout2 = workingtree.WorkingTree.open('target2')
 
207
        self.assertEqual(os.lstat(checkout2.abspath('file')).st_ino,
 
208
                         os.lstat(source.abspath('file')).st_ino)
 
209
 
 
210
    def test_create_mirror(self):
 
211
        source = self.make_branch_and_tree('source')
 
212
        source.commit('message')
 
213
        self.run_bzr('create-mirror source target')
 
214
        target = branch.Branch.open('target')
 
215
        self.assertEqual(source.last_revision(), target.last_revision())
 
216
        self.assertEqual(source.branch.base, target.get_public_branch())
 
217
 
60
218
 
61
219
def test_suite():
62
220
    return makeSuite(TestBzrTools)