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

Better message when failing to import a test suite.

unittest masks them (who knows why?), so load test modules by hand 
and get proper errors.

Add tests for loader and move around meta-tests.

Show diffs side-by-side

added added

removed removed

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