/* * Copyright (c) 2015 Andrew Kelley * * This file is part of zig, which is MIT licensed. * See http://opensource.org/licenses/MIT */ #include "list.hpp" #include "buffer.hpp" #include "os.hpp" #include #include struct TestSourceFile { const char *relative_path; const char *source_code; }; struct TestCase { const char *case_name; const char *output; ZigList source_files; ZigList compile_errors; ZigList compiler_args; ZigList program_args; }; static ZigList test_cases = {0}; static const char *tmp_source_path = ".tmp_source.zig"; static const char *tmp_exe_path = "./.tmp_exe"; static const char *zig_exe = "./zig"; static void add_source_file(TestCase *test_case, const char *path, const char *source) { test_case->source_files.add_one(); test_case->source_files.last().relative_path = path; test_case->source_files.last().source_code = source; } static TestCase *add_simple_case(const char *case_name, const char *source, const char *output) { TestCase *test_case = allocate(1); test_case->case_name = case_name; test_case->output = output; test_case->source_files.resize(1); test_case->source_files.at(0).relative_path = tmp_source_path; test_case->source_files.at(0).source_code = source; test_case->compiler_args.append("build"); test_case->compiler_args.append(tmp_source_path); test_case->compiler_args.append("--export"); test_case->compiler_args.append("exe"); test_case->compiler_args.append("--name"); test_case->compiler_args.append("test"); test_case->compiler_args.append("--output"); test_case->compiler_args.append(tmp_exe_path); test_case->compiler_args.append("--release"); test_case->compiler_args.append("--strip"); test_case->compiler_args.append("--verbose"); test_case->compiler_args.append("--color"); test_case->compiler_args.append("on"); test_cases.append(test_case); return test_case; } static TestCase *add_compile_fail_case(const char *case_name, const char *source, int count, ...) { va_list ap; va_start(ap, count); TestCase *test_case = allocate(1); test_case->case_name = case_name; test_case->source_files.resize(1); test_case->source_files.at(0).relative_path = tmp_source_path; test_case->source_files.at(0).source_code = source; for (int i = 0; i < count; i += 1) { const char *arg = va_arg(ap, const char *); test_case->compile_errors.append(arg); } test_case->compiler_args.append("build"); test_case->compiler_args.append(tmp_source_path); test_case->compiler_args.append("--output"); test_case->compiler_args.append(tmp_exe_path); test_case->compiler_args.append("--release"); test_case->compiler_args.append("--strip"); test_case->compiler_args.append("--verbose"); test_cases.append(test_case); va_end(ap); return test_case; } static void add_compiling_test_cases(void) { add_simple_case("hello world with libc", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { puts(c"Hello, world!"); exit(0); } )SOURCE", "Hello, world!\n"); add_simple_case("function call", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } fn empty_function_1() {} fn empty_function_2() { return; } export fn _start() -> unreachable { empty_function_1(); empty_function_2(); this_is_a_function(); } fn this_is_a_function() -> unreachable { puts(c"OK"); exit(0); } )SOURCE", "OK\n"); add_simple_case("comments", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } /** * multi line doc comment */ fn another_function() {} /// this is a documentation comment /// doc comment line 2 export fn _start() -> unreachable { puts(/* mid-line comment /* nested */ */ c"OK"); exit(0); } )SOURCE", "OK\n"); { TestCase *tc = add_simple_case("multiple files with private function", R"SOURCE( use "libc.zig"; use "foo.zig"; export fn _start() -> unreachable { private_function(); } fn private_function() -> unreachable { print_text(); exit(0); } )SOURCE", "OK\n"); add_source_file(tc, "libc.zig", R"SOURCE( #link("c") extern { pub fn puts(s: &const u8) -> i32; pub fn exit(code: i32) -> unreachable; } )SOURCE"); add_source_file(tc, "foo.zig", R"SOURCE( use "libc.zig"; // purposefully conflicting function with main source file // but it's private so it should be OK fn private_function() { puts(c"OK"); } pub fn print_text() { private_function(); } )SOURCE"); } add_simple_case("if statements", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { if 1 != 0 { puts(c"1 is true"); } else { puts(c"1 is false"); } if 0 != 0 { puts(c"0 is true"); } else if 1 - 1 != 0 { puts(c"1 - 1 is true"); } if !(0 != 0) { puts(c"!0 is true"); } exit(0); } )SOURCE", "1 is true\n!0 is true\n"); add_simple_case("params", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } fn add(a: i32, b: i32) -> i32 { a + b } export fn _start() -> unreachable { if add(22, 11) == 33 { puts(c"pass"); } exit(0); } )SOURCE", "pass\n"); add_simple_case("goto", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } fn loop(a : i32) { if a == 0 { goto done; } puts(c"loop"); loop(a - 1); done: return; } export fn _start() -> unreachable { loop(3); exit(0); } )SOURCE", "loop\nloop\nloop\n"); add_simple_case("local variables", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { let a : i32 = 1; let b = 2 as i32; if (a + b == 3) { puts(c"OK"); } exit(0); } )SOURCE", "OK\n"); add_simple_case("bool literals", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { if (true) { puts(c"OK 1"); } if (false) { puts(c"BAD 1"); } if (!true) { puts(c"BAD 2"); } if (!false) { puts(c"OK 2"); } exit(0); } )SOURCE", "OK 1\nOK 2\n"); add_simple_case("separate block scopes", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { if (true) { let no_conflict : i32 = 5; if (no_conflict == 5) { puts(c"OK 1"); } } let c = { let no_conflict = 10 as i32; no_conflict }; if (c == 10) { puts(c"OK 2"); } exit(0); } )SOURCE", "OK 1\nOK 2\n"); add_simple_case("void parameters", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { void_fun(1, void, 2); exit(0); } fn void_fun(a : i32, b : void, c : i32) { let v = b; let vv : void = if (a == 1) {v} else {}; if (a + c == 3) { puts(c"OK"); } return vv; } )SOURCE", "OK\n"); add_simple_case("mutable local variables", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { let mut zero : i32; if (zero == 0) { puts(c"zero"); } let mut i = 0 as i32; loop_start: if i == 3 { goto done; } puts(c"loop"); i = i + 1; goto loop_start; done: exit(0); } )SOURCE", "zero\nloop\nloop\nloop\n"); add_simple_case("arrays", R"SOURCE( #link("c") extern { fn puts(s: &const u8) -> i32; fn exit(code: i32) -> unreachable; } export fn _start() -> unreachable { let mut array : [i32; 5]; let mut i : i32 = 0; loop_start: if i == 5 { goto loop_end; } array[i] = i + 1; i = array[i]; goto loop_start; loop_end: i = 0; let mut accumulator = 0 as i32; loop_2_start: if i == 5 { goto loop_2_end; } accumulator = accumulator + array[i]; i = i + 1; goto loop_2_start; loop_2_end: if accumulator == 15 { puts(c"OK"); } exit(0); } )SOURCE", "OK\n"); add_simple_case("hello world without libc", R"SOURCE( use "std.zig"; export fn main(argc : isize, argv : &&u8, env : &&u8) -> i32 { print_str("Hello, world!\n" as string); return 0; } )SOURCE", "Hello, world!\n"); add_simple_case("a + b + c", R"SOURCE( use "std.zig"; export fn main(argc : isize, argv : &&u8, env : &&u8) -> i32 { if false || false || false { print_str("BAD 1\n" as string); } if true && true && false { print_str("BAD 2\n" as string); } if 1 | 2 | 4 != 7 { print_str("BAD 3\n" as string); } if 3 ^ 6 ^ 8 != 13 { print_str("BAD 4\n" as string); } if 7 & 14 & 28 != 4 { print_str("BAD 5\n" as string); } if 9 << 1 << 2 != 9 << 3 { print_str("BAD 6\n" as string); } if 90 >> 1 >> 2 != 90 >> 3 { print_str("BAD 7\n" as string); } if 100 - 1 + 1000 != 1099 { print_str("BAD 8\n" as string); } if 5 * 4 / 2 % 3 != 1 { print_str("BAD 9\n" as string); } if 5 as i32 as i32 != 5 { print_str("BAD 10\n" as string); } if !!false { print_str("BAD 11\n" as string); } if 7 != --7 { print_str("BAD 12\n" as string); } print_str("OK\n" as string); return 0; } )SOURCE", "OK\n"); add_simple_case("short circuit", R"SOURCE( use "std.zig"; export fn main(argc : isize, argv : &&u8, env : &&u8) -> i32 { if true || { print_str("BAD 1\n" as string); false } { print_str("OK 1\n" as string); } if false || { print_str("OK 2\n" as string); false } { print_str("BAD 2\n" as string); } if true && { print_str("OK 3\n" as string); false } { print_str("BAD 3\n" as string); } if false && { print_str("BAD 4\n" as string); false } { } else { print_str("OK 4\n" as string); } return 0; } )SOURCE", "OK 1\nOK 2\nOK 3\nOK 4\n"); add_simple_case("modify operators", R"SOURCE( use "std.zig"; export fn main(argc : isize, argv : &&u8, env : &&u8) -> i32 { let mut i : i32 = 0; i += 5; if i != 5 { print_str("BAD +=\n" as string); } i -= 2; if i != 3 { print_str("BAD -=\n" as string); } i *= 20; if i != 60 { print_str("BAD *=\n" as string); } i /= 3; if i != 20 { print_str("BAD /=\n" as string); } i %= 11; if i != 9 { print_str("BAD %=\n" as string); } i <<= 1; if i != 18 { print_str("BAD <<=\n" as string); } i >>= 2; if i != 4 { print_str("BAD >>=\n" as string); } i = 6; i &= 5; if i != 4 { print_str("BAD &=\n" as string); } i ^= 6; if i != 2 { print_str("BAD ^=\n" as string); } i = 6; i |= 3; if i != 7 { print_str("BAD |=\n" as string); } print_str("OK\n" as string); return 0; } )SOURCE", "OK\n"); add_simple_case("structs", R"SOURCE( use "std.zig"; export fn main(argc : isize, argv : &&u8, env : &&u8) -> i32 { let mut foo : Foo; foo.a = foo.a + 1; foo.b = foo.a == 1; test_foo(foo); return 0; } struct Foo { a : i32, b : bool, c : f32, } fn test_foo(foo : Foo) { if foo.b { print_str("OK\n" as string); } } )SOURCE", "OK\n"); } static void add_compile_failure_test_cases(void) { add_compile_fail_case("multiple function definitions", R"SOURCE( fn a() {} fn a() {} )SOURCE", 1, ".tmp_source.zig:3:1: error: redefinition of 'a'"); add_compile_fail_case("bad directive", R"SOURCE( #bogus1("") extern { fn b(); } #bogus2("") fn a() {} )SOURCE", 2, ".tmp_source.zig:2:1: error: invalid directive: 'bogus1'", ".tmp_source.zig:6:1: error: invalid directive: 'bogus2'"); add_compile_fail_case("unreachable with return", R"SOURCE( fn a() -> unreachable {return;} )SOURCE", 1, ".tmp_source.zig:2:24: error: expected type 'unreachable', got 'void'"); add_compile_fail_case("control reaches end of non-void function", R"SOURCE( fn a() -> i32 {} )SOURCE", 1, ".tmp_source.zig:2:15: error: expected type 'i32', got 'void'"); add_compile_fail_case("undefined function call", R"SOURCE( fn a() { b(); } )SOURCE", 1, ".tmp_source.zig:3:5: error: undefined function: 'b'"); add_compile_fail_case("wrong number of arguments", R"SOURCE( fn a() { b(1); } fn b(a: i32, b: i32, c: i32) { } )SOURCE", 1, ".tmp_source.zig:3:6: error: wrong number of arguments. Expected 3, got 1."); add_compile_fail_case("invalid type", R"SOURCE( fn a() -> bogus {} )SOURCE", 1, ".tmp_source.zig:2:11: error: invalid type name: 'bogus'"); add_compile_fail_case("pointer to unreachable", R"SOURCE( fn a() -> &unreachable {} )SOURCE", 1, ".tmp_source.zig:2:11: error: pointer to unreachable not allowed"); add_compile_fail_case("unreachable code", R"SOURCE( fn a() { return; b(); } fn b() {} )SOURCE", 1, ".tmp_source.zig:4:5: error: unreachable code"); add_compile_fail_case("bad version string", R"SOURCE( #version("aoeu") export executable "test"; )SOURCE", 1, ".tmp_source.zig:2:1: error: invalid version string"); add_compile_fail_case("bad import", R"SOURCE( use "bogus-does-not-exist.zig"; )SOURCE", 1, ".tmp_source.zig:2:1: error: unable to find 'bogus-does-not-exist.zig'"); add_compile_fail_case("undeclared identifier", R"SOURCE( fn a() { b + c } )SOURCE", 2, ".tmp_source.zig:3:5: error: use of undeclared identifier 'b'", ".tmp_source.zig:4:5: error: use of undeclared identifier 'c'"); add_compile_fail_case("goto cause unreachable code", R"SOURCE( fn a() { goto done; b(); done: return; } fn b() {} )SOURCE", 1, ".tmp_source.zig:4:5: error: unreachable code"); add_compile_fail_case("parameter redeclaration", R"SOURCE( fn f(a : i32, a : i32) { } )SOURCE", 1, ".tmp_source.zig:2:1: error: redeclaration of parameter 'a'"); add_compile_fail_case("local variable redeclaration", R"SOURCE( fn f() { let a : i32 = 0; let a = 0; } )SOURCE", 1, ".tmp_source.zig:4:5: error: redeclaration of variable 'a'"); add_compile_fail_case("local variable redeclares parameter", R"SOURCE( fn f(a : i32) { let a = 0; } )SOURCE", 1, ".tmp_source.zig:3:5: error: redeclaration of variable 'a'"); add_compile_fail_case("variable has wrong type", R"SOURCE( fn f() -> i32 { let a = c"a"; a } )SOURCE", 1, ".tmp_source.zig:2:15: error: expected type 'i32', got '&const u8'"); add_compile_fail_case("if condition is bool, not int", R"SOURCE( fn f() { if (0) {} } )SOURCE", 1, ".tmp_source.zig:3:9: error: expected type 'bool', got '(u8 literal)'"); add_compile_fail_case("assign unreachable", R"SOURCE( fn f() { let a = return; } )SOURCE", 1, ".tmp_source.zig:3:5: error: variable initialization is unreachable"); add_compile_fail_case("unreachable variable", R"SOURCE( fn f() { let a : unreachable = return; } )SOURCE", 1, ".tmp_source.zig:3:13: error: variable of type 'unreachable' not allowed"); add_compile_fail_case("unreachable parameter", R"SOURCE( fn f(a : unreachable) {} )SOURCE", 1, ".tmp_source.zig:2:10: error: parameter of type 'unreachable' not allowed"); add_compile_fail_case("exporting a void parameter", R"SOURCE( export fn f(a : void) {} )SOURCE", 1, ".tmp_source.zig:2:17: error: parameter of type 'void' not allowed on exported functions"); add_compile_fail_case("unused label", R"SOURCE( fn f() { a_label: } )SOURCE", 1, ".tmp_source.zig:3:1: error: label 'a_label' defined but not used"); add_compile_fail_case("bad assignment target", R"SOURCE( fn f() { 3 = 3; } )SOURCE", 1, ".tmp_source.zig:3:5: error: assignment target must be variable, field, or array element"); add_compile_fail_case("assign to constant variable", R"SOURCE( fn f() { let a = 3; a = 4; } )SOURCE", 1, ".tmp_source.zig:4:5: error: cannot assign to constant variable"); add_compile_fail_case("use of undeclared identifier", R"SOURCE( fn f() { b = 3; } )SOURCE", 1, ".tmp_source.zig:3:5: error: use of undeclared identifier 'b'"); add_compile_fail_case("let is a statement, not an expression", R"SOURCE( fn f() { (let a = 0); } )SOURCE", 1, ".tmp_source.zig:3:6: error: invalid token: 'let'"); add_compile_fail_case("array access errors", R"SOURCE( fn f() { let mut bad : bool; i[i] = i[i]; bad[bad] = bad[bad]; } )SOURCE", 8, ".tmp_source.zig:4:5: error: use of undeclared identifier 'i'", ".tmp_source.zig:4:7: error: use of undeclared identifier 'i'", ".tmp_source.zig:4:12: error: use of undeclared identifier 'i'", ".tmp_source.zig:4:14: error: use of undeclared identifier 'i'", ".tmp_source.zig:5:8: error: array access of non-array", ".tmp_source.zig:5:8: error: array subscripts must be integers", ".tmp_source.zig:5:19: error: array access of non-array", ".tmp_source.zig:5:19: error: array subscripts must be integers"); add_compile_fail_case("variadic functions only allowed in extern", R"SOURCE( fn f(...) {} )SOURCE", 1, ".tmp_source.zig:2:1: error: variadic arguments only allowed in extern functions"); } static void print_compiler_invocation(TestCase *test_case) { printf("%s", zig_exe); for (int i = 0; i < test_case->compiler_args.length; i += 1) { printf(" %s", test_case->compiler_args.at(i)); } printf("\n"); } static void run_test(TestCase *test_case) { for (int i = 0; i < test_case->source_files.length; i += 1) { TestSourceFile *test_source = &test_case->source_files.at(i); os_write_file( buf_create_from_str(test_source->relative_path), buf_create_from_str(test_source->source_code)); } Buf zig_stderr = BUF_INIT; Buf zig_stdout = BUF_INIT; int return_code; os_exec_process(zig_exe, test_case->compiler_args, &return_code, &zig_stderr, &zig_stdout); if (test_case->compile_errors.length) { if (return_code) { for (int i = 0; i < test_case->compile_errors.length; i += 1) { const char *err_text = test_case->compile_errors.at(i); if (!strstr(buf_ptr(&zig_stderr), err_text)) { printf("\n"); printf("========= Expected this compile error: =========\n"); printf("%s\n", err_text); printf("================================================\n"); print_compiler_invocation(test_case); printf("%s\n", buf_ptr(&zig_stderr)); exit(1); } } return; // success } else { printf("\nCompile failed with return code 0 (Expected failure):\n"); print_compiler_invocation(test_case); printf("%s\n", buf_ptr(&zig_stderr)); exit(1); } } if (return_code != 0) { printf("\nCompile failed with return code %d:\n", return_code); print_compiler_invocation(test_case); printf("%s\n", buf_ptr(&zig_stderr)); exit(1); } Buf program_stderr = BUF_INIT; Buf program_stdout = BUF_INIT; os_exec_process(tmp_exe_path, test_case->program_args, &return_code, &program_stderr, &program_stdout); if (return_code != 0) { printf("\nProgram exited with return code %d:\n", return_code); print_compiler_invocation(test_case); printf("%s", tmp_exe_path); for (int i = 0; i < test_case->program_args.length; i += 1) { printf(" %s", test_case->program_args.at(i)); } printf("\n"); printf("%s\n", buf_ptr(&program_stderr)); exit(1); } if (!buf_eql_str(&program_stdout, test_case->output)) { printf("\n"); print_compiler_invocation(test_case); printf("%s", tmp_exe_path); for (int i = 0; i < test_case->program_args.length; i += 1) { printf(" %s", test_case->program_args.at(i)); } printf("\n"); printf("==== Test failed. Expected output: ====\n"); printf("%s\n", test_case->output); printf("========= Actual output: ==============\n"); printf("%s\n", buf_ptr(&program_stdout)); printf("=======================================\n"); exit(1); } for (int i = 0; i < test_case->source_files.length; i += 1) { TestSourceFile *test_source = &test_case->source_files.at(i); remove(test_source->relative_path); } } static void run_all_tests(bool reverse) { if (reverse) { for (int i = test_cases.length - 1; i >= 0; i -= 1) { TestCase *test_case = test_cases.at(i); printf("Test %d/%d %s...", i + 1, test_cases.length, test_case->case_name); run_test(test_case); printf("OK\n"); } } else { for (int i = 0; i < test_cases.length; i += 1) { TestCase *test_case = test_cases.at(i); printf("Test %d/%d %s...", i + 1, test_cases.length, test_case->case_name); run_test(test_case); printf("OK\n"); } } printf("%d tests passed.\n", test_cases.length); } static void cleanup(void) { remove(tmp_source_path); remove(tmp_exe_path); } static int usage(const char *arg0) { fprintf(stderr, "Usage: %s [--reverse]\n", arg0); return 1; } int main(int argc, char **argv) { bool reverse = false; for (int i = 1; i < argc; i += 1) { if (strcmp(argv[i], "--reverse") == 0) { reverse = true; } else { return usage(argv[0]); } } add_compiling_test_cases(); add_compile_failure_test_cases(); run_all_tests(reverse); cleanup(); }