def test_ok_to_continue(self):
# to do this test, I need to first add the current cache file to the array, then try to check ok to continue,
# but then on another thread, wait
download_thread = threading.Thread(
target=TestCacheFileOKToContinue.block_cache_then_release,
args=[self.test_cache_file_path])
c = CF(working_directory=self.temp_dir)
download_thread.start()
self.assertTrue(c.ok_to_continue())
def test_adding_config_to_existing_file(self):
create_workflow_file_in_dir(self.test_cache_file_path)
c = CF(working_directory=self.temp_dir)
c.add_result('existingWorkflow', 'fileA', {'columnA': 'dataA'})
c.add_config('existingWorkflow', 'fileA', {'hey': 'something'})
self.assertIn('existingWorkflow', c.workflow_state[CF.RootKey])
self.assertIn(
'fileA',
c.workflow_state[CF.RootKey]['existingWorkflow'][CF.FilesKey])
self.assertEqual(
'dataA', c.workflow_state[CF.RootKey]['existingWorkflow'][
CF.FilesKey]['fileA'][CF.ResultsKey]['columnA'])
def test_get_files_for_workflow(self):
create_workflow_file_in_dir(self.test_cache_file_path)
c = CF(working_directory=self.temp_dir)
c.add_result('workflowA', 'fileA', {'columnA': 'dataA'})
c.add_result('workflowA', 'fileB', {'columnA': 'dataB'})
c.add_result('workflowF', 'fileA', {'columnA': 'dataC'})
c.add_result('workflowF', 'fileQ', {'columnA': 'dataD'})
files_in_workflow = c.get_files_for_workflow('workflowA')
self.assertIsInstance(files_in_workflow, dict)
self.assertEqual(2, len(files_in_workflow))
self.assertListEqual(['fileA', 'fileB'],
list(files_in_workflow.keys()))
def test_skips_writing_clean_file(self):
create_workflow_file_in_dir(self.test_cache_file_path)
c = CF(working_directory=self.temp_dir)
c.write()
def test_reading_existing_malformed_cache_file(self):
create_workflow_file_in_dir(self.test_cache_file_path,
create_root=False)
c = CF(working_directory=self.temp_dir)
with self.assertRaises(EPLaunchFileException):
c.read()
def test_reading_existing_valid_cache_file(self):
create_workflow_file_in_dir(self.test_cache_file_path)
c = CF(working_directory=self.temp_dir)
c.read()
def test_empty_directory_creates_cache_file(self):
c = CF(working_directory=self.temp_dir)
c.read()
c.write()
self.assertTrue(os.path.exists(self.test_cache_file_path))
def test_never_ok_to_continue(self):
c = CF(working_directory=self.temp_dir)
cache_files_currently_updating_or_writing.append(
self.test_cache_file_path)
self.assertFalse(c.ok_to_continue())
def test_get_files_for_empty_workflow(self):
create_workflow_file_in_dir(self.test_cache_file_path)
c = CF(working_directory=self.temp_dir)
files_in_workflow = c.get_files_for_workflow('workflowA')
self.assertIsInstance(files_in_workflow, dict)
self.assertEqual(0, len(files_in_workflow))