~bzr-pqm/bzr/bzr.dev

4763.2.4 by John Arbash Meinel
merge bzr.2.1 in preparation for NEWS entry.
1
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
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
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
16
17
"""Tests for the win32 walkdir extension."""
18
3504.4.8 by John Arbash Meinel
Some code cleanups.
19
import errno
20
3696.3.2 by John Arbash Meinel
Fix up some of the path joining logic.
21
from bzrlib import (
22
    osutils,
23
    tests,
24
    )
5967.12.1 by Martin Pool
Move all test features into bzrlib.tests.features
25
from bzrlib.tests import (
26
    features,
27
    )
28
29
30
win32_readdir_feature = features.ModuleAvailableFeature('bzrlib._walkdirs_win32')
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
31
32
33
class TestWin32Finder(tests.TestCaseInTempDir):
34
4913.2.6 by John Arbash Meinel
Switch Win32ReadDirFeature, which was probing for the wrong module anyway... :)
35
    _test_needs_features = [win32_readdir_feature]
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
36
3504.4.5 by John Arbash Meinel
Add tests to ensure that you can skip subdirs, start exposing the function.
37
    def setUp(self):
38
        super(TestWin32Finder, self).setUp()
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
39
        from bzrlib._walkdirs_win32 import (
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
40
            Win32ReadDir,
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
41
            )
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
42
        self.reader = Win32ReadDir()
3504.4.5 by John Arbash Meinel
Add tests to ensure that you can skip subdirs, start exposing the function.
43
44
    def _remove_stat_from_dirblock(self, dirblock):
45
        return [info[:3] + info[4:] for info in dirblock]
46
47
    def assertWalkdirs(self, expected, top, prefix=''):
3696.3.2 by John Arbash Meinel
Fix up some of the path joining logic.
48
        old_selected_dir_reader = osutils._selected_dir_reader
49
        try:
50
            osutils._selected_dir_reader = self.reader
51
            finder = osutils._walkdirs_utf8(top, prefix=prefix)
52
            result = []
53
            for dirname, dirblock in finder:
54
                dirblock = self._remove_stat_from_dirblock(dirblock)
55
                result.append((dirname, dirblock))
56
            self.assertEqual(expected, result)
57
        finally:
58
            osutils._selected_dir_reader = old_selected_dir_reader
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
59
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
60
    def assertReadDir(self, expected, prefix, top_unicode):
61
        result = self._remove_stat_from_dirblock(
62
            self.reader.read_dir(prefix, top_unicode))
63
        self.assertEqual(expected, result)
64
65
    def test_top_prefix_to_starting_dir(self):
66
        # preparing an iteration should create a unicode native path.
67
        self.assertEqual(('prefix', None, None, None, u'\x12'),
3696.3.2 by John Arbash Meinel
Fix up some of the path joining logic.
68
            self.reader.top_prefix_to_starting_dir(u'\x12'.encode('utf8'),
69
                                                   'prefix'))
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
70
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
71
    def test_empty_directory(self):
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
72
        self.assertReadDir([], 'prefix', u'.')
3504.4.3 by John Arbash Meinel
Start working on an extension specifically for win32,
73
        self.assertWalkdirs([(('', u'.'), [])], u'.')
3504.4.4 by John Arbash Meinel
We have walkdirs basically working, only without timestamps
74
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
75
    def test_file(self):
3504.4.4 by John Arbash Meinel
We have walkdirs basically working, only without timestamps
76
        self.build_tree(['foo'])
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
77
        self.assertReadDir([('foo', 'foo', 'file', u'./foo')],
78
            '', u'.')
79
5235.2.1 by Martin
Rename test method in bt.test__walkdirs_win32 that clashes with an attribute name
80
    def test_directory(self):
3696.3.1 by Robert Collins
Refactor bzrlib.osutils._walkdirs_utf8 to aid API migration in future.
81
        self.build_tree(['bar/'])
82
        self.assertReadDir([('bar', 'bar', 'directory', u'./bar')],
83
            '', u'.')
84
85
    def test_prefix(self):
86
        self.build_tree(['bar/', 'baf'])
87
        self.assertReadDir([
88
            ('xxx/baf', 'baf', 'file', u'./baf'),
89
            ('xxx/bar', 'bar', 'directory', u'./bar'),
90
            ],
91
            'xxx', u'.')
3504.4.8 by John Arbash Meinel
Some code cleanups.
92
93
    def test_missing_dir(self):
3696.3.3 by John Arbash Meinel
read_dir raises the exception directly, it isn't an iterator/generator
94
        e = self.assertRaises(WindowsError,
95
            self.reader.read_dir, 'prefix', u'no_such_dir')
3504.4.8 by John Arbash Meinel
Some code cleanups.
96
        self.assertEqual(errno.ENOENT, e.errno)
97
        self.assertEqual(3, e.winerror)
98
        self.assertEqual((3, u'no_such_dir/*'), e.args)