-
Notifications
You must be signed in to change notification settings - Fork 53
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
(maint) uncouple from the puppet_litmus gem
- Loading branch information
Showing
23 changed files
with
347 additions
and
249 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,119 @@ | ||
| # frozen_string_literal: true | ||
|
|
||
| require 'yaml' | ||
| require 'delegate' | ||
| require 'json' | ||
|
|
||
| # simple bolt inventory manipulator | ||
| class InventoryHelper < SimpleDelegator | ||
| def initialize(location) | ||
| @location = location | ||
| super(refresh) | ||
| end | ||
|
|
||
| # Load inventory from location in YAML format | ||
| # or generate a default structure | ||
| # | ||
| # @return [Hash] | ||
| def refresh | ||
| x = JSON.parse(YAML.load_file(@location).to_json) if File.file?(@location) | ||
| { 'version' => 2, 'groups' => [] }.merge(x || {}) | ||
| end | ||
|
|
||
| # Save inventory to location in yaml format | ||
| def save | ||
| File.open(@location, 'wb+') { |f| f.write(to_yaml) } | ||
| self | ||
| end | ||
|
|
||
| # Adds a node to a group specified, if group_name exists in inventory hash. | ||
| # | ||
| # @param node [Hash] node to add to the group | ||
| # @param group [String] group of nodes to limit the search for the node_name in | ||
| # @return [Hash] inventory_hash with node added to group if group_name exists in inventory hash. | ||
| def add(node, group) | ||
| # stringify keys | ||
| node = JSON.parse(node.to_json) | ||
| # check if group exists | ||
| if self['groups'].any? { |g| g['name'] == group } | ||
| self['groups'].each do |g| | ||
| g['targets'].push(node) if g['name'] == group | ||
| end | ||
| else | ||
| # add new group | ||
| self['groups'].push({ 'name' => group, 'targets' => [node] }) | ||
| end | ||
|
|
||
| self | ||
| end | ||
|
|
||
| # Lookup a node | ||
| # | ||
| # @param either [String] uri or name of node to find | ||
| # @param uri [String] uri of node to find | ||
| # @param name [String] name of node to find | ||
| # @param group [String] limit search to group | ||
| # @return [Hash] inventory target | ||
| def lookup(either = nil, uri: nil, name: nil, group: nil) | ||
| value = either || uri || name | ||
|
|
||
| keys = [] | ||
| keys << 'uri' if uri || either | ||
| keys << 'name' if name || either | ||
|
|
||
| self['groups'].each do |g| | ||
| next unless (group && group == g['name']) || group.nil? | ||
| g['targets'].each do |t| | ||
| return t if keys.include? t.key(value) | ||
| end | ||
| end | ||
|
|
||
| raise "Failed to lookup target #{value}" | ||
| end | ||
|
|
||
| # Remove node | ||
| # | ||
| # @param node [Hash] | ||
| # @return [Hash] inventory_hash with node of node_name removed. | ||
| def remove(node) | ||
| # stringify keys | ||
| node = JSON.parse(node.to_json) | ||
| self['groups'].map! do |g| | ||
| g['targets'].reject! { |target| target == node } | ||
| g | ||
| end | ||
|
|
||
| self | ||
| end | ||
|
|
||
| class << self | ||
| attr_accessor :instances | ||
|
|
||
| def open(location = nil) | ||
| # Inventory location is an optional task parameter. | ||
| location = location.nil? ? Dir.pwd : location | ||
| location = if File.directory?(location) | ||
| # DEPRECATED: puppet_litmus <= 1.4.0 support | ||
| if Gem.loaded_specs['puppet_litmus'] && Gem.loaded_specs['puppet_litmus'].version <= Gem::Version.new('1.4.0') | ||
| File.join(location, 'spec', 'fixtures', 'litmus_inventory.yaml') | ||
| else | ||
| File.join(location, 'inventory.yaml') | ||
| end | ||
| else | ||
| location | ||
| end | ||
|
|
||
| raise "directory for storing inventory does not exist: #{location}" unless File.exist?(File.dirname(location)) | ||
|
|
||
| @instances ||= {} | ||
| @instances[location] = new(location) unless @instances.key? location | ||
| @instances[location] | ||
| end | ||
| end | ||
|
|
||
| attr_reader :location | ||
|
|
||
| protected | ||
|
|
||
| attr_writer :location | ||
| end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,19 @@ | ||
| require 'fileutils' | ||
|
|
||
| RSpec.configure do |rspec| | ||
| rspec.expect_with :rspec do |c| | ||
| c.max_formatted_output_length = nil | ||
| end | ||
| end | ||
|
|
||
| RSpec.shared_context('with tmpdir') do | ||
| let(:tmpdir) { @tmpdir } # rubocop:disable RSpec/InstanceVariable | ||
|
|
||
| around(:each) do |example| | ||
| Dir.mktmpdir('rspec-provision_test') do |t| | ||
| FileUtils.mkdir_p(File.join(t, 'spec', 'fixtures')) | ||
| @tmpdir = t | ||
| example.run | ||
| end | ||
| end | ||
| end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.