def test_get_task_queue_steps_None():
"""
Given the value of None return `merlin` as the queue name.
"""
steps = None
queue = Step.get_task_queue_from_dict(steps)
assert queue == "merlin"
def test_get_task_queue_run_missing():
"""
Given an empty steps dictionary return `merlin` as the queue name.
"""
steps = {}
queue = Step.get_task_queue_from_dict(steps)
assert queue == "merlin"
def step(self, task_name):
"""Return a Step object for the given task name
:param `task_name`: The task name.
:return: A Merlin Step object.
"""
return Step(self.dag.values[task_name])
def test_get_task_queue_none_str():
"""
Given a steps dictionary where the task_queue is set to the string value
'none', return `merlin` as the queue name.
"""
steps = {"run": {"task_queue": "none"}}
queue = Step.get_task_queue_from_dict(steps)
assert queue == "merlin"
def test_get_task_queue_None():
"""
Given a steps dictionary where the task_queue is set to None, return
`merlin` as the queue name.
"""
steps = {"run": {"task_queue": None}}
queue = Step.get_task_queue_from_dict(steps)
assert queue == "merlin"
def test_get_task_queue_task_queue_missing():
"""
Given a steps dictionary where the run is set to an empty dictionary
return `merlin` as the queue name.
"""
steps = {"run": {}}
queue = Step.get_task_queue_from_dict(steps)
assert queue == "merlin"
def test_get_task_queue_default():
"""
Given a steps dictionary that sets the task queue to `test_queue` return
`test_queue` as the queue name.
"""
steps = {"run": {"task_queue": "test_queue"}}
queue = Step.get_task_queue_from_dict(steps)
assert queue == "test_queue"