Files
cbindgen/test.py
T
2017-11-09 18:04:22 -05:00

105 lines
2.3 KiB
Python
Executable File

#!/bin/python
import os
import glob
import subprocess
import sys
def build_cbindgen():
try:
subprocess.check_output(["cargo", "build"])
return True
except subprocess.CalledProcessError:
return False
def cbindgen(rust_src, out, c, config):
bin = ["target/debug/cbindgen"]
compile = [rust_src, "-o", out]
flags = []
if c:
flags += ["--lang", "c"]
if config != None:
flags += ["--config", config]
command = bin + flags + compile
print(command)
subprocess.check_output(bin + flags + compile)
def gcc(src):
gcc_bin = os.environ.get('CC')
if gcc_bin == None:
gcc_bin = 'gcc'
subprocess.check_output([gcc_bin, "-D", "DEFINED", "-c", src, "-o", "tests/expectations/tmp.o"])
os.remove("tests/expectations/tmp.o")
def gxx(src):
gxx_bin = os.environ.get('CXX')
if gxx_bin == None:
gxx_bin = 'g++'
subprocess.check_output([gxx_bin, "-D", "DEFINED", "-std=c++11", "-c", src, "-o", "tests/expectations/tmp.o"])
os.remove("tests/expectations/tmp.o")
def run_compile_test(rust_src, c):
if c:
out = os.path.join('tests/expectations/', os.path.basename(rust_src).replace(".rs", ".c"))
else:
out = os.path.join('tests/expectations/', os.path.basename(rust_src).replace(".rs", ".cpp"))
config = rust_src.replace(".rs", ".toml")
if not os.path.exists(config):
config = None
try:
cbindgen(rust_src, out, c, config)
if c:
gcc(out)
else:
gxx(out)
except subprocess.CalledProcessError:
return False
return True
if not build_cbindgen():
exit()
args = sys.argv[1:]
files = [x for x in args if not x.startswith("-")]
tests = []
if len(files) == 0:
tests = glob.glob("tests/rust/*.rs")
else:
tests = files
num_pass = 0
num_fail = 0
# C
for test in tests:
if run_compile_test(test, True):
num_pass += 1
print("Pass - %s" % test)
else:
num_fail += 1
print("Fail - %s" % test)
# C++
for test in tests:
if run_compile_test(test, False):
num_pass += 1
print("Pass - %s" % test)
else:
num_fail += 1
print("Fail - %s" % test)
print("Tests complete. %i passed, %i failed." % (num_pass, num_fail))