Exemple #1
0
 def test_binary_write(self):
     """Write binary data with `write`."""
     host = test_base.ftp_host_factory()
     data = b"\000a\001b\r\n\002c\003\n\004\r\005"
     with host.open("dummy", "wb") as output:
         output.write(data)
     child_data = mock_ftplib.content_of("dummy")
     expected_data = data
     self.assertEqual(child_data, expected_data)
Exemple #2
0
 def test_binary_write(self):
     """Write binary data with `write`."""
     host = test_base.ftp_host_factory()
     data = b"\000a\001b\r\n\002c\003\n\004\r\005"
     with host.open("dummy", "wb") as output:
         output.write(data)
     child_data = mock_ftplib.content_of("dummy")
     expected_data = data
     assert child_data == expected_data
Exemple #3
0
 def test_ascii_write(self):
     """Write ASCII text with `write`."""
     host = test_base.ftp_host_factory()
     data = " \nline 2\nline 3"
     with host.open("dummy", "w", newline="\r\n") as output:
         output.write(data)
     child_data = mock_ftplib.content_of("dummy")
     # This corresponds to the byte stream, so expect a `bytes` object.
     expected_data = b" \r\nline 2\r\nline 3"
     self.assertEqual(child_data, expected_data)
Exemple #4
0
 def test_ascii_write(self):
     """Write ASCII text with `write`."""
     host = test_base.ftp_host_factory()
     data = " \nline 2\nline 3"
     with host.open("dummy", "w", newline="\r\n") as output:
         output.write(data)
     child_data = mock_ftplib.content_of("dummy")
     # This corresponds to the byte stream, so expect a `bytes` object.
     expected_data = b" \r\nline 2\r\nline 3"
     assert child_data == expected_data
Exemple #5
0
 def compare_and_delete_downloaded_data(self, file_name):
     """
     Compare content of downloaded file with its source, then
     delete the local target file.
     """
     with open(file_name, "rb") as fobj:
         data = fobj.read()
     # The name `newer` is used by all callers, so use it here, too.
     remote_file_content = mock_ftplib.content_of("newer")
     self.assertEqual(data, remote_file_content)
     # Clean up
     os.unlink(file_name)
Exemple #6
0
 def compare_and_delete_downloaded_data(self, file_name):
     """
     Compare content of downloaded file with its source, then
     delete the local target file.
     """
     with open(file_name, "rb") as fobj:
         data = fobj.read()
     # The name `newer` is used by all callers, so use it here, too.
     remote_file_content = mock_ftplib.content_of("newer")
     self.assertEqual(data, remote_file_content)
     # Clean up
     os.unlink(file_name)
Exemple #7
0
 def test_ascii_writelines(self):
     """Write ASCII text with `writelines`."""
     host = test_base.ftp_host_factory()
     data = [" \n", "line 2\n", "line 3"]
     backup_data = data[:]
     output = host.open("dummy", "w", newline="\r\n")
     output.writelines(data)
     output.close()
     child_data = mock_ftplib.content_of("dummy")
     expected_data = b" \r\nline 2\r\nline 3"
     self.assertEqual(child_data, expected_data)
     # Ensure that the original data was not modified.
     self.assertEqual(data, backup_data)
Exemple #8
0
 def test_ascii_writelines(self):
     """Write ASCII text with `writelines`."""
     host = test_base.ftp_host_factory()
     data = [" \n", "line 2\n", "line 3"]
     backup_data = data[:]
     output = host.open("dummy", "w", newline="\r\n")
     output.writelines(data)
     output.close()
     child_data = mock_ftplib.content_of("dummy")
     expected_data = b" \r\nline 2\r\nline 3"
     assert child_data == expected_data
     # Ensure that the original data was not modified.
     assert data == backup_data
Exemple #9
0
 def test_download(self):
     """Test mode download."""
     local_target = "_test_target_"
     host = test_base.ftp_host_factory(
              session_factory=BinaryDownloadMockSession)
     # Download
     host.download("dummy", local_target)
     # Read file and compare
     with open(local_target, "rb") as fobj:
         data = fobj.read()
     remote_file_content = mock_ftplib.content_of("dummy")
     self.assertEqual(data, remote_file_content)
     # Clean up
     os.unlink(local_target)
Exemple #10
0
 def test_download(self):
     """Test mode download."""
     local_target = "_test_target_"
     host = test_base.ftp_host_factory(
         session_factory=BinaryDownloadMockSession)
     # Download
     host.download("dummy", local_target)
     # Read file and compare
     with open(local_target, "rb") as fobj:
         data = fobj.read()
     remote_file_content = mock_ftplib.content_of("dummy")
     self.assertEqual(data, remote_file_content)
     # Clean up
     os.unlink(local_target)
Exemple #11
0
 def test_conditional_upload(self):
     """Test conditional upload."""
     local_source = "_test_source_"
     data = binary_data()
     self.generate_file(data, local_source)
     # Target is newer, so don't upload.
     host = test_base.ftp_host_factory(
              ftp_host_class=FailingUploadAndDownloadFTPHost)
     flag = host.upload_if_newer(local_source, "/home/newer")
     self.assertEqual(flag, False)
     # Target is older, so upload.
     host = test_base.ftp_host_factory()
     flag = host.upload_if_newer(local_source, "/home/older")
     self.assertEqual(flag, True)
     remote_file_content = mock_ftplib.content_of("older")
     self.assertEqual(data, remote_file_content)
     # Target doesn't exist, so upload.
     host = test_base.ftp_host_factory()
     flag = host.upload_if_newer(local_source, "/home/notthere")
     self.assertEqual(flag, True)
     remote_file_content = mock_ftplib.content_of("notthere")
     self.assertEqual(data, remote_file_content)
     # Clean up.
     os.unlink(local_source)
Exemple #12
0
 def test_conditional_upload(self):
     """Test conditional upload."""
     local_source = "_test_source_"
     data = binary_data()
     self.generate_file(data, local_source)
     # Target is newer, so don't upload.
     host = test_base.ftp_host_factory(
         ftp_host_class=FailingUploadAndDownloadFTPHost)
     flag = host.upload_if_newer(local_source, "/home/newer")
     self.assertEqual(flag, False)
     # Target is older, so upload.
     host = test_base.ftp_host_factory()
     flag = host.upload_if_newer(local_source, "/home/older")
     self.assertEqual(flag, True)
     remote_file_content = mock_ftplib.content_of("older")
     self.assertEqual(data, remote_file_content)
     # Target doesn't exist, so upload.
     host = test_base.ftp_host_factory()
     flag = host.upload_if_newer(local_source, "/home/notthere")
     self.assertEqual(flag, True)
     remote_file_content = mock_ftplib.content_of("notthere")
     self.assertEqual(data, remote_file_content)
     # Clean up.
     os.unlink(local_source)