arcon_codegen 0.1.4

Code generation for Arcon 
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

extern crate arcon;
extern crate arcon_codegen;

use arcon_codegen::*;
use arcon_spec::*;
use std::fs;

pub const RUN_PASS_MODE: &str = "run-pass";
pub const RUN_PASS_PATH: &str = "tests/run-pass";
pub const SPECIFICATION_PATH: &str = "tests/specifications";

#[test]
fn codegen_test() {
    // makes sure that tests/run-pass does not exist
    let _ = fs::remove_dir_all(RUN_PASS_PATH);

    // Fresh start of run-pass tests
    fs::create_dir_all(RUN_PASS_PATH).unwrap();

    let t = trybuild::TestCases::new();

    add_test_spec("basic_dataflow");
    add_test_spec("tumbling_window_dataflow");
    add_test_spec("normalise");
    add_test_spec("pipeline_with_structs");

    // test all generated .rs files
    let specs = format!("{}/{}", RUN_PASS_PATH, "*.rs");
    t.pass(&specs);
}

fn add_test_spec(name: &str) {
    let json_path = format!("{}/{}.json", SPECIFICATION_PATH, name);
    let spec = ArconSpec::load(&json_path).unwrap();
    let generated_code = generate(&spec, true).unwrap();
    let path = format!("{}/{}.rs", RUN_PASS_PATH, name);
    let _ = to_file(generated_code, path.to_string());
}

fn _add_empty_main(path: &str) {
    let main = "fn main() {}";
    use std::fs::OpenOptions;
    use std::io::Write;

    let mut file = OpenOptions::new()
        .write(true)
        .append(true)
        .open(path)
        .unwrap();

    let _ = writeln!(file, "{}", main);
}