diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/stub_sftp.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/tests/stub_sftp.py b/tests/stub_sftp.py index 1679e34e..2eb5f241 100644 --- a/tests/stub_sftp.py +++ b/tests/stub_sftp.py @@ -104,9 +104,15 @@ class StubSFTPServer (SFTPServerInterface): attr._flags &= ~attr.FLAG_PERMISSIONS SFTPServer.set_file_attr(path, attr) if flags & os.O_WRONLY: - fstr = 'wb' + if flags & os.O_APPEND: + fstr = 'ab' + else: + fstr = 'wb' elif flags & os.O_RDWR: - fstr = 'r+b' + if flags & os.O_APPEND: + fstr = 'a+b' + else: + fstr = 'r+b' else: # O_RDONLY (== 0) fstr = 'rb' @@ -114,7 +120,7 @@ class StubSFTPServer (SFTPServerInterface): f = os.fdopen(fd, fstr) except OSError, e: return SFTPServer.convert_errno(e.errno) - fobj = StubSFTPHandle() + fobj = StubSFTPHandle(flags) fobj.filename = path fobj.readfile = f fobj.writefile = f |