jerbs 0.2.0

Command-line work-stealing scheduler.
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
85
86
87
88
89
90
91
92

use anyhow::Result;
use assert_cmd::Command;
use std::path::Path;
use tempfile::NamedTempFile;

fn cmd(db: &Path, args: &[&str]) -> Result<Command> {
    let mut cmd = Command::cargo_bin("jerbs")?;
    cmd.arg(db);
    cmd.args(args);
    Ok(cmd)
}

#[test]
fn test_init() -> Result<()> {
    let db_file = NamedTempFile::new()?;
    let db = db_file.path();
    cmd(db, &["init"])?.assert().success();
    Ok(())
}

#[test]
fn test_take() -> Result<()> {
    let db_file = NamedTempFile::new()?;
    let db = db_file.path();
    cmd(db, &["init"])?.assert().success();
    // create a one-shot job
    cmd(db, &["create", "-c", "1", "-d", "JOBDATA"])?
        .assert()
        .success();
    // take it
    cmd(db, &["take", "WORKERDATA"])?.assert().success();
    // can't take it again
    cmd(db, &["take", "WORKERDATA"])?.assert().failure();
    Ok(())
}

#[test]
fn test_monitor() -> Result<()> {
    let db_file = NamedTempFile::new()?;
    let db = db_file.path();
    cmd(db, &["init"])?.assert().success();
    // create a job with many instances
    cmd(db, &["create", "-c", "12", "-d", "JOBDATA"])?
        .assert()
        .success();
    // worker with no job shouldn't be able to log
    cmd(db, &["log-start", "WORKERDATA1"])?.assert().failure();
    cmd(db, &["log-finish", "WORKERDATA1", "0"])?
        .assert()
        .failure();
    // get a job and then try
    cmd(db, &["take", "WORKERDATA1"])?.assert().success();
    cmd(db, &["log-start", "WORKERDATA1"])?.assert().success();
    cmd(db, &["log-finish", "WORKERDATA1", "0"])?
        .assert()
        .success();
    // no active jobs once again, shouldn't be ready to log
    cmd(db, &["log-start", "WORKERDATA1"])?.assert().failure();
    cmd(db, &["log-finish", "WORKERDATA1", "0"])?
        .assert()
        .failure();
    // monitor should be effectively the same as log-start / log-finish
    cmd(db, &["take", "WORKERDATA1"])?.assert().success();
    cmd(db, &["monitor", "WORKERDATA1", "--", "true"])?
        .assert()
        .success();
    cmd(db, &["log-finish", "WORKERDATA1", "0"])?
        .assert()
        .failure();
    // non-existent command should fail, and log start and finish
    cmd(db, &["take", "WORKERDATA1"])?.assert().success();
    cmd(
        db,
        &["monitor", "WORKERDATA1", "--", "nosuchcommand_foobarbaz"],
    )?
    .assert()
    .failure();
    cmd(db, &["log-start", "WORKERDATA1"])?.assert().failure();
    cmd(db, &["log-finish", "WORKERDATA1", "0"])?
        .assert()
        .failure();
    // monitoring command that returns failure should return failure, and log start and finish
    cmd(db, &["take", "WORKERDATA1"])?.assert().success();
    cmd(db, &["monitor", "WORKERDATA1", "--", "false"])?
        .assert()
        .failure();
    cmd(db, &["log-start", "WORKERDATA1"])?.assert().failure();
    cmd(db, &["log-finish", "WORKERDATA1", "0"])?
        .assert()
        .failure();
    Ok(())
}