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_loose_requirement(version) ⇒ Optional[Gem::Requirement]
private
Construct the loose facter version requirement.
-
.facter_version_to_loose_requirement_string(version) ⇒ String
private
Construct the facter version requirement string.
-
.facter_version_to_strict_requirement(version) ⇒ Gem::Requirement
private
Construct the strict facter version requirement.
-
.facter_version_to_strict_requirement_string(version) ⇒ String
private
Construct the strict facter version requirement string.
-
.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 private
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:
292 293 294 295 296 297 |
# File 'lib/rspec-puppet-facts.rb', line 292 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
271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 |
# File 'lib/rspec-puppet-facts.rb', line 271 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
246 247 248 |
# File 'lib/rspec-puppet-facts.rb', line 246 def self.custom_facts @custom_facts end |
.facter_version_for_puppet_version(puppet_version) ⇒ Object
406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 |
# File 'lib/rspec-puppet-facts.rb', line 406 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_loose_requirement(version) ⇒ Optional[Gem::Requirement]
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.
Construct the loose facter version requirement
385 386 387 388 |
# File 'lib/rspec-puppet-facts.rb', line 385 def self.facter_version_to_loose_requirement(version) string = facter_version_to_loose_requirement_string(version) Gem::Requirement.new(string) if string end |
.facter_version_to_loose_requirement_string(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.
Construct the facter version requirement string
393 394 395 396 397 398 399 400 401 402 403 404 |
# File 'lib/rspec-puppet-facts.rb', line 393 def self.facter_version_to_loose_requirement_string(version) if (m = /\A(?<major>[0-9]+)\.(?<minor>[0-9]+)(?:\.(?<patch>[0-9]+))?\Z/.match(version)) # Interpret 3.1 as < 3.2 and 3.2.1 as < 3.3 "< #{m[:major]}.#{m[:minor].to_i + 1}" elsif /\A[0-9]+\Z/.match?(version) # Interpret 3 as < 4 "< #{version.to_i + 1}" else # This would be the same as the strict requirement nil end end |
.facter_version_to_strict_requirement(version) ⇒ Gem::Requirement
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.
Construct the strict facter version requirement
366 367 368 |
# File 'lib/rspec-puppet-facts.rb', line 366 def self.facter_version_to_strict_requirement(version) Gem::Requirement.new(facter_version_to_strict_requirement_string(version)) end |
.facter_version_to_strict_requirement_string(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.
Construct the strict facter version requirement string
373 374 375 376 377 378 379 380 |
# File 'lib/rspec-puppet-facts.rb', line 373 def self.facter_version_to_strict_requirement_string(version) if /\A[0-9]+(\.[0-9]+)*\Z/.match?(version) # Interpret 3 as ~> 3.0 "~> #{version}.0" else version end 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:
304 305 306 307 308 309 |
# File 'lib/rspec-puppet-facts.rb', line 304 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
318 319 320 321 322 323 |
# File 'lib/rspec-puppet-facts.rb', line 318 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.
330 331 332 333 334 335 336 337 |
# File 'lib/rspec-puppet-facts.rb', line 330 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
342 343 344 |
# File 'lib/rspec-puppet-facts.rb', line 342 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.
212 213 214 215 216 |
# File 'lib/rspec-puppet-facts.rb', line 212 def self.register_custom_fact(name, value, ) @custom_facts ||= {} name = RSpec.configuration.facterdb_string_keys ? name.to_s : name.to_sym @custom_facts[name] = {: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
357 358 359 360 361 |
# File 'lib/rspec-puppet-facts.rb', line 357 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
256 257 258 |
# File 'lib/rspec-puppet-facts.rb', line 256 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.
263 264 265 |
# File 'lib/rspec-puppet-facts.rb', line 263 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
349 350 351 |
# File 'lib/rspec-puppet-facts.rb', line 349 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.
224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 |
# File 'lib/rspec-puppet-facts.rb', line 224 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) value = fact[:value].respond_to?(:call) ? fact[:value].call(os, facts) : fact[:value] # if merge_facts passed, merge supplied facts into facts hash if fact[:options][:merge_facts] facts.deep_merge!({name => value}) else facts[name] = value end 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.
197 198 199 200 201 202 |
# File 'lib/rspec-puppet-facts.rb', line 197 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.
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# File 'lib/rspec-puppet-facts.rb', line 32 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.
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 175 176 177 178 179 |
# File 'lib/rspec-puppet-facts.rb', line 59 def on_supported_os_implementation(opts = {}) unless /\A\d+\.\d+(?:\.\d+)*\z/.match?((facterversion = opts[:facterversion])) 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])}/" case os_sup['operatingsystem'] when /BSD/i hardwaremodel = 'amd64' when /Solaris/i hardwaremodel = 'i86pc' when /AIX/i hardwaremodel = '/^IBM,.*/' os_release_filter = if =~ /\A(\d+)\.(\d+)\Z/ "/^#{$~[1]}#{$~[2]}00-/" else "/^#{}-/" end when /Windows/i hardwaremodel = /^[12]\./.match?(facterversion) ? '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 when /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 strict_requirement = RspecPuppetFacts::facter_version_to_strict_requirement(facterversion) loose_requirement = RspecPuppetFacts::facter_version_to_loose_requirement(facterversion) received_facts = [] # 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| versions = FacterDB.get_facts(filter_spec).to_h { |facts| [Gem::Version.new(facts[:facterversion]), facts] } version, facts = versions.select { |v, _f| strict_requirement =~ v }.max_by { |v, _f| v } unless version version, facts = versions.select { |v, _f| loose_requirement =~ v }.max_by { |v, _f| v } if loose_requirement next unless version if RspecPuppetFacts.spec_facts_strict? raise ArgumentError, "No facts were found in the FacterDB for Facter v#{facterversion} on #{filter_spec}, aborting" end RspecPuppetFacts.warning "No facts were found in the FacterDB for Facter v#{facterversion} on #{filter_spec}, using v#{version} instead" end received_facts << facts end 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 # Todo: refactor the whole block to rely on structured facts and use legacy facts as fallback 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' elsif facts.dig(:os, 'release', 'major') = facts[:os]['release']['major'] elsif facts.dig(:os, 'distro', 'release', 'major') = facts[:os]['distro']['release']['major'] 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
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.
182 183 184 |
# File 'lib/rspec-puppet-facts.rb', line 182 def stringify_keys(hash) hash.to_h { |k, v| [k.to_s, v.is_a?(Hash) ? stringify_keys(v) : v] } end |