def test_parse_check_if_else_nest_level():
code = """
int main (int argc, char *argv[]) {
char *space;
space = malloc(sizeof(char) * 10);
if (space == NULL) {
if (1 > 2) {
if (3 == 4) {
return -1;
}
} else {
if (1 == 1) {
return -1;
} else {
if (1 == 2) {
return -1;
} else {
if (1 == 2) {
return -1;
} else {
return -1;
}
}
}
}
return -1;
}
return 0;
}
"""
parsed_content = parse(code, "<string>")
assert parsed_content
assert 1 == len(parsed_content.functions)
assert "main" == parsed_content.functions[0].name
assert 5 == parsed_content.functions[0].if_nest_level()
def test_parse_check_functions_calling_malloc():
code = """
int main (int argc, char *argv[]) {
char *space;
space = malloc(sizeof(char) * 10);
if (space == NULL) {
return -1;
}
return 0;
}
"""
parsed_content = parse(code, "<string>")
assert parsed_content
assert 1 == len(parsed_content.functions)
assert "main" == parsed_content.functions[0].name
assert "main" == [function.name for function
in parsed_content.functions
if function.it_calls("malloc")][0]
def test_parse_check_functions():
parsed_content = parse("int main (int argc, char *argv[]) { return 0; }",
"<string>")
assert parsed_content
assert 1 == len(parsed_content.functions)
assert "main" == parsed_content.functions[0].name
def test_parse():
parsed_content = parse("int main (int argc, char *argv[]) { return 0; }",
"<string>")
assert parsed_content