shpool 0.5.0

shpool is a mechanism for establishing lightweight persistant shell sessions to gracefully handle network disconnects. 
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use std::process::Command;

use anyhow::Context;
use ntest::timeout;
use regex::Regex;

mod support;

#[test]
#[timeout(30000)]
fn empty() -> anyhow::Result<()> {
    support::dump_err(|| {
        let mut daemon_proc =
            support::daemon::Proc::new("norc.toml", false).context("starting daemon proc")?;
        let out = daemon_proc.list()?;
        assert!(out.status.success(), "list proc did not exit successfully");

        let stderr = String::from_utf8_lossy(&out.stderr[..]);
        assert_eq!(stderr.len(), 0, "expected no stderr");

        let stdout = String::from_utf8_lossy(&out.stdout[..]);
        assert!(stdout.contains("NAME"));
        assert!(stdout.contains("STARTED_AT"));

        Ok(())
    })
}

#[test]
#[timeout(30000)]
fn no_daemon() -> anyhow::Result<()> {
    support::dump_err(|| {
        let out = Command::new(support::shpool_bin()?)
            .arg("--socket")
            .arg("/fake/does/not/exist/shpool.socket")
            .arg("list")
            .output()
            .context("spawning list proc")?;

        assert!(!out.status.success(), "list proc exited successfully");

        let stderr = String::from_utf8_lossy(&out.stderr[..]);
        assert!(stderr.contains("could not connect to daemon"));

        Ok(())
    })
}

#[test]
#[timeout(30000)]
fn one_session() -> anyhow::Result<()> {
    support::dump_err(|| {
        let mut daemon_proc =
            support::daemon::Proc::new("norc.toml", true).context("starting daemon proc")?;
        let bidi_enter_w = daemon_proc.events.take().unwrap().waiter(["daemon-bidi-stream-enter"]);

        let _sess1 = daemon_proc.attach("sh1", Default::default())?;

        daemon_proc.events = Some(bidi_enter_w.wait_final_event("daemon-bidi-stream-enter")?);

        let out = daemon_proc.list()?;
        assert!(out.status.success(), "list proc did not exit successfully");

        let stderr = String::from_utf8_lossy(&out.stderr[..]);
        assert_eq!(stderr.len(), 0, "expected no stderr");

        let stdout = String::from_utf8_lossy(&out.stdout[..]);
        assert!(stdout.contains("sh1"));

        Ok(())
    })
}

#[test]
#[timeout(30000)]
fn two_sessions() -> anyhow::Result<()> {
    support::dump_err(|| {
        let mut daemon_proc =
            support::daemon::Proc::new("norc.toml", true).context("starting daemon proc")?;
        let mut bidi_enter_w = daemon_proc.events.take().unwrap().waiter([
            "daemon-bidi-stream-enter",
            "daemon-bidi-stream-enter",
            "daemon-bidi-stream-done",
        ]);

        let _sess1 = daemon_proc.attach("sh1", Default::default())?;

        bidi_enter_w.wait_event("daemon-bidi-stream-enter")?;

        {
            let _sess2 = daemon_proc.attach("sh2", Default::default())?;

            bidi_enter_w.wait_event("daemon-bidi-stream-enter")?;

            let out = daemon_proc.list()?;
            assert!(out.status.success(), "list proc did not exit successfully");

            let stderr = String::from_utf8_lossy(&out.stderr[..]);
            assert_eq!(stderr.len(), 0, "expected no stderr");

            let stdout = String::from_utf8_lossy(&out.stdout[..]);
            assert!(stdout.contains("sh1"));
            assert!(stdout.contains("sh2"));
        }

        // wait for the hangup to complete
        bidi_enter_w.wait_event("daemon-bidi-stream-done")?;

        let out = daemon_proc.list()?;
        assert!(out.status.success(), "list proc did not exit successfully");

        let sh1_re = Regex::new("sh1.*attached")?;
        let sh2_re = Regex::new("sh2.*disconnected")?;
        let stdout = String::from_utf8_lossy(&out.stdout[..]);
        dbg!(&stdout);
        assert!(sh1_re.is_match(&stdout));
        assert!(sh2_re.is_match(&stdout));

        Ok(())
    })
}