← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_bad_files.py

Don't encode unicode messages to UTF-8 in mutter() (the stream writer does it).

Use a codec wrapped log file in tests to match the real environment.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_bad_files.py
1
 
# Copyright (C) 2005, 2007 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 by Canonical Ltd
 
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
18
 
"""Tests being able to ignore bad filetypes."""
 
18
"""Tests being able to ignore mad filetypes.
 
19
"""
19
20
 
20
 
from cStringIO import StringIO
 
21
from bzrlib.tests import TestCaseInTempDir
 
22
from bzrlib.errors import BadFileKindError
21
23
import os
22
24
 
23
 
from bzrlib import (
24
 
    add,
25
 
    errors,
26
 
    )
27
 
from bzrlib.status import show_tree_status
28
 
from bzrlib.tests import TestCaseWithTransport
29
 
 
30
 
 
31
 
def verify_status(tester, tree, value):
32
 
    """Verify the output of show_tree_status"""
 
25
def verify_status(tester, branch, value):
 
26
    from bzrlib.status import show_status
 
27
    from cStringIO import StringIO
 
28
 
33
29
    tof = StringIO()
34
 
    show_tree_status(tree, to_file=tof)
 
30
    show_status(branch, to_file=tof)
35
31
    tof.seek(0)
36
 
    tester.assertEqual(value, tof.readlines())
37
 
 
38
 
 
39
 
class TestBadFiles(TestCaseWithTransport):
40
 
 
41
 
    def test_bad_files(self):
 
32
    tester.assertEquals(tof.readlines(), value)
 
33
 
 
34
 
 
35
class TestBadFiles(TestCaseInTempDir):
 
36
    
 
37
    def test_bad_files(self): 
42
38
        """Test that bzr will ignore files it doesn't like"""
43
 
        if getattr(os, 'mkfifo', None) is None:
44
 
            # TODO: Ultimately this should be TestSkipped
45
 
            # or PlatformDeficiency
46
 
            return
47
 
 
48
 
        wt = self.make_branch_and_tree('.')
49
 
        b = wt.branch
50
 
 
51
 
        files = ['one', 'two', 'three']
52
 
        file_ids = ['one-id', 'two-id', 'three-id']
53
 
        self.build_tree(files)
54
 
        wt.add(files, file_ids)
55
 
        wt.commit("Commit one", rev_id="a@u-0-0")
 
39
        from bzrlib.commit import commit
 
40
        from bzrlib.add import smart_add
 
41
        from bzrlib.branch import Branch
 
42
 
 
43
        b = Branch.initialize(u'.')
 
44
 
 
45
        self.build_tree(['one', 'two', 'three'])
 
46
        smart_add(u'.')
 
47
        commit(b, "Commit one", rev_id="a@u-0-0")
 
48
        self.build_tree(['four'])
 
49
        smart_add(u'.')
 
50
        commit(b, "Commit two", rev_id="a@u-0-1")
 
51
        self.build_tree(['five'])
 
52
        smart_add(u'.')
 
53
        commit(b, "Commit three", rev_id="a@u-0-2")
56
54
 
57
55
        # We should now have a few files, lets try to
58
56
        # put some bogus stuff in the tree
59
57
 
60
 
        # status with nothing changed
61
 
        verify_status(self, wt, [])
 
58
        # We can only continue if we have mkfifo
 
59
        if not hasattr(os, 'mkfifo'):
 
60
            return
 
61
 
 
62
        # status with nothing
 
63
        verify_status(self, b, [])
62
64
 
63
65
        os.mkfifo('a-fifo')
64
66
        self.build_tree(['six'])
65
67
 
66
 
        verify_status(self, wt,
67
 
                          ['unknown:\n',
68
 
                           '  a-fifo\n',
69
 
                           '  six\n'
70
 
                           ])
71
 
 
72
 
        # We should raise an error if we are adding a bogus file
73
 
        self.assertRaises(errors.BadFileKindError,
74
 
                          add.smart_add_tree, wt, ['a-fifo'])
75
 
 
76
 
        # And the list of files shouldn't have been modified
77
 
        verify_status(self, wt,
78
 
                          ['unknown:\n',
79
 
                           '  a-fifo\n',
80
 
                           '  six\n'
81
 
                           ])
82
 
 
 
68
        verify_status(self, b,
 
69
                          ['unknown:\n',
 
70
                           '  a-fifo\n',
 
71
                           '  six\n'
 
72
                           ])
 
73
        
83
74
        # Make sure smart_add can handle having a bogus
84
75
        # file in the way
85
 
        add.smart_add_tree(wt, ['.'])
86
 
        verify_status(self, wt,
 
76
        smart_add(u'.')
 
77
        verify_status(self, b,
87
78
                          ['added:\n',
88
79
                           '  six\n',
89
80
                           'unknown:\n',
90
81
                           '  a-fifo\n',
91
82
                           ])
92
 
        wt.commit("Commit four", rev_id="a@u-0-3")
 
83
        commit(b, "Commit four", rev_id="a@u-0-3")
93
84
 
94
 
        verify_status(self, wt,
 
85
        verify_status(self, b,
95
86
                          ['unknown:\n',
96
87
                           '  a-fifo\n',
97
88
                           ])
 
89
 
 
90
        # We should raise an error if we are adding a bogus file
 
91
        # Is there a way to test the actual error that should be raised?
 
92
        self.run_bzr('add', 'a-fifo', retcode=3)
 
93