Module: RspecPuppetFacts
- Defined in:
- lib/rspec-puppet-facts.rb,
lib/rspec-puppet-facts/version.rb
Overview
The purpose of this module is to simplify the Puppet module’s RSpec tests by looping through all supported OS’es and their facts data which is received from the FacterDB.
Defined Under Namespace
Modules: Version
Constant Summary collapse
- FACTS_CACHE =
{}
Class Method Summary collapse
-
.augeas? ⇒ Boolean
private
Determine if the Augeas gem is available.
-
.common_facts ⇒ Hash <Symbol => String>
private
These facts are common for all OS’es and will be added to the facts retrieved from the FacterDB.
-
.custom_facts ⇒ nil, Hash
private
Get custom facts.
- .facter_version_for_puppet_version(puppet_version) ⇒ Object
-
.facter_version_to_filter(version) ⇒ String
private
Generates a JGrep statement expression for a specific facter version.
-
.mcollective? ⇒ Boolean
private
Determine if the mcollective gem is available :nocov:.
-
.meta_supported_os ⇒ Array<Hash>
private
Get the “operatingsystem_support” structure from the parsed metadata.json file in the metadata.
-
.metadata ⇒ Hash
private
Read the metadata file and parse its JSON content.
-
.metadata_file ⇒ String
private
This file contains the Puppet module’s metadata.
-
.register_custom_fact(name, value, options) ⇒ Object
private
Adds a custom fact to the @custom_facts variable.
-
.reset ⇒ Object
private
Reset the memoization to make the saved structures be generated again.
-
.spec_facts_os_filter ⇒ nil, String
private
If provided this filter can be used to limit the set of retrieved facts only to the matched OS names.
-
.spec_facts_strict? ⇒ Boolean
private
If SPEC_FACTS_STRICT is set to ‘yes`, RspecPuppetFacts will error on missing FacterDB entries, instead of warning & skipping the tests, or using an older facter version.
-
.warning(message) ⇒ Object
private
Print a warning message to the console.
-
.with_custom_facts(os, facts) ⇒ Hash
private
Adds any custom facts according to the rules defined for the operating system with the given facts.
Instance Method Summary collapse
-
#add_custom_fact(name, value, options = {}) ⇒ Object
Register a custom fact that will be included in the facts hash.
-
#on_supported_os(opts = {}) ⇒ Hash <String => Hash>
Use the provided options or the data from the metadata.json file to find a set of matching facts in the FacterDB.
-
#on_supported_os_implementation(opts = {}) ⇒ Object
private
The real implementation of on_supported_os.
- #stringify_keys(hash) ⇒ Object
Class Method Details
.augeas? ⇒ Boolean
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Determine if the Augeas gem is available. :nocov:
279 280 281 282 283 284 |
# File 'lib/rspec-puppet-facts.rb', line 279 def self.augeas? require 'augeas' true rescue LoadError false end |
.common_facts ⇒ Hash <Symbol => String>
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
These facts are common for all OS’es and will be added to the facts retrieved from the FacterDB
258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 |
# File 'lib/rspec-puppet-facts.rb', line 258 def self.common_facts return @common_facts if @common_facts @common_facts = { :puppetversion => Puppet.version, :rubysitedir => RbConfig::CONFIG['sitelibdir'], :rubyversion => RUBY_VERSION, } @common_facts[:mco_version] = MCollective::VERSION if mcollective? if augeas? @common_facts[:augeasversion] = Augeas.open(nil, nil, Augeas::NO_MODL_AUTOLOAD).get('/augeas/version') end @common_facts end |
.custom_facts ⇒ nil, Hash
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Get custom facts
233 234 235 |
# File 'lib/rspec-puppet-facts.rb', line 233 def self.custom_facts @custom_facts end |
.facter_version_for_puppet_version(puppet_version) ⇒ Object
359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 |
# File 'lib/rspec-puppet-facts.rb', line 359 def self.facter_version_for_puppet_version(puppet_version) return Facter.version if puppet_version.nil? json_path = File.(File.join(__dir__, '..', 'ext', 'puppet_agent_components.json')) unless File.file?(json_path) && File.readable?(json_path) warning "#{json_path} does not exist or is not readable, defaulting to Facter #{Facter.version}" return Facter.version end fd = File.open(json_path, 'rb:UTF-8') data = JSON.parse(fd.read) version_map = data.map { |_, versions| if versions['puppet'].nil? || versions['facter'].nil? nil else [Gem::Version.new(versions['puppet']), versions['facter']] end }.compact puppet_gem_version = Gem::Version.new(puppet_version) applicable_versions = version_map.select { |p, _| puppet_gem_version >= p } if applicable_versions.empty? warning "Unable to find Puppet #{puppet_version} in #{json_path}, defaulting to Facter #{Facter.version}" return Facter.version end applicable_versions.sort { |a, b| b.first <=> a.first }.first.last rescue JSON::ParserError warning "#{json_path} contains invalid JSON, defaulting to Facter #{Facter.version}" Facter.version ensure fd.close if fd end |
.facter_version_to_filter(version) ⇒ String
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Generates a JGrep statement expression for a specific facter version
354 355 356 357 |
# File 'lib/rspec-puppet-facts.rb', line 354 def self.facter_version_to_filter(version) major, minor = version.split('.') "/\\A#{major}\\.#{minor}\\./" end |
.mcollective? ⇒ Boolean
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Determine if the mcollective gem is available :nocov:
291 292 293 294 295 296 |
# File 'lib/rspec-puppet-facts.rb', line 291 def self.mcollective? require 'mcollective' true rescue LoadError false end |
.meta_supported_os ⇒ Array<Hash>
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Get the “operatingsystem_support” structure from the parsed metadata.json file in the metadata
305 306 307 308 309 310 |
# File 'lib/rspec-puppet-facts.rb', line 305 def self. unless ['operatingsystem_support'].is_a? Array fail StandardError, 'Unknown operatingsystem support in the metadata file!' end ['operatingsystem_support'] end |
.metadata ⇒ Hash
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Read the metadata file and parse its JSON content.
317 318 319 320 321 322 323 324 |
# File 'lib/rspec-puppet-facts.rb', line 317 def self. return @metadata if @metadata unless File.file? fail StandardError, "Can't find metadata.json... dunno why" end content = File.read @metadata = JSON.parse content end |
.metadata_file ⇒ String
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
This file contains the Puppet module’s metadata
329 330 331 |
# File 'lib/rspec-puppet-facts.rb', line 329 def self. 'metadata.json' end |
.register_custom_fact(name, value, options) ⇒ Object
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Adds a custom fact to the @custom_facts variable.
206 207 208 209 |
# File 'lib/rspec-puppet-facts.rb', line 206 def self.register_custom_fact(name, value, ) @custom_facts ||= {} @custom_facts[name.to_s] = {:options => , :value => value} end |
.reset ⇒ Object
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Reset the memoization to make the saved structures be generated again
344 345 346 347 348 |
# File 'lib/rspec-puppet-facts.rb', line 344 def self.reset @custom_facts = nil @common_facts = nil @metadata = nil end |
.spec_facts_os_filter ⇒ nil, String
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
If provided this filter can be used to limit the set of retrieved facts only to the matched OS names. The value is being taken from the SPEC_FACTS_OS environment variable and
243 244 245 |
# File 'lib/rspec-puppet-facts.rb', line 243 def self.spec_facts_os_filter ENV['SPEC_FACTS_OS'] end |
.spec_facts_strict? ⇒ Boolean
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
If SPEC_FACTS_STRICT is set to ‘yes`, RspecPuppetFacts will error on missing FacterDB entries, instead of warning & skipping the tests, or using an older facter version.
250 251 252 |
# File 'lib/rspec-puppet-facts.rb', line 250 def self.spec_facts_strict? ENV['SPEC_FACTS_STRICT'] == 'yes' end |
.warning(message) ⇒ Object
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Print a warning message to the console
336 337 338 |
# File 'lib/rspec-puppet-facts.rb', line 336 def self.warning() STDERR.puts end |
.with_custom_facts(os, facts) ⇒ Hash
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
Adds any custom facts according to the rules defined for the operating system with the given facts.
217 218 219 220 221 222 223 224 225 226 227 228 |
# File 'lib/rspec-puppet-facts.rb', line 217 def self.with_custom_facts(os, facts) return facts unless @custom_facts @custom_facts.each do |name, fact| next if fact[:options][:confine] && !fact[:options][:confine].include?(os) next if fact[:options][:exclude] && fact[:options][:exclude].include?(os) facts[name] = fact[:value].respond_to?(:call) ? fact[:value].call(os, facts) : fact[:value] end facts end |
Instance Method Details
#add_custom_fact(name, value, options = {}) ⇒ Object
Register a custom fact that will be included in the facts hash. If it should be limited to a particular OS, pass a :confine option that contains the operating system(s) to confine to. If it should be excluded on a particular OS, use :exclude.
191 192 193 194 195 196 |
# File 'lib/rspec-puppet-facts.rb', line 191 def add_custom_fact(name, value, = {}) [:confine] = [[:confine]] if [:confine].is_a?(String) [:exclude] = [[:exclude]] if [:exclude].is_a?(String) RspecPuppetFacts.register_custom_fact(name, value, ) end |
#on_supported_os(opts = {}) ⇒ Hash <String => Hash>
Use the provided options or the data from the metadata.json file to find a set of matching facts in the FacterDB. OS names and facts can be used in the Puppet RSpec tests to run the examples against all supported facts combinations.
The list of received OS facts can also be filtered by the SPEC_FACTS_OS environment variable. For example, if the variable is set to “debian” only the OS names which start with “debian” will be returned. It allows a user to quickly run the tests only on a single facts set without any file modifications.
select facts from, e.g.: ‘3.6’ will be used instead of the “operatingsystem_support” section if the metadata file even if the file is missing.
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
# File 'lib/rspec-puppet-facts.rb', line 31 def on_supported_os(opts = {}) opts[:hardwaremodels] ||= ['x86_64'] opts[:hardwaremodels] = [opts[:hardwaremodels]] unless opts[:hardwaremodels].is_a? Array opts[:supported_os] ||= RspecPuppetFacts. opts[:facterversion] ||= RSpec.configuration.default_facter_version # This should list all variables that on_supported_os_implementation uses cache_key = [ opts.to_s, RspecPuppetFacts.custom_facts.to_s, RspecPuppetFacts.spec_facts_os_filter, RspecPuppetFacts.spec_facts_strict?, ] result = FACTS_CACHE[cache_key] ||= on_supported_os_implementation(opts) # Marshalling is used to get unique instances which is needed for test # isolation when facts are overridden. Marshal.load(Marshal.dump(result)) end |
#on_supported_os_implementation(opts = {}) ⇒ Object
This method is part of a private API. You should avoid using this method if possible, as it may be removed or be changed in the future.
The real implementation of on_supported_os.
Generating facts is slow - this allows memoization of the facts between multiple calls.
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 |
# File 'lib/rspec-puppet-facts.rb', line 58 def on_supported_os_implementation(opts = {}) unless (facterversion = opts[:facterversion]) =~ /\A\d+\.\d+(?:\.\d+)*\z/ raise ArgumentError, ":facterversion must be in the format 'n.n' or " \ "'n.n.n' (n is numeric), not '#{facterversion}'" end filter = [] opts[:supported_os].map do |os_sup| if os_sup['operatingsystemrelease'] Array(os_sup['operatingsystemrelease']).map do || opts[:hardwaremodels].each do |hardwaremodel| os_release_filter = "/^#{Regexp.escape(.split(' ')[0])}/" if os_sup['operatingsystem'] =~ /BSD/i hardwaremodel = 'amd64' elsif os_sup['operatingsystem'] =~ /Solaris/i hardwaremodel = 'i86pc' elsif os_sup['operatingsystem'] =~ /AIX/i hardwaremodel = '/^IBM,.*/' os_release_filter = if =~ /\A(\d+)\.(\d+)\Z/ "/^#{$~[1]}#{$~[2]}00-/" else "/^#{}-/" end elsif os_sup['operatingsystem'] =~ /Windows/i hardwaremodel = facterversion =~ /^[12]\./ ? 'x64' : 'x86_64' os_sup['operatingsystem'] = os_sup['operatingsystem'].downcase = [/\A(?:Server )?(.+)/i, 1] # force quoting because windows releases can contain spaces os_release_filter = "\"#{}\"" if == '2016' && Puppet::Util::Package.versioncmp(facterversion, '3.4') < 0 os_release_filter = '/^10\\.0\\./' end elsif os_sup['operatingsystem'] =~ /Amazon/i # Tighten the regex for Amazon Linux 2 so that we don't pick up Amazon Linux 2016 or 2017 facts os_release_filter = "/^2$/" if == '2' end filter << { :operatingsystem => os_sup['operatingsystem'], :operatingsystemrelease => os_release_filter, :hardwaremodel => hardwaremodel, } end end else opts[:hardwaremodels].each do |hardwaremodel| filter << { :operatingsystem => os_sup['operatingsystem'], :hardwaremodel => hardwaremodel, } end end end # FacterDB may have newer versions of facter data for which it contains a subset of all possible # facter data (see FacterDB 0.5.2 for Facter releases 3.8 and 3.9). In this situation we need to # cycle through and downgrade Facter versions per platform type until we find matching Facter data. filter.each do |filter_spec| facter_version_filter = RspecPuppetFacts.facter_version_to_filter(facterversion) db = FacterDB.get_facts(filter_spec.merge({ :facterversion => facter_version_filter })) if db.empty? if RspecPuppetFacts.spec_facts_strict? raise ArgumentError, "No facts were found in the FacterDB for Facter v#{facterversion} on #{filter_spec}, aborting" end facterversion_obj = Gem::Version.new(facterversion) version = FacterDB.get_facts(filter_spec).map { |facts| Gem::Version.new(facts[:facterversion]) }.sort.reverse.detect { |v| v <= facterversion_obj } next unless version version = version.to_s facter_version_filter = "/\\A#{Regexp.escape(version)}/" unless version == facterversion RspecPuppetFacts.warning "No facts were found in the FacterDB for Facter v#{facterversion} on #{filter_spec}, using v#{version} instead" end end filter_spec[:facterversion] = facter_version_filter end received_facts = FacterDB::get_facts(filter) unless received_facts.any? RspecPuppetFacts.warning "No facts were found in the FacterDB for: #{filter.inspect}" return {} end os_facts_hash = {} received_facts.map do |facts| # Fix facter bug if facts[:operatingsystem] == 'Ubuntu' = facts[:operatingsystemrelease].split('.')[0..1].join('.') elsif facts[:operatingsystem] == 'OpenBSD' = facts[:operatingsystemrelease] elsif facts[:operatingsystem] == 'windows' && facts[:operatingsystemrelease].start_with?('10.0.') = '2016' else if facts[:operatingsystemmajrelease].nil? = facts[:operatingsystemrelease].split('.')[0] else = facts[:operatingsystemmajrelease] end end os = "#{facts[:operatingsystem].downcase}-#{}-#{facts[:hardwaremodel]}" next unless os.start_with? RspecPuppetFacts.spec_facts_os_filter if RspecPuppetFacts.spec_facts_os_filter facts.merge! RspecPuppetFacts.common_facts os_facts_hash[os] = RspecPuppetFacts.with_custom_facts(os, facts) end return stringify_keys(os_facts_hash) if RSpec.configuration.facterdb_string_keys os_facts_hash end |
#stringify_keys(hash) ⇒ Object
176 177 178 |
# File 'lib/rspec-puppet-facts.rb', line 176 def stringify_keys(hash) Hash[hash.collect { |k,v| [k.to_s, v.is_a?(Hash) ? stringify_keys(v) : v] }] end |