nifi-rust-client 0.12.0

Apache NiFi REST API client library
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

#![cfg(not(feature = "dynamic"))]

use bytes::Bytes;
use futures_util::StreamExt;
use nifi_rust_client::NifiClientBuilder;
use wiremock::matchers::{method, path_regex};
use wiremock::{Mock, MockServer, ResponseTemplate};

/// Asserts a streamed download concatenates to the same bytes as the
/// buffered variant. Uses `/controller/nar-manager/nars/{id}/content`
/// because its emitted accessor chain is the simplest bytes path.
#[tokio::test]
async fn stream_chunks_concat_match_payload() {
    let server = MockServer::start().await;
    let payload: Vec<u8> = (0..=255u8).cycle().take(16_384).collect();
    Mock::given(method("GET"))
        .and(path_regex(
            r"^/nifi-api/controller/nar-manager/nars/[^/]+/content$",
        ))
        .respond_with(
            ResponseTemplate::new(200)
                .set_body_bytes(payload.clone())
                .insert_header("content-type", "application/octet-stream"),
        )
        .expect(1)
        .mount(&server)
        .await;

    let client = NifiClientBuilder::new(&server.uri())
        .unwrap()
        .build()
        .unwrap();

    let mut stream = client
        .controller()
        .download_nar_stream("nar-id-123")
        .await
        .expect("start stream");
    let mut collected: Vec<u8> = Vec::new();
    while let Some(chunk) = stream.next().await {
        let chunk: Bytes = chunk.expect("stream chunk");
        collected.extend_from_slice(&chunk);
    }
    assert_eq!(collected, payload);
}

#[tokio::test]
async fn stream_forwards_range_header_and_accepts_206() {
    use wiremock::matchers::header;

    let server = MockServer::start().await;
    let payload: Vec<u8> = (0..128u8).collect();

    Mock::given(method("GET"))
        .and(path_regex(
            r"^/nifi-api/provenance-events/\d+/content/input$",
        ))
        .and(header("Range", "bytes=0-63"))
        .respond_with(
            ResponseTemplate::new(206)
                .set_body_bytes(payload[..64].to_vec())
                .insert_header("content-type", "application/octet-stream")
                .insert_header("content-range", "bytes 0-63/128"),
        )
        .expect(1)
        .mount(&server)
        .await;

    let client = NifiClientBuilder::new(&server.uri())
        .unwrap()
        .build()
        .unwrap();

    let mut stream = client
        .provenanceevents()
        .get_input_content_stream("12345", None, Some("bytes=0-63"))
        .await
        .expect("partial content stream");
    let mut collected = Vec::new();
    while let Some(chunk) = stream.next().await {
        collected.extend_from_slice(&chunk.unwrap());
    }
    assert_eq!(collected, payload[..64]);
}