spec_helper_acceptance_local.rb

# frozen_string_literal: true

require 'puppet_litmus'
require 'puppet_litmus/util'
require 'singleton'

class Helper
  include Singleton
  include PuppetLitmus
end

def install_pwsh
  script_folder = File.expand_path(File.join(File.dirname(__FILE__), 'fixtures/scripts/install_pwsh'))
  script_path = case os[:family]
                when 'ubuntu'
                  File.join(script_folder, "ubuntu_#{os[:release].to_f}.sh")
                when 'redhat'
                  File.join(script_folder, 'rhel.sh')
                when 'debian'
                  File.join(script_folder, "debian_#{os[:release].to_i}.sh")
                when 'darwin'
                  File.join(script_folder, 'darwin.sh')
                when 'windows'
                  File.join(script_folder, 'windows.ps1')
                end
  Helper.instance.bolt_run_script(script_path)
end

def uninstall_pwsh
  command = case os[:family]
            when 'ubuntu', 'debian'
              'apt-get remove powershell -y'
            when 'redhat'
              'yum remove powershell -y'
            when 'darwin'
              'brew cask uninstall powershell'
            when 'windows'
              PuppetLitmus::Util.interpolate_powershell('Get-Command pwsh | ForEach-Object { Remove-Item -Path (Split-Path -Parent $_.Path) -Recurse -Force }')
            end
  Helper.instance.run_shell(command)
end

def pwsh_installed?
  Helper.instance.run_shell('pwsh -v', expect_failures: true).exit_code == 0
end

def cleanup_files
  absent_files_manifest = if os[:family] == 'windows'
                            'file{["c:/services.txt","c:/process.txt","c:/try_success.txt","c:/catch_shouldntexist.txt","c:/try_shouldntexist.txt","c:/catch_success.txt"]: ensure => absent }'
                          else
                            'file{["/tmp/services.txt","/tmp/process.txt","/tmp/try_success.txt","/tmp/catch_shouldntexist.txt","/tmp/try_shouldntexist.txt","/tmp/catch_success.txt"]: ensure => absent }'
                          end
  Helper.instance.apply_manifest(absent_files_manifest, catch_failures: true)
end

def relative_folder(relative_path)
  expanded_path = File.expand_path(File.join(File.dirname(__FILE__), relative_path))
  Dir.open(expanded_path) if File.exist?(expanded_path)
end

def remove_pwshlib
  uninstall_command = 'puppet module uninstall puppetlabs/pwshlib --force'
  uninstall_command += " --modulepath #{relative_folder('fixtures/modules').path}" if ENV['TARGET_HOST'] == 'localhost'
  Helper.instance.run_shell(uninstall_command, expect_failures: true) do |result|
    raise 'Failed to uninstall puppetlabs/pwshlib' unless result.stderr.include?("Module 'puppetlabs-pwshlib' is not installed") || result.exit_code == 0
  end
end

def install_pwshlib
  install_command = 'puppet module install puppetlabs/pwshlib'
  install_command += " --modulepath #{relative_folder('fixtures/modules').path}" if ENV['TARGET_HOST'] == 'localhost'
  Helper.instance.run_shell(install_command)
end

def localhost_windows?
  os[:family] == 'windows' && ENV['TARGET_HOST'] == 'localhost'
end

RSpec.configure do |c|
  c.before :suite do
    install_pwshlib
    cleanup_files
  end
end