mirror of
https://github.com/inspec/inspec
synced 2024-12-03 01:49:37 +00:00
77 lines
1.6 KiB
Ruby
77 lines
1.6 KiB
Ruby
require 'azure_conn'
|
|
require 'azure_mgmt_compute'
|
|
|
|
require_relative 'common/resource_groups'
|
|
|
|
class AzureVmImage < Inspec.resource(1)
|
|
name 'azurevm_image'
|
|
|
|
desc "
|
|
This resource gathers information about which image the vm was created from
|
|
"
|
|
|
|
example "
|
|
describe azurevm_image_sku(host: 'acme-test-01', resource_group: 'ACME') do
|
|
its('sku') { should eq '16.04.0-LTS'}
|
|
end
|
|
"
|
|
|
|
# Load the configuration file on initialisation
|
|
def initialize(opts)
|
|
@opts = opts
|
|
end
|
|
|
|
def sku
|
|
vm = get_vm(@opts[:host], @opts[:resource_group])
|
|
|
|
if vm.instance_of?(String)
|
|
vm
|
|
else
|
|
vm.storage_profile.image_reference.sku
|
|
end
|
|
end
|
|
|
|
def publisher
|
|
|
|
vm = get_vm(@opts[:host], @opts[:resource_group])
|
|
|
|
if vm.instance_of?(String)
|
|
vm
|
|
else
|
|
vm.storage_profile.image_reference.publisher
|
|
end
|
|
end
|
|
|
|
def offer
|
|
vm = get_vm(@opts[:host], @opts[:resource_group])
|
|
|
|
if vm.instance_of?(String)
|
|
vm
|
|
else
|
|
vm.storage_profile.image_reference.offer
|
|
end
|
|
end
|
|
|
|
# Retrieve the named virtual machine from Azure
|
|
def get_vm(name, rg_name)
|
|
# Azure connection
|
|
azure = AzureConnection.new
|
|
|
|
client = Azure::ARM::Compute::ComputeManagementClient.new(azure.connection)
|
|
client.subscription_id = azure.subscription_id
|
|
|
|
# Ensure that the resource group exists
|
|
rg = ResourceGroups.new(azure)
|
|
|
|
unless rg.exists(rg_name)
|
|
throw "The Resource group cannot be found: #{rg_name}"
|
|
end
|
|
|
|
# get a vm from the named resource group
|
|
begin
|
|
client.virtual_machines.get(rg_name, name)
|
|
rescue => e
|
|
e.error_message
|
|
end
|
|
end
|
|
end
|