inspec/lib/resources/aws/aws_ec2_instances.rb
Clinton Wolfe 8683c54510 Update core resources with filtertable API changes (#3117)
* Search and replace filtertable methods to use new names, and rely on automatic methods
* Remove spurious exists? matchers - see https://relishapp.com/rspec/rspec-expectations/docs/built-in-matchers/exist-matcher
* Revert removing exists? - we'll do it on a separate PR
* Gah, didn't save before resolving conflict
* Add back name column on aws cloudtrail trails

Signed-off-by: Clinton Wolfe <clintoncwolfe@gmail.com>
2018-06-26 15:14:21 -04:00

63 lines
1.7 KiB
Ruby

class AwsEc2Instances < Inspec.resource(1)
name 'aws_ec2_instances'
desc 'Verifies settings for AWS EC2 Instances in bulk'
example '
describe aws_ec2_instances do
it { should exist }
end
'
supports platform: 'aws'
include AwsPluralResourceMixin
def validate_params(resource_params)
unless resource_params.empty?
raise ArgumentError, 'aws_ec2_instances does not accept resource parameters.'
end
resource_params
end
# Underlying FilterTable implementation.
filter = FilterTable.create
filter.register_custom_matcher(:exists?) { |x| !x.entries.empty? }
filter.register_column(:instance_ids, field: :instance_id)
filter.install_filter_methods_on_resource(self, :table)
def to_s
'EC2 Instances'
end
def fetch_from_api
backend = BackendFactory.create(inspec_runner)
@table = []
pagination_opts = {}
loop do
api_result = backend.describe_instances(pagination_opts)
@table += unpack_describe_instances_response(api_result.reservations)
break unless api_result.next_token
pagination_opts = { next_token: api_result.next_token }
end
end
def unpack_describe_instances_response(reservations)
instance_rows = []
reservations.each do |res|
instance_rows += res.instances.map do |instance_struct|
{
instance_id: instance_struct.instance_id,
}
end
end
instance_rows
end
class Backend
class AwsClientApi < AwsBackendBase
BackendFactory.set_default_backend(self)
self.aws_client_class = Aws::EC2::Client
def describe_instances(query)
aws_service_client.describe_instances(query)
end
end
end
end