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