1
# Copyright (C) 2005 by Canonical Ltd
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.
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.
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
"""Tests of status command.
20
Most of these depend on the particular formatting used.
21
As such they really are blackbox tests even though some of the
22
tests are not using self.capture. If we add tests for the programmatic
23
interface later, they will be non blackbox tests.
27
from cStringIO import StringIO
29
from tempfile import TemporaryFile
32
from bzrlib.clone import copy_branch
33
from bzrlib.branch import Branch
34
from bzrlib.builtins import merge
35
from bzrlib.revisionspec import RevisionSpec
36
from bzrlib.status import show_status
37
from bzrlib.tests import TestCaseInTempDir
38
from bzrlib.workingtree import WorkingTree
41
class BranchStatus(TestCaseInTempDir):
43
def test_branch_status(self):
44
"""Test basic branch status"""
45
wt = WorkingTree.create_standalone('.')
50
show_status(b, to_file=tof)
51
self.assertEquals(tof.getvalue(), "")
54
self.build_tree(['hello.c', 'bye.c'])
55
wt.add_pending_merge('pending@pending-0-0')
56
show_status(b, to_file=tof)
58
self.assertEquals(tof.readlines(),
63
' pending@pending-0-0\n'
66
def test_branch_status_revisions(self):
67
"""Tests branch status with revisions"""
68
wt = WorkingTree.create_standalone('.')
72
self.build_tree(['hello.c', 'bye.c'])
75
wt.commit('Test message')
79
revs.append(RevisionSpec(0))
81
show_status(b, to_file=tof, revision=revs)
84
self.assertEquals(tof.readlines(),
89
self.build_tree(['more.c'])
91
wt.commit('Another test message')
94
revs.append(RevisionSpec(1))
96
show_status(b, to_file=tof, revision=revs)
99
self.assertEquals(tof.readlines(),
104
def status_string(self, branch):
105
# use a real file rather than StringIO because it doesn't handle
107
tof = codecs.getwriter('utf-8')(TemporaryFile())
108
show_status(branch, to_file=tof)
110
return tof.read().decode('utf-8')
112
def test_pending(self):
113
"""Pending merges display works, including Unicode"""
115
wt = WorkingTree.create_standalone('branch')
117
wt.commit("Empty commit 1")
118
b_2 = b.clone('./copy')
119
wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
120
merge(["./branch", -1], [None, None], this_dir = './copy')
121
message = self.status_string(b_2)
122
self.assert_(message.startswith("pending merges:\n"))
123
self.assert_(message.endswith("Empty commit 2\n"))
124
b_2.working_tree().commit("merged")
125
# must be long to make sure we see elipsis at the end
126
b.working_tree().commit("Empty commit 3 " +
127
"blah blah blah blah " * 10)
128
merge(["./branch", -1], [None, None], this_dir = './copy')
129
message = self.status_string(b_2)
130
self.assert_(message.startswith("pending merges:\n"))
131
self.assert_("Empty commit 3" in message)
132
self.assert_(message.endswith("...\n"))
134
def test_branch_status_specific_files(self):
135
"""Tests branch status with given specific files"""
136
wt = WorkingTree.create_standalone('.')
139
self.build_tree(['directory/','directory/hello.c', 'bye.c','test.c','dir2/'])
145
show_status(b, to_file=tof)
147
self.assertEquals(tof.readlines(),
151
' directory/hello.c\n'
155
show_status(b, specific_files=['bye.c','test.c','absent.c'], to_file=tof)
157
self.assertEquals(tof.readlines(),
163
show_status(b, specific_files=['directory'], to_file=tof)
165
self.assertEquals(tof.readlines(),
167
' directory/hello.c\n'
170
show_status(b, specific_files=['dir2'], to_file=tof)
172
self.assertEquals(tof.readlines(),