2
0
Fork 0
mirror of https://github.com/inspec/inspec synced 2024-12-26 21:13:25 +00:00
inspec/test/unit/resources/aws_billing_backend.rb

102 lines
3.4 KiB
Ruby
Raw Normal View History

require 'resource_support/aws'
module MockAwsBillingReports
class Empty < AwsBackendBase
def describe_report_definitions(_query)
Aws::CostandUsageReportService::Types::DescribeReportDefinitionsResponse.new(report_definitions: [])
end
end
class Basic < AwsBackendBase
def describe_report_definitions(_query)
Aws::CostandUsageReportService::Types::DescribeReportDefinitionsResponse
.new(report_definitions:
[
Aws::CostandUsageReportService::Types::ReportDefinition.new(
report_name: 'inspec1',
time_unit: 'HOURLY',
format: 'textORcsv',
compression: 'ZIP',
s3_bucket: 'inspec1-s3-bucket',
s3_prefix: 'inspec1/accounting',
s3_region: 'us-east-1',
),
Aws::CostandUsageReportService::Types::ReportDefinition.new(
report_name: 'inspec2',
time_unit: 'DAILY',
format: 'textORcsv',
compression: 'GZIP',
s3_bucket: 'inspec2-s3-bucket',
s3_prefix: 'inspec2/accounting',
s3_region: 'us-west-1',
),
])
end
end
# This backend will always repond with 5 reports, as if the `max_results` option was passed to
# `#describe_report_definitions`. I chose 5 because when using `max_results` in the real world
# it seems to only accept a value of 5.
#
# == Returns:
# A Aws::CostandUsageReportService::Types::DescribeReportDefinitionsRespons object with two instance
# properties:
# `report_definitions` An Array that includes a single page of 5 Reports.
# `next_token` A String set to the start of the next page. When `next_token` is nil, there are no more pages.
#
class Paginated < AwsBackendBase
# Generate a set of report data, and shuffle their order.
def generate_definitions
definitions = []
definitions << Aws::CostandUsageReportService::Types::ReportDefinition.new(
report_name: 'inspec1',
time_unit: 'HOURLY',
format: 'textORcsv',
compression: 'ZIP',
s3_bucket: 'inspec1-s3-bucket',
s3_prefix: 'inspec1/accounting',
s3_region: 'us-east-1')
definitions << Aws::CostandUsageReportService::Types::ReportDefinition.new(
report_name: 'inspec2',
time_unit: 'DAILY',
format: 'textORcsv',
compression: 'GZIP',
s3_bucket: 'inspec2-s3-bucket',
s3_prefix: 'inspec2/accounting',
s3_region: 'us-west-1')
(3..12).each do |i|
definitions <<
Aws::CostandUsageReportService::Types::ReportDefinition.new(
report_name: "inspec#{i}",
time_unit: %w{HOURLY DAILY}.sample,
format: 'textORcsv',
compression: %w{ZIP GZIP}.sample,
s3_bucket: "inspec#{i}-s3-bucket",
s3_prefix: "inspec#{i}",
s3_region: 'us-east-1'
)
end
definitions.shuffle
end
def describe_report_definitions(options = {})
@definitions ||= generate_definitions
starting_position = options.fetch(:next_token, 0)
selected_definitions = @definitions.slice(starting_position, 5).compact
next_token = starting_position + 5
next_token = @definitions.count < next_token ? nil : next_token
response = Aws::CostandUsageReportService::Types::DescribeReportDefinitionsResponse
.new(report_definitions: selected_definitions)
response.next_token = next_token
response
end
end
end