def testNetworkSpec(self):
zone = 'us-west-1a'
spec = aws_network.AwsNetworkSpec(zone)
self.assertEqual(zone, spec.zone)
self.assertIsNone(spec.vpc_id)
self.assertIsNone(spec.subnet_id)
self.assertIsNone(spec.cidr_block)
def __init__(self, spec):
super(EksCluster, self).__init__(spec)
self.zones = sorted(FLAGS.eks_zones)
self.zone = ','.join(self.zones)
self.region = util.GetRegionFromZone(self.zones[0])
self.networks = [
aws_network.AwsNetwork.GetNetworkFromNetworkSpec(
aws_network.AwsNetworkSpec(zone)) for zone in self.zones
]
self.eks_control_plane = _EksControlPlane(self.name, self.region,
self.networks)
self.eks_workers = _EksWorkers('%s-worker-nodes' % self.name,
self.region)
def __init__(self, dpb_service_spec):
super(AwsDpbEmr, self).__init__(dpb_service_spec)
self.dpb_service_type = AwsDpbEmr.SERVICE_TYPE
self.project = None
self.cmd_prefix = list(util.AWS_PREFIX)
if self.dpb_service_zone:
region = util.GetRegionFromZone(self.dpb_service_zone)
self.cmd_prefix += ['--region', region]
self.network = aws_network.AwsNetwork.GetNetworkFromNetworkSpec(
aws_network.AwsNetworkSpec(zone=self.dpb_service_zone))
else:
raise errors.Setup.InvalidSetupError(
'dpb_service_zone must be provided, for provisioning.')
self.bucket_to_delete = None
self.dpb_version = FLAGS.dpb_emr_release_label
def testNetworkSpecWithVpcId(self):
zone = 'us-west-1a'
vpc_id = 'vpc-1234'
subnet_id = 'subnet-1234'
cidr = '10.0.0.0/24'
res = {
'Subnets': [{
'VpcId': vpc_id,
'SubnetId': subnet_id,
'CidrBlock': cidr
}]
}
util.IssueRetryableCommand.return_value = json.dumps(res), None
spec = aws_network.AwsNetworkSpec(zone, vpc_id)
self.assertEqual(zone, spec.zone)
self.assertEqual(vpc_id, spec.vpc_id)
self.assertEqual(subnet_id, spec.subnet_id)
self.assertEqual(cidr, spec.cidr_block)
def __init__(self, dpb_service_spec):
super(AwsDpbEmr, self).__init__(dpb_service_spec)
self.dpb_service_type = AwsDpbEmr.SERVICE_TYPE
self.project = None
self.cmd_prefix = list(util.AWS_PREFIX)
if self.dpb_service_zone:
self.region = util.GetRegionFromZone(self.dpb_service_zone)
else:
raise errors.Setup.InvalidSetupError(
'dpb_service_zone must be provided, for provisioning.')
self.cmd_prefix += ['--region', self.region]
self.network = aws_network.AwsNetwork.GetNetworkFromNetworkSpec(
aws_network.AwsNetworkSpec(zone=self.dpb_service_zone))
self.storage_service = s3.S3Service()
self.storage_service.PrepareService(self.region)
self.persistent_fs_prefix = 's3://'
self.bucket_to_delete = None
self.dpb_version = FLAGS.dpb_emr_release_label or self.dpb_version
self._cluster_create_time = None
if not self.dpb_version:
raise errors.Setup.InvalidSetupError(
'dpb_service.version must be provided.')
def network(self):
network_spec = aws_network.AwsNetworkSpec(self.zone)
return aws_network.AwsNetwork.GetNetworkFromNetworkSpec(network_spec)
