← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/stub_sftp.py

  • Committer: Robert Collins
  • Date: 2006-07-20 13:00:31 UTC
  • mto: (1852.9.1 Tree.compare().)
  • mto: This revision was merged to the branch mainline in revision 1890.
  • Revision ID: robertc@robertcollins.net-20060720130031-d26103a427ea10f3
Start tree implementation tests.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/stub_sftp.py
1
1
# Copyright (C) 2005 Robey Pointer <robey@lag.net>, 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
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
 
# 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
"""
18
18
A stub SFTP server for loopback SFTP testing.
71
71
        if home is None:
72
72
            self.home = ''
73
73
        else:
74
 
            if not home.startswith(self.root):
75
 
                raise AssertionError(
76
 
                    "home must be a subdirectory of root (%s vs %s)"
77
 
                    % (home, root))
 
74
            assert home.startswith(self.root), \
 
75
                    "home must be a subdirectory of root (%s vs %s)" \
 
76
                    % (home, root)
78
77
            self.home = home[len(self.root):]
79
78
        if self.home.startswith('/'):
80
79
            self.home = self.home[1:]
83
82
    def _realpath(self, path):
84
83
        # paths returned from self.canonicalize() always start with
85
84
        # a path separator. So if 'root' is just '/', this would cause
86
 
        # a double slash at the beginning '//home/dir'.
 
85
        # a double slash at the beginning '//home/dir'. 
87
86
        if self.root == '/':
88
87
            return self.canonicalize(path)
89
88
        return self.root + self.canonicalize(path)
96
95
            #     /h:/foo/bar
97
96
            # and relative paths stay the same:
98
97
            #     foo/bar
99
 
            # win32 needs to use the Unicode APIs. so we require the
 
98
            # win32 needs to use the Unicode APIs. so we require the 
100
99
            # paths to be utf8 (Linux just uses bytestreams)
101
100
            thispath = path.decode('utf8')
102
101
            if path.startswith('/'):
124
123
            out = [ ]
125
124
            # TODO: win32 incorrectly lists paths with non-ascii if path is not
126
125
            # unicode. However on Linux the server should only deal with
127
 
            # bytestreams and posix.listdir does the right thing
 
126
            # bytestreams and posix.listdir does the right thing 
128
127
            if sys.platform == 'win32':
129
128
                flist = [f.encode('utf8') for f in os.listdir(path)]
130
129
            else:
154
153
    def open(self, path, flags, attr):
155
154
        path = self._realpath(path)
156
155
        try:
157
 
            flags |= getattr(os, 'O_BINARY', 0)
 
156
            if hasattr(os, 'O_BINARY'):
 
157
                flags |= os.O_BINARY
158
158
            if getattr(attr, 'st_mode', None):
159
159
                fd = os.open(path, flags, attr.st_mode)
160
160
            else:
161
 
                # os.open() defaults to 0777 which is
162
 
                # an odd default mode for files
163
 
                fd = os.open(path, flags, 0666)
 
161
                fd = os.open(path, flags)
164
162
        except OSError, e:
165
163
            return SFTPServer.convert_errno(e.errno)
166
164