Changeset 119

Timestamp:

2002-03-30 16:48:07 (7 years ago)

Author:

schwa

Message:

MockSession: added methods _remove_trailing_slash and transfercmd.
MockSession/MockSocket: provided code so that a call to
    MockSession.transfercmd returns an appropriate MockSocket object
    whose makefile method uses the file content determined by
    MockSession.file_content. This means that this intended file content
    is passed through from MockSession to MockSocket.

Files:

• trunk/_mock_ftplib.py (modified) (7 diffs)

trunk/_mock_ftplib.py

@@ -30 +30 @@
-# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-
-7 2002/03/30 14:53:01
+8 2002/03/30 16:48:07
-
-"""
@@ -41 +41 @@
-
-import ftplib
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
-
-
@@ -323 +310 @@
-
-
+DEBUG = 0
+
+# class MockFileObject(StringIO.StringIO):
+#     """
+#     Mock class for the file objects _contained in_ _FTPFile
+#     objects (not for _FTPFile objects themselves!).
+#     """
+#     contents = ''
+# 
+#     def __init__(self, contents, mode='r'):
+        
+    
+class MockSocket:
+    """
+    Mock class which is used to return something from
+    MockSession.transfercmd.
+    """
+    def __init__(self, mock_file_content=''):
+        self.mock_file_content = mock_file_content
+
+    def makefile(self, mode):
+        return StringIO.StringIO(self.mock_file_content)
+
+
-class MockSession:
-
+
+
+
+
-    # used by MockSession.cwd and MockSession.pwd
-    current_dir = '/home/sschwarzer'
@@ -344 +358 @@
-drwxr-sr-x   6 45854    200           512 Sep 20  1999 scios2"""}
-
+    # file content to be used (indirectly) with transfercmd
+    mock_file_content = ''
+    
-    def __init__(self, host='', user='', password=''):
-
-
+
+
+
+
+
+
+
-    def voidcmd(self, cmd):
+        if DEBUG:
+            print cmd
-        if cmd == 'STAT':
-            return 'MockSession server awaiting your commands ;-)'
+        elif cmd.startswith('TYPE '):
+            return
-        else:
-            raise ftplib.error_perm
@@ -356 +382 @@
-        return self.current_dir
-
+    def cwd(self, path):
+        path = self._remove_trailing_slash(path)
+        self.current_dir = path
+
-    def dir(self, path, callback=None):
+        "Provide a callback function with each line of a directory listing."
-        if DEBUG:
-            print 'dir: %s' % path
-
-
+
-        if not self.dir_contents.has_key(path):
-            raise ftplib.error_perm
@@ -370 +400 @@
-                callback(line)
-
+    def transfercmd(self, cmd):
+        """
+        Return a MockSocket object whose makefile method will return
+        a mock file object.
+        """
+        if DEBUG:
+            print cmd
+        return MockSocket(self.mock_file_content)
+
-
-class FailOnLoginSession(MockSession):
@@ -375 +414 @@
-        raise ftplib.error_perm
-
+