Changeset 135

Timestamp:

2002-03-30 21:50:25 (7 years ago)

Author:

schwa

Message:

Module: the path of a file is now passed from MockSession.transfercmd to
    MockSocket.makefile to MockFile's constructor. This makes it
    possible to access the MockFile object at any time via the
    module-global dictionary mock_files. The convenience function
    content_of(path) can be used to get the content for the requested path.

Files:

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

trunk/_mock_ftplib.py

@@ -30 +30 @@
-# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-
-2 2002/03/30 21:20:28
+3 2002/03/30 21:50:25
-
-"""
@@ -312 +312 @@
-DEBUG = 0
-
+# Use a global dictionary of the form {path: mock_file, ...}
+#  to make "volatile" mock files accessible. This is used for
+#  testing the contents of a file after an FTPHost.upload call.
+mock_files = {}
+
+def content_of(path):
+    return mock_files[path].getvalue()
+
-class MockFile(StringIO.StringIO):
-    """
@@ -317 +325 @@
-    objects (not for _FTPFile objects themselves!).
-    """
-
+
+
+
-        StringIO.StringIO.__init__(self, content)
-
@@ -330 +340 @@
-            self._value_after_close = StringIO.StringIO.getvalue(self)
-        StringIO.StringIO.close(self)
-
+
-    
-class MockSocket:
@@ -337 +347 @@
-    MockSession.transfercmd.
-    """
-
+path, 
-        if DEBUG:
-            print 'File content: *%s*' % mock_file_content
+        self.file_path = path
-        self.mock_file_content = mock_file_content
-
-    def makefile(self, mode):
-
+file_path, self.
-
-    def close(self):
@@ -432 +443 @@
-        # fail if path isn't available (this name is hard-coded here
-        #  and has to be used for the corresponding tests)
-cmd == 'RETR' and path == 'notthere'
+(cmd, path) == ('RETR', 'notthere')
-            raise ftplib.error_perm
-
-
+path, 
+