def test_fetch_latest_task(mock_client_factory, tasks):
mock_client = MagicMock()
mock_client.list_tasks_paginated = MagicMock(return_value=(tasks, 0))
mock_client_factory.return_value = mock_client
task = engine.FlyteEngineFactory().fetch_latest_task(_common_models.NamedEntityIdentifier("p", "d", "n"))
if tasks:
assert task.id == tasks[0].id
else:
assert not task
mock_client.list_tasks_paginated.assert_called_once_with(
_common_models.NamedEntityIdentifier("p", "d", "n"),
limit=1,
sort_by=_common.Sort("created_at", _common.Sort.Direction.DESCENDING),
)
def test_fetch_active_launch_plan(mock_client_factory):
mock_client = MagicMock()
mock_client.get_active_launch_plan = MagicMock(
return_value=_launch_plan_models.LaunchPlan(
identifier.Identifier(identifier.ResourceType.LAUNCH_PLAN, "p1", "d1", "n1", "v1"),
MagicMock(),
MagicMock(),
)
)
mock_client_factory.return_value = mock_client
lp = engine.FlyteEngineFactory().fetch_launch_plan(
identifier.Identifier(identifier.ResourceType.LAUNCH_PLAN, "p", "d", "n", "")
)
assert lp.id == identifier.Identifier(identifier.ResourceType.LAUNCH_PLAN, "p1", "d1", "n1", "v1")
mock_client.get_active_launch_plan.assert_called_once_with(_common_models.NamedEntityIdentifier("p", "d", "n"))
