mirror of
https://github.com/inspec/inspec
synced 2024-11-24 05:33:17 +00:00
2857d07151
The opening and closing mechanic varied between all the various resources. This changes them all to use a HEREDOC with a tilde to remove leading whitespace. This removes the need for the special method to trim the `#print_example` method from shell. Signed-off-by: Franklin Webber <franklin.webber@gmail.com>
81 lines
2.3 KiB
Ruby
81 lines
2.3 KiB
Ruby
class AwsElb < Inspec.resource(1)
|
|
name 'aws_elb'
|
|
desc 'Verifies settings for AWS Elastic Load Balancer'
|
|
example <<~EXAMPLE
|
|
describe aws_elb('myelb') do
|
|
it { should exist }
|
|
end
|
|
EXAMPLE
|
|
supports platform: 'aws'
|
|
|
|
include AwsSingularResourceMixin
|
|
attr_reader :availability_zones, :dns_name, :elb_name, :external_ports,
|
|
:instance_ids, :internal_ports, :security_group_ids,
|
|
:subnet_ids, :vpc_id
|
|
|
|
def to_s
|
|
"AWS ELB #{elb_name}"
|
|
end
|
|
|
|
private
|
|
|
|
def validate_params(raw_params)
|
|
validated_params = check_resource_param_names(
|
|
raw_params: raw_params,
|
|
allowed_params: [:elb_name],
|
|
allowed_scalar_name: :elb_name,
|
|
allowed_scalar_type: String,
|
|
)
|
|
|
|
if validated_params.empty?
|
|
raise ArgumentError, 'You must provide a elb_name to aws_elb.'
|
|
end
|
|
|
|
validated_params
|
|
end
|
|
|
|
def fetch_from_api
|
|
backend = BackendFactory.create(inspec_runner)
|
|
begin
|
|
lbs = backend.describe_load_balancers(load_balancer_names: [elb_name]).load_balancer_descriptions
|
|
@exists = true
|
|
# Load balancer names are uniq; we will either have 0 or 1 result
|
|
unpack_describe_elbs_response(lbs.first)
|
|
rescue Aws::ElasticLoadBalancing::Errors::LoadBalancerNotFound
|
|
@exists = false
|
|
populate_as_missing
|
|
end
|
|
end
|
|
|
|
def unpack_describe_elbs_response(lb_struct)
|
|
@availability_zones = lb_struct.availability_zones
|
|
@dns_name = lb_struct.dns_name
|
|
@external_ports = lb_struct.listener_descriptions.map { |ld| ld.listener.load_balancer_port }
|
|
@instance_ids = lb_struct.instances.map(&:instance_id)
|
|
@internal_ports = lb_struct.listener_descriptions.map { |ld| ld.listener.instance_port }
|
|
@elb_name = lb_struct.load_balancer_name
|
|
@security_group_ids = lb_struct.security_groups
|
|
@subnet_ids = lb_struct.subnets
|
|
@vpc_id = lb_struct.vpc_id
|
|
end
|
|
|
|
def populate_as_missing
|
|
@availability_zones = []
|
|
@external_ports = []
|
|
@instance_ids = []
|
|
@internal_ports = []
|
|
@security_group_ids = []
|
|
@subnet_ids = []
|
|
end
|
|
|
|
class Backend
|
|
class AwsClientApi < AwsBackendBase
|
|
BackendFactory.set_default_backend(self)
|
|
self.aws_client_class = Aws::ElasticLoadBalancing::Client
|
|
|
|
def describe_load_balancers(query = {})
|
|
aws_service_client.describe_load_balancers(query)
|
|
end
|
|
end
|
|
end
|
|
end
|