inspec 2.0.16 → 2.0.17

data/lib/resources/azure/azure_generic_resource.rb

@@ -1,59 +1,59 @@
-# encoding: utf-8
-
-require 'resources/azure/azure_backend'
-require 'utils/filter'
-
-module Inspec::Resources
-  class AzureGenericResource < AzureResourceBase
-    name 'azure_generic_resource'
-
-    desc '
-      Inspec Resource to interrogate any Resource type in Azure
-    '
-
-    supports platform: 'azure'
-
-    attr_accessor :filter, :total, :counts, :name, :type, :location, :probes
-
-    def initialize(opts = {})
-      # Call the parent class constructor
-      super(opts)
-
-      # Get the resource group
-      resource_group
-
-      # Get the resources
-      resources
-
-      # Create the tag methods
-      create_tag_methods
-    end
-
-    # Define the filter table so that it can be interrogated
-    @filter = FilterTable.create
-    @filter.add_accessor(:count)
-           .add_accessor(:entries)
-           .add_accessor(:where)
-           .add_accessor(:contains)
-           .add(:exist?, field: 'exist?')
-           .add(:type, field: 'type')
-           .add(:name, field: 'name')
-           .add(:location, field: 'location')
-           .add(:properties, field: 'properties')
-
-    @filter.connect(self, :probes)
-
-    def parse_resource(resource)
-      # return a hash of information
-      parsed = {
-        'location' => resource.location,
-        'name' => resource.name,
-        'type' => resource.type,
-        'exist?' => true,
-        'properties' => AzureResourceProbe.new(resource.properties),
-      }
-
-      parsed
-    end
-  end
-end
+# encoding: utf-8
+
+require 'resources/azure/azure_backend'
+require 'utils/filter'
+
+module Inspec::Resources
+  class AzureGenericResource < AzureResourceBase
+    name 'azure_generic_resource'
+
+    desc '
+      Inspec Resource to interrogate any Resource type in Azure
+    '
+
+    supports platform: 'azure'
+
+    attr_accessor :filter, :total, :counts, :name, :type, :location, :probes
+
+    def initialize(opts = {})
+      # Call the parent class constructor
+      super(opts)
+
+      # Get the resource group
+      resource_group
+
+      # Get the resources
+      resources
+
+      # Create the tag methods
+      create_tag_methods
+    end
+
+    # Define the filter table so that it can be interrogated
+    @filter = FilterTable.create
+    @filter.add_accessor(:count)
+           .add_accessor(:entries)
+           .add_accessor(:where)
+           .add_accessor(:contains)
+           .add(:exist?, field: 'exist?')
+           .add(:type, field: 'type')
+           .add(:name, field: 'name')
+           .add(:location, field: 'location')
+           .add(:properties, field: 'properties')
+
+    @filter.connect(self, :probes)
+
+    def parse_resource(resource)
+      # return a hash of information
+      parsed = {
+        'location' => resource.location,
+        'name' => resource.name,
+        'type' => resource.type,
+        'exist?' => true,
+        'properties' => AzureResourceProbe.new(resource.properties),
+      }
+
+      parsed
+    end
+  end
+end

data/lib/resources/azure/azure_resource_group.rb

@@ -1,152 +1,152 @@
-# encoding: utf-8
-
-require 'resources/azure/azure_backend'
-
-module Inspec::Resources
-  class AzureResourceGroup < AzureResourceBase
-    name 'azure_resource_group'
-
-    desc '
-      Inspec Resource to get metadata about a specific Resource Group
-    '
-
-    supports platform: 'azure'
-
-    attr_reader :name, :location, :id, :total, :counts, :mapping
-
-    # Constructor to get the resource group itself and perform some analysis on the
-    # resources that in the resource group.
-    #
-    # This analysis is defined by the the mapping hashtable which is used to define
-    # the 'has_xxx?' methods (see AzureResourceGroup#create_has_methods) and return
-    # the counts for each type
-    #
-    # @author Russell Seymour
-    def initialize(opts)
-      opts.key?(:name) ? opts[:group_name] = opts[:name] : false
-      # Ensure that the opts only have the name of the resource group set
-      opts.select! { |k, _v| k == :group_name }
-      super(opts)
-
-      # set the mapping for the Azure Resources
-      @mapping = {
-        nic: 'Microsoft.Network/networkInterfaces',
-        vm: 'Microsoft.Compute/virtualMachines',
-        extension: 'Microsoft.Compute/virtualMachines/extensions',
-        nsg: 'Microsoft.Network/networkSecurityGroups',
-        vnet: 'Microsoft.Network/virtualNetworks',
-        managed_disk: 'Microsoft.Compute/disks',
-        managed_disk_image: 'Microsoft.Compute/images',
-        sa: 'Microsoft.Storage/storageAccounts',
-        public_ip: 'Microsoft.Network/publicIPAddresses',
-      }
-
-      # Get information about the resource group itself
-      resource_group
-
-      # Get information about the resources in the resource group
-      resources
-
-      # Call method to create the has_xxxx? methods
-      create_has_methods
-
-      # Call method to allow access to the tag values
-      create_tag_methods
-    end
-
-    # Return the provisioning state of the resource group
-    #
-    # @author Russell Seymour
-    def provisioning_state
-      properties.provisioningState
-    end
-
-    # Analyze the fully qualified id of the resource group to return the subscription id
-    # that this resource group is part of
-    #
-    # The format of the id is
-    #   /subscriptions/<SUBSCRIPTION_ID>/resourceGroups/<RESOURCE_GROUP_NAME>
-    #
-    # @author Russell Seymour
-    def subscription_id
-      id.split(%r{\/}).reject(&:empty?)[1]
-    end
-
-    # Method to parse the resources that have been returned
-    # This allows the calculations of the amount of resources to be determined
-    #
-    # @author Russell Seymour
-    #
-    # @param [Hash] resource A hashtable representing the resource group
-    def parse_resource(resource)
-      # return a hash of information
-      parsed = {
-        'name' => resource.name,
-        'type' => resource.type,
-      }
-
-      parsed
-    end
-
-    # This method catches the xxx_count calls that are made on the resource.
-    #
-    # The method that is called is stripped of '_count' and then compared with the
-    # mappings table. If that type exists then the number of those items is returned.
-    # However if that type is not in the Resource Group then the method will return
-    # a NoMethodError exception
-    #
-    # @author Russell Seymour
-    #
-    # @param [Symbol] method_id The name of the method that was called
-    def method_missing(method_id)
-      # Determine the mapping_key based on the method_id
-      mapping_key = method_id.to_s.chomp('_count').to_sym
-
-      if mapping.key?(mapping_key)
-        # based on the method id get the
-        namespace, type_name = mapping[mapping_key].split(/\./)
-
-        # check that the type_name is defined, if not return 0
-        if send(namespace).methods.include?(type_name.to_sym)
-          # return the count for the method id
-          send(namespace).send(type_name)
-        else
-          0
-        end
-      else
-        msg = format('undefined method `%s` for %s', method_id, self.class)
-        raise NoMethodError, msg
-      end
-    end
-
-    private
-
-    # For each of the mappings this method creates the has_xxx? method. This allows the use
-    # of the following type of test
-    #
-    #   it { should have_nics }
-    #
-    # For example, it will create a has_nics? method that returns a boolean to state of the
-    # resource group has any nics at all.
-    #
-    # @author Russell Seymour
-    # @private
-    def create_has_methods
-      return if failed_resource?
-      # Create the has methods for each of the mappings
-      # This is a quick test to show that the resource group has at least one of these things
-      mapping.each do |name, type|
-        # Determine the name of the method name
-        method_name = format('has_%ss?', name)
-        namespace, type_name = type.split(/\./)
-
-        # use the namespace and the type_name to determine if the resource group has this type or not
-        result = send(namespace).methods.include?(type_name.to_sym) ? true : false
-
-        define_singleton_method method_name do
-          result
-        end
-      end
-    end
-  end
-end
+# encoding: utf-8
+
+require 'resources/azure/azure_backend'
+
+module Inspec::Resources
+  class AzureResourceGroup < AzureResourceBase
+    name 'azure_resource_group'
+
+    desc '
+      Inspec Resource to get metadata about a specific Resource Group
+    '
+
+    supports platform: 'azure'
+
+    attr_reader :name, :location, :id, :total, :counts, :mapping
+
+    # Constructor to get the resource group itself and perform some analysis on the
+    # resources that in the resource group.
+    #
+    # This analysis is defined by the the mapping hashtable which is used to define
+    # the 'has_xxx?' methods (see AzureResourceGroup#create_has_methods) and return
+    # the counts for each type
+    #
+    # @author Russell Seymour
+    def initialize(opts)
+      opts.key?(:name) ? opts[:group_name] = opts[:name] : false
+      # Ensure that the opts only have the name of the resource group set
+      opts.select! { |k, _v| k == :group_name }
+      super(opts)
+
+      # set the mapping for the Azure Resources
+      @mapping = {
+        nic: 'Microsoft.Network/networkInterfaces',
+        vm: 'Microsoft.Compute/virtualMachines',
+        extension: 'Microsoft.Compute/virtualMachines/extensions',
+        nsg: 'Microsoft.Network/networkSecurityGroups',
+        vnet: 'Microsoft.Network/virtualNetworks',
+        managed_disk: 'Microsoft.Compute/disks',
+        managed_disk_image: 'Microsoft.Compute/images',
+        sa: 'Microsoft.Storage/storageAccounts',
+        public_ip: 'Microsoft.Network/publicIPAddresses',
+      }
+
+      # Get information about the resource group itself
+      resource_group
+
+      # Get information about the resources in the resource group
+      resources
+
+      # Call method to create the has_xxxx? methods
+      create_has_methods
+
+      # Call method to allow access to the tag values
+      create_tag_methods
+    end
+
+    # Return the provisioning state of the resource group
+    #
+    # @author Russell Seymour
+    def provisioning_state
+      properties.provisioningState
+    end
+
+    # Analyze the fully qualified id of the resource group to return the subscription id
+    # that this resource group is part of
+    #
+    # The format of the id is
+    #   /subscriptions/<SUBSCRIPTION_ID>/resourceGroups/<RESOURCE_GROUP_NAME>
+    #
+    # @author Russell Seymour
+    def subscription_id
+      id.split(%r{\/}).reject(&:empty?)[1]
+    end
+
+    # Method to parse the resources that have been returned
+    # This allows the calculations of the amount of resources to be determined
+    #
+    # @author Russell Seymour
+    #
+    # @param [Hash] resource A hashtable representing the resource group
+    def parse_resource(resource)
+      # return a hash of information
+      parsed = {
+        'name' => resource.name,
+        'type' => resource.type,
+      }
+
+      parsed
+    end
+
+    # This method catches the xxx_count calls that are made on the resource.
+    #
+    # The method that is called is stripped of '_count' and then compared with the
+    # mappings table. If that type exists then the number of those items is returned.
+    # However if that type is not in the Resource Group then the method will return
+    # a NoMethodError exception
+    #
+    # @author Russell Seymour
+    #
+    # @param [Symbol] method_id The name of the method that was called
+    def method_missing(method_id)
+      # Determine the mapping_key based on the method_id
+      mapping_key = method_id.to_s.chomp('_count').to_sym
+
+      if mapping.key?(mapping_key)
+        # based on the method id get the
+        namespace, type_name = mapping[mapping_key].split(/\./)
+
+        # check that the type_name is defined, if not return 0
+        if send(namespace).methods.include?(type_name.to_sym)
+          # return the count for the method id
+          send(namespace).send(type_name)
+        else
+          0
+        end
+      else
+        msg = format('undefined method `%s` for %s', method_id, self.class)
+        raise NoMethodError, msg
+      end
+    end
+
+    private
+
+    # For each of the mappings this method creates the has_xxx? method. This allows the use
+    # of the following type of test
+    #
+    #   it { should have_nics }
+    #
+    # For example, it will create a has_nics? method that returns a boolean to state of the
+    # resource group has any nics at all.
+    #
+    # @author Russell Seymour
+    # @private
+    def create_has_methods
+      return if failed_resource?
+      # Create the has methods for each of the mappings
+      # This is a quick test to show that the resource group has at least one of these things
+      mapping.each do |name, type|
+        # Determine the name of the method name
+        method_name = format('has_%ss?', name)
+        namespace, type_name = type.split(/\./)
+
+        # use the namespace and the type_name to determine if the resource group has this type or not
+        result = send(namespace).methods.include?(type_name.to_sym) ? true : false
+
+        define_singleton_method method_name do
+          result
+        end
+      end
+    end
+  end
+end

data/lib/resources/azure/azure_virtual_machine.rb

@@ -1,264 +1,264 @@
-# encoding: utf-8
-
-require 'resources/azure/azure_backend'
-
-module Inspec::Resources
-  class AzureVirtualMachine < AzureResourceBase
-    name 'azure_virtual_machine'
-
-    desc '
-      Inspec Resource to test Azure Virtual Machines
-    '
-
-    supports platform: 'azure'
-
-    # Constructor for the resource. This calls the parent constructor to
-    # get the generic resource for the specified machine. This will provide
-    # static methods that are documented
-    #
-    # @author Russell Seymour
-    def initialize(opts = {})
-      # The generic resource needs to pass back a Microsoft.Compute/virtualMachines object so force it
-      opts[:type] = 'Microsoft.Compute/virtualMachines'
-      super(opts)
-
-      # Find the virtual machines
-      resources
-
-      create_tag_methods
-    end
-
-    # Method to catch calls that are not explicitly defined.
-    # This allows the simple attributes of the virtual machine to be read without having
-    # to define each one in turn.
-    #
-    # rubocop:disable Metrics/AbcSize
-    #
-    # @param symobl method_id The symbol of the method that has been called
-    #
-    # @return Value of attribute that has been called
-    def method_missing(method_id)
-      # Depending on the method that has been called, determine what value should be returned
-      # These are set as camel case methods to comply with rubocop
-      image_reference_attrs = %w{sku publisher offer}
-      osdisk_attrs = %w{os_type caching create_option disk_size_gb}
-      hardware_profile_attrs = %w{vm_size}
-      os_profile_attrs = %w{computer_name admin_username}
-      osdisk_managed_disk_attrs = %w{storage_account_type}
-
-      # determine the method name to call by converting the snake_case to camelCase
-      # method_name = self.camel_case(method_id.to_s)
-      method_name = method_id.to_s.split('_').inject([]) { |buffer, e| buffer.push(buffer.empty? ? e : e.capitalize) }.join
-      method_name.end_with?('Gb') ? method_name.gsub!(/Gb/, &:upcase) : false
-
-      if image_reference_attrs.include?(method_id.to_s)
-        properties.storageProfile.imageReference.send(method_name)
-      elsif osdisk_attrs.include?(method_id.to_s)
-        properties.storageProfile.osDisk.send(method_name)
-      elsif hardware_profile_attrs.include?(method_id.to_s)
-        properties.hardwareProfile.send(method_name)
-      elsif os_profile_attrs.include?(method_id.to_s)
-        properties.osProfile.send(method_name)
-      elsif osdisk_managed_disk_attrs.include?(method_id.to_s)
-        properties.storageProfile.osDisk.managedDisk.send(method_name)
-      end
-    end
-
-    # Return the name of the os disk
-    #
-    # @return string Name of the OS disk
-    def os_disk_name
-      properties.storageProfile.osDisk.name
-    end
-
-    # Determine if the OS disk is a managed disk
-    #
-    # @return boolean
-    def has_managed_osdisk?
-      defined?(properties.storageProfile.osDisk.managedDisk)
-    end
-
-    # Does the machine have any NICs connected
-    #
-    # @return boolean
-    def has_nics?
-      properties.networkProfile.networkInterfaces.count != 0
-    end
-
-    # How many NICs are connected to the machine
-    #
-    # @return integer
-    def nic_count
-      properties.networkProfile.networkInterfaces.count
-    end
-
-    # Return an array of the connected NICs so that it can be tested to ensure
-    # the machine is connected properly
-    #
-    # @return array Array of NIC names connected to the machine
-    def connected_nics
-      nic_names = []
-      properties.networkProfile.networkInterfaces.each do |nic|
-        nic_names << nic.id.split(%r{/}).last
-      end
-      nic_names
-    end
-
-    # Whether the machine has data disks or not
-    #
-    # @return boolean
-    def has_data_disks?
-      properties.storageProfile.dataDisks.count != 0
-    end
-
-    # How many data disks are connected
-    #
-    # @return integer
-    def data_disk_count
-      properties.storageProfile.dataDisks.count
-    end
-
-    # Does the machine allow password authentication
-    #
-    # This allows the use of
-    #   it { should have_password_authentication }
-    # within the Inspec profile
-    #
-    # @return boolean
-    def has_password_authentication?
-      password_authentication?
-    end
-
-    # Deteremine if the machine allows password authentication
-    #
-    # @return boolean
-    def password_authentication?
-      # if the osProfile property has a linuxConfiguration section then interrogate that
-      # otherwise it is a Windows machine and that always has password auth
-      if defined?(properties.osProfile.linuxConfiguration)
-        !properties.osProfile.linuxConfiguration.disablePasswordAuthentication
-      else
-        true
-      end
-    end
-
-    # Has the machine been given Custom Data at creation
-    #
-    # This allows the use of
-    #    it { should have_custom_data }
-    # within the Inspec Profile
-    #
-    # @return boolean
-    def has_custom_data?
-      custom_data?
-    end
-
-    # Determine if custom data has been set
-    #
-    # @return boolean
-    def custom_data?
-      if defined?(properties.osProfile.CustomData)
-        true
-      else
-        false
-      end
-    end
-
-    # Are any SSH Keys assigned to the machine
-    #
-    # This allows the use of
-    #    it { should have_ssh_keys }
-    # within the Inspec Profile
-    #
-    # @return boolean
-    def has_ssh_keys?
-      ssh_keys?
-    end
-
-    # Determine if any ssh keys have been asigned to the machine
-    #
-    # @return boolean
-    def ssh_keys?
-      if defined?(properties.osProfile.linuxConfiguration.ssh)
-        properties.osProfile.linuxConfiguration.ssh.publicKeys != 0
-      else
-        false
-      end
-    end
-
-    # Return the number of ssh keys that have been assigned to the machine
-    #
-    # @return integer
-    def ssh_key_count
-      if defined?(properties.osProfile.linuxConfiguration.ssh)
-        properties.osProfile.linuxConfiguration.ssh.publicKeys.count
-      else
-        0
-      end
-    end
-
-    # Determine is the specified key is in the ssh_keys list
-    #
-    # @return array Array of the public keys that are assigned to allow for testing of that key
-    def ssh_keys
-      # iterate around the keys
-      keys = []
-      properties.osProfile.linuxConfiguration.ssh.publicKeys.each do |key|
-        keys << key.keyData
-      end
-      keys
-    end
-
-    # Does the machine have boot diagnostics enabled
-    #
-    # @return boolean
-    def has_boot_diagnostics?
-      if defined?(properties.diagnosticsProfile)
-        properties.diagnosticsProfile.bootDiagnostics.enabled
-      else
-        false
-      end
-    end
-
-    # Return the URI that has been set for the boot diagnostics storage
-    #
-    # @return string
-    def boot_diagnostics_storage_uri
-      properties.diagnosticsProfile.bootDiagnostics.storageUri
-    end
-
-    # If this is a windows machine, returns whether the agent was provisioned or not
-    #
-    # @return boolean
-    def has_provision_vmagent?
-      if defined?(properties.osProfile.windowsConfiguration)
-        properties.osProfile.windowsConfiguration.provisionVMAgent
-      else
-        false
-      end
-    end
-
-    # If a windows machine see if automatic updates for the agent are enabled
-    #
-    # @return boolean
-    def has_automatic_agent_update?
-      if defined?(properties.osProfile.windowsConfiguration)
-        properties.osProfile.windowsConfiguration.enableAutomaticUpdates
-      else
-        false
-      end
-    end
-
-    # If this is a windows machine return a boolean to state of the WinRM options
-    # have been set
-    #
-    # @return boolean
-    def has_winrm_options?
-      if defined?(properties.osProfile.windowsConfiguration) && defined?(properties.osProfile.windowsConfiguration.winrm)
-        properties.osProfile.windowsConfiguration.winrm.protocol
-      else
-        false
-      end
-    end
-  end
-end
+# encoding: utf-8
+
+require 'resources/azure/azure_backend'
+
+module Inspec::Resources
+  class AzureVirtualMachine < AzureResourceBase
+    name 'azure_virtual_machine'
+
+    desc '
+      Inspec Resource to test Azure Virtual Machines
+    '
+
+    supports platform: 'azure'
+
+    # Constructor for the resource. This calls the parent constructor to
+    # get the generic resource for the specified machine. This will provide
+    # static methods that are documented
+    #
+    # @author Russell Seymour
+    def initialize(opts = {})
+      # The generic resource needs to pass back a Microsoft.Compute/virtualMachines object so force it
+      opts[:type] = 'Microsoft.Compute/virtualMachines'
+      super(opts)
+
+      # Find the virtual machines
+      resources
+
+      create_tag_methods
+    end
+
+    # Method to catch calls that are not explicitly defined.
+    # This allows the simple attributes of the virtual machine to be read without having
+    # to define each one in turn.
+    #
+    # rubocop:disable Metrics/AbcSize
+    #
+    # @param symobl method_id The symbol of the method that has been called
+    #
+    # @return Value of attribute that has been called
+    def method_missing(method_id)
+      # Depending on the method that has been called, determine what value should be returned
+      # These are set as camel case methods to comply with rubocop
+      image_reference_attrs = %w{sku publisher offer}
+      osdisk_attrs = %w{os_type caching create_option disk_size_gb}
+      hardware_profile_attrs = %w{vm_size}
+      os_profile_attrs = %w{computer_name admin_username}
+      osdisk_managed_disk_attrs = %w{storage_account_type}
+
+      # determine the method name to call by converting the snake_case to camelCase
+      # method_name = self.camel_case(method_id.to_s)
+      method_name = method_id.to_s.split('_').inject([]) { |buffer, e| buffer.push(buffer.empty? ? e : e.capitalize) }.join
+      method_name.end_with?('Gb') ? method_name.gsub!(/Gb/, &:upcase) : false
+
+      if image_reference_attrs.include?(method_id.to_s)
+        properties.storageProfile.imageReference.send(method_name)
+      elsif osdisk_attrs.include?(method_id.to_s)
+        properties.storageProfile.osDisk.send(method_name)
+      elsif hardware_profile_attrs.include?(method_id.to_s)
+        properties.hardwareProfile.send(method_name)
+      elsif os_profile_attrs.include?(method_id.to_s)
+        properties.osProfile.send(method_name)
+      elsif osdisk_managed_disk_attrs.include?(method_id.to_s)
+        properties.storageProfile.osDisk.managedDisk.send(method_name)
+      end
+    end
+
+    # Return the name of the os disk
+    #
+    # @return string Name of the OS disk
+    def os_disk_name
+      properties.storageProfile.osDisk.name
+    end
+
+    # Determine if the OS disk is a managed disk
+    #
+    # @return boolean
+    def has_managed_osdisk?
+      defined?(properties.storageProfile.osDisk.managedDisk)
+    end
+
+    # Does the machine have any NICs connected
+    #
+    # @return boolean
+    def has_nics?
+      properties.networkProfile.networkInterfaces.count != 0
+    end
+
+    # How many NICs are connected to the machine
+    #
+    # @return integer
+    def nic_count
+      properties.networkProfile.networkInterfaces.count
+    end
+
+    # Return an array of the connected NICs so that it can be tested to ensure
+    # the machine is connected properly
+    #
+    # @return array Array of NIC names connected to the machine
+    def connected_nics
+      nic_names = []
+      properties.networkProfile.networkInterfaces.each do |nic|
+        nic_names << nic.id.split(%r{/}).last
+      end
+      nic_names
+    end
+
+    # Whether the machine has data disks or not
+    #
+    # @return boolean
+    def has_data_disks?
+      properties.storageProfile.dataDisks.count != 0
+    end
+
+    # How many data disks are connected
+    #
+    # @return integer
+    def data_disk_count
+      properties.storageProfile.dataDisks.count
+    end
+
+    # Does the machine allow password authentication
+    #
+    # This allows the use of
+    #   it { should have_password_authentication }
+    # within the Inspec profile
+    #
+    # @return boolean
+    def has_password_authentication?
+      password_authentication?
+    end
+
+    # Deteremine if the machine allows password authentication
+    #
+    # @return boolean
+    def password_authentication?
+      # if the osProfile property has a linuxConfiguration section then interrogate that
+      # otherwise it is a Windows machine and that always has password auth
+      if defined?(properties.osProfile.linuxConfiguration)
+        !properties.osProfile.linuxConfiguration.disablePasswordAuthentication
+      else
+        true
+      end
+    end
+
+    # Has the machine been given Custom Data at creation
+    #
+    # This allows the use of
+    #    it { should have_custom_data }
+    # within the Inspec Profile
+    #
+    # @return boolean
+    def has_custom_data?
+      custom_data?
+    end
+
+    # Determine if custom data has been set
+    #
+    # @return boolean
+    def custom_data?
+      if defined?(properties.osProfile.CustomData)
+        true
+      else
+        false
+      end
+    end
+
+    # Are any SSH Keys assigned to the machine
+    #
+    # This allows the use of
+    #    it { should have_ssh_keys }
+    # within the Inspec Profile
+    #
+    # @return boolean
+    def has_ssh_keys?
+      ssh_keys?
+    end
+
+    # Determine if any ssh keys have been asigned to the machine
+    #
+    # @return boolean
+    def ssh_keys?
+      if defined?(properties.osProfile.linuxConfiguration.ssh)
+        properties.osProfile.linuxConfiguration.ssh.publicKeys != 0
+      else
+        false
+      end
+    end
+
+    # Return the number of ssh keys that have been assigned to the machine
+    #
+    # @return integer
+    def ssh_key_count
+      if defined?(properties.osProfile.linuxConfiguration.ssh)
+        properties.osProfile.linuxConfiguration.ssh.publicKeys.count
+      else
+        0
+      end
+    end
+
+    # Determine is the specified key is in the ssh_keys list
+    #
+    # @return array Array of the public keys that are assigned to allow for testing of that key
+    def ssh_keys
+      # iterate around the keys
+      keys = []
+      properties.osProfile.linuxConfiguration.ssh.publicKeys.each do |key|
+        keys << key.keyData
+      end
+      keys
+    end
+
+    # Does the machine have boot diagnostics enabled
+    #
+    # @return boolean
+    def has_boot_diagnostics?
+      if defined?(properties.diagnosticsProfile)
+        properties.diagnosticsProfile.bootDiagnostics.enabled
+      else
+        false
+      end
+    end
+
+    # Return the URI that has been set for the boot diagnostics storage
+    #
+    # @return string
+    def boot_diagnostics_storage_uri
+      properties.diagnosticsProfile.bootDiagnostics.storageUri
+    end
+
+    # If this is a windows machine, returns whether the agent was provisioned or not
+    #
+    # @return boolean
+    def has_provision_vmagent?
+      if defined?(properties.osProfile.windowsConfiguration)
+        properties.osProfile.windowsConfiguration.provisionVMAgent
+      else
+        false
+      end
+    end
+
+    # If a windows machine see if automatic updates for the agent are enabled
+    #
+    # @return boolean
+    def has_automatic_agent_update?
+      if defined?(properties.osProfile.windowsConfiguration)
+        properties.osProfile.windowsConfiguration.enableAutomaticUpdates
+      else
+        false
+      end
+    end
+
+    # If this is a windows machine return a boolean to state of the WinRM options
+    # have been set
+    #
+    # @return boolean
+    def has_winrm_options?
+      if defined?(properties.osProfile.windowsConfiguration) && defined?(properties.osProfile.windowsConfiguration.winrm)
+        properties.osProfile.windowsConfiguration.winrm.protocol
+      else
+        false
+      end
+    end
+  end
+end