def test_download(self, mock_container, mock_exists): files_audit.audit_temp_path = os.path.join( storage_settings.AUDIT_TEMP_PATH) file_contents = ['fake', 'file', 'content'] mock_obj = mock.Mock() mock_obj.fetch.return_value = iter(file_contents) mock_container.get_object.return_value = mock_obj mock_exists.return_value = False version = FileVersionFactory() version.metadata = { 'sha256': 'dff32002043d7a4da7173d2034cb2f6856d10549bd8cc6d7e16d62f1304681f8' } # fakefilecontent mock_open = mock.mock_open() with mock.patch('scripts.osfstorage.files_audit.open', mock_open, create=True): download_from_cloudfiles(version) mock_container.get_object.assert_called_with( version.location['object']) mock_open.assert_called_once_with( os.path.join(os.path.join(storage_settings.AUDIT_TEMP_PATH), version.location['object']), 'wb') handle = mock_open() assert_equal(handle.write.call_count, 3) for content in file_contents: handle.write.assert_any_call(content)
def test_download(self, mock_container, mock_exists): mock_object = mock.Mock() mock_container.get_object.return_value = mock_object mock_exists.return_value = False version = FileVersionFactory() download_from_cloudfiles(version) mock_container.get_object.assert_called_with(version.location['object']) mock_object.download.assert_called_with(storage_settings.AUDIT_TEMP_PATH)
def test_download(self, mock_exists): files_audit.audit_temp_path = os.path.join(storage_settings.AUDIT_TEMP_PATH) file_contents = ['fake', 'file', 'content'] self.ctx.container_primary = mock.Mock() self.ctx.container_primary.get_object.return_value.fetch.return_value = iter(file_contents) version = FileVersionFactory() version.metadata = {'sha256': 'dff32002043d7a4da7173d2034cb2f6856d10549bd8cc6d7e16d62f1304681f8'} # fakefilecontent mock_open = mock.mock_open() with mock.patch('scripts.osfstorage.files_audit.open', mock_open, create=True): download_from_cloudfiles(self.ctx, version) self.ctx.container_primary.get_object.assert_called_with(version.location['object']) mock_open.assert_called_once_with(os.path.join(os.path.join(storage_settings.AUDIT_TEMP_PATH), version.location['object']), 'wb') handle = mock_open() assert_equal(handle.write.call_count, 3) for content in file_contents: handle.write.assert_any_call(content)