Ejemplos de read_db_connect_info en Python

Ejemplo n.º 1

Archivo: requests_db.py Proyecto: ghrcdaac/operational-recovery-cloud-archive

def get_dbconnect_info():
    """
    Gets the dbconnection info
    """
    dbconnect_info = {}
    db_params = {}
    db_params["db_host"] = {"ssm": "drdb-host"}
    db_params["db_port"] = {"env": "DATABASE_PORT"}
    db_params["db_name"] = {"env": "DATABASE_NAME"}
    db_params["db_user"] = {"env": "DATABASE_USER"}
    db_params["db_pw"] = {"ssm": "drdb-user-pass"}
    dbconnect_info = database.read_db_connect_info(db_params)
    return dbconnect_info

Ejemplo n.º 2

Archivo: requests_db.py Proyecto: rizbi-hassan/cumulus-orca

def get_dbconnect_info():
    """
    Gets the dbconnection info. PREFIX environment variable expected to be set.
    """
    dbconnect_info = {}
    db_params = {}
    prefix = os.environ['PREFIX']
    db_params['db_host'] = {"secretsmanager": f"{prefix}-drdb-host"}
    db_params['db_port'] = {'env': 'DATABASE_PORT'}
    db_params['db_name'] = {'env': 'DATABASE_NAME'}
    db_params['db_user'] = {'env': 'DATABASE_USER'}
    db_params['db_pw'] = {'secretsmanager': f"{prefix}-drdb-user-pass"}
    dbconnect_info = database.read_db_connect_info(db_params)
    return dbconnect_info

Ejemplo n.º 3

Archivo: test_database.py Proyecto: ghrcdaac/operational-recovery-cloud-archive

    def setUp(self):

        os.environ["DATABASE_HOST"] = "my.db.host.gov"
        os.environ["DATABASE_PORT"] = "50"
        os.environ["DATABASE_NAME"] = "sndbx"
        os.environ["DATABASE_USER"] = "******"
        os.environ["DATABASE_PW"] = "unittestdbpw"

        self.mock_single_query = database.single_query
        self.mock_utcnow = database.get_utc_now_iso
        self.mock_uuid = database.uuid_generator
        self.mock_boto3 = boto3.client
        boto3.client = Mock()
        s3_cli = boto3.client('ssm')
        db_host = {"Parameter": {"Value": os.environ['DATABASE_HOST']}}
        db_pw = {"Parameter": {"Value": os.environ['DATABASE_PW']}}
        s3_cli.get_parameter = Mock(side_effect=[db_host, db_pw])
        db_params = {}
        db_params["db_host"] = {"ssm": "drdb-host"}
        db_params["db_port"] = {"env": "DATABASE_PORT"}
        db_params["db_name"] = {"env": "DATABASE_NAME"}
        db_params["db_user"] = {"env": "DATABASE_USER"}
        db_params["db_pw"] = {"ssm": "drdb-user-pass"}
        self.dbconnect_info = database.read_db_connect_info(db_params)

Ejemplo n.º 4

    def setUp(self):

        os.environ['DATABASE_HOST'] = 'localhost'
        os.environ['DATABASE_PORT'] = '5432'
        os.environ['DATABASE_NAME'] = 'postgres'
        os.environ['DATABASE_USER'] = '******'
        os.environ['DATABASE_PW'] = 'postgres'

        self.mock_single_query = database.single_query
        self.mock_utcnow = database.get_utc_now_iso
        self.mock_uuid = database.uuid_generator
        self.mock_boto3 = boto3.client
        boto3.client = Mock()
        secretsmanager_cli = boto3.client('secretsmanager')
        db_host = {'SecretString': os.environ['DATABASE_HOST']}
        db_pw = {'SecretString': os.environ['DATABASE_PW']}
        secretsmanager_cli.get_secret_value = Mock(
            side_effect=[db_host, db_pw])
        db_params = {
            'db_host': {
                'secretsmanager': 'drdb-host'
            },
            'db_port': {
                'env': 'DATABASE_PORT'
            },
            'db_name': {
                'env': 'DATABASE_NAME'
            },
            'db_user': {
                'env': 'DATABASE_USER'
            },
            'db_pw': {
                'secretsmanager': 'drdb-user-pass'
            }
        }
        self.dbconnect_info = database.read_db_connect_info(db_params)