inspec/test/unit/dsl/control_test.rb

require 'helper'
require 'inspec/config'

describe 'controls' do
  def load(content)
    data = {
      'inspec.yml' => "name: mock",
      'controls/mock.rb' => "control '1' do\n#{content}\nend\n",
    }
    opts = { test_collector: Inspec::RunnerMock.new, backend: Inspec::Backend.create(Inspec::Config.mock) }
    Inspec::Profile.for_target(data, opts)
                   .params[:controls]['1']
  end

  let(:rand_string) { rand.to_s }

  it 'adds a title' do
    load("title #{rand_string.inspect}")[:title].must_equal rand_string
  end

  it 'adds a description' do
    load("desc #{rand_string.inspect}")[:descriptions][:default].must_equal rand_string
  end

  it 'allows multiple desc with keys and values' do
    lines = <<-DESC_EXAMPLES
      desc 'this is default'
      desc 'example', 'this is foo'
      desc 'a space', 'this label has a space'
      desc 'newline', 'this value has
                       a newline'
      desc 'heredoc', <<-EOF
        This is a heredoc
      EOF
    DESC_EXAMPLES

    result = load(lines)
    result[:descriptions][:default].must_equal('this is default')
    result[:descriptions][:example].must_equal('this is foo')
    result[:descriptions][:'a space'].must_equal('this label has a space')
    result[:descriptions][:newline].must_match(/this value has\n\s*a newline/)
    result[:descriptions][:heredoc].must_equal('This is a heredoc')
  end

  it 'adds a multiline description' do
    t = rand_string + "\n" + rand_string
    load("desc #{t.inspect}")[:descriptions][:default].must_equal t
  end

  it 'strips empty lines and spaces in description at start and end' do
    t = "   \n" + rand_string + "\n   "
    load("desc #{t.inspect}")[:descriptions][:default].must_equal rand_string
  end

  it 'unindents properly' do
    t = "\n    #{rand_string}\n  \n\t\t    #{rand_string}\n "
    load("desc #{t.inspect}")[:descriptions][:default].must_equal "#{rand_string}\n  \n  #{rand_string}"
  end

  it 'works with empty refs' do
    load('ref')[:refs].must_be :empty?
  end

  it 'defines a simple ref' do
    s = rand.to_s
    load("ref #{s.inspect}")[:refs].must_equal [{:ref=>s}]
  end

  it 'defines a ref with url' do
    s = rand.to_s
    u = rand.to_s
    load("ref #{s.inspect}, url: #{u.inspect}")[:refs].must_equal [{ref: s, url: u}]
  end

  it 'defines a ref without content but with url' do
    u = rand.to_s
    load("ref url: #{u.inspect}")[:refs].must_equal [{url: u}]
  end

  it 'works with empty tags' do
    load('tag')[:tags].must_be :empty?
  end

  it 'defines a simple tag' do
    s = rand.to_s
    load("tag #{s.inspect}")[:tags].must_equal({ s => nil })
  end

  it 'define multiple tags' do
    a, b, c = rand.to_s, rand.to_s, rand.to_s
    load("tag #{a.inspect}, #{b.inspect}, #{c.inspect}")[:tags].must_equal(
      { a => nil, b => nil, c => nil })
  end

  it 'tag by key=value' do
    a, b = rand.to_s, rand.to_s
    load("tag #{a.inspect} => #{b.inspect}")[:tags].must_equal(
      { a => b })
  end
end