azure-functions-durable 1.2.9__py3-none-any.whl → 1.3.0__py3-none-any.whl

tests/orchestrator/test_serialization.py

@@ -1,31 +1,31 @@
-from azure.durable_functions.models.ReplaySchema import ReplaySchema
-from tests.test_utils.ContextBuilder import ContextBuilder
-from .orchestrator_test_utils \
-    import get_orchestration_state_result, assert_orchestration_state_equals, assert_valid_schema
-from azure.durable_functions.models.OrchestratorState import OrchestratorState
-
-def base_expected_state(output=None, replay_schema: ReplaySchema = ReplaySchema.V1) -> OrchestratorState:
-    return OrchestratorState(is_done=False, actions=[], output=output, replay_schema=replay_schema.value)
-
-def generator_function(context):
-    return False
-
-def test_serialization_of_False():
-    """Test that an orchestrator can return False."""
-
-    context_builder = ContextBuilder("serialize False")
-
-    result = get_orchestration_state_result(
-        context_builder, generator_function)
-
-    expected_state = base_expected_state(output=False)
-
-    expected_state._is_done = True
-    expected = expected_state.to_json()
-
-    # Since we're essentially testing the `to_json` functionality,
-    # we explicitely ensure that the output is set
-    expected["output"] = False
-
-    assert_valid_schema(result)
+from azure.durable_functions.models.ReplaySchema import ReplaySchema
+from tests.test_utils.ContextBuilder import ContextBuilder
+from .orchestrator_test_utils \
+    import get_orchestration_state_result, assert_orchestration_state_equals, assert_valid_schema
+from azure.durable_functions.models.OrchestratorState import OrchestratorState
+
+def base_expected_state(output=None, replay_schema: ReplaySchema = ReplaySchema.V1) -> OrchestratorState:
+    return OrchestratorState(is_done=False, actions=[], output=output, replay_schema=replay_schema.value)
+
+def generator_function(context):
+    return False
+
+def test_serialization_of_False():
+    """Test that an orchestrator can return False."""
+
+    context_builder = ContextBuilder("serialize False")
+
+    result = get_orchestration_state_result(
+        context_builder, generator_function)
+
+    expected_state = base_expected_state(output=False)
+
+    expected_state._is_done = True
+    expected = expected_state.to_json()
+
+    # Since we're essentially testing the `to_json` functionality,
+    # we explicitely ensure that the output is set
+    expected["output"] = False
+
+    assert_valid_schema(result)
     assert_orchestration_state_equals(expected, result)

tests/orchestrator/test_sub_orchestrator.py

@@ -1,95 +1,95 @@
-from azure.durable_functions.models.ReplaySchema import ReplaySchema
-from .orchestrator_test_utils \
-    import assert_orchestration_state_equals, get_orchestration_state_result, assert_valid_schema
-from tests.test_utils.ContextBuilder import ContextBuilder
-from azure.durable_functions.models.OrchestratorState import OrchestratorState
-from azure.durable_functions.models.actions.CallSubOrchestratorAction \
-    import CallSubOrchestratorAction
-import azure.durable_functions as df
-import azure.functions as func
-
-app = df.DFApp(http_auth_level=func.AuthLevel.ANONYMOUS)
-
-def generator_function(context):
-    outputs = []
-    task1 = yield context.call_sub_orchestrator("HelloSubOrchestrator", "Tokyo")
-    task2 = yield context.call_sub_orchestrator("HelloSubOrchestrator", "Seattle")
-    task3 = yield context.call_sub_orchestrator("HelloSubOrchestrator", "London")
-
-    outputs.append(task1)
-    outputs.append(task2)
-    outputs.append(task3)
-
-    return outputs
-
-def generator_function_call_by_function_name(context):
-    outputs = []
-    task1 = yield context.call_sub_orchestrator(HelloSubOrchestrator, "Tokyo")
-    task2 = yield context.call_sub_orchestrator(HelloSubOrchestrator, "Seattle")
-    task3 = yield context.call_sub_orchestrator(HelloSubOrchestrator, "London")
-
-    outputs.append(task1)
-    outputs.append(task2)
-    outputs.append(task3)
-
-    return outputs
-
-@app.orchestration_trigger(context_name="context")
-def HelloSubOrchestrator(context):
-    return "Hello" + context
-
-def base_expected_state(output=None, replay_schema: ReplaySchema = ReplaySchema.V1) -> OrchestratorState:
-    return OrchestratorState(is_done=False, actions=[], output=output, replay_schema=replay_schema.value)
-
-
-def add_hello_suborch_action(state: OrchestratorState, input_: str):
-    action = CallSubOrchestratorAction(function_name='HelloSubOrchestrator', _input=input_)
-    state.actions.append([action])
-
-
-def add_hello_suborch_completed_events(
-        context_builder: ContextBuilder, id_: int, result: str):
-    context_builder.add_sub_orchestrator_started_event(name="HelloSubOrchestrator", id_=id_,input_="")
-    context_builder.add_orchestrator_completed_event()
-    context_builder.add_orchestrator_started_event()
-    context_builder.add_sub_orchestrator_completed_event(result=result, id_=id_)
-
-def test_tokyo_and_seattle_and_london_state():
-    context_builder = ContextBuilder('test_simple_function')
-    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
-    add_hello_suborch_completed_events(context_builder, 1, "\"Hello Seattle!\"")
-    add_hello_suborch_completed_events(context_builder, 2, "\"Hello London!\"")
-
-    result = get_orchestration_state_result(
-        context_builder, generator_function)
-
-    expected_state = base_expected_state(
-        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
-    add_hello_suborch_action(expected_state, 'Tokyo')
-    add_hello_suborch_action(expected_state, 'Seattle')
-    add_hello_suborch_action(expected_state, 'London')
-    expected_state._is_done = True
-    expected = expected_state.to_json()
-
-    #assert_valid_schema(result)
-    assert_orchestration_state_equals(expected, result)
-
-def test_call_suborchestrator_by_name():
-    context_builder = ContextBuilder('test_call_suborchestrator_by_name')
-    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
-    add_hello_suborch_completed_events(context_builder, 1, "\"Hello Seattle!\"")
-    add_hello_suborch_completed_events(context_builder, 2, "\"Hello London!\"")
-
-    result = get_orchestration_state_result(
-        context_builder, generator_function_call_by_function_name)
-
-    expected_state = base_expected_state(
-        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
-    add_hello_suborch_action(expected_state, 'Tokyo')
-    add_hello_suborch_action(expected_state, 'Seattle')
-    add_hello_suborch_action(expected_state, 'London')
-    expected_state._is_done = True
-    expected = expected_state.to_json()
-
-    #assert_valid_schema(result)
-    assert_orchestration_state_equals(expected, result)
+from azure.durable_functions.models.ReplaySchema import ReplaySchema
+from .orchestrator_test_utils \
+    import assert_orchestration_state_equals, get_orchestration_state_result, assert_valid_schema
+from tests.test_utils.ContextBuilder import ContextBuilder
+from azure.durable_functions.models.OrchestratorState import OrchestratorState
+from azure.durable_functions.models.actions.CallSubOrchestratorAction \
+    import CallSubOrchestratorAction
+import azure.durable_functions as df
+import azure.functions as func
+
+app = df.DFApp(http_auth_level=func.AuthLevel.ANONYMOUS)
+
+def generator_function(context):
+    outputs = []
+    task1 = yield context.call_sub_orchestrator("HelloSubOrchestrator", "Tokyo")
+    task2 = yield context.call_sub_orchestrator("HelloSubOrchestrator", "Seattle")
+    task3 = yield context.call_sub_orchestrator("HelloSubOrchestrator", "London")
+
+    outputs.append(task1)
+    outputs.append(task2)
+    outputs.append(task3)
+
+    return outputs
+
+def generator_function_call_by_function_name(context):
+    outputs = []
+    task1 = yield context.call_sub_orchestrator(HelloSubOrchestrator, "Tokyo")
+    task2 = yield context.call_sub_orchestrator(HelloSubOrchestrator, "Seattle")
+    task3 = yield context.call_sub_orchestrator(HelloSubOrchestrator, "London")
+
+    outputs.append(task1)
+    outputs.append(task2)
+    outputs.append(task3)
+
+    return outputs
+
+@app.orchestration_trigger(context_name="context")
+def HelloSubOrchestrator(context):
+    return "Hello" + context
+
+def base_expected_state(output=None, replay_schema: ReplaySchema = ReplaySchema.V1) -> OrchestratorState:
+    return OrchestratorState(is_done=False, actions=[], output=output, replay_schema=replay_schema.value)
+
+
+def add_hello_suborch_action(state: OrchestratorState, input_: str):
+    action = CallSubOrchestratorAction(function_name='HelloSubOrchestrator', _input=input_)
+    state.actions.append([action])
+
+
+def add_hello_suborch_completed_events(
+        context_builder: ContextBuilder, id_: int, result: str):
+    context_builder.add_sub_orchestrator_started_event(name="HelloSubOrchestrator", id_=id_,input_="")
+    context_builder.add_orchestrator_completed_event()
+    context_builder.add_orchestrator_started_event()
+    context_builder.add_sub_orchestrator_completed_event(result=result, id_=id_)
+
+def test_tokyo_and_seattle_and_london_state():
+    context_builder = ContextBuilder('test_simple_function')
+    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
+    add_hello_suborch_completed_events(context_builder, 1, "\"Hello Seattle!\"")
+    add_hello_suborch_completed_events(context_builder, 2, "\"Hello London!\"")
+
+    result = get_orchestration_state_result(
+        context_builder, generator_function)
+
+    expected_state = base_expected_state(
+        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
+    add_hello_suborch_action(expected_state, 'Tokyo')
+    add_hello_suborch_action(expected_state, 'Seattle')
+    add_hello_suborch_action(expected_state, 'London')
+    expected_state._is_done = True
+    expected = expected_state.to_json()
+
+    #assert_valid_schema(result)
+    assert_orchestration_state_equals(expected, result)
+
+def test_call_suborchestrator_by_name():
+    context_builder = ContextBuilder('test_call_suborchestrator_by_name')
+    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
+    add_hello_suborch_completed_events(context_builder, 1, "\"Hello Seattle!\"")
+    add_hello_suborch_completed_events(context_builder, 2, "\"Hello London!\"")
+
+    result = get_orchestration_state_result(
+        context_builder, generator_function_call_by_function_name)
+
+    expected_state = base_expected_state(
+        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
+    add_hello_suborch_action(expected_state, 'Tokyo')
+    add_hello_suborch_action(expected_state, 'Seattle')
+    add_hello_suborch_action(expected_state, 'London')
+    expected_state._is_done = True
+    expected = expected_state.to_json()
+
+    #assert_valid_schema(result)
+    assert_orchestration_state_equals(expected, result)

tests/orchestrator/test_sub_orchestrator_with_retry.py

@@ -1,130 +1,130 @@
-from azure.durable_functions.models.ReplaySchema import ReplaySchema
-from .orchestrator_test_utils \
-    import assert_orchestration_state_equals, get_orchestration_state_result, assert_valid_schema
-from tests.test_utils.ContextBuilder import ContextBuilder
-from azure.durable_functions.models.OrchestratorState import OrchestratorState
-from azure.durable_functions.models.RetryOptions import RetryOptions
-from azure.durable_functions.models.actions.CallSubOrchestratorWithRetryAction \
-    import CallSubOrchestratorWithRetryAction
-
-RETRY_OPTIONS = RetryOptions(5000, 3)
-
-def generator_function(context):
-    outputs = []
-    retry_options = RETRY_OPTIONS
-
-    task1 = yield context.call_sub_orchestrator_with_retry("HelloSubOrchestrator", retry_options, "Tokyo")
-    task2 = yield context.call_sub_orchestrator_with_retry("HelloSubOrchestrator", retry_options, "Seattle")
-    task3 = yield context.call_sub_orchestrator_with_retry("HelloSubOrchestrator", retry_options, "London")
-
-    outputs.append(task1)
-    outputs.append(task2)
-    outputs.append(task3)
-
-    return outputs
-
-def base_expected_state(output=None, replay_schema: ReplaySchema = ReplaySchema.V1) -> OrchestratorState:
-    return OrchestratorState(is_done=False, actions=[], output=output, replay_schema=replay_schema.value)
-
-
-def add_hello_suborch_action(state: OrchestratorState, input_: str):
-    retry_options = RETRY_OPTIONS
-    action = CallSubOrchestratorWithRetryAction(function_name='HelloSubOrchestrator',
-        retry_options=retry_options,_input=input_)
-    state.actions.append([action])
-
-
-def add_hello_suborch_completed_events(
-        context_builder: ContextBuilder, id_: int, result: str):
-    context_builder.add_sub_orchestrator_started_event(name="HelloSubOrchestrator", id_=id_,input_="")
-    context_builder.add_orchestrator_completed_event()
-    context_builder.add_orchestrator_started_event()
-    context_builder.add_sub_orchestrator_completed_event(result=result, id_=id_)
-
-def add_hello_suborch_failed_events(
-        context_builder: ContextBuilder, id_: int, reason: str, details: str):
-    context_builder.add_sub_orchestrator_started_event(name="HelloSubOrchestrator", id_=id_,input_="")
-    context_builder.add_orchestrator_completed_event()
-    context_builder.add_orchestrator_started_event()
-    context_builder.add_sub_orchestrator_failed_event(id_=id_, reason=reason, details=details)
-
-def add_retry_timer_events(context_builder: ContextBuilder, id_: int):
-    fire_at = context_builder.add_timer_created_event(id_)
-    context_builder.add_orchestrator_completed_event()
-    context_builder.add_orchestrator_started_event()
-    context_builder.add_timer_fired_event(id_=id_, fire_at=fire_at)
-
-def test_tokyo_and_seattle_and_london_state():
-    context_builder = ContextBuilder('test_simple_function')
-    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
-    add_hello_suborch_completed_events(context_builder, 1, "\"Hello Seattle!\"")
-    add_hello_suborch_completed_events(context_builder, 2, "\"Hello London!\"")
-
-    result = get_orchestration_state_result(
-        context_builder, generator_function)
-
-    expected_state = base_expected_state(
-        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
-    add_hello_suborch_action(expected_state, 'Tokyo')
-    add_hello_suborch_action(expected_state, 'Seattle')
-    add_hello_suborch_action(expected_state, 'London')
-    expected_state._is_done = True
-    expected = expected_state.to_json()
-
-    #assert_valid_schema(result)
-    assert_orchestration_state_equals(expected, result)
-
-
-def test_tokyo_and_seattle_and_london_state_partial_failure():
-    failed_reason = 'Reasons'
-    failed_details = 'Stuff and Things'
-    context_builder = ContextBuilder('test_simple_function')
-    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
-    add_hello_suborch_failed_events(context_builder, 1, failed_reason, failed_details)
-    add_retry_timer_events(context_builder, 2)
-    add_hello_suborch_completed_events(context_builder, 3, "\"Hello Seattle!\"")
-    add_hello_suborch_completed_events(context_builder, 4, "\"Hello London!\"")
-
-    result = get_orchestration_state_result(
-        context_builder, generator_function)
-
-    expected_state = base_expected_state(
-        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
-    add_hello_suborch_action(expected_state, 'Tokyo')
-    add_hello_suborch_action(expected_state, 'Seattle')
-    add_hello_suborch_action(expected_state, 'London')
-    expected_state._is_done = True
-    expected = expected_state.to_json()
-
-    #assert_valid_schema(result)
-    assert_orchestration_state_equals(expected, result)
-
-def test_tokyo_and_seattle_and_london_state_all_failed():
-    failed_reason = 'Reasons'
-    failed_details = 'Stuff and Things'
-    context_builder = ContextBuilder('test_simple_function')
-    add_hello_suborch_failed_events(context_builder, 0, failed_reason, failed_details)
-    add_retry_timer_events(context_builder, 1)
-    add_hello_suborch_failed_events(context_builder, 2, failed_reason, failed_details)
-    add_retry_timer_events(context_builder, 3)
-    add_hello_suborch_failed_events(context_builder, 4, failed_reason, failed_details)
-    add_retry_timer_events(context_builder, 5)
-
-    try:
-        result = get_orchestration_state_result(
-            context_builder, generator_function)
-        # Should have error'ed out
-        assert False
-    except Exception as e:
-        error_label = "\n\n$OutOfProcData$:"
-        error_str = str(e)
-    
-        expected_state = base_expected_state()
-        add_hello_suborch_action(expected_state, 'Tokyo')
-
-        error_msg = f'{failed_reason} \n {failed_details}'
-        expected_state._error = error_msg
-        state_str = expected_state.to_json_string()
-        
-        expected_error_str = f"{error_msg}{error_label}{state_str}"
+from azure.durable_functions.models.ReplaySchema import ReplaySchema
+from .orchestrator_test_utils \
+    import assert_orchestration_state_equals, get_orchestration_state_result, assert_valid_schema
+from tests.test_utils.ContextBuilder import ContextBuilder
+from azure.durable_functions.models.OrchestratorState import OrchestratorState
+from azure.durable_functions.models.RetryOptions import RetryOptions
+from azure.durable_functions.models.actions.CallSubOrchestratorWithRetryAction \
+    import CallSubOrchestratorWithRetryAction
+
+RETRY_OPTIONS = RetryOptions(5000, 3)
+
+def generator_function(context):
+    outputs = []
+    retry_options = RETRY_OPTIONS
+
+    task1 = yield context.call_sub_orchestrator_with_retry("HelloSubOrchestrator", retry_options, "Tokyo")
+    task2 = yield context.call_sub_orchestrator_with_retry("HelloSubOrchestrator", retry_options, "Seattle")
+    task3 = yield context.call_sub_orchestrator_with_retry("HelloSubOrchestrator", retry_options, "London")
+
+    outputs.append(task1)
+    outputs.append(task2)
+    outputs.append(task3)
+
+    return outputs
+
+def base_expected_state(output=None, replay_schema: ReplaySchema = ReplaySchema.V1) -> OrchestratorState:
+    return OrchestratorState(is_done=False, actions=[], output=output, replay_schema=replay_schema.value)
+
+
+def add_hello_suborch_action(state: OrchestratorState, input_: str):
+    retry_options = RETRY_OPTIONS
+    action = CallSubOrchestratorWithRetryAction(function_name='HelloSubOrchestrator',
+        retry_options=retry_options,_input=input_)
+    state.actions.append([action])
+
+
+def add_hello_suborch_completed_events(
+        context_builder: ContextBuilder, id_: int, result: str):
+    context_builder.add_sub_orchestrator_started_event(name="HelloSubOrchestrator", id_=id_,input_="")
+    context_builder.add_orchestrator_completed_event()
+    context_builder.add_orchestrator_started_event()
+    context_builder.add_sub_orchestrator_completed_event(result=result, id_=id_)
+
+def add_hello_suborch_failed_events(
+        context_builder: ContextBuilder, id_: int, reason: str, details: str):
+    context_builder.add_sub_orchestrator_started_event(name="HelloSubOrchestrator", id_=id_,input_="")
+    context_builder.add_orchestrator_completed_event()
+    context_builder.add_orchestrator_started_event()
+    context_builder.add_sub_orchestrator_failed_event(id_=id_, reason=reason, details=details)
+
+def add_retry_timer_events(context_builder: ContextBuilder, id_: int):
+    fire_at = context_builder.add_timer_created_event(id_)
+    context_builder.add_orchestrator_completed_event()
+    context_builder.add_orchestrator_started_event()
+    context_builder.add_timer_fired_event(id_=id_, fire_at=fire_at)
+
+def test_tokyo_and_seattle_and_london_state():
+    context_builder = ContextBuilder('test_simple_function')
+    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
+    add_hello_suborch_completed_events(context_builder, 1, "\"Hello Seattle!\"")
+    add_hello_suborch_completed_events(context_builder, 2, "\"Hello London!\"")
+
+    result = get_orchestration_state_result(
+        context_builder, generator_function)
+
+    expected_state = base_expected_state(
+        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
+    add_hello_suborch_action(expected_state, 'Tokyo')
+    add_hello_suborch_action(expected_state, 'Seattle')
+    add_hello_suborch_action(expected_state, 'London')
+    expected_state._is_done = True
+    expected = expected_state.to_json()
+
+    #assert_valid_schema(result)
+    assert_orchestration_state_equals(expected, result)
+
+
+def test_tokyo_and_seattle_and_london_state_partial_failure():
+    failed_reason = 'Reasons'
+    failed_details = 'Stuff and Things'
+    context_builder = ContextBuilder('test_simple_function')
+    add_hello_suborch_completed_events(context_builder, 0, "\"Hello Tokyo!\"")
+    add_hello_suborch_failed_events(context_builder, 1, failed_reason, failed_details)
+    add_retry_timer_events(context_builder, 2)
+    add_hello_suborch_completed_events(context_builder, 3, "\"Hello Seattle!\"")
+    add_hello_suborch_completed_events(context_builder, 4, "\"Hello London!\"")
+
+    result = get_orchestration_state_result(
+        context_builder, generator_function)
+
+    expected_state = base_expected_state(
+        ['Hello Tokyo!', 'Hello Seattle!', 'Hello London!'])
+    add_hello_suborch_action(expected_state, 'Tokyo')
+    add_hello_suborch_action(expected_state, 'Seattle')
+    add_hello_suborch_action(expected_state, 'London')
+    expected_state._is_done = True
+    expected = expected_state.to_json()
+
+    #assert_valid_schema(result)
+    assert_orchestration_state_equals(expected, result)
+
+def test_tokyo_and_seattle_and_london_state_all_failed():
+    failed_reason = 'Reasons'
+    failed_details = 'Stuff and Things'
+    context_builder = ContextBuilder('test_simple_function')
+    add_hello_suborch_failed_events(context_builder, 0, failed_reason, failed_details)
+    add_retry_timer_events(context_builder, 1)
+    add_hello_suborch_failed_events(context_builder, 2, failed_reason, failed_details)
+    add_retry_timer_events(context_builder, 3)
+    add_hello_suborch_failed_events(context_builder, 4, failed_reason, failed_details)
+    add_retry_timer_events(context_builder, 5)
+
+    try:
+        result = get_orchestration_state_result(
+            context_builder, generator_function)
+        # Should have error'ed out
+        assert False
+    except Exception as e:
+        error_label = "\n\n$OutOfProcData$:"
+        error_str = str(e)
+    
+        expected_state = base_expected_state()
+        add_hello_suborch_action(expected_state, 'Tokyo')
+
+        error_msg = f'{failed_reason} \n {failed_details}'
+        expected_state._error = error_msg
+        state_str = expected_state.to_json_string()
+        
+        expected_error_str = f"{error_msg}{error_label}{state_str}"
         assert expected_error_str == error_str

tests/orchestrator/test_task_any.py

@@ -1,61 +1,61 @@
-from tests.orchestrator.test_fan_out_fan_in import add_completed_event, add_failed_event, base_expected_state, add_multi_actions
-from tests.orchestrator.orchestrator_test_utils import assert_orchestration_state_equals, get_orchestration_state_result
-from tests.test_utils.ContextBuilder import ContextBuilder
-
-def generator_function(context):
-    task1 = context.call_activity("Hello", "0")
-    task2 = context.call_activity("Hello", "1")
-    task3 = context.call_activity("Hello", "2")
-    task4 = context.task_any([task1, task2, task3])
-    first_completed_task = yield task4
-    try:
-        result = yield first_completed_task
-        return result
-    except:
-        return "exception"
-
-def generator_function_no_activity(context):
-    yield context.task_any([])
-    return "Done!"
-
-def test_continues_on_zero_inner_tasks():
-    context_builder = ContextBuilder()
-    result = get_orchestration_state_result(
-        context_builder, generator_function_no_activity)
-    expected_state = base_expected_state("Done!")
-    expected_state._is_done = True
-    expected = expected_state.to_json()
-    assert_orchestration_state_equals(expected, result)
-
-def test_continues_on_zero_results():
-    context_builder = ContextBuilder()
-    result = get_orchestration_state_result(
-        context_builder, generator_function)
-    expected_state = base_expected_state()
-    add_multi_actions(expected_state, function_name='Hello', volume=3)
-    expected = expected_state.to_json()
-    assert_orchestration_state_equals(expected, result)
-
-def test_continues_on_one_failure():
-    context_builder = ContextBuilder()
-    add_failed_event(context_builder, 0, "Hello", reason="", details="")
-    result = get_orchestration_state_result(
-        context_builder, generator_function)
-    add_failed_event(context_builder, 0, "Hello", reason="", details="")
-    expected_state = base_expected_state("exception")
-    add_multi_actions(expected_state, function_name='Hello', volume=3)
-    expected_state._is_done = True
-    expected = expected_state.to_json()
-    assert_orchestration_state_equals(expected, result)
-
-def test_succeeds_on_one_result():
-    context_builder = ContextBuilder()
-    add_completed_event(context_builder, 0, "Hello", result="1")
-    result = get_orchestration_state_result(
-        context_builder, generator_function)
-    add_completed_event(context_builder, 2, "Hello", "3")
-    expected_state = base_expected_state("1")
-    add_multi_actions(expected_state, function_name='Hello', volume=3)
-    expected_state._is_done = True
-    expected = expected_state.to_json()
+from tests.orchestrator.test_fan_out_fan_in import add_completed_event, add_failed_event, base_expected_state, add_multi_actions
+from tests.orchestrator.orchestrator_test_utils import assert_orchestration_state_equals, get_orchestration_state_result
+from tests.test_utils.ContextBuilder import ContextBuilder
+
+def generator_function(context):
+    task1 = context.call_activity("Hello", "0")
+    task2 = context.call_activity("Hello", "1")
+    task3 = context.call_activity("Hello", "2")
+    task4 = context.task_any([task1, task2, task3])
+    first_completed_task = yield task4
+    try:
+        result = yield first_completed_task
+        return result
+    except:
+        return "exception"
+
+def generator_function_no_activity(context):
+    yield context.task_any([])
+    return "Done!"
+
+def test_continues_on_zero_inner_tasks():
+    context_builder = ContextBuilder()
+    result = get_orchestration_state_result(
+        context_builder, generator_function_no_activity)
+    expected_state = base_expected_state("Done!")
+    expected_state._is_done = True
+    expected = expected_state.to_json()
+    assert_orchestration_state_equals(expected, result)
+
+def test_continues_on_zero_results():
+    context_builder = ContextBuilder()
+    result = get_orchestration_state_result(
+        context_builder, generator_function)
+    expected_state = base_expected_state()
+    add_multi_actions(expected_state, function_name='Hello', volume=3)
+    expected = expected_state.to_json()
+    assert_orchestration_state_equals(expected, result)
+
+def test_continues_on_one_failure():
+    context_builder = ContextBuilder()
+    add_failed_event(context_builder, 0, "Hello", reason="", details="")
+    result = get_orchestration_state_result(
+        context_builder, generator_function)
+    add_failed_event(context_builder, 0, "Hello", reason="", details="")
+    expected_state = base_expected_state("exception")
+    add_multi_actions(expected_state, function_name='Hello', volume=3)
+    expected_state._is_done = True
+    expected = expected_state.to_json()
+    assert_orchestration_state_equals(expected, result)
+
+def test_succeeds_on_one_result():
+    context_builder = ContextBuilder()
+    add_completed_event(context_builder, 0, "Hello", result="1")
+    result = get_orchestration_state_result(
+        context_builder, generator_function)
+    add_completed_event(context_builder, 2, "Hello", "3")
+    expected_state = base_expected_state("1")
+    add_multi_actions(expected_state, function_name='Hello', volume=3)
+    expected_state._is_done = True
+    expected = expected_state.to_json()
     assert_orchestration_state_equals(expected, result)