89 lines
3.3 KiB
Python
89 lines
3.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from fabric.api import *
|
||
|
|
|
||
|
|
from common import manager_fsim_dir, manager_ci_dir, manager_fsim_pem, set_fabric_firesim_pem
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def run_build_recipes_ini_api_tests():
|
||
|
|
""" Test config_{build, build_recipes}.ini APIs """
|
||
|
|
|
||
|
|
def commands_to_run(commands, opts):
|
||
|
|
""" Run a list of commands with the specified opts """
|
||
|
|
for command in commands:
|
||
|
|
with prefix('cd {} && source sourceme-f1-manager.sh'.format(manager_fsim_dir)):
|
||
|
|
rc = 0
|
||
|
|
with settings(warn_only=True):
|
||
|
|
rc = run("{} {}".format(command, opts)).return_code
|
||
|
|
if rc == 0:
|
||
|
|
print("{} passed unexpectedly.".format(command))
|
||
|
|
|
||
|
|
# exit since passing is not wanted
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
def run_test(name):
|
||
|
|
# test files should exist on the manager already
|
||
|
|
test_dir = "{}/ini-tests/failing-buildafi-files/{}".format(manager_ci_dir, name)
|
||
|
|
|
||
|
|
commands_to_run(
|
||
|
|
["firesim buildafi"],
|
||
|
|
"-b {}/sample_config_build.ini -r {}/sample_config_build_recipes.ini".format(test_dir, test_dir))
|
||
|
|
|
||
|
|
run_test("invalid-build-section")
|
||
|
|
run_test("invalid-recipe-inst-type")
|
||
|
|
|
||
|
|
# test invalid config_build.ini
|
||
|
|
commands_to_run(["firesim buildafi"], "-b ~/GHOST_FILE")
|
||
|
|
|
||
|
|
# test invalid config_build_recipes.ini
|
||
|
|
commands_to_run(["firesim buildafi"], "-r ~/GHOST_FILE")
|
||
|
|
|
||
|
|
def run_runtime_hwdb_ini_api_tests():
|
||
|
|
""" Test config_{runtime, hwdb}.ini APIs """
|
||
|
|
|
||
|
|
def commands_to_run(commands, opts):
|
||
|
|
""" Run a list of commands with the specified opts """
|
||
|
|
for command in commands:
|
||
|
|
with prefix('cd {} && source sourceme-f1-manager.sh'.format(manager_fsim_dir)):
|
||
|
|
rc = 0
|
||
|
|
with settings(warn_only=True):
|
||
|
|
rc = run("{} {}".format(command, opts)).return_code
|
||
|
|
if rc == 0:
|
||
|
|
print("{} passed unexpectedly.".format(command))
|
||
|
|
|
||
|
|
# test passed so make sure to terminate runfarm
|
||
|
|
run("firesim terminaterunfarm -q {}".format(opts))
|
||
|
|
|
||
|
|
# exit since passing is not wanted
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
def run_test(name):
|
||
|
|
# test files should exist on the manager already
|
||
|
|
test_dir = "{}/ini-tests/failing-runtime-files/{}".format(manager_ci_dir, name)
|
||
|
|
|
||
|
|
commands_to_run(
|
||
|
|
["firesim launchrunfarm", "firesim infrasetup", "firesim runworkload", "firesim terminaterunfarm -q"],
|
||
|
|
"-c {}/sample_config_runtime.ini -a {}/sample_config_hwdb.ini".format(test_dir, test_dir))
|
||
|
|
|
||
|
|
run_test("hwdb-invalid-afi")
|
||
|
|
run_test("runtime-invalid-hwconfig")
|
||
|
|
run_test("runtime-invalid-topology")
|
||
|
|
run_test("runtime-invalid-workloadname")
|
||
|
|
|
||
|
|
# test invalid config_runtime.ini
|
||
|
|
commands_to_run(["firesim launchrunfarm", "firesim infrasetup", "firesim runworkload", "firesim terminaterunfarm -q"], "-c ~/GHOST_FILE")
|
||
|
|
|
||
|
|
# test invalid config_hwdb.ini
|
||
|
|
commands_to_run(["firesim launchrunfarm", "firesim infrasetup", "firesim runworkload", "firesim terminaterunfarm -q"], "-a ~/GHOST_FILE")
|
||
|
|
|
||
|
|
def run_ini_api_tests():
|
||
|
|
""" Test manager .ini file APIs """
|
||
|
|
|
||
|
|
run_build_recipes_ini_api_tests()
|
||
|
|
run_runtime_hwdb_ini_api_tests()
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
set_fabric_firesim_pem()
|
||
|
|
execute(run_ini_api_tests, hosts=["localhost"])
|