def test_upload_writes_iv_and_encrypts_file_data(self, mockDropbox,
mockUploadCursor):
uploadSessionStarter = MagicMock()
uploadSessionAppender = MagicMock()
mockDropbox.return_value.files_upload_session_start = uploadSessionStarter
mockDropbox.return_value.files_upload_session_finish = uploadSessionAppender
testSecretData = b"secret test data"
testFileHandle = BytesIO(testSecretData)
testPartName = "testFile__1__1.enc"
testClientModified = datetime.datetime(2020, 1, 1, 10, 5, 30)
testFileData = {
"userTimezone": "+0200",
"utcModified": testClientModified.timestamp(),
"path": "subDir"
}
testTask = Task(taskType=MessageTypes.UPLOAD_FILE, data=testFileData)
cloudAccount = DropboxAccountWrapper(self.testAccountData)
result = cloudAccount.upload(testFileHandle, len(testSecretData),
testPartName, testTask)
cipher = AES.new(self.testAccountData.cryptoKey.encode(),
AES.MODE_CFB,
iv=uploadSessionStarter.call_args[0][0])
encoded = mockDropbox.return_value.files_upload_session_finish.call_args[
0][0]
self.assertNotEqual(encoded, testSecretData)
self.assertEqual(cipher.decrypt(encoded), testSecretData)
def test_download_file_does_nothing_if_task_is_stale(self, mockRequest):
secretData = b"secret test data"
testEncoder = AES.new(self.testAccountData.cryptoKey.encode(),
AES.MODE_CFB)
testIV = testEncoder.iv
ivResponse = MagicMock()
ivResponse.content = testIV
encryptedDataResponse = MagicMock()
encryptedDataResponse.content = testEncoder.encrypt(secretData)
mockRequest.side_effect = [ivResponse, encryptedDataResponse]
testDownloadFileHandle = BytesIO()
testFilePartInfo = FilePart(
filename="apple.txt__1__1.enc",
modified=int(
datetime.datetime(2020, 1, 1, 10, 10, 30).timestamp()),
size=len(encryptedDataResponse.content) + len(testIV),
path="",
fullPath="apple.txt__1__1.enc",
storingAccountID=self.testAccountData.id,
extraInfo={})
testDownloadFileTask = Task(taskType=MessageTypes.DOWNLOAD_FILE,
stale=True)
cloudAccount = DropboxAccountWrapper(self.testAccountData)
cloudAccount.download(testDownloadFileHandle, testFilePartInfo,
testDownloadFileTask)
testDownloadFileHandle.seek(0)
self.assertEqual(b"", testDownloadFileHandle.read())
def test_upload_does_nothing_if_task_is_stale(self, mockDropbox,
mockUploadCursor):
uploadSessionStarter = MagicMock()
uploadSessionAppender = MagicMock()
mockDropbox.return_value.files_upload_session_start = uploadSessionStarter
mockDropbox.return_value.files_upload_session_finish = uploadSessionAppender
testSecretData = b"secret test data"
testFileHandle = BytesIO(testSecretData)
testPartName = "testFile__1__1.enc"
testClientModified = datetime.datetime(2020, 1, 1, 10, 5, 30)
testFileData = {
"userTimezone": "+0200",
"utcModified": testClientModified.timestamp(),
"path": "subDir"
}
testTask = Task(taskType=MessageTypes.UPLOAD_FILE,
data=testFileData,
stale=True)
cloudAccount = DropboxAccountWrapper(self.testAccountData)
result = cloudAccount.upload(testFileHandle, len(testSecretData),
testPartName, testTask)
self.assertEqual(uploadSessionStarter.call_count, 0)
self.assertEqual(uploadSessionAppender.call_count, 0)
self.assertIsNone(result)
def test_getFileList_returns_encrypted_files_recursively(
self, mockDropbox):
testCloudFiles = [
FileMetadata(name="apple.txt__1__1.enc",
size=5,
client_modified=datetime.datetime(
2020, 1, 1, 6, 10, 15),
path_display="/apple.txt__1__1.enc"),
FolderMetadata(name="thisFolderShouldBeIgnored"),
FileMetadata(name="pear.txt__1__2.enc",
size=10,
client_modified=datetime.datetime(
2020, 1, 1, 7, 10, 15),
path_display="/subDir/pear.txt__1__2.enc"),
FileMetadata(name="notAnEncodedFile.jpeg",
size=20,
client_modified=datetime.datetime(
2020, 1, 1, 7, 10, 15),
path_display="/notAnEncodedFile.jpeg"),
]
mockDropbox.return_value.files_list_folder.return_value.entries = testCloudFiles
cloudAccount = DropboxAccountWrapper(self.testAccountData)
files = cloudAccount.getFileList()
self.assertEqual(len(files), 2)
self.assertEqual(
mockDropbox.return_value.files_list_folder.call_args[0][0], "")
self.assertEqual(
mockDropbox.return_value.files_list_folder.call_args[1],
{"recursive": True})
self.assertEqual(files[0].filename, testCloudFiles[0].name)
self.assertEqual(files[0].modified,
int(testCloudFiles[0].client_modified.timestamp()))
self.assertEqual(files[0].size, testCloudFiles[0].size)
self.assertEqual(files[0].path, "")
self.assertEqual(files[0].fullPath, "apple.txt__1__1.enc")
self.assertEqual(files[0].storingAccountID, self.testAccountData.id)
self.assertEqual(files[1].filename, testCloudFiles[2].name)
self.assertEqual(files[1].modified,
int(testCloudFiles[2].client_modified.timestamp()))
self.assertEqual(files[1].size, testCloudFiles[2].size)
self.assertEqual(files[1].path, "subDir")
self.assertEqual(files[1].fullPath, "subDir/pear.txt__1__2.enc")
self.assertEqual(files[1].storingAccountID, self.testAccountData.id)
def test_deleteFile_delegates_call_properly(self, mockDropbox):
mockDropbox.return_value.files_delete = MagicMock()
testFilePartInfo = FilePart(filename="apple.txt__1__1.enc",
modified=int(
datetime.datetime(
2020, 1, 1, 10, 10,
30).timestamp()),
size=32,
path="subDir",
fullPath="subDir/apple.txt__1__1.enc",
storingAccountID=self.testAccountData.id,
extraInfo={})
cloudAccount = DropboxAccountWrapper(self.testAccountData)
cloudAccount.deleteFile(testFilePartInfo)
self.assertEqual(mockDropbox.return_value.files_delete.call_count, 1)
self.assertEqual(mockDropbox.return_value.files_delete.call_args[0][0],
f"/{testFilePartInfo.fullPath}")