Support multiple extensions in lld test generation (WebAssembly#1417)

aheejin · web-flow · commit 8ddbac2c642e · 2018-02-12T13:43:25.000-08:00
diff --git a/auto_update_tests.py b/auto_update_tests.py
@@ -76,22 +76,35 @@
 
 print '\n[ checking wasm-link-metadata testcases... ]\n'
 
+link_metadata_extension_arg_map = {
+  '.json': [],
+  '.jscall.json': ['--emscripten-reserved-function-pointers=3'],
+}
 for obj_path in files_with_pattern('test', 'lld', '*.o'):
   print '..', obj_path
-  json_path = obj_path.replace('.o', '.json')
-
-  cmd = WASM_LINK_METADATA + [obj_path]
-  actual = run_command(cmd)
-  with open(json_path, 'w') as o: o.write(actual)
+  for ext, ext_args in link_metadata_extension_arg_map.items():
+    json_path = obj_path.replace('.o', ext)
+    if ext != 'json' and not os.path.exists(json_path):
+      continue
+    cmd = WASM_LINK_METADATA + [obj_path] + ext_args
+    actual = run_command(cmd)
+    with open(json_path, 'w') as o: o.write(actual)
 
 print '\n[ checking wasm-emscripten-finalize testcases... ]\n'
 
+emscripten_finalize_extension_arg_map = {
+  '.out': [],
+  '.jscall.out': ['--emscripten-reserved-function-pointers=3'],
+}
 for wast_path in files_with_pattern('test', 'lld', '*.wast'):
   print '..', wast_path
-  out_path = wast_path + '.out'
-  cmd = WASM_EMSCRIPTEN_FINALIZE + [wast_path, '-S']
-  actual = run_command(cmd)
-  with open(out_path, 'w') as o: o.write(actual)
+  for ext, ext_args in emscripten_finalize_extension_arg_map.items():
+    out_path = wast_path + ext
+    if ext != '.out' and not os.path.exists(out_path):
+      continue
+    cmd = WASM_EMSCRIPTEN_FINALIZE + [wast_path, '-S'] + ext_args
+    actual = run_command(cmd)
+    with open(out_path, 'w') as o: o.write(actual)
 
 for t in sorted(os.listdir(os.path.join('test', 'print'))):
   if t.endswith('.wast'):
