mirror of
https://github.com/inspec/inspec
synced 2024-11-27 07:00:39 +00:00
87 lines
2.4 KiB
Ruby
87 lines
2.4 KiB
Ruby
require "resource_support/aws/aws_singular_resource_mixin"
|
|
require "resource_support/aws/aws_backend_base"
|
|
require "aws-sdk-ecs"
|
|
|
|
class AwsEcsCluster < Inspec.resource(1)
|
|
name "aws_ecs_cluster"
|
|
desc "Verifies settings for an ECS cluster"
|
|
|
|
example <<~EXAMPLE
|
|
describe aws_ecs_cluster('default') do
|
|
it { should exist }
|
|
end
|
|
EXAMPLE
|
|
supports platform: "aws"
|
|
|
|
include AwsSingularResourceMixin
|
|
attr_reader :cluster_arn, :cluster_name, :status,
|
|
:registered_container_instances_count, :running_tasks_count,
|
|
:pending_tasks_count, :active_services_count, :statistics
|
|
|
|
def to_s
|
|
"AWS ECS cluster #{cluster_name}"
|
|
end
|
|
|
|
private
|
|
|
|
def validate_params(raw_params)
|
|
validated_params = check_resource_param_names(
|
|
raw_params: raw_params,
|
|
allowed_params: [:cluster_name],
|
|
allowed_scalar_name: :cluster_name,
|
|
allowed_scalar_type: String
|
|
)
|
|
|
|
validated_params
|
|
end
|
|
|
|
def fetch_from_api
|
|
backend = BackendFactory.create(inspec_runner)
|
|
|
|
# Use default cluster if no cluster name is specified
|
|
params = cluster_name.nil? ? {} : { clusters: [cluster_name] }
|
|
clusters = backend.describe_clusters(params).clusters
|
|
|
|
# Cluster name is unique, we either get back one cluster, or none
|
|
if clusters.length == 1
|
|
@exists = true
|
|
unpack_describe_clusters_response(clusters.first)
|
|
else
|
|
@exists = false
|
|
populate_as_missing
|
|
end
|
|
end
|
|
|
|
def unpack_describe_clusters_response(cluster_struct)
|
|
@cluster_arn = cluster_struct.cluster_arn
|
|
@cluster_name = cluster_struct.cluster_name
|
|
@status = cluster_struct.status
|
|
@registered_container_instances_count = cluster_struct.registered_container_instances_count
|
|
@running_tasks_count = cluster_struct.running_tasks_count
|
|
@pending_tasks_count = cluster_struct.pending_tasks_count
|
|
@active_services_count = cluster_struct.active_services_count
|
|
@statistics = cluster_struct.statistics
|
|
end
|
|
|
|
def populate_as_missing
|
|
@cluster_arn = ""
|
|
@cluster_name = ""
|
|
@status = ""
|
|
@registered_container_instances_count = 0
|
|
@running_tasks_count = 0
|
|
@pending_tasks_count = 0
|
|
@active_services_count = 0
|
|
@statistics = []
|
|
end
|
|
|
|
class Backend
|
|
class AwsClientApi < AwsBackendBase
|
|
BackendFactory.set_default_backend(self)
|
|
self.aws_client_class = Aws::ECS::Client
|
|
|
|
def describe_clusters(query = {})
|
|
aws_service_client.describe_clusters(query)
|
|
end
|
|
end
|
|
end
|
|
end
|