Changeset 150

Timestamp:

2002-03-31 23:18:06 (7 years ago)

Author:

schwa

Message:

TestUploadAndDownload: added tests for condiitional upload and download.

Files:

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

trunk/_test_ftputil.py

@@ -30 +30 @@
-# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-
-4 2002/03/31 20:36:28
+5 2002/03/31 23:18:06
-
-import unittest
@@ -89 +89 @@
-
-#
+# customized FTPHost class for conditional upload/download tests
+#
+class FailingUploadAndDownloadFTPHost(ftputil.FTPHost):
+    def upload(self, source, target, mode=''):
+        assert 0, "FTPHost.upload should not have been called"
+    
+    def download(self, source, target, mode=''):
+        assert 0, "FTPHost.download should not have been called"
+
+
+#
-# factory to produce FTPHost-like classes from a given FTPHost
-#  class and a given MockSession class
@@ -135 +146 @@
-
-    def test_lstat_one_file(self):
-with
+for
-        host = ftp_host_factory()
-        stat_result = host.lstat('/home/sschwarzer/index.html')
@@ -142 +153 @@
-
-    def test_lstat_one_dir(self):
-with
+for
-        # some directory
-        host = ftp_host_factory()
@@ -220 +231 @@
-
-    def test_caching(self):
-if
+whether
-        host = ftp_host_factory()
-        self.assertEqual( len(host._children), 0 )
@@ -392 +403 @@
-    """Test ascii upload and binary download as examples."""
-
+    def generate_ascii_file(self, data, filename):
+        """Generate an ASCII data file."""
+        source_file = open(filename, 'w')
+        source_file.write(data)
+        source_file.close()
+        
-    def test_ascii_upload(self):
-        """Test ASCII mode upload."""
-        local_source = '__test_source'
-        # generate file
-        data = ascii_data()
-
-
-
+
-        # upload
-        host = ftp_host_factory()
@@ -425 +439 @@
-        os.unlink(local_target)
-
+    def test_conditional_upload(self):
+        """Test conditional ASCII mode upload."""
+        local_source = '__test_source'
+        data = ascii_data()
+        self.generate_ascii_file(data, local_source)
+        # target is newer, so don't upload
+        host = ftp_host_factory(
+               ftp_host_class=FailingUploadAndDownloadFTPHost)
+        host.upload_if_newer(local_source, '/home/newer')
+        # target is older, so upload
+        host = ftp_host_factory()
+        host.upload_if_newer(local_source, '/home/older')
+        # check uploaded content
+        # the data which was uploaded has its line endings converted
+        #  so the conversion must also be applied to 'data'
+        data = data.replace('\n', '\r\n')
+        remote_file_content = _mock_ftplib.content_of('/home/older')
+        self.assertEqual(data, remote_file_content)
+        # target doesn't exist, so upload
+        host = ftp_host_factory()
+        host.upload_if_newer(local_source, '/home/notthere')
+        remote_file_content = _mock_ftplib.content_of('/home/notthere')
+        self.assertEqual(data, remote_file_content)
+        # clean up
+        os.unlink(local_source)
+        
+    def compare_and_delete_downloaded_data(self, filename):
+        """Compare content of downloaded file with its source, then
+        delete the local target file."""
+        data = open(filename, 'rb').read()
+        remote_file_content = _mock_ftplib.content_of('/home/newer')
+        self.assertEqual(data, remote_file_content)
+        # clean up
+        os.unlink(filename)
+
+    def test_conditional_download_without_target(self):
+        "Test conditional binary mode download when no target file exists."
+        local_target = '__test_target'
+        # target does not exist, so download
+        host = ftp_host_factory(session_factory=BinaryDownloadMockSession)
+        host.download_if_newer('/home/newer', local_target, 'b')
+        self.compare_and_delete_downloaded_data(local_target)
+
+    def test_conditional_download_with_older_target(self):
+        """Test conditional binary mode download with newer source file."""
+        local_target = '__test_target'
+        # make target file
+        open(local_target, 'w').close()
+        # source is newer, so download
+        host = ftp_host_factory(session_factory=BinaryDownloadMockSession)
+        host.download_if_newer('/home/newer', local_target, 'b')
+        self.compare_and_delete_downloaded_data(local_target)
+
+    def test_conditional_download_with_newer_target(self):
+        """Test conditional binary mode download with older source file."""
+        local_target = '__test_target'
+        # make target file
+        open(local_target, 'w').close()
+        # source is older, so don't download
+        host = ftp_host_factory(session_factory=BinaryDownloadMockSession)
+        host = ftp_host_factory(
+               ftp_host_class=FailingUploadAndDownloadFTPHost,
+               session_factory=BinaryDownloadMockSession)
+        host.download_if_newer('/home/older', local_target, 'b')
+        # remove target file
+        os.unlink(local_target)
+
-
-if __name__ == '__main__':