mirror of https://github.com/apache/kafka.git
57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# ===================================
|
||
|
|
# system_test_env.py
|
||
|
|
# ===================================
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
class SystemTestEnv():
|
||
|
|
|
||
|
|
# private:
|
||
|
|
_cwdFullPath = os.getcwd()
|
||
|
|
_thisScriptFullPathName = os.path.realpath(__file__)
|
||
|
|
_thisScriptBaseDir = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0])))
|
||
|
|
|
||
|
|
# public:
|
||
|
|
SYSTEM_TEST_BASE_DIR = os.path.abspath(_thisScriptBaseDir)
|
||
|
|
SYSTEM_TEST_UTIL_DIR = os.path.abspath(SYSTEM_TEST_BASE_DIR + "/utils")
|
||
|
|
SYSTEM_TEST_SUITE_SUFFIX = "_testsuite"
|
||
|
|
SYSTEM_TEST_CASE_PREFIX = "testcase_"
|
||
|
|
SYSTEM_TEST_MODULE_EXT = ".py"
|
||
|
|
CLUSTER_CONFIG_FILENAME = "cluster_config.json"
|
||
|
|
CLUSTER_CONFIG_PATHNAME = os.path.abspath(SYSTEM_TEST_BASE_DIR + "/" + CLUSTER_CONFIG_FILENAME)
|
||
|
|
|
||
|
|
clusterEntityConfigDictList = []
|
||
|
|
systemTestResultsList = []
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
"Create an object with this system test session environment"
|
||
|
|
|
||
|
|
# retrieve each entity's data from cluster config json file
|
||
|
|
# as "dict" and enter them into a "list"
|
||
|
|
jsonFileContent = open(self.CLUSTER_CONFIG_PATHNAME, "r").read()
|
||
|
|
jsonData = json.loads(jsonFileContent)
|
||
|
|
for key, cfgList in jsonData.items():
|
||
|
|
if key == "cluster_config":
|
||
|
|
for cfg in cfgList:
|
||
|
|
self.clusterEntityConfigDictList.append(cfg)
|
||
|
|
|
||
|
|
|
||
|
|
def getSystemTestEnvDict(self):
|
||
|
|
envDict = {}
|
||
|
|
envDict["system_test_base_dir"] = self.SYSTEM_TEST_BASE_DIR
|
||
|
|
envDict["system_test_util_dir"] = self.SYSTEM_TEST_UTIL_DIR
|
||
|
|
envDict["cluster_config_pathname"] = self.CLUSTER_CONFIG_PATHNAME
|
||
|
|
envDict["system_test_suite_suffix"] = self.SYSTEM_TEST_SUITE_SUFFIX
|
||
|
|
envDict["system_test_case_prefix"] = self.SYSTEM_TEST_CASE_PREFIX
|
||
|
|
envDict["system_test_module_ext"] = self.SYSTEM_TEST_MODULE_EXT
|
||
|
|
envDict["cluster_config_pathname"] = self.CLUSTER_CONFIG_PATHNAME
|
||
|
|
envDict["cluster_entity_config_dict_list"] = self.clusterEntityConfigDictList
|
||
|
|
envDict["system_test_results_list"] = self.systemTestResultsList
|
||
|
|
return envDict
|
||
|
|
|
||
|
|
|