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

Branch now uses BzrDir reasonably sanely.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_status.py
 
1
# Copyright (C) 2005 by 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
"""Tests of status command.
 
19
 
 
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.
 
24
"""
 
25
 
 
26
 
 
27
from cStringIO import StringIO
 
28
from os import mkdir
 
29
from tempfile import TemporaryFile
 
30
import codecs
 
31
 
 
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
 
39
 
 
40
 
 
41
class BranchStatus(TestCaseInTempDir):
 
42
    
 
43
    def test_branch_status(self): 
 
44
        """Test basic branch status"""
 
45
        wt = WorkingTree.create_standalone('.')
 
46
        b = wt.branch
 
47
 
 
48
        # status with nothing
 
49
        tof = StringIO()
 
50
        show_status(b, to_file=tof)
 
51
        self.assertEquals(tof.getvalue(), "")
 
52
 
 
53
        tof = StringIO()
 
54
        self.build_tree(['hello.c', 'bye.c'])
 
55
        wt.add_pending_merge('pending@pending-0-0')
 
56
        show_status(b, to_file=tof)
 
57
        tof.seek(0)
 
58
        self.assertEquals(tof.readlines(),
 
59
                          ['unknown:\n',
 
60
                           '  bye.c\n',
 
61
                           '  hello.c\n',
 
62
                           'pending merges:\n',
 
63
                           '  pending@pending-0-0\n'
 
64
                           ])
 
65
 
 
66
    def test_branch_status_revisions(self):
 
67
        """Tests branch status with revisions"""
 
68
        wt = WorkingTree.create_standalone('.')
 
69
        b = wt.branch
 
70
 
 
71
        tof = StringIO()
 
72
        self.build_tree(['hello.c', 'bye.c'])
 
73
        wt.add('hello.c')
 
74
        wt.add('bye.c')
 
75
        wt.commit('Test message')
 
76
 
 
77
        tof = StringIO()
 
78
        revs =[]
 
79
        revs.append(RevisionSpec(0))
 
80
        
 
81
        show_status(b, to_file=tof, revision=revs)
 
82
        
 
83
        tof.seek(0)
 
84
        self.assertEquals(tof.readlines(),
 
85
                          ['added:\n',
 
86
                           '  bye.c\n',
 
87
                           '  hello.c\n'])
 
88
 
 
89
        self.build_tree(['more.c'])
 
90
        wt.add('more.c')
 
91
        wt.commit('Another test message')
 
92
        
 
93
        tof = StringIO()
 
94
        revs.append(RevisionSpec(1))
 
95
        
 
96
        show_status(b, to_file=tof, revision=revs)
 
97
        
 
98
        tof.seek(0)
 
99
        self.assertEquals(tof.readlines(),
 
100
                          ['added:\n',
 
101
                           '  bye.c\n',
 
102
                           '  hello.c\n'])
 
103
 
 
104
    def status_string(self, branch):
 
105
        # use a real file rather than StringIO because it doesn't handle
 
106
        # Unicode very well.
 
107
        tof = codecs.getwriter('utf-8')(TemporaryFile())
 
108
        show_status(branch, to_file=tof)
 
109
        tof.seek(0)
 
110
        return tof.read().decode('utf-8')
 
111
 
 
112
    def test_pending(self):
 
113
        """Pending merges display works, including Unicode"""
 
114
        mkdir("./branch")
 
115
        wt = WorkingTree.create_standalone('branch')
 
116
        b = wt.branch
 
117
        wt.commit("Empty commit 1")
 
118
        b_2 = b.clone('./copy')
 
119
        wt2 = WorkingTree('copy', b_2)
 
120
        wt.commit(u"\N{TIBETAN DIGIT TWO} Empty commit 2")
 
121
        merge(["./branch", -1], [None, None], this_dir = './copy')
 
122
        message = self.status_string(b_2)
 
123
        self.assert_(message.startswith("pending merges:\n"))
 
124
        self.assert_(message.endswith("Empty commit 2\n")) 
 
125
        wt2.commit("merged")
 
126
        # must be long to make sure we see elipsis at the end
 
127
        wt.commit("Empty commit 3 " + 
 
128
                   "blah blah blah blah " * 10)
 
129
        merge(["./branch", -1], [None, None], this_dir = './copy')
 
130
        message = self.status_string(b_2)
 
131
        self.assert_(message.startswith("pending merges:\n"))
 
132
        self.assert_("Empty commit 3" in message)
 
133
        self.assert_(message.endswith("...\n")) 
 
134
 
 
135
    def test_branch_status_specific_files(self): 
 
136
        """Tests branch status with given specific files"""
 
137
        wt = WorkingTree.create_standalone('.')
 
138
        b = wt.branch
 
139
 
 
140
        self.build_tree(['directory/','directory/hello.c', 'bye.c','test.c','dir2/'])
 
141
        wt.add('directory')
 
142
        wt.add('test.c')
 
143
        wt.commit('testing')
 
144
        
 
145
        tof = StringIO()
 
146
        show_status(b, to_file=tof)
 
147
        tof.seek(0)
 
148
        self.assertEquals(tof.readlines(),
 
149
                          ['unknown:\n',
 
150
                           '  bye.c\n',
 
151
                           '  dir2\n',
 
152
                           '  directory/hello.c\n'
 
153
                           ])
 
154
 
 
155
        tof = StringIO()
 
156
        show_status(b, specific_files=['bye.c','test.c','absent.c'], to_file=tof)
 
157
        tof.seek(0)
 
158
        self.assertEquals(tof.readlines(),
 
159
                          ['unknown:\n',
 
160
                           '  bye.c\n'
 
161
                           ])
 
162
        
 
163
        tof = StringIO()
 
164
        show_status(b, specific_files=['directory'], to_file=tof)
 
165
        tof.seek(0)
 
166
        self.assertEquals(tof.readlines(),
 
167
                          ['unknown:\n',
 
168
                           '  directory/hello.c\n'
 
169
                           ])
 
170
        tof = StringIO()
 
171
        show_status(b, specific_files=['dir2'], to_file=tof)
 
172
        tof.seek(0)
 
173
        self.assertEquals(tof.readlines(),
 
174
                          ['unknown:\n',
 
175
                           '  dir2\n'
 
176
                           ])