mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-10 22:16:41 +00:00
Bernd Hahnebach
330 lines
10 KiB
Python
330 lines
10 KiB
Python
import behave.formatter.pretty # Needed for pyinstaller to package it
|
|
import fileinput
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import webbrowser
|
|
from behave.__main__ import main as behave_main
|
|
|
|
|
|
# get bimtester source code module path
|
|
bimtester_path = os.path.dirname(os.path.realpath(__file__))
|
|
# print(bimtester_path)
|
|
locale_path = os.path.join(bimtester_path, "locale")
|
|
|
|
|
|
try:
|
|
# PyInstaller creates a temp folder and stores path in _MEIPASS
|
|
base_path = sys._MEIPASS
|
|
except Exception:
|
|
base_path = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
|
|
def get_resource_path(relative_path):
|
|
return os.path.join(base_path, relative_path)
|
|
|
|
|
|
def run_tests(args):
|
|
if not get_features(args):
|
|
print("No features could be found to check.")
|
|
return False
|
|
behave_args = [get_resource_path("features")]
|
|
if args["advanced_arguments"]:
|
|
behave_args.extend(args["advanced_arguments"].split())
|
|
elif not args["console"]:
|
|
behave_args.extend([
|
|
"--format",
|
|
"json.pretty",
|
|
"--outfile",
|
|
"report/report.json"
|
|
])
|
|
behave_args.extend([
|
|
"--define",
|
|
"localedir={}".format(locale_path)
|
|
])
|
|
if args["ifcfile"]:
|
|
behave_args.extend(["--define", "ifcfile={}".format(args["ifcfile"])])
|
|
if args["path"]:
|
|
behave_args.extend(["--define", "path={}".format(args["path"])])
|
|
behave_main(behave_args)
|
|
print("# All tests are finished.")
|
|
return True
|
|
|
|
|
|
def get_features(args):
|
|
# current_path = os.path.abspath(".")
|
|
features_dir = get_resource_path("features")
|
|
for f in os.listdir(features_dir):
|
|
if f.endswith(".feature"):
|
|
os.remove(os.path.join(features_dir, f))
|
|
if args["feature"]:
|
|
shutil.copyfile(
|
|
args["feature"],
|
|
os.path.join(
|
|
get_resource_path("features"),
|
|
os.path.basename(args["feature"])
|
|
)
|
|
)
|
|
return True
|
|
if os.path.exists("features"):
|
|
shutil.copytree("features", get_resource_path("features"))
|
|
return True
|
|
has_features = False
|
|
for f in os.listdir("."):
|
|
if not f.endswith(".feature"):
|
|
continue
|
|
if args["feature"] and args["feature"] != f:
|
|
continue
|
|
has_features = True
|
|
shutil.copyfile(
|
|
f,
|
|
os.path.join(get_resource_path("features"), os.path.basename(f))
|
|
)
|
|
return has_features
|
|
|
|
|
|
"""
|
|
# clean logs to be able to run tests
|
|
# once again but on another building model and in another directory
|
|
# somehow does not work, thus test will be run in the same directory
|
|
# on each new run, directory will be deleted before each new run
|
|
# https://github.com/behave/behave/issues/871
|
|
# run bimtester
|
|
# copy manually this code, run bimtester again,
|
|
# does not work on two directories
|
|
from behave.runner_util import reset_runtime
|
|
reset_runtime()
|
|
|
|
"""
|
|
|
|
|
|
# TODO: if the ifc file name or path contains special character
|
|
# like German Umlaute behave gives an error
|
|
|
|
|
|
def run_intmp_tests(args={}):
|
|
|
|
"""
|
|
run bimtester unit test in a temporary directory
|
|
features, steps and environment.py are copied to a temp directory
|
|
|
|
Keys of parameter args
|
|
----------------------
|
|
features: optional (ATM mandatory)
|
|
the path the features directory with feature files is in
|
|
ifcfile: optional (ATM mandatory)
|
|
the ifc file
|
|
advanced_arguments: optional
|
|
they will be directly passed to the behave call
|
|
|
|
features and ifcfile are given:
|
|
the ifcfile in feature files is replaced
|
|
|
|
features only is given (TODO):
|
|
the ifcfile provided in the feature files is used
|
|
|
|
ifcfile only is given (TODO):
|
|
features = ifcfile directory
|
|
the ifcfile in feature files is replaced
|
|
|
|
none of both is given (TODO):
|
|
the current directory = features
|
|
the ifcfile provided in the feature files is used
|
|
|
|
TODO: if the above is implemented adapt signature of run_all
|
|
"""
|
|
|
|
from behave import __version__ as behave_version
|
|
# https://github.com/behave/behave/issues/871
|
|
if behave_version == "1.2.5":
|
|
print(
|
|
"At least behave version 1.2.6 is needed, but version {} found."
|
|
.format(behave_version)
|
|
)
|
|
return False
|
|
|
|
# get the features_path, the dir where the feature files to test are in
|
|
if "features" in args and args["features"] != "":
|
|
the_features_path = os.path.join(args["features"], "features")
|
|
if not os.path.isdir(the_features_path):
|
|
print(
|
|
"The features directory does not exist: {}"
|
|
.format(the_features_path)
|
|
)
|
|
return False
|
|
else:
|
|
# TODO assume features beside ifc thus use ifc path
|
|
print("No features path was given.")
|
|
return False
|
|
|
|
# get ifc path and ifc filename
|
|
if "ifcfile" in args and args["ifcfile"] != "":
|
|
ifcfile = args["ifcfile"]
|
|
ifc_path = os.path.dirname(os.path.realpath(ifcfile))
|
|
if os.path.isdir(ifc_path) is False:
|
|
print("ifc path does not exist.")
|
|
return False
|
|
if os.path.isfile(ifcfile) is True:
|
|
ifc_filename = os.path.basename(ifcfile)
|
|
else:
|
|
print("ifc file '{}' does not exist.".format(ifcfile))
|
|
return False
|
|
else:
|
|
# TODO use ifc path from feature files
|
|
print("No ifc file was given.")
|
|
return False
|
|
|
|
# set up paths
|
|
# a unique temp path should not be used
|
|
# behave raises an ambiguous step exception
|
|
# run_path = tempfile.mkdtemp()
|
|
# thus use the same path on every run
|
|
# but delete it if exists
|
|
run_path = os.path.join(tempfile.gettempdir(), "bimtesterfc")
|
|
if os.path.isdir(run_path):
|
|
from shutil import rmtree
|
|
rmtree(run_path) # fails on read only files
|
|
if os.path.isdir(run_path):
|
|
print("Delete former beimtester run dir {} failed".format(run_path))
|
|
return False
|
|
os.mkdir(run_path)
|
|
report_path = os.path.join(run_path, "report")
|
|
copy_features_path = os.path.join(run_path, "features")
|
|
|
|
# copy features path from bimtester source code
|
|
srccode_features_path = os.path.join(
|
|
bimtester_path,
|
|
"features",
|
|
)
|
|
# print(srccode_features_path)
|
|
if os.path.exists(srccode_features_path):
|
|
shutil.copytree(srccode_features_path, copy_features_path)
|
|
else:
|
|
print(
|
|
"Bimtester source code features directory {} not found."
|
|
.format(srccode_features_path)
|
|
)
|
|
return False
|
|
|
|
# copy features files
|
|
# print(the_features_path)
|
|
# print(copy_features_path)
|
|
# parameter dirs_exist_ok=True, from py 3.8
|
|
# if os.path.exists(the_features_path):
|
|
# shutil.copytree(
|
|
# the_features_path,
|
|
# copy_features_path,
|
|
# dirs_exist_ok=True
|
|
# )
|
|
|
|
# copy feature files and replace ifcpath in feature files
|
|
# replaceing is IMHO better than copy the ifc file which could be 500 MB
|
|
feature_files = os.listdir(the_features_path)
|
|
# print(feature_files)
|
|
for feature_file in feature_files:
|
|
cp_feature_file = os.path.join(copy_features_path, feature_file)
|
|
# print(feature_file)
|
|
# copy file
|
|
shutil.copyfile(
|
|
os.path.join(the_features_path, feature_file),
|
|
cp_feature_file
|
|
)
|
|
# search the line
|
|
ff = open(cp_feature_file, "r")
|
|
lines = ff.readlines()
|
|
ff.close()
|
|
theline = ""
|
|
for line in lines:
|
|
if "* The IFC file" in line and "must be provided" in line:
|
|
theline = line
|
|
if ifc_filename is None:
|
|
ifc_filename = os.path.basename(theline.split('"')[1])
|
|
newifcline = (
|
|
' * The IFC file "{}" must be provided\n'
|
|
.format(os.path.join(ifc_path, ifc_filename))
|
|
)
|
|
# print(newifcline)
|
|
break
|
|
else:
|
|
print("The line which sets the ifc file to test was not found.")
|
|
newifcline = ""
|
|
# replace the line
|
|
if newifcline != "":
|
|
# https://stackoverflow.com/a/290494
|
|
for line in fileinput.input(cp_feature_file, inplace=True):
|
|
# the print replaces the line in the file
|
|
print(line.replace(theline, newifcline), end="")
|
|
|
|
# get advanced args
|
|
# print to console from inside step files, add "--no-capture" flag
|
|
# https://github.com/behave/behave/issues/346
|
|
behave_args = [copy_features_path]
|
|
if "advanced_arguments" in args:
|
|
behave_args.extend(args["advanced_arguments"].split())
|
|
elif "console" not in args:
|
|
behave_args.extend([
|
|
# redirect prints in step methods
|
|
# if step fails some output is catched, thus might not be printed
|
|
"--no-capture",
|
|
# next two lines are one arg
|
|
"--format",
|
|
"json.pretty",
|
|
# next two lines are one arg
|
|
"--outfile",
|
|
os.path.join(report_path, "report.json"),
|
|
# next two lines are one arg
|
|
"--define",
|
|
"ifcbasename={}".format(os.path.splitext(ifc_filename)[0]),
|
|
# next two lines are one arg
|
|
"--define",
|
|
"localedir={}".format(locale_path)
|
|
])
|
|
print(behave_args)
|
|
|
|
# run tests
|
|
from behave.__main__ import main as behave_main
|
|
behave_main(behave_args)
|
|
print("All tests are finished.")
|
|
|
|
# delete steps
|
|
# shutil.rmtree(steps_path)
|
|
|
|
return run_path
|
|
|
|
|
|
def run_all(the_features_path, the_ifcfile):
|
|
|
|
# run bimtester
|
|
runpath = run_intmp_tests({
|
|
"features": the_features_path,
|
|
"ifcfile": the_ifcfile
|
|
})
|
|
print(runpath)
|
|
|
|
# check if it worked out well
|
|
if runpath is False:
|
|
print("BIMTester behave tests returned False.")
|
|
return False
|
|
|
|
if not os.path.isdir(runpath):
|
|
print("runpath does not exist. This should not happen. Debug")
|
|
return False
|
|
|
|
# create html report and open in webbrowser
|
|
from .reports import generate_report
|
|
generate_report(runpath)
|
|
# get the feature files
|
|
feature_files = os.listdir(
|
|
os.path.join(the_features_path, "features")
|
|
)
|
|
# print(feature_files)
|
|
for ff in feature_files:
|
|
webbrowser.open(os.path.join(
|
|
runpath,
|
|
"report",
|
|
ff + ".html"
|
|
))
|
|
|
|
return True
|