integration/tests/streaming/mod.rs

use bytes::Bytes;
use iggy::models::polled_messages::{MessageState, PolledMessage};
use iggy::utils::checksum;

mod common;
mod consumer_group;
mod consumer_offset;
mod messages;
mod partition;
mod personal_access_token;
mod segment;
mod stream;
mod system;
mod topic;
mod topic_messages;
mod user;

fn create_messages() -> Vec<PolledMessage> {
    vec![
        create_message(0, 1, "message 1"),
        create_message(1, 2, "message 2"),
        create_message(2, 3, "message 3"),
        create_message(3, 2, "message 3.2"),
        create_message(4, 1, "message 1.2"),
        create_message(5, 3, "message 3.3"),
    ]
}

fn create_message(offset: u64, id: u128, payload: &str) -> PolledMessage {
    let payload = Bytes::from(payload.to_string());
    let checksum = checksum::calculate(payload.as_ref());
    PolledMessage::create(
        offset,
        MessageState::Available,
        1,
        id,
        payload,
        checksum,
        None,
    )
}