80 lines
1.9 KiB
OCaml
80 lines
1.9 KiB
OCaml
let valid_program =
|
|
{|
|
|
struct Item {
|
|
int value;
|
|
};
|
|
|
|
int fold(int[] xs) {
|
|
int total = 0;
|
|
foreach (int x in xs) {
|
|
total = total + x;
|
|
}
|
|
return total;
|
|
}
|
|
|
|
int main() {
|
|
int[] xs;
|
|
struct Item it;
|
|
int y = 2 + 3 * 4;
|
|
it.value = y;
|
|
if (y >= 0) {
|
|
y = fold(xs);
|
|
} else {
|
|
y = 0;
|
|
}
|
|
return y;
|
|
}
|
|
|}
|
|
|
|
let invalid_program =
|
|
{|
|
|
int main() {
|
|
bool flag = true;
|
|
int x = 1;
|
|
x = flag;
|
|
return x;
|
|
}
|
|
|}
|
|
|
|
let test_valid_program () =
|
|
match Spooky.parse_and_type_check valid_program with
|
|
| Ok _ -> ()
|
|
| Error msg -> failwith ("expected valid program, got: " ^ msg)
|
|
|
|
let test_invalid_program () =
|
|
match Spooky.parse_and_type_check invalid_program with
|
|
| Ok _ -> failwith "expected type error, but got success"
|
|
| Error _ -> ()
|
|
|
|
let write_file path content =
|
|
let oc = open_out_bin path in
|
|
Fun.protect ~finally:(fun () -> close_out oc) (fun () -> output_string oc content)
|
|
|
|
let test_imports () =
|
|
let base = Filename.concat (Filename.get_temp_dir_name ()) "spooky_import_test" in
|
|
let modules_dir = Filename.concat base "modules" in
|
|
Unix.mkdir base 0o755;
|
|
Unix.mkdir modules_dir 0o755;
|
|
let cleanup () =
|
|
(try Sys.remove (Filename.concat modules_dir "math.spooky") with _ -> ());
|
|
(try Sys.remove (Filename.concat base "main.spooky") with _ -> ());
|
|
(try Unix.rmdir modules_dir with _ -> ());
|
|
(try Unix.rmdir base with _ -> ())
|
|
in
|
|
Fun.protect
|
|
~finally:cleanup
|
|
(fun () ->
|
|
write_file (Filename.concat modules_dir "math.spooky")
|
|
"int add(int a, int b) { return a + b; }\n";
|
|
write_file (Filename.concat base "main.spooky")
|
|
"import \"modules/math.spooky\";\nint main() { return add(1, 2); }\n";
|
|
match Spooky.parse_and_type_check_file (Filename.concat base "main.spooky") with
|
|
| Ok _ -> ()
|
|
| Error msg -> failwith ("expected valid import program, got: " ^ msg))
|
|
|
|
let () =
|
|
test_valid_program ();
|
|
test_invalid_program ();
|
|
test_imports ();
|
|
print_endline "All parser/type-check tests passed."
|