/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 breezy/tests/blackbox/test_status.py

  • Committer: Jelmer Vernooij
  • Date: 2017-08-07 11:49:46 UTC
  • mto: (6747.3.4 avoid-set-revid-3)
  • mto: This revision was merged to the branch mainline in revision 6750.
  • Revision ID: jelmer@jelmer.uk-20170807114946-luclmxuawyzhpiot
Avoid setting revision_ids.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005-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
 
 
17
"""Tests of status command.
 
18
 
 
19
Most of these depend on the particular formatting used.
 
20
As such they really are blackbox tests even though some of the
 
21
tests are not using self.capture. If we add tests for the programmatic
 
22
interface later, they will be non blackbox tests.
 
23
"""
 
24
 
 
25
import codecs
 
26
from os import mkdir, chdir, rmdir, unlink
 
27
import sys
 
28
 
 
29
from ... import (
 
30
    conflicts,
 
31
    errors,
 
32
    osutils,
 
33
    status,
 
34
    )
 
35
from breezy.bzr import (
 
36
    bzrdir,
 
37
    )
 
38
import breezy.branch
 
39
from ...osutils import pathjoin
 
40
from ...revisionspec import RevisionSpec
 
41
from ...sixish import (
 
42
    BytesIO,
 
43
    )
 
44
from ...status import show_tree_status
 
45
from .. import TestCaseWithTransport, TestSkipped
 
46
from ...workingtree import WorkingTree
 
47
 
 
48
 
 
49
class BranchStatus(TestCaseWithTransport):
 
50
 
 
51
    def setUp(self):
 
52
        super(BranchStatus, self).setUp()
 
53
        # As TestCase.setUp clears all hooks, we install this default
 
54
        # post_status hook handler for the test.
 
55
        status.hooks.install_named_hook('post_status',
 
56
            status._show_shelve_summary,
 
57
            'brz status')
 
58
 
 
59
    def assertStatus(self, expected_lines, working_tree, specific_files=None,
 
60
        revision=None, short=False, pending=True, verbose=False):
 
61
        """Run status in working_tree and look for output.
 
62
 
 
63
        :param expected_lines: The lines to look for.
 
64
        :param working_tree: The tree to run status in.
 
65
        """
 
66
        output_string = self.status_string(working_tree, specific_files, revision, short,
 
67
                pending, verbose)
 
68
        self.assertEqual(expected_lines, output_string.splitlines(True))
 
69
 
 
70
    def status_string(self, wt, specific_files=None, revision=None,
 
71
        short=False, pending=True, verbose=False):
 
72
        uio = self.make_utf8_encoded_stringio()
 
73
        show_tree_status(wt, specific_files=specific_files, to_file=uio,
 
74
                revision=revision, short=short, show_pending=pending,
 
75
                verbose=verbose)
 
76
        return uio.getvalue().decode('utf-8')
 
77
 
 
78
    def test_branch_status(self):
 
79
        """Test basic branch status"""
 
80
        wt = self.make_branch_and_tree('.')
 
81
 
 
82
        # status with no commits or files - it must
 
83
        # work and show no output. We do this with no
 
84
        # commits to be sure that it's not going to fail
 
85
        # as a corner case.
 
86
        self.assertStatus([], wt)
 
87
 
 
88
        self.build_tree(['hello.c', 'bye.c'])
 
89
        self.assertStatus([
 
90
                'unknown:\n',
 
91
                '  bye.c\n',
 
92
                '  hello.c\n',
 
93
            ],
 
94
            wt)
 
95
        self.assertStatus([
 
96
                '?   bye.c\n',
 
97
                '?   hello.c\n',
 
98
            ],
 
99
            wt, short=True)
 
100
 
 
101
        # add a commit to allow showing pending merges.
 
102
        wt.commit('create a parent to allow testing merge output')
 
103
 
 
104
        wt.add_parent_tree_id('pending@pending-0-0')
 
105
        self.assertStatus([
 
106
                'unknown:\n',
 
107
                '  bye.c\n',
 
108
                '  hello.c\n',
 
109
                'pending merge tips: (use -v to see all merge revisions)\n',
 
110
                '  (ghost) pending@pending-0-0\n',
 
111
            ],
 
112
            wt)
 
113
        self.assertStatus([
 
114
                'unknown:\n',
 
115
                '  bye.c\n',
 
116
                '  hello.c\n',
 
117
                'pending merges:\n',
 
118
                '  (ghost) pending@pending-0-0\n',
 
119
            ],
 
120
            wt, verbose=True)
 
121
        self.assertStatus([
 
122
                '?   bye.c\n',
 
123
                '?   hello.c\n',
 
124
                'P   (ghost) pending@pending-0-0\n',
 
125
            ],
 
126
            wt, short=True)
 
127
        self.assertStatus([
 
128
                'unknown:\n',
 
129
                '  bye.c\n',
 
130
                '  hello.c\n',
 
131
            ],
 
132
            wt, pending=False)
 
133
        self.assertStatus([
 
134
                '?   bye.c\n',
 
135
                '?   hello.c\n',
 
136
            ],
 
137
            wt, short=True, pending=False)
 
138
 
 
139
    def test_branch_status_revisions(self):
 
140
        """Tests branch status with revisions"""
 
141
        wt = self.make_branch_and_tree('.')
 
142
 
 
143
        self.build_tree(['hello.c', 'bye.c'])
 
144
        wt.add('hello.c')
 
145
        wt.add('bye.c')
 
146
        wt.commit('Test message')
 
147
 
 
148
        revs = [RevisionSpec.from_string('0')]
 
149
        self.assertStatus([
 
150
                'added:\n',
 
151
                '  bye.c\n',
 
152
                '  hello.c\n'
 
153
            ],
 
154
            wt,
 
155
            revision=revs)
 
156
 
 
157
        self.build_tree(['more.c'])
 
158
        wt.add('more.c')
 
159
        wt.commit('Another test message')
 
160
 
 
161
        revs.append(RevisionSpec.from_string('1'))
 
162
        self.assertStatus([
 
163
                'added:\n',
 
164
                '  bye.c\n',
 
165
                '  hello.c\n',
 
166
            ],
 
167
            wt,
 
168
            revision=revs)
 
169
 
 
170
    def test_pending(self):
 
171
        """Pending merges display works, including Unicode"""
 
172
        mkdir("./branch")
 
173
        wt = self.make_branch_and_tree('branch')
 
174
        b = wt.branch
 
175
        wt.commit("Empty commit 1")
 
176
        b_2_dir = b.controldir.sprout('./copy')
 
177
        b_2 = b_2_dir.open_branch()
 
178
        wt2 = b_2_dir.open_workingtree()
 
179
        wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
 
180
        wt2.merge_from_branch(wt.branch)
 
181
        message = self.status_string(wt2, verbose=True)
 
182
        self.assertStartsWith(message, "pending merges:\n")
 
183
        self.assertEndsWith(message, "Empty commit 2\n")
 
184
        wt2.commit("merged")
 
185
        # must be long to make sure we see elipsis at the end
 
186
        wt.commit("Empty commit 3 " +
 
187
                   "blah blah blah blah " * 100)
 
188
        wt2.merge_from_branch(wt.branch)
 
189
        message = self.status_string(wt2, verbose=True)
 
190
        self.assertStartsWith(message, "pending merges:\n")
 
191
        self.assertTrue("Empty commit 3" in message)
 
192
        self.assertEndsWith(message, "...\n")
 
193
 
 
194
    def test_tree_status_ignores(self):
 
195
        """Tests branch status with ignores"""
 
196
        wt = self.make_branch_and_tree('.')
 
197
        self.run_bzr('ignore *~')
 
198
        wt.commit('commit .bzrignore')
 
199
        self.build_tree(['foo.c', 'foo.c~'])
 
200
        self.assertStatus([
 
201
                'unknown:\n',
 
202
                '  foo.c\n',
 
203
                ],
 
204
                wt)
 
205
        self.assertStatus([
 
206
                '?   foo.c\n',
 
207
                ],
 
208
                wt, short=True)
 
209
 
 
210
    def test_tree_status_specific_files(self):
 
211
        """Tests branch status with given specific files"""
 
212
        wt = self.make_branch_and_tree('.')
 
213
        b = wt.branch
 
214
 
 
215
        self.build_tree(['directory/','directory/hello.c',
 
216
                         'bye.c','test.c','dir2/',
 
217
                         'missing.c'])
 
218
        wt.add('directory')
 
219
        wt.add('test.c')
 
220
        wt.commit('testing')
 
221
        wt.add('missing.c')
 
222
        unlink('missing.c')
 
223
 
 
224
        self.assertStatus([
 
225
                'missing:\n',
 
226
                '  missing.c\n',
 
227
                'unknown:\n',
 
228
                '  bye.c\n',
 
229
                '  dir2/\n',
 
230
                '  directory/hello.c\n'
 
231
                ],
 
232
                wt)
 
233
 
 
234
        self.assertStatus([
 
235
                '?   bye.c\n',
 
236
                '?   dir2/\n',
 
237
                '+!  missing.c\n',
 
238
                '?   directory/hello.c\n'
 
239
                ],
 
240
                wt, short=True)
 
241
 
 
242
        tof = BytesIO()
 
243
        self.assertRaises(errors.PathsDoNotExist,
 
244
                          show_tree_status,
 
245
                          wt, specific_files=['bye.c','test.c','absent.c'],
 
246
                          to_file=tof)
 
247
 
 
248
        tof = BytesIO()
 
249
        show_tree_status(wt, specific_files=['directory'], to_file=tof)
 
250
        tof.seek(0)
 
251
        self.assertEqual(tof.readlines(),
 
252
                          ['unknown:\n',
 
253
                           '  directory/hello.c\n'
 
254
                           ])
 
255
        tof = BytesIO()
 
256
        show_tree_status(wt, specific_files=['directory'], to_file=tof,
 
257
                         short=True)
 
258
        tof.seek(0)
 
259
        self.assertEqual(tof.readlines(), ['?   directory/hello.c\n'])
 
260
 
 
261
        tof = BytesIO()
 
262
        show_tree_status(wt, specific_files=['dir2'], to_file=tof)
 
263
        tof.seek(0)
 
264
        self.assertEqual(tof.readlines(),
 
265
                          ['unknown:\n',
 
266
                           '  dir2/\n'
 
267
                           ])
 
268
        tof = BytesIO()
 
269
        show_tree_status(wt, specific_files=['dir2'], to_file=tof, short=True)
 
270
        tof.seek(0)
 
271
        self.assertEqual(tof.readlines(), ['?   dir2/\n'])
 
272
 
 
273
        tof = BytesIO()
 
274
        revs = [RevisionSpec.from_string('0'), RevisionSpec.from_string('1')]
 
275
        show_tree_status(wt, specific_files=['test.c'], to_file=tof,
 
276
                         short=True, revision=revs)
 
277
        tof.seek(0)
 
278
        self.assertEqual(tof.readlines(), ['+N  test.c\n'])
 
279
 
 
280
        tof = BytesIO()
 
281
        show_tree_status(wt, specific_files=['missing.c'], to_file=tof)
 
282
        tof.seek(0)
 
283
        self.assertEqual(tof.readlines(),
 
284
                          ['missing:\n',
 
285
                           '  missing.c\n'])
 
286
 
 
287
        tof = BytesIO()
 
288
        show_tree_status(wt, specific_files=['missing.c'], to_file=tof,
 
289
                         short=True)
 
290
        tof.seek(0)
 
291
        self.assertEqual(tof.readlines(),
 
292
                          ['+!  missing.c\n'])
 
293
 
 
294
    def test_specific_files_conflicts(self):
 
295
        tree = self.make_branch_and_tree('.')
 
296
        self.build_tree(['dir2/'])
 
297
        tree.add('dir2')
 
298
        tree.commit('added dir2')
 
299
        tree.set_conflicts(conflicts.ConflictList(
 
300
            [conflicts.ContentsConflict('foo')]))
 
301
        tof = BytesIO()
 
302
        show_tree_status(tree, specific_files=['dir2'], to_file=tof)
 
303
        self.assertEqualDiff('', tof.getvalue())
 
304
        tree.set_conflicts(conflicts.ConflictList(
 
305
            [conflicts.ContentsConflict('dir2')]))
 
306
        tof = BytesIO()
 
307
        show_tree_status(tree, specific_files=['dir2'], to_file=tof)
 
308
        self.assertEqualDiff('conflicts:\n  Contents conflict in dir2\n',
 
309
                             tof.getvalue())
 
310
 
 
311
        tree.set_conflicts(conflicts.ConflictList(
 
312
            [conflicts.ContentsConflict('dir2/file1')]))
 
313
        tof = BytesIO()
 
314
        show_tree_status(tree, specific_files=['dir2'], to_file=tof)
 
315
        self.assertEqualDiff('conflicts:\n  Contents conflict in dir2/file1\n',
 
316
                             tof.getvalue())
 
317
 
 
318
    def _prepare_nonexistent(self):
 
319
        wt = self.make_branch_and_tree('.')
 
320
        self.assertStatus([], wt)
 
321
        self.build_tree(['FILE_A', 'FILE_B', 'FILE_C', 'FILE_D', 'FILE_E', ])
 
322
        wt.add('FILE_A')
 
323
        wt.add('FILE_B')
 
324
        wt.add('FILE_C')
 
325
        wt.add('FILE_D')
 
326
        wt.add('FILE_E')
 
327
        wt.commit('Create five empty files.')
 
328
        with open('FILE_B', 'w') as f: f.write('Modification to file FILE_B.')
 
329
        with open('FILE_C', 'w') as f: f.write('Modification to file FILE_C.')
 
330
        unlink('FILE_E')  # FILE_E will be versioned but missing
 
331
        with open('FILE_Q', 'w') as f: f.write('FILE_Q is added but not committed.')
 
332
        wt.add('FILE_Q')  # FILE_Q will be added but not committed
 
333
        open('UNVERSIONED_BUT_EXISTING', 'w')
 
334
        return wt
 
335
 
 
336
    def test_status_nonexistent_file(self):
 
337
        # files that don't exist in either the basis tree or working tree
 
338
        # should give an error
 
339
        wt = self._prepare_nonexistent()
 
340
        self.assertStatus([
 
341
            'removed:\n',
 
342
            '  FILE_E\n',
 
343
            'added:\n',
 
344
            '  FILE_Q\n',
 
345
            'modified:\n',
 
346
            '  FILE_B\n',
 
347
            '  FILE_C\n',
 
348
            'unknown:\n',
 
349
            '  UNVERSIONED_BUT_EXISTING\n',
 
350
            ],
 
351
            wt)
 
352
        self.assertStatus([
 
353
            ' M  FILE_B\n',
 
354
            ' M  FILE_C\n',
 
355
            ' D  FILE_E\n',
 
356
            '+N  FILE_Q\n',
 
357
            '?   UNVERSIONED_BUT_EXISTING\n',
 
358
            ],
 
359
            wt, short=True)
 
360
 
 
361
        # Okay, everything's looking good with the existent files.
 
362
        # Let's see what happens when we throw in non-existent files.
 
363
 
 
364
        # brz st [--short] NONEXISTENT '
 
365
        expected = [
 
366
          'nonexistent:\n',
 
367
          '  NONEXISTENT\n',
 
368
          ]
 
369
        out, err = self.run_bzr('status NONEXISTENT', retcode=3)
 
370
        self.assertEqual(expected, out.splitlines(True))
 
371
        self.assertContainsRe(err,
 
372
                              r'.*ERROR: Path\(s\) do not exist: '
 
373
                              'NONEXISTENT.*')
 
374
        expected = [
 
375
          'X:   NONEXISTENT\n',
 
376
          ]
 
377
        out, err = self.run_bzr('status --short NONEXISTENT', retcode=3)
 
378
        self.assertContainsRe(err,
 
379
                              r'.*ERROR: Path\(s\) do not exist: '
 
380
                              'NONEXISTENT.*')
 
381
 
 
382
    def test_status_nonexistent_file_with_others(self):
 
383
        # brz st [--short] NONEXISTENT ...others..
 
384
        wt = self._prepare_nonexistent()
 
385
        expected = [
 
386
          'removed:\n',
 
387
          '  FILE_E\n',
 
388
          'modified:\n',
 
389
          '  FILE_B\n',
 
390
          '  FILE_C\n',
 
391
          'nonexistent:\n',
 
392
          '  NONEXISTENT\n',
 
393
          ]
 
394
        out, err = self.run_bzr('status NONEXISTENT '
 
395
                                'FILE_A FILE_B FILE_C FILE_D FILE_E',
 
396
                                retcode=3)
 
397
        self.assertEqual(expected, out.splitlines(True))
 
398
        self.assertContainsRe(err,
 
399
                              r'.*ERROR: Path\(s\) do not exist: '
 
400
                              'NONEXISTENT.*')
 
401
        expected = [
 
402
          ' D  FILE_E\n',
 
403
          ' M  FILE_C\n',
 
404
          ' M  FILE_B\n',
 
405
          'X   NONEXISTENT\n',
 
406
          ]
 
407
        out, err = self.run_bzr('status --short NONEXISTENT '
 
408
                                'FILE_A FILE_B FILE_C FILE_D FILE_E',
 
409
                                retcode=3)
 
410
        self.assertEqual(expected, out.splitlines(True))
 
411
        self.assertContainsRe(err,
 
412
                              r'.*ERROR: Path\(s\) do not exist: '
 
413
                              'NONEXISTENT.*')
 
414
 
 
415
    def test_status_multiple_nonexistent_files(self):
 
416
        # brz st [--short] NONEXISTENT ... ANOTHER_NONEXISTENT ...
 
417
        wt = self._prepare_nonexistent()
 
418
        expected = [
 
419
          'removed:\n',
 
420
          '  FILE_E\n',
 
421
          'modified:\n',
 
422
          '  FILE_B\n',
 
423
          '  FILE_C\n',
 
424
          'nonexistent:\n',
 
425
          '  ANOTHER_NONEXISTENT\n',
 
426
          '  NONEXISTENT\n',
 
427
          ]
 
428
        out, err = self.run_bzr('status NONEXISTENT '
 
429
                                'FILE_A FILE_B ANOTHER_NONEXISTENT '
 
430
                                'FILE_C FILE_D FILE_E', retcode=3)
 
431
        self.assertEqual(expected, out.splitlines(True))
 
432
        self.assertContainsRe(err,
 
433
                              r'.*ERROR: Path\(s\) do not exist: '
 
434
                              'ANOTHER_NONEXISTENT NONEXISTENT.*')
 
435
        expected = [
 
436
          ' D  FILE_E\n',
 
437
          ' M  FILE_C\n',
 
438
          ' M  FILE_B\n',
 
439
          'X   ANOTHER_NONEXISTENT\n',
 
440
          'X   NONEXISTENT\n',
 
441
          ]
 
442
        out, err = self.run_bzr('status --short NONEXISTENT '
 
443
                                'FILE_A FILE_B ANOTHER_NONEXISTENT '
 
444
                                'FILE_C FILE_D FILE_E', retcode=3)
 
445
        self.assertEqual(expected, out.splitlines(True))
 
446
        self.assertContainsRe(err,
 
447
                              r'.*ERROR: Path\(s\) do not exist: '
 
448
                              'ANOTHER_NONEXISTENT NONEXISTENT.*')
 
449
 
 
450
    def test_status_nonexistent_file_with_unversioned(self):
 
451
        # brz st [--short] NONEXISTENT A B UNVERSIONED_BUT_EXISTING C D E Q
 
452
        wt = self._prepare_nonexistent()
 
453
        expected = [
 
454
          'removed:\n',
 
455
          '  FILE_E\n',
 
456
          'added:\n',
 
457
          '  FILE_Q\n',
 
458
          'modified:\n',
 
459
          '  FILE_B\n',
 
460
          '  FILE_C\n',
 
461
          'unknown:\n',
 
462
          '  UNVERSIONED_BUT_EXISTING\n',
 
463
          'nonexistent:\n',
 
464
          '  NONEXISTENT\n',
 
465
          ]
 
466
        out, err = self.run_bzr('status NONEXISTENT '
 
467
                                'FILE_A FILE_B UNVERSIONED_BUT_EXISTING '
 
468
                                'FILE_C FILE_D FILE_E FILE_Q', retcode=3)
 
469
        self.assertEqual(expected, out.splitlines(True))
 
470
        self.assertContainsRe(err,
 
471
                              r'.*ERROR: Path\(s\) do not exist: '
 
472
                              'NONEXISTENT.*')
 
473
        expected = sorted([
 
474
          '+N  FILE_Q\n',
 
475
          '?   UNVERSIONED_BUT_EXISTING\n',
 
476
          ' D  FILE_E\n',
 
477
          ' M  FILE_C\n',
 
478
          ' M  FILE_B\n',
 
479
          'X   NONEXISTENT\n',
 
480
          ])
 
481
        out, err = self.run_bzr('status --short NONEXISTENT '
 
482
                                'FILE_A FILE_B UNVERSIONED_BUT_EXISTING '
 
483
                                'FILE_C FILE_D FILE_E FILE_Q', retcode=3)
 
484
        actual = out.splitlines(True)
 
485
        actual.sort()
 
486
        self.assertEqual(expected, actual)
 
487
        self.assertContainsRe(err,
 
488
                              r'.*ERROR: Path\(s\) do not exist: '
 
489
                              'NONEXISTENT.*')
 
490
 
 
491
    def test_status_out_of_date(self):
 
492
        """Simulate status of out-of-date tree after remote push"""
 
493
        tree = self.make_branch_and_tree('.')
 
494
        self.build_tree_contents([('a', 'foo\n')])
 
495
        tree.lock_write()
 
496
        try:
 
497
            tree.add(['a'])
 
498
            tree.commit('add test file')
 
499
            # simulate what happens after a remote push
 
500
            tree.set_last_revision("0")
 
501
        finally:
 
502
            # before run another commands we should unlock tree
 
503
            tree.unlock()
 
504
        out, err = self.run_bzr('status')
 
505
        self.assertEqual("working tree is out of date, run 'brz update'\n",
 
506
                         err)
 
507
 
 
508
    def test_status_on_ignored(self):
 
509
        """Tests branch status on an unversioned file which is considered ignored.
 
510
 
 
511
        See https://bugs.launchpad.net/bzr/+bug/40103
 
512
        """
 
513
        tree = self.make_branch_and_tree('.')
 
514
 
 
515
        self.build_tree(['test1.c', 'test1.c~', 'test2.c~'])
 
516
        result = self.run_bzr('status')[0]
 
517
        self.assertContainsRe(result, "unknown:\n  test1.c\n")
 
518
        short_result = self.run_bzr('status --short')[0]
 
519
        self.assertContainsRe(short_result, "\?   test1.c\n")
 
520
 
 
521
        result = self.run_bzr('status test1.c')[0]
 
522
        self.assertContainsRe(result, "unknown:\n  test1.c\n")
 
523
        short_result = self.run_bzr('status --short test1.c')[0]
 
524
        self.assertContainsRe(short_result, "\?   test1.c\n")
 
525
 
 
526
        result = self.run_bzr('status test1.c~')[0]
 
527
        self.assertContainsRe(result, "ignored:\n  test1.c~\n")
 
528
        short_result = self.run_bzr('status --short test1.c~')[0]
 
529
        self.assertContainsRe(short_result, "I   test1.c~\n")
 
530
 
 
531
        result = self.run_bzr('status test1.c~ test2.c~')[0]
 
532
        self.assertContainsRe(result, "ignored:\n  test1.c~\n  test2.c~\n")
 
533
        short_result = self.run_bzr('status --short test1.c~ test2.c~')[0]
 
534
        self.assertContainsRe(short_result, "I   test1.c~\nI   test2.c~\n")
 
535
 
 
536
        result = self.run_bzr('status test1.c test1.c~ test2.c~')[0]
 
537
        self.assertContainsRe(result, "unknown:\n  test1.c\nignored:\n  test1.c~\n  test2.c~\n")
 
538
        short_result = self.run_bzr('status --short test1.c test1.c~ test2.c~')[0]
 
539
        self.assertContainsRe(short_result, "\?   test1.c\nI   test1.c~\nI   test2.c~\n")
 
540
 
 
541
    def test_status_write_lock(self):
 
542
        """Test that status works without fetching history and
 
543
        having a write lock.
 
544
 
 
545
        See https://bugs.launchpad.net/bzr/+bug/149270
 
546
        """
 
547
        mkdir('branch1')
 
548
        wt = self.make_branch_and_tree('branch1')
 
549
        b = wt.branch
 
550
        wt.commit('Empty commit 1')
 
551
        wt2 = b.controldir.sprout('branch2').open_workingtree()
 
552
        wt2.commit('Empty commit 2')
 
553
        out, err = self.run_bzr('status branch1 -rbranch:branch2')
 
554
        self.assertEqual('', out)
 
555
 
 
556
    def test_status_with_shelves(self):
 
557
        """Ensure that _show_shelve_summary handler works.
 
558
        """
 
559
        wt = self.make_branch_and_tree('.')
 
560
        self.build_tree(['hello.c'])
 
561
        wt.add('hello.c')
 
562
        self.run_bzr(['shelve', '--all', '-m', 'foo'])
 
563
        self.build_tree(['bye.c'])
 
564
        wt.add('bye.c')
 
565
        self.assertStatus([
 
566
                'added:\n',
 
567
                '  bye.c\n',
 
568
                '1 shelf exists. See "brz shelve --list" for details.\n',
 
569
            ],
 
570
            wt)
 
571
        self.run_bzr(['shelve', '--all', '-m', 'bar'])
 
572
        self.build_tree(['eggs.c', 'spam.c'])
 
573
        wt.add('eggs.c')
 
574
        wt.add('spam.c')
 
575
        self.assertStatus([
 
576
                'added:\n',
 
577
                '  eggs.c\n',
 
578
                '  spam.c\n',
 
579
                '2 shelves exist. See "brz shelve --list" for details.\n',
 
580
            ],
 
581
            wt)
 
582
        self.assertStatus([
 
583
                'added:\n',
 
584
                '  spam.c\n',
 
585
            ],
 
586
            wt,
 
587
            specific_files=['spam.c'])
 
588
 
 
589
 
 
590
class CheckoutStatus(BranchStatus):
 
591
 
 
592
    def setUp(self):
 
593
        super(CheckoutStatus, self).setUp()
 
594
        mkdir('codir')
 
595
        chdir('codir')
 
596
 
 
597
    def make_branch_and_tree(self, relpath):
 
598
        source = self.make_branch(pathjoin('..', relpath))
 
599
        checkout = bzrdir.BzrDirMetaFormat1().initialize(relpath)
 
600
        checkout.set_branch_reference(source)
 
601
        return checkout.create_workingtree()
 
602
 
 
603
 
 
604
class TestStatus(TestCaseWithTransport):
 
605
 
 
606
    def test_status_plain(self):
 
607
        tree = self.make_branch_and_tree('.')
 
608
 
 
609
        self.build_tree(['hello.txt'])
 
610
        result = self.run_bzr("status")[0]
 
611
        self.assertContainsRe(result, "unknown:\n  hello.txt\n")
 
612
 
 
613
        tree.add("hello.txt")
 
614
        result = self.run_bzr("status")[0]
 
615
        self.assertContainsRe(result, "added:\n  hello.txt\n")
 
616
 
 
617
        tree.commit(message="added")
 
618
        result = self.run_bzr("status -r 0..1")[0]
 
619
        self.assertContainsRe(result, "added:\n  hello.txt\n")
 
620
 
 
621
        result = self.run_bzr("status -c 1")[0]
 
622
        self.assertContainsRe(result, "added:\n  hello.txt\n")
 
623
 
 
624
        self.build_tree(['world.txt'])
 
625
        result = self.run_bzr("status -r 0")[0]
 
626
        self.assertContainsRe(result, "added:\n  hello.txt\n" \
 
627
                                      "unknown:\n  world.txt\n")
 
628
        result2 = self.run_bzr("status -r 0..")[0]
 
629
        self.assertEqual(result2, result)
 
630
 
 
631
    def test_status_short(self):
 
632
        tree = self.make_branch_and_tree('.')
 
633
 
 
634
        self.build_tree(['hello.txt'])
 
635
        result = self.run_bzr("status --short")[0]
 
636
        self.assertContainsRe(result, "[?]   hello.txt\n")
 
637
 
 
638
        tree.add("hello.txt")
 
639
        result = self.run_bzr("status --short")[0]
 
640
        self.assertContainsRe(result, "[+]N  hello.txt\n")
 
641
 
 
642
        tree.commit(message="added")
 
643
        result = self.run_bzr("status --short -r 0..1")[0]
 
644
        self.assertContainsRe(result, "[+]N  hello.txt\n")
 
645
 
 
646
        self.build_tree(['world.txt'])
 
647
        result = self.run_bzr("status -S -r 0")[0]
 
648
        self.assertContainsRe(result, "[+]N  hello.txt\n" \
 
649
                                      "[?]   world.txt\n")
 
650
        result2 = self.run_bzr("status -S -r 0..")[0]
 
651
        self.assertEqual(result2, result)
 
652
 
 
653
    def test_status_versioned(self):
 
654
        tree = self.make_branch_and_tree('.')
 
655
 
 
656
        self.build_tree(['hello.txt'])
 
657
        result = self.run_bzr("status --versioned")[0]
 
658
        self.assertNotContainsRe(result, "unknown:\n  hello.txt\n")
 
659
 
 
660
        tree.add("hello.txt")
 
661
        result = self.run_bzr("status --versioned")[0]
 
662
        self.assertContainsRe(result, "added:\n  hello.txt\n")
 
663
 
 
664
        tree.commit("added")
 
665
        result = self.run_bzr("status --versioned -r 0..1")[0]
 
666
        self.assertContainsRe(result, "added:\n  hello.txt\n")
 
667
 
 
668
        self.build_tree(['world.txt'])
 
669
        result = self.run_bzr("status --versioned -r 0")[0]
 
670
        self.assertContainsRe(result, "added:\n  hello.txt\n")
 
671
        self.assertNotContainsRe(result, "unknown:\n  world.txt\n")
 
672
        result2 = self.run_bzr("status --versioned -r 0..")[0]
 
673
        self.assertEqual(result2, result)
 
674
 
 
675
    def test_status_SV(self):
 
676
        tree = self.make_branch_and_tree('.')
 
677
 
 
678
        self.build_tree(['hello.txt'])
 
679
        result = self.run_bzr("status -SV")[0]
 
680
        self.assertNotContainsRe(result, "hello.txt")
 
681
 
 
682
        tree.add("hello.txt")
 
683
        result = self.run_bzr("status -SV")[0]
 
684
        self.assertContainsRe(result, "[+]N  hello.txt\n")
 
685
 
 
686
        tree.commit(message="added")
 
687
        result = self.run_bzr("status -SV -r 0..1")[0]
 
688
        self.assertContainsRe(result, "[+]N  hello.txt\n")
 
689
 
 
690
        self.build_tree(['world.txt'])
 
691
        result = self.run_bzr("status -SV -r 0")[0]
 
692
        self.assertContainsRe(result, "[+]N  hello.txt\n")
 
693
 
 
694
        result2 = self.run_bzr("status -SV -r 0..")[0]
 
695
        self.assertEqual(result2, result)
 
696
 
 
697
    def assertStatusContains(self, pattern, short=False):
 
698
        """Run status, and assert it contains the given pattern"""
 
699
        if short:
 
700
            result = self.run_bzr("status --short")[0]
 
701
        else:
 
702
            result = self.run_bzr("status")[0]
 
703
        self.assertContainsRe(result, pattern)
 
704
 
 
705
    def test_kind_change_plain(self):
 
706
        tree = self.make_branch_and_tree('.')
 
707
        self.build_tree(['file'])
 
708
        tree.add('file')
 
709
        tree.commit('added file')
 
710
        unlink('file')
 
711
        self.build_tree(['file/'])
 
712
        self.assertStatusContains('kind changed:\n  file \(file => directory\)')
 
713
        tree.rename_one('file', 'directory')
 
714
        self.assertStatusContains('renamed:\n  file/ => directory/\n' \
 
715
                                  'modified:\n  directory/\n')
 
716
        rmdir('directory')
 
717
        self.assertStatusContains('removed:\n  file\n')
 
718
 
 
719
    def test_kind_change_short(self):
 
720
        tree = self.make_branch_and_tree('.')
 
721
        self.build_tree(['file'])
 
722
        tree.add('file')
 
723
        tree.commit('added file')
 
724
        unlink('file')
 
725
        self.build_tree(['file/'])
 
726
        self.assertStatusContains('K  file => file/',
 
727
                                   short=True)
 
728
        tree.rename_one('file', 'directory')
 
729
        self.assertStatusContains('RK  file => directory/',
 
730
                                   short=True)
 
731
        rmdir('directory')
 
732
        self.assertStatusContains('RD  file => directory',
 
733
                                   short=True)
 
734
 
 
735
    def test_status_illegal_revision_specifiers(self):
 
736
        out, err = self.run_bzr('status -r 1..23..123', retcode=3)
 
737
        self.assertContainsRe(err, 'one or two revision specifiers')
 
738
 
 
739
    def test_status_no_pending(self):
 
740
        a_tree = self.make_branch_and_tree('a')
 
741
        self.build_tree(['a/a'])
 
742
        a_tree.add('a')
 
743
        a_tree.commit('a')
 
744
        b_tree = a_tree.controldir.sprout('b').open_workingtree()
 
745
        self.build_tree(['b/b'])
 
746
        b_tree.add('b')
 
747
        b_tree.commit('b')
 
748
 
 
749
        self.run_bzr('merge ../b', working_dir='a')
 
750
        out, err = self.run_bzr('status --no-pending', working_dir='a')
 
751
        self.assertEqual(out, "added:\n  b\n")
 
752
 
 
753
    def test_pending_specific_files(self):
 
754
        """With a specific file list, pending merges are not shown."""
 
755
        tree = self.make_branch_and_tree('tree')
 
756
        self.build_tree_contents([('tree/a', 'content of a\n')])
 
757
        tree.add('a')
 
758
        r1_id = tree.commit('one')
 
759
        alt = tree.controldir.sprout('alt').open_workingtree()
 
760
        self.build_tree_contents([('alt/a', 'content of a\nfrom alt\n')])
 
761
        alt_id = alt.commit('alt')
 
762
        tree.merge_from_branch(alt.branch)
 
763
        output = self.make_utf8_encoded_stringio()
 
764
        show_tree_status(tree, to_file=output)
 
765
        self.assertContainsRe(output.getvalue(), 'pending merge')
 
766
        out, err = self.run_bzr('status tree/a')
 
767
        self.assertNotContainsRe(out, 'pending merge')
 
768
 
 
769
 
 
770
class TestStatusEncodings(TestCaseWithTransport):
 
771
 
 
772
    def make_uncommitted_tree(self):
 
773
        """Build a branch with uncommitted unicode named changes in the cwd."""
 
774
        working_tree = self.make_branch_and_tree(u'.')
 
775
        filename = u'hell\u00d8'
 
776
        try:
 
777
            self.build_tree_contents([(filename, 'contents of hello')])
 
778
        except UnicodeEncodeError:
 
779
            raise TestSkipped("can't build unicode working tree in "
 
780
                "filesystem encoding %s" % sys.getfilesystemencoding())
 
781
        working_tree.add(filename)
 
782
        return working_tree
 
783
 
 
784
    def test_stdout_ascii(self):
 
785
        self.overrideAttr(osutils, '_cached_user_encoding', 'ascii')
 
786
        working_tree = self.make_uncommitted_tree()
 
787
        stdout, stderr = self.run_bzr("status")
 
788
 
 
789
        self.assertEqual(stdout, """\
 
790
added:
 
791
  hell?
 
792
""")
 
793
 
 
794
    def test_stdout_latin1(self):
 
795
        self.overrideAttr(osutils, '_cached_user_encoding', 'latin-1')
 
796
        working_tree = self.make_uncommitted_tree()
 
797
        stdout, stderr = self.run_bzr('status')
 
798
 
 
799
        self.assertEqual(stdout, u"""\
 
800
added:
 
801
  hell\u00d8
 
802
""".encode('latin-1'))
 
803