a577ea8202
Adding a config file for security related items. Only item to start is dns name. Change-Id: Ia55060198c705fdd0c9114aba199857d8196beb5
391 lines
10 KiB
Python
391 lines
10 KiB
Python
from optparse import OptionParser
|
|
|
|
from _pytest.main import Session
|
|
|
|
|
|
class ConfigurationFileLocationsManager:
|
|
"""
|
|
Class to hold all the file locations for different configs.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.lab_config_file = None
|
|
self.k8s_config_file = None
|
|
self.ptp_config_file = None
|
|
self.logger_config_file = None
|
|
self.docker_config_file = None
|
|
self.web_config_file = None
|
|
self.database_config_file = None
|
|
self.rest_api_config_file = None
|
|
self.security_config_file = None
|
|
|
|
def set_configs_from_pytest_args(self, session: Session):
|
|
"""
|
|
Sets the configs from pytest args.
|
|
|
|
Args:
|
|
session (Session): the pytest session
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
lab_config_file = session.config.getoption("--lab_config_file")
|
|
if lab_config_file:
|
|
self.set_lab_config_file(lab_config_file)
|
|
|
|
k8s_config_file = session.config.getoption("--k8s_config_file")
|
|
if k8s_config_file:
|
|
self.set_k8s_config_file(k8s_config_file)
|
|
|
|
ptp_config_file = session.config.getoption("--ptp_config_file")
|
|
if ptp_config_file:
|
|
self.set_ptp_config_file(ptp_config_file)
|
|
|
|
logger_config_file = session.config.getoption("--logger_config_file")
|
|
if logger_config_file:
|
|
self.set_logger_config_file(logger_config_file)
|
|
|
|
docker_config_file = session.config.getoption("--docker_config_file")
|
|
if docker_config_file:
|
|
self.set_docker_config_file(docker_config_file)
|
|
|
|
web_config_file = session.config.getoption("--web_config_file")
|
|
if web_config_file:
|
|
self.set_web_config_file(web_config_file)
|
|
|
|
database_config_file = session.config.getoption("--database_config_file")
|
|
if database_config_file:
|
|
self.set_database_config_file(database_config_file)
|
|
|
|
rest_api_config_file = session.config.getoption("--rest_api_config_file")
|
|
if rest_api_config_file:
|
|
self.set_rest_api_config_file(rest_api_config_file)
|
|
|
|
security_config_file = session.config.getoption("--security_config_file")
|
|
if security_config_file:
|
|
self.set_security_config_file(security_config_file)
|
|
|
|
def set_configs_from_options_parser(self, parser: OptionParser = None):
|
|
"""
|
|
Sets the config files from options parser.
|
|
|
|
Args:
|
|
parser (OptionParser): This is needed for the case where we have other command line options besides just config,
|
|
in these cases the parser will need to be instantiated and the other options added before passing to
|
|
config location manager.
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
options = self._add_options(parser)
|
|
|
|
lab_config_file = options.lab_config_file
|
|
if lab_config_file:
|
|
self.set_lab_config_file(lab_config_file)
|
|
|
|
k8s_config_file = options.k8s_config_file
|
|
if k8s_config_file:
|
|
self.set_k8s_config_file(k8s_config_file)
|
|
|
|
ptp_config_file = options.ptp_config_file
|
|
if ptp_config_file:
|
|
self.set_ptp_config_file(ptp_config_file)
|
|
|
|
logger_config_file = options.logger_config_file
|
|
if logger_config_file:
|
|
self.set_logger_config_file(logger_config_file)
|
|
|
|
docker_config_file = options.docker_config_file
|
|
if docker_config_file:
|
|
self.set_docker_config_file(docker_config_file)
|
|
|
|
web_config_file = options.web_config_file
|
|
if web_config_file:
|
|
self.set_web_config_file(web_config_file)
|
|
|
|
database_config_file = options.database_config_file
|
|
if database_config_file:
|
|
self.set_database_config_file(database_config_file)
|
|
|
|
rest_api_config_file = options.rest_api_config_file
|
|
if rest_api_config_file:
|
|
self.set_rest_api_config_file(rest_api_config_file)
|
|
|
|
security_config_file = options.security_config_file
|
|
if security_config_file:
|
|
self.set_security_config_file(security_config_file)
|
|
|
|
def _add_options(self, parser: OptionParser):
|
|
"""
|
|
Adds the command line options we can expect.
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
if not parser:
|
|
parser = OptionParser()
|
|
|
|
parser.add_option(
|
|
"--lab_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="lab_config_file",
|
|
help="the lab file used for scanning",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--k8s_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="k8s_config_file",
|
|
help="the k8s config file",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--ptp_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="ptp_config_file",
|
|
help="the PTP config file",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--logger_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="logger_config_file",
|
|
help="the logger config file",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--docker_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="docker_config_file",
|
|
help="the docker config file",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--web_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="web_config_file",
|
|
help="The Web config file",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--database_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="database_config_file",
|
|
help="The database config file",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--rest_api_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="rest_api_config_file",
|
|
help="The rest api config file",
|
|
)
|
|
|
|
parser.add_option(
|
|
"--security_config_file",
|
|
action="store",
|
|
type="str",
|
|
dest="security_config_file",
|
|
help="The security config file",
|
|
)
|
|
|
|
options, args = parser.parse_args()
|
|
|
|
return options
|
|
|
|
def set_lab_config_file(self, lab_config_file: str):
|
|
"""
|
|
Setter for lab config files.
|
|
|
|
Args:
|
|
lab_config_file (str): the location of the lab config file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.lab_config_file = lab_config_file
|
|
|
|
def get_lab_config_file(self) -> str:
|
|
"""
|
|
Getter for lab config file.
|
|
|
|
Returns: the lab config file
|
|
|
|
"""
|
|
return self.lab_config_file
|
|
|
|
def set_k8s_config_file(self, k8s_config_file: str):
|
|
"""
|
|
Setter for k8s config file.
|
|
|
|
Args:
|
|
k8s_config_file (str): the location of the k8s config file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.k8s_config_file = k8s_config_file
|
|
|
|
def get_k8s_config_file(self) -> str:
|
|
"""
|
|
Getter for k8s config file.
|
|
|
|
Returns: the k8s config file
|
|
|
|
"""
|
|
return self.k8s_config_file
|
|
|
|
def set_ptp_config_file(self, ptp_config_file: str):
|
|
"""
|
|
Setter for ptp config file.
|
|
|
|
Args:
|
|
ptp_config_file (str): the location of the ptp config file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.ptp_config_file = ptp_config_file
|
|
|
|
def get_ptp_config_file(self) -> str:
|
|
"""
|
|
Getter for ptp config file.
|
|
|
|
Returns: the ptp config file
|
|
|
|
"""
|
|
return self.ptp_config_file
|
|
|
|
def set_logger_config_file(self, logger_config_file: str):
|
|
"""
|
|
Setter for logger config file.
|
|
|
|
Args:
|
|
logger_config_file (str): the logger config file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.logger_config_file = logger_config_file
|
|
|
|
def get_logger_config_file(self) -> str:
|
|
"""
|
|
Getter for logger config file.
|
|
|
|
Returns: the logger config file
|
|
|
|
"""
|
|
return self.logger_config_file
|
|
|
|
def set_docker_config_file(self, docker_config_file: str):
|
|
"""
|
|
Setter for docker config file.
|
|
|
|
Args:
|
|
docker_config_file (str): the docker config file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.docker_config_file = docker_config_file
|
|
|
|
def get_docker_config_file(self) -> str:
|
|
"""
|
|
Getter for docker config file.
|
|
|
|
Returns: the docker config file
|
|
|
|
"""
|
|
return self.docker_config_file
|
|
|
|
def set_web_config_file(self, web_config_file: str):
|
|
"""
|
|
Setter for web config file.
|
|
|
|
Args:
|
|
web_config_file (str): the web config file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.web_config_file = web_config_file
|
|
|
|
def get_web_config_file(self) -> str:
|
|
"""
|
|
Getter for web config file.
|
|
|
|
Returns: the web config file
|
|
|
|
"""
|
|
return self.web_config_file
|
|
|
|
def set_database_config_file(self, database_config_file: str):
|
|
"""
|
|
Setter for database config file.
|
|
|
|
Args:
|
|
database_config_file (str): the database config file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.database_config_file = database_config_file
|
|
|
|
def get_database_config_file(self) -> str:
|
|
"""
|
|
Getter for database config file.
|
|
|
|
Returns: the database config file
|
|
|
|
"""
|
|
return self.database_config_file
|
|
|
|
def set_rest_api_config_file(self, rest_api_config_file: str):
|
|
"""
|
|
Setter for rest_api_config_file.
|
|
|
|
Args:
|
|
rest_api_config_file (str): the rest_api_config_file
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
self.rest_api_config_file = rest_api_config_file
|
|
|
|
def get_rest_api_config_file(self) -> str:
|
|
"""
|
|
Getter for rest_api_config_file.
|
|
|
|
Returns: the rest_api_config_file
|
|
|
|
"""
|
|
return self.rest_api_config_file
|
|
|
|
def get_security_config_file(self) -> str:
|
|
"""
|
|
Getter for security config file
|
|
|
|
Returns:
|
|
str: the security config file
|
|
|
|
"""
|
|
return self.security_config_file
|
|
|
|
def set_security_config_file(self, security_config_file: str):
|
|
"""
|
|
Setter for security config file
|
|
|
|
Args:
|
|
security_config_file (str): the security config file
|
|
|
|
"""
|
|
self.security_config_file = security_config_file
|