Cleanup default parameters.
This commit is contained in:
parent
93bd852419
commit
72a7946126
@ -12,29 +12,38 @@ class Cluster(object):
|
|||||||
preferred_maintenance_window, cluster_parameter_group_name,
|
preferred_maintenance_window, cluster_parameter_group_name,
|
||||||
automated_snapshot_retention_period, port, cluster_version,
|
automated_snapshot_retention_period, port, cluster_version,
|
||||||
allow_version_upgrade, number_of_nodes, publicly_accessible,
|
allow_version_upgrade, number_of_nodes, publicly_accessible,
|
||||||
encrypted):
|
encrypted, region):
|
||||||
self.cluster_identifier = cluster_identifier
|
self.cluster_identifier = cluster_identifier
|
||||||
self.node_type = node_type
|
self.node_type = node_type
|
||||||
self.master_username = master_username
|
self.master_username = master_username
|
||||||
self.master_user_password = master_user_password
|
self.master_user_password = master_user_password
|
||||||
self.db_name = db_name
|
self.db_name = db_name if db_name else "dev"
|
||||||
self.cluster_security_groups = cluster_security_groups
|
self.cluster_security_groups = cluster_security_groups
|
||||||
self.vpc_security_group_ids = vpc_security_group_ids
|
self.vpc_security_group_ids = vpc_security_group_ids
|
||||||
self.cluster_subnet_group_name = cluster_subnet_group_name
|
self.cluster_subnet_group_name = cluster_subnet_group_name
|
||||||
self.availability_zone = availability_zone
|
|
||||||
self.preferred_maintenance_window = preferred_maintenance_window
|
|
||||||
self.cluster_parameter_group_name = cluster_parameter_group_name
|
self.cluster_parameter_group_name = cluster_parameter_group_name
|
||||||
self.automated_snapshot_retention_period = automated_snapshot_retention_period
|
|
||||||
self.port = port
|
|
||||||
self.cluster_version = cluster_version
|
|
||||||
self.allow_version_upgrade = allow_version_upgrade
|
|
||||||
self.publicly_accessible = publicly_accessible
|
self.publicly_accessible = publicly_accessible
|
||||||
self.encrypted = encrypted
|
self.encrypted = encrypted
|
||||||
|
|
||||||
|
self.allow_version_upgrade = allow_version_upgrade if allow_version_upgrade is not None else True
|
||||||
|
self.cluster_version = cluster_version if cluster_version else "1.0"
|
||||||
|
self.port = port if port else 5439
|
||||||
|
self.automated_snapshot_retention_period = automated_snapshot_retention_period if automated_snapshot_retention_period else 1
|
||||||
|
self.preferred_maintenance_window = preferred_maintenance_window if preferred_maintenance_window else "Mon:03:00-Mon:03:30"
|
||||||
|
|
||||||
|
if availability_zone:
|
||||||
|
self.availability_zone = availability_zone
|
||||||
|
else:
|
||||||
|
# This could probably be smarter, but there doesn't appear to be a
|
||||||
|
# way to pull AZs for a region in boto
|
||||||
|
self.availability_zone = region + "a"
|
||||||
|
|
||||||
if cluster_type == 'single-node':
|
if cluster_type == 'single-node':
|
||||||
self.number_of_nodes = 1
|
self.number_of_nodes = 1
|
||||||
else:
|
elif number_of_nodes:
|
||||||
self.number_of_nodes = number_of_nodes
|
self.number_of_nodes = number_of_nodes
|
||||||
|
else:
|
||||||
|
self.number_of_nodes = 1
|
||||||
|
|
||||||
def to_json(self):
|
def to_json(self):
|
||||||
return {
|
return {
|
||||||
|
@ -33,6 +33,7 @@ class RedshiftResponse(BaseResponse):
|
|||||||
"number_of_nodes": self._get_int_param('NumberOfNodes'),
|
"number_of_nodes": self._get_int_param('NumberOfNodes'),
|
||||||
"publicly_accessible": self._get_param("PubliclyAccessible"),
|
"publicly_accessible": self._get_param("PubliclyAccessible"),
|
||||||
"encrypted": self._get_param("Encrypted"),
|
"encrypted": self._get_param("Encrypted"),
|
||||||
|
"region": self.region,
|
||||||
}
|
}
|
||||||
|
|
||||||
cluster = self.redshift_backend.create_cluster(**cluster_kwargs)
|
cluster = self.redshift_backend.create_cluster(**cluster_kwargs)
|
||||||
|
@ -76,6 +76,35 @@ def test_create_single_node_cluster():
|
|||||||
cluster['NumberOfNodes'].should.equal(1)
|
cluster['NumberOfNodes'].should.equal(1)
|
||||||
|
|
||||||
|
|
||||||
|
@mock_redshift
|
||||||
|
def test_default_cluster_attibutes():
|
||||||
|
conn = boto.redshift.connect_to_region("us-east-1")
|
||||||
|
cluster_identifier = 'my_cluster'
|
||||||
|
|
||||||
|
conn.create_cluster(
|
||||||
|
cluster_identifier,
|
||||||
|
node_type="dw.hs1.xlarge",
|
||||||
|
master_username="username",
|
||||||
|
master_user_password="password",
|
||||||
|
)
|
||||||
|
|
||||||
|
cluster_response = conn.describe_clusters(cluster_identifier)
|
||||||
|
cluster = cluster_response['DescribeClustersResponse']['DescribeClustersResult']['Clusters'][0]
|
||||||
|
|
||||||
|
cluster['DBName'].should.equal("dev")
|
||||||
|
# cluster['ClusterSecurityGroups'].should.equal([])
|
||||||
|
# cluster['VpcSecurityGroups'].should.equal([])
|
||||||
|
# cluster['ClusterSubnetGroupName'].should.equal(None)
|
||||||
|
assert "us-east-" in cluster['AvailabilityZone']
|
||||||
|
cluster['PreferredMaintenanceWindow'].should.equal("Mon:03:00-Mon:03:30")
|
||||||
|
# cluster['ClusterParameterGroups'].should.equal([])
|
||||||
|
cluster['AutomatedSnapshotRetentionPeriod'].should.equal(1)
|
||||||
|
cluster['Port'].should.equal(5439)
|
||||||
|
cluster['ClusterVersion'].should.equal("1.0")
|
||||||
|
cluster['AllowVersionUpgrade'].should.equal(True)
|
||||||
|
cluster['NumberOfNodes'].should.equal(1)
|
||||||
|
|
||||||
|
|
||||||
@mock_redshift
|
@mock_redshift
|
||||||
def test_describe_non_existant_cluster():
|
def test_describe_non_existant_cluster():
|
||||||
conn = boto.redshift.connect_to_region("us-east-1")
|
conn = boto.redshift.connect_to_region("us-east-1")
|
||||||
|
Loading…
Reference in New Issue
Block a user