use ast_convert;
use ast_dump;
use python_tb;
use python_parse;
use std::io::Read;
use std::fs::File;

fn test_parse_file(filename: &str){
    let code = {
        let mut file = match File::open(filename) {
            Ok(file) => file,
            Err(err) => {
                panic!(format!("Can't find {}", filename));
            }
        };
        let mut code = String::new();
        file.read_to_string(&mut code).unwrap();
        code
    };

    let module = match python_parse::parse_ast(code.clone()) {
        Ok(module) => module,
        Err(err) => {
            python_tb::traceback(err);
            panic!(format!("Error while parsing file “{}”:", filename));
        }
    };

    let module_ast = ast_convert::convert_ast("__main__".to_string(), &module);
    let dumped_ast = ast_dump::dump_ast(&module_ast);
    assert_eq!(code, dumped_ast + "\n");
}

#[test]
fn test_assign(){
    test_parse_file("tests/test_parse_files/test_assign.py");
}