Example #1
0
def test_overrides_respected():
    """Ensure provided overrides are respected."""
    protobuf_strategies = modules_to_strategies(im_pb2, **{
        full_field_name(im_pb2.InstantMessage, 'message'): st.just('test message')
    })
    instant_message_strategy = protobuf_strategies[im_pb2.InstantMessage]
    instant_message_example = instant_message_strategy.example()
    assert instant_message_example.message == 'test message'
def test_overrides_respected_nested():
    """Ensure provided overrides are respected."""
    protobuf_strategies = modules_to_strategies(
        im_nested_pb2, **{
            full_field_name(im_nested_pb2.InstantMessage, 'timestamp'):
            st.just(101),
            full_field_name(im_nested_pb2.InstantMessage, 'message'):
            st.just('test message'),
            full_field_name(im_nested_pb2.InstantMessage, 'nested1'):
            st.just(3),
            full_field_name(im_nested_pb2.MetaData, 'latency'):
            st.just(10),
            full_field_name(im_nested_pb2.MetaData.InnerData, 'nested1'):
            st.just(2),
        })
    instant_message_strategy = protobuf_strategies[
        im_nested_pb2.InstantMessage]
    instant_message_example = instant_message_strategy.example()
    assert instant_message_example.timestamp == 101
    assert instant_message_example.message == 'test message'
    assert instant_message_example.nested1 == 3
    assert instant_message_example.metadata.latency == 10
    assert instant_message_example.metadata.inner_data.nested1 == 2