Function test_set_and_get_state

tests/units/test_app.py:490–524

Test setting and getting the state of an app with different tokens. Args: test_state: The default state.

(test_state: type[ATestState])

@pytest.mark.asyncio
async def test_set_and_get_state(test_state: type[ATestState]):
    """Test setting and getting the state of an app with different tokens.

    Args:
        test_state: The default state.
    """
    app = App(_state=test_state)

    # Create two tokens.
    token1 = BaseStateToken(ident=str(uuid.uuid4()), cls=test_state)
    token2 = BaseStateToken(ident=str(uuid.uuid4()), cls=test_state)

    # Get the default state for each token.
    state1 = await app.state_manager.get_state(token1)
    state2 = await app.state_manager.get_state(token2)
    assert isinstance(state1, test_state)
    assert isinstance(state2, test_state)
    assert state1.var == 0
    assert state2.var == 0

    # Set the vars to different values.
    state1.var = 1
    state2.var = 2
    await app.state_manager.set_state(token1, state1)
    await app.state_manager.set_state(token2, state2)

    # Get the states again and check the values.
    state1 = await app.state_manager.get_state(token1)
    state2 = await app.state_manager.get_state(token2)
    assert isinstance(state1, test_state)
    assert isinstance(state2, test_state)
    assert state1.var == 1
    assert state2.var == 2

    await app.state_manager.close()


@pytest.mark.asyncio