def test_get_execution_outputs(mock_client_factory, execution_data_locations):
mock_client = MagicMock()
mock_client.get_execution_data = MagicMock(
return_value=_execution_models.WorkflowExecutionGetDataResponse(
execution_data_locations[0],
execution_data_locations[1]
)
)
mock_client_factory.return_value = mock_client
m = MagicMock()
type(m).id = PropertyMock(
return_value=identifier.WorkflowExecutionIdentifier(
"project",
"domain",
"name",
)
)
inputs = engine.FlyteWorkflowExecution(m).get_outputs()
assert len(inputs.literals) == 1
assert inputs.literals['b'].scalar.primitive.integer == 2
mock_client.get_execution_data.assert_called_once_with(
identifier.WorkflowExecutionIdentifier("project", "domain", "name")
)
def test_get_full_execution_inputs(mock_client_factory):
mock_client = MagicMock()
mock_client.get_execution_data = MagicMock(
return_value=_execution_models.WorkflowExecutionGetDataResponse(
None,
None,
_INPUT_MAP,
_OUTPUT_MAP,
)
)
mock_client_factory.return_value = mock_client
m = MagicMock()
type(m).id = PropertyMock(
return_value=identifier.WorkflowExecutionIdentifier(
"project",
"domain",
"name",
)
)
inputs = engine.FlyteWorkflowExecution(m).get_inputs()
assert len(inputs.literals) == 1
assert inputs.literals["a"].scalar.primitive.integer == 1
mock_client.get_execution_data.assert_called_once_with(
identifier.WorkflowExecutionIdentifier("project", "domain", "name")
)