fix benchmark_test

Author: ry

deno_typescript/lib.rs

@@ -292,9 +292,9 @@ fn snapshot_to_env(
   let snapshot_slice =
     unsafe { std::slice::from_raw_parts(snapshot.data_ptr, snapshot.data_len) };
   println!("snapshot bytes {}", snapshot_slice.len());
-  //
+
   let out_dir = PathBuf::from(std::env::var_os("OUT_DIR").unwrap());
-  let snapshot_path = out_dir.join(env_var);
+  let snapshot_path = out_dir.join(String::from(env_var) + ".bin");
 
   fs::write(&snapshot_path, snapshot_slice)?;
   println!("snapshot path {} ", snapshot_path.display());

tools/benchmark.py

@@ -11,7 +11,7 @@
 import json
 import time
 import shutil
-from util import root_path, run, run_output, build_path, executable_suffix
+from util import find_exts, root_path, run, run_output, build_path, executable_suffix
 import tempfile
 import http_server
 import throughput_benchmark
@@ -59,22 +59,26 @@ def import_data_from_gh_pages():
 
 
 def get_binary_sizes(build_dir):
+    # Because cargo's OUT_DIR is not predictable, we have to search the build
+    # tree for these files...
+    files = find_exts([build_dir], ["js", "map", "bin"])
     path_dict = {
-        "deno":
-        os.path.join(build_dir, "deno" + executable_suffix),
-        "main.js":
-        os.path.join(build_dir, "gen/cli/bundle/main.js"),
-        "main.js.map":
-        os.path.join(build_dir, "gen/cli/bundle/main.js.map"),
-        "compiler.js":
-        os.path.join(build_dir, "gen/cli/bundle/compiler.js"),
-        "compiler.js.map":
-        os.path.join(build_dir, "gen/cli/bundle/compiler.js.map"),
-        "snapshot_deno.bin":
-        os.path.join(build_dir, "gen/cli/snapshot_deno.bin"),
-        "snapshot_compiler.bin":
-        os.path.join(build_dir, "gen/cli/snapshot_compiler.bin")
+        "deno": os.path.join(build_dir, "deno" + executable_suffix),
     }
+    for f in files:
+        if f.endswith("CLI_SNAPSHOT.js"):
+            path_dict["CLI_SNAPSHOT.js"] = f
+        elif f.endswith("CLI_SNAPSHOT.js.map"):
+            path_dict["CLI_SNAPSHOT.js.map"] = f
+        elif f.endswith("CLI_SNAPSHOT.bin"):
+            path_dict["CLI_SNAPSHOT.bin"] = f
+        elif f.endswith("COMPILER_SNAPSHOT.js"):
+            path_dict["COMPILER_SNAPSHOT.js"] = f
+        elif f.endswith("COMPILER_SNAPSHOT.js.map"):
+            path_dict["COMPILER_SNAPSHOT.js.map"] = f
+        elif f.endswith("COMPILER_SNAPSHOT.bin"):
+            path_dict["COMPILER_SNAPSHOT.bin"] = f
+
     sizes = {}
     for name, path in path_dict.items():
         assert os.path.exists(path)

tools/benchmark_test.py

@@ -32,9 +32,9 @@ def test_max_mem_parse(self):
     def test_binary_size(self):
         binary_size_dict = benchmark.get_binary_sizes(self.build_dir)
         assert binary_size_dict["deno"] > 0
-        assert binary_size_dict["main.js"] > 0
-        assert binary_size_dict["main.js.map"] > 0
-        assert binary_size_dict["snapshot_deno.bin"] > 0
+        assert binary_size_dict["CLI_SNAPSHOT.js"] > 0
+        assert binary_size_dict["CLI_SNAPSHOT.js.map"] > 0
+        assert binary_size_dict["CLI_SNAPSHOT.bin"] > 0
 
     @unittest.skipIf("linux" not in sys.platform,
                      "strace only supported on linux")