stressed 1.1.2

Universal CLI stress tester for competitive programming
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

use crate::communicator::Communicator;
use crate::utils::SeedType;
use std::error::Error;
use std::path::PathBuf;

pub struct Sampler {
    c: Communicator,
    use_stdin: bool,
}

impl Sampler {
    pub fn new(executable: PathBuf, use_stdin: bool) -> Sampler {
        Sampler {
            c: Communicator::new(executable, false),
            use_stdin,
        }
    }

    pub async fn sample(&self, seed: SeedType) -> Result<String, Box<dyn Error>> {
        let seed_string = seed.to_string();
        let ans = if self.use_stdin {
            self.c.communicate_result(Some(&seed_string), None).await?
        } else {
            self.c
                .communicate_result(None, Some(&[&seed_string]))
                .await?
        };
        Ok(ans)
    }
}

#[cfg(all(target_os = "linux", test))]
mod tests {
    use super::*;
    #[tokio::test]
    async fn seed_as_arg() {
        let arg_sampler = Sampler::new(PathBuf::from("echo"), false);
        for seed in 0..100 {
            let ans = arg_sampler.sample(seed as SeedType).await;
            let correct_ans = format!("{seed}\n");
            let incorrect_ans = format!("{}\n", seed + 1);
            let ans = ans.unwrap();
            assert_eq!(ans, correct_ans);
            assert_ne!(ans, incorrect_ans);
        }
    }
    #[tokio::test]
    async fn seed_as_stdin() {
        let arg_sampler = Sampler::new(PathBuf::from("cat"), true);
        for seed in 0..100 {
            let ans = arg_sampler.sample(seed as SeedType).await;
            let correct_ans = format!("{seed}\n");
            let ans = ans.unwrap();
            assert_eq!(ans, correct_ans);
        }
    }
}