Module: PuppetLitmus

Extended by:
InventoryManipulation
Includes:
BoltSpec::Run, InventoryManipulation, PuppetHelpers, RakeHelper
Defined in:
lib/puppet_litmus.rb,
lib/puppet_litmus.rb,
lib/puppet_litmus/version.rb,
lib/puppet_litmus/inventory_manipulation.rb,
lib/puppet_litmus/spec_helper_acceptance.rb

Overview

Helper methods for testing puppet content

Defined Under Namespace

Modules: InventoryManipulation, PuppetHelpers, RakeHelper, Util

Constant Summary collapse

VERSION = 
'2.8.0'

Class Method Summary collapse

Class Method Details

.configure! ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
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
 
# File 'lib/puppet_litmus/spec_helper_acceptance.rb', line 7

def self.configure!
  require 'serverspec'

  RSpec.configure do |config|
    config.include PuppetLitmus
    config.extend PuppetLitmus
  end

  if ENV['TARGET_HOST'].nil? || ENV['TARGET_HOST'] == 'localhost'
    puts 'Running tests against this machine !'
    if Gem.win_platform?
      set :backend, :cmd
    else
      set :backend, :exec
    end
  else
    # load inventory
    inventory_hash = inventory_hash_from_inventory_file
    node_config = config_from_node(inventory_hash, ENV.fetch('TARGET_HOST', nil))

    if target_in_group(inventory_hash, ENV.fetch('TARGET_HOST', nil), 'docker_nodes')
      host = ENV.fetch('TARGET_HOST', nil)
      set :backend, :dockercli
      set :docker_container, host
    elsif target_in_group(inventory_hash, ENV.fetch('TARGET_HOST', nil), 'lxd_nodes')
      host = ENV.fetch('TARGET_HOST', nil)
      set :backend, :lxd
      set :login_shell, true
      set :lxd_remote, node_config.dig('lxd', 'remote') unless node_config.dig('lxd', 'remote').nil?
      set :lxd_instance, host
    elsif target_in_group(inventory_hash, ENV.fetch('TARGET_HOST', nil), 'ssh_nodes')
      set :backend, :ssh
      options = Net::SSH::Config.for(host)
      options[:user] = node_config.dig('ssh', 'user') unless node_config.dig('ssh', 'user').nil?
      options[:port] = node_config.dig('ssh', 'port') unless node_config.dig('ssh', 'port').nil?
      options[:keys] = node_config.dig('ssh', 'private-key') unless node_config.dig('ssh', 'private-key').nil?
      options[:password] = node_config.dig('ssh', 'password') unless node_config.dig('ssh', 'password').nil?
      run_as = node_config.dig('ssh', 'run-as') unless node_config.dig('ssh', 'run-as').nil?
      # Support both net-ssh 4 and 5.
      # rubocop:disable Metrics/BlockNesting
      options[:verify_host_key] = if node_config.dig('ssh', 'host-key-check').nil?
                                    # Fall back to SSH behavior. This variable will only be set in net-ssh 5.3+.
                                    if @strict_host_key_checking.nil? || @strict_host_key_checking
                                      Net::SSH::Verifiers::Always.new
                                    else
                                      # SSH's behavior with StrictHostKeyChecking=no: adds new keys to known_hosts.
                                      # If known_hosts points to /dev/null, then equivalent to :never where it
                                      # accepts any key beacuse they're all new.
                                      Net::SSH::Verifiers::AcceptNewOrLocalTunnel.new
                                    end
                                  elsif node_config.dig('ssh', 'host-key-check')
                                    if defined?(Net::SSH::Verifiers::Always)
                                      Net::SSH::Verifiers::Always.new
                                    else
                                      Net::SSH::Verifiers::Secure.new
                                    end
                                  elsif defined?(Net::SSH::Verifiers::Never)
                                    Net::SSH::Verifiers::Never.new
                                  else
                                    Net::SSH::Verifiers::Null.new
                                  end
      # rubocop:enable Metrics/BlockNesting
      host = if ENV['TARGET_HOST'].include?(':')
               ENV['TARGET_HOST'].split(':').first
             else
               ENV['TARGET_HOST']
             end
      set :host,        options[:host_name] || host
      set :ssh_options, options
      set :request_pty, false
      set :sudo_password, options[:password] if run_as
      puts "Running tests as #{run_as}" if run_as
    elsif target_in_group(inventory_hash, ENV.fetch('TARGET_HOST', nil), 'winrm_nodes')
      require 'winrm'

      set :backend, :winrm
      set :os, family: 'windows'
      user = node_config.dig('winrm', 'user') unless node_config.dig('winrm', 'user').nil?
      pass = node_config.dig('winrm', 'password') unless node_config.dig('winrm', 'password').nil?
      endpoint = URI("http://#{ENV.fetch('TARGET_HOST', nil)}")
      endpoint.port = 5985 if endpoint.port == 80
      endpoint.path = '/wsman'

      opts = {
        user:,
        password: pass,
        endpoint:,
        operation_timeout: 300
      }

      winrm = WinRM::Connection.new opts
      Specinfra.configuration.winrm = winrm
    end
  end
end