Function test_get_state

tests/units/test_state.py:3489–3600

Test that a get_state populates the top level state and delta calculation is correct. Args: token: A token. attached_mock_event_context: An event context with a state manager that has a TestState instance corresponding to the token.

(token: str, attached_mock_event_context: EventContext)

@pytest.mark.asyncio
async def test_get_state(token: str, attached_mock_event_context: EventContext):
    """Test that a get_state populates the top level state and delta calculation is correct.

    Args:
        token: A token.
        attached_mock_event_context: An event context with a state manager that has a TestState instance corresponding to the token.
    """
    state_manager = attached_mock_event_context.state_manager

    # Get instance of ChildState2.
    test_state = await state_manager.get_state(
        BaseStateToken(ident=token, cls=ChildState2)
    )
    assert isinstance(test_state, TestState)
    if isinstance(state_manager, (StateManagerMemory, StateManagerDisk)):
        # All substates are available
        assert tuple(sorted(test_state.substates)) == (
            ChildState.get_name(),
            ChildState2.get_name(),
            ChildState3.get_name(),
        )
    else:
        # Sibling states are only populated if they have computed vars
        assert tuple(sorted(test_state.substates)) == (
            ChildState2.get_name(),
            ChildState3.get_name(),
        )

    # Because ChildState3 has a computed var, it is always dirty, and always populated.
    grandchild_state3 = test_state.substates[ChildState3.get_name()].substates[
        GrandchildState3.get_name()
    ]
    assert isinstance(grandchild_state3, GrandchildState3)
    assert grandchild_state3.computed == ""

    # Get the child_state2 directly.
    child_state2_direct = test_state.get_substate([ChildState2.get_name()])
    child_state2_get_state = await test_state.get_state(ChildState2)
    # These should be the same object.
    assert child_state2_direct is child_state2_get_state

    # Get arbitrary GrandchildState.
    grandchild_state = await child_state2_get_state.get_state(GrandchildState)
    assert isinstance(grandchild_state, GrandchildState)

    # Now the original root should have all substates populated.
    assert tuple(sorted(test_state.substates)) == (
        ChildState.get_name(),
        ChildState2.get_name(),
        ChildState3.get_name(),
    )

    # ChildState should be retrievable
    child_state_direct = test_state.get_substate([ChildState.get_name()])
    child_state_get_state = await test_state.get_state(ChildState)
    # These should be the same object.
    assert child_state_direct is child_state_get_state