Add max memory benchmark (#2061)

1 files changed, 26 insertions, 0 deletions

diff --git a/tools/benchmark_test.py b/tools/benchmark_test.py
index b2c4aa9a6..bc3ce8a71 100644..100755
--- a/tools/benchmark_test.py
+++ b/tools/benchmark_test.py
@@ -1,7 +1,9 @@
+#!/usr/bin/env python
 # Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
 import sys
 import os
 import benchmark
+from util import build_path, executable_suffix
 
 
 def strace_parse_test():
@@ -20,6 +22,12 @@ def strace_parse_test():
         assert summary["total"]["calls"] == 704
 
 
+def max_mem_parse_test():
+    with open(os.path.join(sys.path[0], "testdata/time.out"), "r") as f:
+        data = f.read()
+        assert benchmark.find_max_mem_in_bytes(data) == 120380 * 1024
+
+
 def binary_size_test(build_dir):
     binary_size_dict = benchmark.get_binary_sizes(build_dir)
     assert binary_size_dict["deno"] > 0
@@ -43,6 +51,24 @@ def syscall_count_test(deno_path):
 def benchmark_test(build_dir, deno_path):
     strace_parse_test()
     binary_size_test(build_dir)
+    max_mem_parse_test()
     if "linux" in sys.platform:
         thread_count_test(deno_path)
         syscall_count_test(deno_path)
+
+
+# This test assumes tools/http_server.py is running in the background.
+def main():
+    if len(sys.argv) == 2:
+        build_dir = sys.argv[1]
+    elif len(sys.argv) == 1:
+        build_dir = build_path()
+    else:
+        print "Usage: tools/benchmark_test.py [build_dir]"
+        sys.exit(1)
+    deno_exe = os.path.join(build_dir, "deno" + executable_suffix)
+    benchmark_test(build_dir, deno_exe)
+
+
+if __name__ == '__main__':
+    main()