def test_execution_notification_soft_overrides(mock_client_factory):
mock_client = MagicMock()
mock_client.create_execution = MagicMock(
return_value=identifier.WorkflowExecutionIdentifier('xp', 'xd', 'xn'))
mock_client_factory.return_value = mock_client
m = MagicMock()
type(m).id = PropertyMock(return_value=identifier.Identifier(
identifier.ResourceType.LAUNCH_PLAN, "project", "domain", "name",
"version"))
notification = _common_models.Notification(
[0, 1, 2], email=_common_models.EmailNotification(["*****@*****.**"]))
engine.FlyteLaunchPlan(m).execute('xp',
'xd',
'xn',
literals.LiteralMap({}),
notification_overrides=[notification])
mock_client.create_execution.assert_called_once_with(
'xp', 'xd', 'xn',
_execution_models.ExecutionSpec(
identifier.Identifier(identifier.ResourceType.LAUNCH_PLAN,
"project", "domain", "name", "version"),
literals.LiteralMap({}),
_execution_models.ExecutionMetadata(
_execution_models.ExecutionMetadata.ExecutionMode.MANUAL,
'sdk', 0),
notifications=_execution_models.NotificationList([notification]),
))
def test_execution_notification_overrides(mock_client_factory):
mock_client = MagicMock()
mock_client.create_execution = MagicMock(
return_value=identifier.WorkflowExecutionIdentifier('xp', 'xd', 'xn'))
mock_client_factory.return_value = mock_client
m = MagicMock()
type(m).id = PropertyMock(return_value=identifier.Identifier(
identifier.ResourceType.LAUNCH_PLAN, "project", "domain", "name",
"version"))
engine.FlyteLaunchPlan(m).execute('xp',
'xd',
'xn',
literals.LiteralMap({}),
notification_overrides=[])
mock_client.create_execution.assert_called_once_with(
'xp', 'xd', 'xn',
_execution_models.ExecutionSpec(
identifier.Identifier(identifier.ResourceType.LAUNCH_PLAN,
"project", "domain", "name", "version"),
literals.LiteralMap({}),
_execution_models.ExecutionMetadata(
_execution_models.ExecutionMetadata.ExecutionMode.MANUAL,
'sdk', 0),
disable_all=True,
))
def test_execution_annotation_overrides(mock_client_factory):
mock_client = MagicMock()
mock_client.create_execution = MagicMock(
return_value=identifier.WorkflowExecutionIdentifier("xp", "xd", "xn"))
mock_client_factory.return_value = mock_client
m = MagicMock()
type(m).id = PropertyMock(return_value=identifier.Identifier(
identifier.ResourceType.LAUNCH_PLAN, "project", "domain", "name",
"version"))
annotations = _common_models.Annotations({"my": "annotation"})
engine.FlyteLaunchPlan(m).launch(
"xp",
"xd",
"xn",
literals.LiteralMap({}),
notification_overrides=[],
annotation_overrides=annotations,
)
mock_client.create_execution.assert_called_once_with(
"xp",
"xd",
"xn",
_execution_models.ExecutionSpec(
identifier.Identifier(
identifier.ResourceType.LAUNCH_PLAN,
"project",
"domain",
"name",
"version",
),
_execution_models.ExecutionMetadata(
_execution_models.ExecutionMetadata.ExecutionMode.MANUAL,
"sdk", 0),
disable_all=True,
annotations=annotations,
),
literals.LiteralMap({}),
)
