fix #118

Author: TomWambsgans

crates/lean_compiler/src/parser/parsers/program.rs

@@ -53,11 +53,9 @@ impl Parse<Program> for ProgramParser {
                         let saved_file_id = ctx.current_file_id;
                         ctx.current_filepath = filepath.clone();
                         ctx.imported_filepaths.insert(filepath.clone());
-                        let file_id = ctx.get_next_file_id();
-                        ctx.current_file_id = file_id;
-                        filepaths.insert(file_id, filepath.clone());
-                        ctx.current_source_code = ProgramSource::Filepath(filepath).get_content(&ctx.flags).unwrap();
-                        source_code.insert(file_id, ctx.current_source_code.clone());
+                        ctx.current_source_code = ProgramSource::Filepath(filepath.clone())
+                            .get_content(&ctx.flags)
+                            .unwrap();
                         let subprogram = parse_program_helper(ctx)?;
                         functions.extend(subprogram.functions);
                         function_locations.extend(subprogram.function_locations);

crates/lean_compiler/tests/test_compiler.rs

@@ -1322,3 +1322,13 @@ fn test_const_array_element_exponentiation() {
         false,
     );
 }
+
+#[test]
+fn test_num_files() {
+    let expected_num_files = 3; // program_6.snark imports foo.snark and bar.snark
+    let manifest_dir = env!("CARGO_MANIFEST_DIR");
+    let path = format!("{manifest_dir}/tests/program_6.snark");
+    let bytecode = compile_program(&ProgramSource::Filepath(path));
+    assert_eq!(bytecode.filepaths.len(), expected_num_files);
+    assert_eq!(bytecode.source_code.len(), expected_num_files);
+}