wireframe 0.3.0

Simplify building servers and clients for custom binary protocols.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

//! Shared request builders for streaming helper tests.

use futures::TryStreamExt;

use super::streaming_infra::{
    CorrelationId,
    MessageId,
    Payload,
    TestStreamEnvelope,
    setup_streaming_test,
};
use crate::client::{ClientError, StreamingResponseExt};

pub(super) fn build_request(correlation_id: CorrelationId) -> TestStreamEnvelope {
    TestStreamEnvelope::data(MessageId::new(99), correlation_id, Payload::new(vec![]))
}

pub(super) async fn collect_typed_items<F>(
    cid: CorrelationId,
    frames: Vec<TestStreamEnvelope>,
    mapper: F,
) -> Result<Vec<Vec<u8>>, Box<dyn std::error::Error + Send + Sync>>
where
    F: FnMut(TestStreamEnvelope) -> Result<Option<Vec<u8>>, ClientError> + Send + Unpin + 'static,
{
    let (mut client, _server) = setup_streaming_test(frames).await?;
    let items: Vec<Vec<u8>> = client
        .call_streaming(build_request(cid))
        .await?
        .typed_with(mapper)
        .try_collect()
        .await?;
    Ok(items)
}