← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/ftp_server.py

Fix ftp transport so that it handles the 'mode' parameter when provided.

* ftp.py:
(FtpTransport.put_file, FtpTransport.mkdir,
FtpTransport._try_append): Use _setmode.
(FtpTransport._setmode): Fix implementation (parameters were
reversed and mode not converted).

* ftp_server.py:
(test_filesystem): New filesystem implementing chmod.
(test_authorizer.authorize): Use our filesystem when authorized.
(ftp_channel.cmd_site): Implement SITE CHOWN.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/ftp_server.py
37
37
    )
38
38
 
39
39
 
 
40
class test_filesystem(medusa.filesys.os_filesystem):
 
41
    """A custom filesystem wrapper to add missing functionalities."""
 
42
 
 
43
    def chmod(self, path, mode):
 
44
        p = self.normalize (self.path_module.join (self.wd, path))
 
45
        return os.chmod(self.translate(p), mode)
 
46
 
 
47
 
 
48
 
40
49
class test_authorizer(object):
41
50
    """A custom Authorizer object for running the test suite.
42
51
 
64
73
            and password != self.secured_password):
65
74
            return 0, 'Password invalid.', None
66
75
        else:
67
 
            return 1, 'OK.', medusa.filesys.os_filesystem(self.root)
 
76
            return 1, 'OK.', test_filesystem(self.root)
68
77
 
69
78
 
70
79
class ftp_channel(medusa.ftp_server.ftp_channel):
153
162
            except:
154
163
                self.respond ('550 error creating directory.')
155
164
 
 
165
    def cmd_site(self, line):
 
166
        command, args = line[1].split(' ', 1)
 
167
        if command.lower() == 'chmod':
 
168
            try:
 
169
                mode, path = args.split()
 
170
                mode = int(mode, 8)
 
171
            except ValueError:
 
172
                self.command_not_understood(' '.join(line))
 
173
                return
 
174
            try:
 
175
                # Yes path and mode are reversed
 
176
                self.filesystem.chmod(path, mode)
 
177
                self.respond('200 SITE CHMOD command successful')
 
178
            except AttributeError:
 
179
                # The chmod method is not available in read-only
 
180
                self.command_not_authorized(' '.join(line))
 
181
        else:
 
182
            self.command_not_understood(' '.join(line))
 
183
 
156
184
 
157
185
class ftp_server(medusa.ftp_server.ftp_server):
158
186
    """Customize the behavior of the Medusa ftp_server.