processmanager 0.6.0

manage process lifecycles, graceful shutdown and process faults
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

//! Integration-style tests for the public `ProcessManagerBuilder` API.
//
//! The goal is to ensure that compile-time plumbing between the builder, the
//! resulting `ProcessManager` instance and its runtime metadata works as
//! expected.

use std::sync::Arc;
use std::time::Duration;

use processmanager::{
    CtrlFuture, ProcFuture, ProcessControlHandler, ProcessManagerBuilder, Runnable, RuntimeError,
};

/// A no-op service that terminates immediately and successfully.
///
/// We mostly need this to satisfy the `pre_insert` type parameter; the process
/// never actually runs inside the test.
#[derive(Default)]
struct NoopSvc;

impl Runnable for NoopSvc {
    fn process_start(&self) -> ProcFuture<'_> {
        Box::pin(async { Ok::<(), RuntimeError>(()) })
    }

    fn process_handle(&self) -> Arc<dyn ProcessControlHandler> {
        // The test never calls control methods, so a stub handle is sufficient.
        Arc::new(StubHandle)
    }
}

/// A minimal `ProcessControlHandler` implementation that does nothing.
struct StubHandle;

impl ProcessControlHandler for StubHandle {
    fn shutdown(&self) -> CtrlFuture<'_> {
        Box::pin(async {})
    }

    fn reload(&self) -> CtrlFuture<'_> {
        Box::pin(async {})
    }
}

#[test]
fn builder_sets_custom_name() {
    let mgr = ProcessManagerBuilder::default()
        .name("my-supervisor")
        .pre_insert(NoopSvc)
        .build();

    assert_eq!(mgr.process_name(), "my-supervisor");
}

#[test]
fn builder_sets_shutdown_grace_period() {
    let mgr = ProcessManagerBuilder::default()
        .shutdown_grace_period(Duration::from_millis(250))
        .pre_insert(NoopSvc)
        .build();

    assert_eq!(mgr.shutdown_grace_period(), Duration::from_millis(250));
}