29
29
from tempfile import TemporaryFile
38
from bzrlib.osutils import pathjoin
39
from bzrlib.revisionspec import RevisionSpec
40
from bzrlib.status import show_tree_status
41
from bzrlib.tests import TestCaseWithTransport, TestSkipped
42
from bzrlib.workingtree import WorkingTree
39
from brzlib.osutils import pathjoin
40
from brzlib.revisionspec import RevisionSpec
41
from brzlib.status import show_tree_status
42
from brzlib.tests import TestCaseWithTransport, TestSkipped
43
from brzlib.workingtree import WorkingTree
45
46
class BranchStatus(TestCaseWithTransport):
47
def assertStatus(self, expected_lines, working_tree,
49
super(BranchStatus, self).setUp()
50
# As TestCase.setUp clears all hooks, we install this default
51
# post_status hook handler for the test.
52
status.hooks.install_named_hook('post_status',
53
status._show_shelve_summary,
56
def assertStatus(self, expected_lines, working_tree, specific_files=None,
48
57
revision=None, short=False, pending=True, verbose=False):
49
58
"""Run status in working_tree and look for output.
51
60
:param expected_lines: The lines to look for.
52
61
:param working_tree: The tree to run status in.
54
output_string = self.status_string(working_tree, revision, short,
63
output_string = self.status_string(working_tree, specific_files, revision, short,
56
65
self.assertEqual(expected_lines, output_string.splitlines(True))
58
def status_string(self, wt, revision=None, short=False, pending=True,
67
def status_string(self, wt, specific_files=None, revision=None,
68
short=False, pending=True, verbose=False):
60
69
# use a real file rather than StringIO because it doesn't handle
61
70
# Unicode very well.
62
71
tof = codecs.getwriter('utf-8')(TemporaryFile())
63
show_tree_status(wt, to_file=tof, revision=revision, short=short,
64
show_pending=pending, verbose=verbose)
72
show_tree_status(wt, specific_files=specific_files, to_file=tof,
73
revision=revision, short=short, show_pending=pending,
66
76
return tof.read().decode('utf-8')
178
188
wt2.merge_from_branch(wt.branch)
179
189
message = self.status_string(wt2, verbose=True)
180
190
self.assertStartsWith(message, "pending merges:\n")
181
self.assert_("Empty commit 3" in message)
191
self.assertTrue("Empty commit 3" in message)
182
192
self.assertEndsWith(message, "...\n")
184
194
def test_tree_status_ignores(self):
239
256
show_tree_status(wt, specific_files=['directory'], to_file=tof,
242
self.assertEquals(tof.readlines(), ['? directory/hello.c\n'])
259
self.assertEqual(tof.readlines(), ['? directory/hello.c\n'])
245
262
show_tree_status(wt, specific_files=['dir2'], to_file=tof)
247
self.assertEquals(tof.readlines(),
264
self.assertEqual(tof.readlines(),
252
269
show_tree_status(wt, specific_files=['dir2'], to_file=tof, short=True)
254
self.assertEquals(tof.readlines(), ['? dir2/\n'])
271
self.assertEqual(tof.readlines(), ['? dir2/\n'])
257
274
revs = [RevisionSpec.from_string('0'), RevisionSpec.from_string('1')]
258
275
show_tree_status(wt, specific_files=['test.c'], to_file=tof,
259
276
short=True, revision=revs)
261
self.assertEquals(tof.readlines(), ['+N test.c\n'])
278
self.assertEqual(tof.readlines(), ['+N test.c\n'])
281
show_tree_status(wt, specific_files=['missing.c'], to_file=tof)
283
self.assertEqual(tof.readlines(),
288
show_tree_status(wt, specific_files=['missing.c'], to_file=tof,
291
self.assertEqual(tof.readlines(),
263
294
def test_specific_files_conflicts(self):
264
295
tree = self.make_branch_and_tree('.')
296
327
wt.commit('Create five empty files.')
297
open('FILE_B', 'w').write('Modification to file FILE_B.')
298
open('FILE_C', 'w').write('Modification to file FILE_C.')
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.')
299
330
unlink('FILE_E') # FILE_E will be versioned but missing
300
open('FILE_Q', 'w').write('FILE_Q is added but not committed.')
331
with open('FILE_Q', 'w') as f: f.write('FILE_Q is added but not committed.')
301
332
wt.add('FILE_Q') # FILE_Q will be added but not committed
302
333
open('UNVERSIONED_BUT_EXISTING', 'w')
448
479
'X NONEXISTENT\n',
450
482
out, err = self.run_bzr('status --short NONEXISTENT '
451
483
'FILE_A FILE_B UNVERSIONED_BUT_EXISTING '
452
484
'FILE_C FILE_D FILE_E FILE_Q', retcode=3)
453
self.assertEqual(expected, out.splitlines(True))
485
actual = out.splitlines(True)
487
self.assertEqual(expected, actual)
454
488
self.assertContainsRe(err,
455
489
r'.*ERROR: Path\(s\) do not exist: '
520
554
out, err = self.run_bzr('status branch1 -rbranch:branch2')
521
555
self.assertEqual('', out)
557
def test_status_with_shelves(self):
558
"""Ensure that _show_shelve_summary handler works.
560
wt = self.make_branch_and_tree('.')
561
self.build_tree(['hello.c'])
563
self.run_bzr(['shelve', '--all', '-m', 'foo'])
564
self.build_tree(['bye.c'])
569
'1 shelf exists. See "bzr shelve --list" for details.\n',
572
self.run_bzr(['shelve', '--all', '-m', 'bar'])
573
self.build_tree(['eggs.c', 'spam.c'])
580
'2 shelves exist. See "bzr shelve --list" for details.\n',
588
specific_files=['spam.c'])
524
591
class CheckoutStatus(BranchStatus):
561
627
self.assertContainsRe(result, "added:\n hello.txt\n" \
562
628
"unknown:\n world.txt\n")
563
629
result2 = self.run_bzr("status -r 0..")[0]
564
self.assertEquals(result2, result)
630
self.assertEqual(result2, result)
566
632
def test_status_short(self):
567
633
tree = self.make_branch_and_tree('.')
579
645
self.assertContainsRe(result, "[+]N hello.txt\n")
581
647
self.build_tree(['world.txt'])
582
result = self.run_bzr("status --short -r 0")[0]
648
result = self.run_bzr("status -S -r 0")[0]
583
649
self.assertContainsRe(result, "[+]N hello.txt\n" \
584
650
"[?] world.txt\n")
585
result2 = self.run_bzr("status --short -r 0..")[0]
586
self.assertEquals(result2, result)
651
result2 = self.run_bzr("status -S -r 0..")[0]
652
self.assertEqual(result2, result)
588
654
def test_status_versioned(self):
589
655
tree = self.make_branch_and_tree('.')
605
671
self.assertContainsRe(result, "added:\n hello.txt\n")
606
672
self.assertNotContainsRe(result, "unknown:\n world.txt\n")
607
673
result2 = self.run_bzr("status --versioned -r 0..")[0]
608
self.assertEquals(result2, result)
674
self.assertEqual(result2, result)
610
676
def test_status_SV(self):
611
677
tree = self.make_branch_and_tree('.')
627
693
self.assertContainsRe(result, "[+]N hello.txt\n")
629
695
result2 = self.run_bzr("status -SV -r 0..")[0]
630
self.assertEquals(result2, result)
696
self.assertEqual(result2, result)
632
698
def assertStatusContains(self, pattern, short=False):
633
699
"""Run status, and assert it contains the given pattern"""
684
750
self.run_bzr('merge ../b', working_dir='a')
685
751
out, err = self.run_bzr('status --no-pending', working_dir='a')
686
self.assertEquals(out, "added:\n b\n")
752
self.assertEqual(out, "added:\n b\n")
688
754
def test_pending_specific_files(self):
689
755
"""With a specific file list, pending merges are not shown."""
705
771
class TestStatusEncodings(TestCaseWithTransport):
708
TestCaseWithTransport.setUp(self)
709
self.user_encoding = osutils._cached_user_encoding
710
self.stdout = sys.stdout
713
osutils._cached_user_encoding = self.user_encoding
714
sys.stdout = self.stdout
715
TestCaseWithTransport.tearDown(self)
717
773
def make_uncommitted_tree(self):
718
774
"""Build a branch with uncommitted unicode named changes in the cwd."""
719
775
working_tree = self.make_branch_and_tree(u'.')
727
783
return working_tree
729
785
def test_stdout_ascii(self):
730
sys.stdout = StringIO()
731
osutils._cached_user_encoding = 'ascii'
786
self.overrideAttr(osutils, '_cached_user_encoding', 'ascii')
732
787
working_tree = self.make_uncommitted_tree()
733
788
stdout, stderr = self.run_bzr("status")
735
self.assertEquals(stdout, """\
790
self.assertEqual(stdout, """\
740
795
def test_stdout_latin1(self):
741
sys.stdout = StringIO()
742
osutils._cached_user_encoding = 'latin-1'
796
self.overrideAttr(osutils, '_cached_user_encoding', 'latin-1')
743
797
working_tree = self.make_uncommitted_tree()
744
798
stdout, stderr = self.run_bzr('status')
746
self.assertEquals(stdout, u"""\
800
self.assertEqual(stdout, u"""\
749
803
""".encode('latin-1'))