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
28
from os import mkdir, chdir
29
from tempfile import TemporaryFile
33
from bzrlib.builtins import merge
34
import bzrlib.bzrdir as bzrdir
35
from bzrlib.osutils import pathjoin
36
from bzrlib.revisionspec import RevisionSpec
37
from bzrlib.status import show_tree_status
38
from bzrlib.tests import TestCaseWithTransport
39
from bzrlib.workingtree import WorkingTree
42
class BranchStatus(TestCaseWithTransport):
44
def test_branch_status(self):
45
"""Test basic branch status"""
46
wt = self.make_branch_and_tree('.')
51
show_tree_status(wt, to_file=tof)
52
self.assertEquals(tof.getvalue(), "")
55
self.build_tree(['hello.c', 'bye.c'])
56
wt.add_pending_merge('pending@pending-0-0')
57
show_tree_status(wt, to_file=tof)
59
self.assertEquals(tof.readlines(),
64
' pending@pending-0-0\n'
67
def test_branch_status_revisions(self):
68
"""Tests branch status with revisions"""
69
wt = self.make_branch_and_tree('.')
73
self.build_tree(['hello.c', 'bye.c'])
76
wt.commit('Test message')
80
revs.append(RevisionSpec(0))
82
show_tree_status(wt, to_file=tof, revision=revs)
85
self.assertEquals(tof.readlines(),
90
self.build_tree(['more.c'])
92
wt.commit('Another test message')
95
revs.append(RevisionSpec(1))
97
show_tree_status(wt, to_file=tof, revision=revs)
100
self.assertEquals(tof.readlines(),
105
def status_string(self, wt):
106
# use a real file rather than StringIO because it doesn't handle
108
tof = codecs.getwriter('utf-8')(TemporaryFile())
109
show_tree_status(wt, to_file=tof)
111
return tof.read().decode('utf-8')
113
def test_pending(self):
114
"""Pending merges display works, including Unicode"""
116
wt = self.make_branch_and_tree('branch')
118
wt.commit("Empty commit 1")
119
b_2_dir = b.bzrdir.sprout('./copy')
120
b_2 = b_2_dir.open_branch()
121
wt2 = b_2_dir.open_workingtree()
122
wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
123
merge(["./branch", -1], [None, None], this_dir = './copy')
124
message = self.status_string(wt2)
125
self.assert_(message.startswith("pending merges:\n"))
126
self.assert_(message.endswith("Empty commit 2\n"))
128
# must be long to make sure we see elipsis at the end
129
wt.commit("Empty commit 3 " +
130
"blah blah blah blah " * 10)
131
merge(["./branch", -1], [None, None], this_dir = './copy')
132
message = self.status_string(wt2)
133
self.assert_(message.startswith("pending merges:\n"))
134
self.assert_("Empty commit 3" in message)
135
self.assert_(message.endswith("...\n"))
137
def test_branch_status_specific_files(self):
138
"""Tests branch status with given specific files"""
139
wt = self.make_branch_and_tree('.')
142
self.build_tree(['directory/','directory/hello.c', 'bye.c','test.c','dir2/'])
148
show_tree_status(wt, to_file=tof)
150
self.assertEquals(tof.readlines(),
154
' directory/hello.c\n'
158
show_tree_status(wt, specific_files=['bye.c','test.c','absent.c'],
161
self.assertEquals(tof.readlines(),
167
show_tree_status(wt, specific_files=['directory'], to_file=tof)
169
self.assertEquals(tof.readlines(),
171
' directory/hello.c\n'
174
show_tree_status(wt, specific_files=['dir2'], to_file=tof)
176
self.assertEquals(tof.readlines(),
181
def test_status_nonexistent_file(self):
182
# files that don't exist in either the basis tree or working tree
183
# should give an error
184
wt = self.make_branch_and_tree('.')
185
out, err = self.run_bzr('status', 'does-not-exist', retcode=3)
186
self.assertContainsRe(err, r'do not exist.*does-not-exist')
189
class CheckoutStatus(BranchStatus):
192
super(CheckoutStatus, self).setUp()
196
def make_branch_and_tree(self, relpath):
197
source = self.make_branch(pathjoin('..', relpath))
198
checkout = bzrdir.BzrDirMetaFormat1().initialize(relpath)
199
bzrlib.branch.BranchReferenceFormat().initialize(checkout, source)
200
return checkout.create_workingtree()
203
class TestStatus(TestCaseWithTransport):
205
def test_status(self):
207
self.build_tree(['hello.txt'])
208
result = self.run_bzr("status")[0]
209
self.assert_("unknown:\n hello.txt\n" in result, result)
210
self.run_bzr("add", "hello.txt")
211
result = self.run_bzr("status")[0]
212
self.assert_("added:\n hello.txt\n" in result, result)
213
self.run_bzr("commit", "-m", "added")
214
result = self.run_bzr("status", "-r", "0..1")[0]
215
self.assert_("added:\n hello.txt\n" in result, result)
216
self.build_tree(['world.txt'])
217
result = self.run_bzr("status", "-r", "0")[0]
218
self.assert_("added:\n hello.txt\n" \
219
"unknown:\n world.txt\n" in result, result)
221
result2 = self.run_bzr("status", "-r", "0..")[0]
222
self.assertEquals(result2, result)
added
removed
bzrlib/tests/blackbox/test_status.py
