Mercurial > hg > orthanc-tests
diff PerfsDb/run.py @ 156:f1a75985caa8
first Db test framework - work in progress
author | am@osimis.io |
---|---|
date | Thu, 16 Aug 2018 17:13:32 +0200 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/PerfsDb/run.py Thu Aug 16 17:13:32 2018 +0200 @@ -0,0 +1,111 @@ +import argparse +from ConfigFileBuilder import ConfigFileBuilder +from TestConfig import TestConfig +from DbServer import DbServer +from DbType import DbType +from DbSize import DbSize + +testConfigs = { + "mysql-small" : TestConfig(label= "mysql-small", dbSize=DbSize.Small, dbServer=DbServer(dbType=DbType.MySQL, port=2000)), + "mysql-large" : TestConfig(label= "mysql-large", dbSize=DbSize.Large, dbServer=DbServer(dbType=DbType.MySQL, port=2001)), + "sqlite-small": TestConfig(label= "sqlite-small", dbSize=DbSize.Small, dbType=DbType.Sqlite), + "pg9-small": TestConfig(label= "pg9-small", dbSize=DbSize.Small, dbServer=DbServer(dbType=DbType.PG9, port=2002)), +} + +selectedTestConfigs = [] + +parser = argparse.ArgumentParser(description = "Initializes/Runs/Clears PerfsDb setup.") + +# create a cli option for each config +for testConfigName in testConfigs.keys(): + parser.add_argument("--" + testConfigName, action = "store_true") + +parser.add_argument("--init", help = "initializes DBs", action = "store_true") +parser.add_argument("--run", help = "runs tests", action = "store_true") +parser.add_argument("--clear", help = "clear DBs", action = "store_true") + +parser.add_argument("--orthanc-path", help = "path to the folder containing Orthanc executable", default=".") +parser.add_argument("--plugins-path", help = "path to the folder containing Orthanc executable", default=".") +parser.add_argument("--repeat", help = "number of times to repeat each test to average timings", type=int, default=50) + +args = parser.parse_args() + +for testConfigName in testConfigs.keys(): + if args.__dict__[testConfigName.replace("-", "_")]: + selectedTestConfigs.append(testConfigName) + +# if no test config specified, take them all +if len(selectedTestConfigs) == 0: + selectedTestConfigs = testConfigs.keys() + +# if no action specified, it means only run +if not (args.init | args.run | args.clear): + args.init = False + args.run = True + args.clear = False + +print("***** Orthanc *******") +print("path :", args.orthanc_path) + + +# libOrthancMySQLIndex.so +# libOrthancMySQLStorage.so +# libOrthancPostgreSQLIndex.so +# libOrthancPostgreSQLStorage.so +# libOrthancMSSQLIndex.so + +results = {} + +for configName in selectedTestConfigs: + testConfig = testConfigs[configName] + testConfig.setName(configName) + testConfig.setRepeatCount(args.repeat) + + print("======= " + configName + " ========") + + if args.clear: + print("** Clearing Db") + testConfig.clearDb() + + if args.init or args.run: + print("** Generating config files") + testConfig.generateOrthancConfigurationFile(args.plugins_path) + + print("** Launching DbServer") + testConfig.launchDbServer() + + print("** Launching Orthanc") + testConfig.launchOrthanc(args.orthanc_path) + + if args.init: + testConfig.initializeDb() + + if args.run: + print("** Runnnig tests") + results[configName] = testConfig.runTests() + print("** Stoping Orthanc") + testConfig.stopOrthanc() + +print("++++++++++++++ results summary +++++++++++++++") +testNames = set() +resultsByTestName = {} +for configName, configResult in results.items(): + for result in configResult: + testNames.add(result.name) + if not result.name in resultsByTestName: + resultsByTestName[result.name] = {} + resultsByTestName[result.name][configName] = result + +headerLine = "{empty:<40}|".format(empty="") +for configName in selectedTestConfigs: + headerLine += "{configName:^15}|".format(configName=configName) + +print(headerLine) + +for testName in sorted(testNames): + resultLine = "{name:<40}|".format(name=testName) + for configName in selectedTestConfigs: + resultLine += "{avg:>11.2f} ms |".format(avg = resultsByTestName[testName][configName].averageTimeInMs) + print(resultLine) + +print("** Done")