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

  • Committer: Andrew Bennetts
  • Date: 2008-08-12 14:53:26 UTC
  • mto: This revision was merged to the branch mainline in revision 3624.
  • Revision ID: andrew.bennetts@canonical.com-20080812145326-yx693x2jc4rcovb7
Move the notes on writing tests out of HACKING into a new file, and improve
them.

Many of the testing notes in the HACKING file were in duplicated in two places
in that file!  This change removes that duplication.  It also adds new sections
on “Where should I put a new test?” and “TestCase and its subclasses”, and
others like “Test feature dependencies” have been expanded.  The whole document
has generally been edited to be a bit more coherent. 

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test__walkdirs_win32.py
1
 
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2008 Canonical Ltd
2
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
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
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Tests for the win32 walkdir extension."""
18
18
 
19
19
import errno
20
20
 
21
 
from bzrlib import (
22
 
    osutils,
23
 
    tests,
24
 
    )
25
 
 
26
 
 
27
 
win32_readdir_feature = tests.ModuleAvailableFeature('bzrlib._walkdirs_win32')
 
21
from bzrlib import tests
 
22
 
 
23
 
 
24
class _WalkdirsWin32Feature(tests.Feature):
 
25
 
 
26
    def _probe(self):
 
27
        try:
 
28
            import bzrlib._walkdirs_win32
 
29
        except ImportError:
 
30
            return False
 
31
        else:
 
32
            return True
 
33
 
 
34
    def feature_name(self):
 
35
        return 'bzrlib._walkdirs_win32'
 
36
 
 
37
WalkdirsWin32Feature = _WalkdirsWin32Feature()
28
38
 
29
39
 
30
40
class TestWin32Finder(tests.TestCaseInTempDir):
31
41
 
32
 
    _test_needs_features = [win32_readdir_feature]
 
42
    _test_needs_features = [WalkdirsWin32Feature]
33
43
 
34
44
    def setUp(self):
35
45
        super(TestWin32Finder, self).setUp()
36
46
        from bzrlib._walkdirs_win32 import (
37
 
            Win32ReadDir,
 
47
            _walkdirs_utf8_win32_find_file
38
48
            )
39
 
        self.reader = Win32ReadDir()
 
49
        self.walkdirs_utf8 = _walkdirs_utf8_win32_find_file
40
50
 
41
51
    def _remove_stat_from_dirblock(self, dirblock):
42
52
        return [info[:3] + info[4:] for info in dirblock]
43
53
 
44
54
    def assertWalkdirs(self, expected, top, prefix=''):
45
 
        old_selected_dir_reader = osutils._selected_dir_reader
46
 
        try:
47
 
            osutils._selected_dir_reader = self.reader
48
 
            finder = osutils._walkdirs_utf8(top, prefix=prefix)
49
 
            result = []
50
 
            for dirname, dirblock in finder:
51
 
                dirblock = self._remove_stat_from_dirblock(dirblock)
52
 
                result.append((dirname, dirblock))
53
 
            self.assertEqual(expected, result)
54
 
        finally:
55
 
            osutils._selected_dir_reader = old_selected_dir_reader
56
 
 
57
 
    def assertReadDir(self, expected, prefix, top_unicode):
58
 
        result = self._remove_stat_from_dirblock(
59
 
            self.reader.read_dir(prefix, top_unicode))
 
55
        finder = self.walkdirs_utf8(top, prefix=prefix)
 
56
        result = []
 
57
        for dirname, dirblock in finder:
 
58
            result.append((dirname, self._remove_stat_from_dirblock(dirblock)))
60
59
        self.assertEqual(expected, result)
61
60
 
62
 
    def test_top_prefix_to_starting_dir(self):
63
 
        # preparing an iteration should create a unicode native path.
64
 
        self.assertEqual(('prefix', None, None, None, u'\x12'),
65
 
            self.reader.top_prefix_to_starting_dir(u'\x12'.encode('utf8'),
66
 
                                                   'prefix'))
67
 
 
68
61
    def test_empty_directory(self):
69
 
        self.assertReadDir([], 'prefix', u'.')
70
62
        self.assertWalkdirs([(('', u'.'), [])], u'.')
71
63
 
72
 
    def test_file(self):
 
64
    def test_file_in_dir(self):
73
65
        self.build_tree(['foo'])
74
 
        self.assertReadDir([('foo', 'foo', 'file', u'./foo')],
75
 
            '', u'.')
76
 
 
77
 
    def test_dir(self):
78
 
        self.build_tree(['bar/'])
79
 
        self.assertReadDir([('bar', 'bar', 'directory', u'./bar')],
80
 
            '', u'.')
81
 
 
82
 
    def test_prefix(self):
83
 
        self.build_tree(['bar/', 'baf'])
84
 
        self.assertReadDir([
85
 
            ('xxx/baf', 'baf', 'file', u'./baf'),
86
 
            ('xxx/bar', 'bar', 'directory', u'./bar'),
87
 
            ],
88
 
            'xxx', u'.')
 
66
        self.assertWalkdirs([
 
67
            (('', u'.'), [('foo', 'foo', 'file', u'./foo')])
 
68
            ], u'.')
 
69
 
 
70
    def test_subdir(self):
 
71
        self.build_tree(['foo', 'bar/', 'bar/baz'])
 
72
        self.assertWalkdirs([
 
73
            (('', u'.'), [('bar', 'bar', 'directory', u'./bar'),
 
74
                          ('foo', 'foo', 'file', u'./foo'),
 
75
                         ]),
 
76
            (('bar', u'./bar'), [('bar/baz', 'baz', 'file', u'./bar/baz')]),
 
77
            ], '.')
 
78
        self.assertWalkdirs([
 
79
            (('xxx', u'.'), [('xxx/bar', 'bar', 'directory', u'./bar'),
 
80
                             ('xxx/foo', 'foo', 'file', u'./foo'),
 
81
                            ]),
 
82
            (('xxx/bar', u'./bar'), [('xxx/bar/baz', 'baz', 'file', u'./bar/baz')]),
 
83
            ], '.', prefix='xxx')
 
84
        self.assertWalkdirs([
 
85
            (('', u'bar'), [('baz', 'baz', 'file', u'bar/baz')]),
 
86
            ], 'bar')
 
87
 
 
88
    def test_skip_subdir(self): 
 
89
        self.build_tree(['a/', 'b/', 'c/', 'a/aa', 'b/bb', 'c/cc'])
 
90
        base_dirblock = [('a', 'a', 'directory', u'./a'),
 
91
                          ('b', 'b', 'directory', u'./b'),
 
92
                          ('c', 'c', 'directory', u'./c'),
 
93
                         ]
 
94
        self.assertWalkdirs([
 
95
            (('', u'.'), base_dirblock),
 
96
            (('a', u'./a'), [('a/aa', 'aa', 'file', u'./a/aa')]),
 
97
            (('b', u'./b'), [('b/bb', 'bb', 'file', u'./b/bb')]),
 
98
            (('c', u'./c'), [('c/cc', 'cc', 'file', u'./c/cc')]),
 
99
            ], '.')
 
100
 
 
101
        walker = self.walkdirs_utf8('.')
 
102
        dir_info, first_dirblock = walker.next()
 
103
        self.assertEqual(('', u'.'), dir_info)
 
104
        self.assertEqual(base_dirblock,
 
105
                         self._remove_stat_from_dirblock(first_dirblock))
 
106
        # Now, remove 'b' and it should be skipped on the next round
 
107
        del first_dirblock[1]
 
108
        dir_info, second_dirblock = walker.next()
 
109
        second_dirblock = self._remove_stat_from_dirblock(second_dirblock)
 
110
        self.assertEqual(('a', u'./a'), dir_info)
 
111
        self.assertEqual([('a/aa', 'aa', 'file', u'./a/aa')], second_dirblock)
 
112
        dir_info, third_dirblock = walker.next()
 
113
        third_dirblock = self._remove_stat_from_dirblock(third_dirblock)
 
114
        self.assertEqual(('c', u'./c'), dir_info)
 
115
        self.assertEqual([('c/cc', 'cc', 'file', u'./c/cc')], third_dirblock)
89
116
 
90
117
    def test_missing_dir(self):
91
 
        e = self.assertRaises(WindowsError,
92
 
            self.reader.read_dir, 'prefix', u'no_such_dir')
 
118
        e = self.assertRaises(WindowsError, list,
 
119
                                self.walkdirs_utf8(u'no_such_dir'))
93
120
        self.assertEqual(errno.ENOENT, e.errno)
94
121
        self.assertEqual(3, e.winerror)
95
122
        self.assertEqual((3, u'no_such_dir/*'), e.args)