def test_construction_of_get_query_sql_text_with_parameters_replaced(self):
cluster = RedshiftCluster(cluster_endpoint='test')
db_name = 'testdb'
cluster.set_db(db_name)
sql = DBResource(rs_cluster=cluster).get_query_sql_text_with_parameters_replaced(GET_DATABASE_NAME_OWNER_ACL)
expected_sql = GET_DATABASE_NAME_OWNER_ACL.format(db=db_name)
self.assertEquals(sql, expected_sql)
def get_cluster_from_cluster_dict(cluster_dict, kms_region):
cluster = RedshiftCluster(cluster_dict['clusterEndpoint'])
cluster.set_port(cluster_dict['clusterPort'])
cluster.set_user(cluster_dict['connectUser'])
cluster.set_host(cluster_dict['clusterEndpoint'])
cluster.set_db(cluster_dict['db'])
if 'connectPwd' in cluster_dict:
if kms_region is None:
kms_region = cluster.get_region_name()
kms_helper = KMSHelper(kms_region)
cluster.set_password(kms_helper.decrypt(cluster_dict['connectPwd']))
cluster.set_user_auto_create(False)
if 'userAutoCreate' in cluster_dict \
and cluster_dict['userAutoCreate'].lower() == 'true':
cluster.set_user_auto_create(True)
cluster.user_db_groups = []
if 'userDbGroups' in cluster_dict:
cluster.set_user_db_groups(cluster_dict['userDbGroups'])
return cluster
def get_cluster_from_cluster_dict(cluster_dict, kms_region):
cluster = RedshiftCluster(cluster_dict['clusterEndpoint'])
cluster.set_port(cluster_dict['clusterPort'])
cluster.set_user(cluster_dict['connectUser'])
cluster.set_host(cluster_dict['clusterEndpoint'])
cluster.set_db(cluster_dict['db'])
if 'connectPwd' in cluster_dict:
if kms_region is None:
kms_region = cluster.get_region_name()
kms_helper = KMSHelper(kms_region)
cluster.set_password(kms_helper.decrypt(
cluster_dict['connectPwd']))
cluster.set_user_auto_create(False)
if 'userAutoCreate' in cluster_dict \
and cluster_dict['userAutoCreate'].lower() == 'true':
cluster.set_user_auto_create(True)
cluster.user_db_groups = []
if 'userDbGroups' in cluster_dict:
cluster.set_user_db_groups(cluster_dict['userDbGroups'])
return cluster