def test_get_workflow_tokens(self, repository_mock):
repository = mock.Mock()
repository_mock.return_value = repository
repository.get_job_names.return_value = ['some_job']
job_config = JobConfig()
job_config.workflow = 'some_workflow'
job_config.job = 'some_job'
job_config.is_condition = False
job_config.template = ('tests.pinball.parser.'
'repository_config_parser_test.SomeJobTemplate')
job_config.template_params = {'some_param': 'some_value'}
job_config.parents = []
job_config.emails = [
'*****@*****.**', '*****@*****.**'
]
job_config.max_attempts = 10
job_config.retry_delay_sec = 20
job_config.priority = 100
repository.get_job.return_value = job_config
parser = RepositoryConfigParser()
workflow_tokens = parser.get_workflow_tokens('some_workflow')
self.assertEqual(2, len(workflow_tokens))
# Verify the triggering event token.
if Name.from_event_token_name(workflow_tokens[0].name).workflow:
event_token = workflow_tokens[0]
else:
event_token = workflow_tokens[1]
event_name = Name.from_event_token_name(event_token.name)
self.assertEqual('some_workflow', event_name.workflow)
self.assertEqual('some_job', event_name.job)
self.assertEqual('__WORKFLOW_START__', event_name.input)
event = pickle.loads(event_token.data)
self.assertEqual('repository_config_parser', event.creator)
repository.get_job_names.assert_called_once_with('some_workflow')
repository.get_job.assert_called_once_with('some_workflow', 'some_job')
def test_get_workflow_tokens(self, repository_mock):
repository = mock.Mock()
repository_mock.return_value = repository
repository.get_job_names.return_value = ['some_job']
job_config = JobConfig()
job_config.workflow = 'some_workflow'
job_config.job = 'some_job'
job_config.is_condition = False
job_config.template = ('tests.pinball.parser.'
'repository_config_parser_test.SomeJobTemplate')
job_config.template_params = {'some_param': 'some_value'}
job_config.parents = []
job_config.emails = ['*****@*****.**',
'*****@*****.**']
job_config.max_attempts = 10
job_config.retry_delay_sec = 20
job_config.priority = 100
repository.get_job.return_value = job_config
parser = RepositoryConfigParser()
workflow_tokens = parser.get_workflow_tokens('some_workflow')
self.assertEqual(2, len(workflow_tokens))
# Verify the triggering event token.
if Name.from_event_token_name(workflow_tokens[0].name).workflow:
event_token = workflow_tokens[0]
else:
event_token = workflow_tokens[1]
event_name = Name.from_event_token_name(event_token.name)
self.assertEqual('some_workflow', event_name.workflow)
self.assertEqual('some_job', event_name.job)
self.assertEqual('__WORKFLOW_START__', event_name.input)
event = pickle.loads(event_token.data)
self.assertEqual('repository_config_parser', event.creator)
repository.get_job_names.assert_called_once_with('some_workflow')
repository.get_job.assert_called_once_with('some_workflow', 'some_job')
