/src/etc/combine-tests.py
http://github.com/jruderman/rust · Python · 70 lines · 56 code · 10 blank · 4 comment · 15 complexity · 85c06a891d1d1b1341be1792367f946c MD5 · raw file
- #!/usr/bin/env python
- # this combines all the working run-pass tests into a single large crate so we
- # can run it "fast": spawning zillions of windows processes is our major build
- # bottleneck (and it doesn't hurt to run faster on other platforms as well).
- import sys, os, re, codecs
- def scrub(b):
- if sys.version_info >= (3,) and type(b) == bytes:
- return b.decode('ascii')
- else:
- return b
- src_dir = scrub(os.getenv("CFG_SRC_DIR"))
- if not src_dir:
- raise Exception("missing env var CFG_SRC_DIR")
- run_pass = os.path.join(src_dir, "src", "test", "run-pass")
- run_pass = os.path.abspath(run_pass)
- stage2_tests = []
- take_args = {}
- for t in os.listdir(run_pass):
- if t.endswith(".rs") and not (
- t.startswith(".") or t.startswith("#") or t.startswith("~")):
- f = codecs.open(os.path.join(run_pass, t), "r", "utf8")
- s = f.read()
- if not ("xfail-test" in s or
- "xfail-fast" in s or
- "xfail-win32" in s):
- stage2_tests.append(t)
- if "main(args: ~[~str])" in s:
- take_args[t] = True
- f.close()
- stage2_tests.sort()
- c = open("tmp/run_pass_stage2.rc", "w")
- i = 0
- c.write("// AUTO-GENERATED FILE: DO NOT EDIT\n")
- c.write("#[link(name=\"run_pass_stage2\", vers=\"0.1\")];\n")
- for t in stage2_tests:
- p = os.path.join(run_pass, t)
- p = p.replace("\\", "\\\\")
- c.write("#[path = \"%s\"]" % p);
- c.write("mod t_%d;\n" % i)
- i += 1
- c.close()
- d = open("tmp/run_pass_stage2_driver.rs", "w")
- d.write("// AUTO-GENERATED FILE: DO NOT EDIT\n")
- d.write("use std;\n")
- d.write("use run_pass_stage2;\n")
- d.write("import run_pass_stage2::*;\n")
- d.write("import io::writer_util;\n");
- d.write("fn main() {\n");
- d.write(" let out = io::stdout();\n");
- i = 0
- for t in stage2_tests:
- p = os.path.join("test", "run-pass", t)
- p = p.replace("\\", "\\\\")
- d.write(" out.write_str(~\"run-pass [stage2]: %s\\n\");\n" % p)
- if t in take_args:
- d.write(" t_%d::main(~[~\"arg0\"]);\n" % i)
- else:
- d.write(" t_%d::main();\n" % i)
- i += 1
- d.write("}\n")