mirror of
https://github.com/inspec/inspec
synced 2024-11-23 21:23:29 +00:00
6c3ab70dd1
* json resource (et. al.): allow inspec check to succeed when using command When using the `json` resource (or any of the resources that subclass JsonConfig), `inspec check` would fail if the content was supplied with the `command` option. This is because the `command` resource is mocked and an empty string would be returned for `stdout`. That content would be blindly passed to the `parse` method would which raise an exception and cause `inspec check` to fail. This change refactors JsonConfig to be a bit cleaner and use some helper methods. Additionally, we use the new Exceptions to properly raise errors which are naturally caught by Inspec::Profile, etc. Signed-off-by: Adam Leff <adam@leff.co> * Make `resource_base_name` method private Signed-off-by: Adam Leff <adam@leff.co>
58 lines
1.8 KiB
Ruby
58 lines
1.8 KiB
Ruby
# encoding: utf-8
|
|
# author: Christoph Hartmann
|
|
# author: Dominik Richter
|
|
|
|
# Parses a csv document
|
|
# This implementation was inspired by a blog post
|
|
# @see http://technicalpickles.com/posts/parsing-csv-with-ruby
|
|
module Inspec::Resources
|
|
class CsvConfig < JsonConfig
|
|
name 'csv'
|
|
desc 'Use the csv InSpec audit resource to test configuration data in a CSV file.'
|
|
example "
|
|
describe csv('example.csv') do
|
|
its('name') { should eq(['John', 'Alice']) }
|
|
end
|
|
"
|
|
|
|
# override the parse method from JsonConfig
|
|
# Assuming a header row of name,col1,col2, it will output an array of hashes like so:
|
|
# [
|
|
# { 'name' => 'row1', 'col1' => 'value1', 'col2' => 'value2' },
|
|
# { 'name' => 'row2', 'col1' => 'value3', 'col2' => 'value4' }
|
|
# ]
|
|
def parse(content)
|
|
require 'csv'
|
|
|
|
# convert empty field to nil
|
|
CSV::Converters[:blank_to_nil] = lambda do |field|
|
|
field && field.empty? ? nil : field
|
|
end
|
|
|
|
# implicit conversion of values
|
|
csv = CSV.new(content, headers: true, converters: [:all, :blank_to_nil])
|
|
|
|
# convert to hash
|
|
csv.to_a.map(&:to_hash)
|
|
rescue => e
|
|
raise Inspec::Exceptions::ResourceFailed, "Unable to parse CSV: #{e.message}"
|
|
end
|
|
|
|
# override the value method from JsonConfig
|
|
# The format of the CSV hash as created by #parse is very different
|
|
# than what the YAML, JSON, and INI resources create, so using the
|
|
# #value method from JsonConfig (which uses ObjectTraverser.extract_value)
|
|
# doesn't make sense here.
|
|
def value(key)
|
|
@params.map { |x| x[key.first.to_s] }.compact
|
|
end
|
|
|
|
private
|
|
|
|
# used by JsonConfig to build up a full to_s method
|
|
# based on whether a file path, content, or command was supplied.
|
|
def resource_base_name
|
|
'CSV'
|
|
end
|
|
end
|
|
end
|