def test_remove_files(self, subproc_mock, os_mock): """S3Payload - Shred Temp Directory""" S3Payload._shred_temp_directory() subproc_mock.assert_called_with( ['shred', '--force', '--iterations=1', '--remove', self.temp_file]) os_mock.assert_called_with(self.temp_dir)
def test_read_local_s3_obj_non_gz(): """S3Payload - Read S3 Object On Disk, non-gzipped""" temp_file_path = os.path.join(tempfile.gettempdir(), 's3_test.json') with open(temp_file_path, 'w') as temp_file: temp_file.write('test line of data') for line_num, line in S3Payload._read_downloaded_s3_object(temp_file_path): assert_equal(line_num, 1) assert_equal(line, 'test line of data')
def test_read_s3_failed_remove(_, log_mock): """S3Payload - Read S3 Object On Disk, Removal Failure""" temp_file_path = os.path.join(tempfile.gettempdir(), 's3_test.json') with open(temp_file_path, 'w') as temp_file: temp_file.write('test line of data') _ = [_ for _ in S3Payload._read_downloaded_s3_object(temp_file_path)] log_mock.assert_called_with('Failed to remove temp S3 file: %s', temp_file_path)