mirror of
https://github.com/inspec/inspec
synced 2024-11-27 07:00:39 +00:00
77 lines
1.6 KiB
Ruby
77 lines
1.6 KiB
Ruby
# encoding: utf-8
|
|
|
|
require 'docker'
|
|
require 'yaml'
|
|
require_relative '../lib/vulcano'
|
|
|
|
tests = ARGV
|
|
if tests.empty?
|
|
puts 'Nothing to do.'
|
|
exit 0
|
|
end
|
|
|
|
class DockerTester
|
|
def initialize(tests)
|
|
@tests = tests
|
|
@images = docker_images_by_tag
|
|
@conf = tests_conf
|
|
end
|
|
|
|
def run
|
|
puts ['Running tests:', @tests].flatten.join("\n- ")
|
|
puts ''
|
|
# test all images
|
|
@conf['images'].each do |n|
|
|
test_image(n)
|
|
end.all? or fail 'Test failures'
|
|
end
|
|
|
|
def docker_images_by_tag
|
|
# get all docker image tags
|
|
images = {}
|
|
Docker::Image.all.map do |img|
|
|
Array(img.info['RepoTags']).each do |tag|
|
|
images[tag] = img
|
|
end
|
|
end
|
|
images
|
|
end
|
|
|
|
def tests_conf
|
|
# get the test configuration
|
|
conf_path = File.join(File.dirname(__FILE__), '..', '.tests.yaml')
|
|
fail "Can't find tests config in #{conf_path}" unless File.file?(conf_path)
|
|
YAML.load(File.read(conf_path))
|
|
end
|
|
|
|
def test_container(container_id)
|
|
opts = { 'target' => "specinfra+docker://#{container_id}" }
|
|
runner = Vulcano::Runner.new(nil, opts)
|
|
runner.add_tests(@tests)
|
|
runner.run
|
|
end
|
|
|
|
def test_image(name)
|
|
dname = "docker-#{name}:latest"
|
|
image = @images[dname]
|
|
fail "Can't find docker image #{dname}" if image.nil?
|
|
|
|
puts "--> start docker #{name}"
|
|
container = Docker::Container.create(
|
|
'Cmd' => %w{ /bin/bash },
|
|
'Image' => image.id,
|
|
'OpenStdin' => true
|
|
)
|
|
container.start
|
|
|
|
puts "--> run test on docker #{name}"
|
|
res = test_container(container.id)
|
|
|
|
puts "--> killrm docker #{name}"
|
|
container.kill
|
|
container.delete(force: true)
|
|
res
|
|
end
|
|
end
|
|
|
|
DockerTester.new(tests).run
|