Function parser_handles_split_multiple_messages

src/protocol/tests.rs:768–806

()

#[test]
fn parser_handles_split_multiple_messages() -> Result<()> {
    let data_row = buffers::data_row(&[Some("!")]);
    let ready_for_query = buffers::ready_for_query();
    let mut combined = data_row.clone();
    combined.extend_from_slice(&ready_for_query);

    let verify_messages = |messages: &[BackendMessage]| {
        assert_eq!(messages.len(), 2);
        assert_data_row_fields(&messages[0], &[Some("!")]);
        match &messages[1] {
            BackendMessage::ReadyForQuery(msg) => assert_eq!(msg.status, b'I'),
            other => panic!("unexpected message: {:?}", other.name()),
        }
    };

    verify_messages(&parse_vec_chunks(vec![combined.clone()])?);

    let splits = [
        11,
        combined.len().saturating_sub(1),
        combined.len().saturating_sub(4),
        combined.len().saturating_sub(6),
        8,
        1,
    ];

    for split in splits {
        if split == 0 || split >= combined.len() {
            continue;
        }
        let first_len = combined.len() - split;
        let first = combined[..first_len].to_vec();
        let second = combined[first_len..].to_vec();
        let messages = parse_vec_chunks(vec![first, second])?;
        verify_messages(&messages);
    }

    Ok(())
}

#[test]
fn parser_respects_buffer_views() -> Result<()> {