def test_create_persistence_job(self, mock_uuid, mock_pickle):
file_name = '1'
mock_pickle.dump.return_value = True
mock_uuid.return_value = file_name
exporter_config = ExporterConfig(self.config)
try:
persistence = PicklePersistence(
exporter_config.persistence_options, meta())
self.assertIsInstance(persistence, PicklePersistence)
persistence.close()
finally:
remove_if_exists('/tmp/'+file_name)
def test_create_persistence_job(self, mock_uuid, mock_pickle):
file_name = '1'
mock_pickle.dump.return_value = True
mock_uuid.return_value = file_name
exporter_config = ExporterConfig(self.config)
try:
persistence = PicklePersistence(
exporter_config.persistence_options, meta())
self.assertIsInstance(persistence, PicklePersistence)
persistence.close()
finally:
remove_if_exists('/tmp/' + file_name)
def delete(self):
remove_if_exists(self.persistence_file_name)
def tearDown(self):
self.mock_s3.stop()
remove_if_exists(self.tmp_bypass_resume_file)
def clean_tmp_files(self, write_info):
remove_if_exists(write_info.get('path'))
remove_if_exists(write_info.get('file_path'))
def delete(self):
remove_if_exists(self.persistence_file_name)
def clean_tmp_files(self, write_info):
remove_if_exists(write_info.get('path'))
remove_if_exists(write_info.get('file_path'))
def tearDown(self):
remove_if_exists(self.data_dir + self.tmp_bypass_resume_file)
def tearDown(self):
self.mock_s3.stop()
remove_if_exists(self.tmp_bypass_resume_file)
def tearDown(self):
remove_if_exists(self.data_dir + self.tmp_bypass_resume_file)
