module Beaker module DSL module InstallUtils module PEClientTools def install_pe_client_tools_on(hosts, opts = {}) product = 'pe-client-tools' required_keys = [:puppet_collection, :pe_client_tools_sha, :pe_client_tools_version] unless required_keys.all? { |opt| opts.keys.include?(opt) && opts[opt]} raise ArgumentError, "The keys #{required_keys.to_s} are required in the opts hash" end urls = { :dev_builds_url => "http://builds.delivery.puppetlabs.net", } opts = urls.merge(opts) block_on hosts do |host| variant, version, arch, codename = host['platform'].to_array package_name = '' case host['platform'] when /win/ package_name << product release_path = "#{opts[:dev_builds_url]}/#{product}/#{ opts[:pe_client_tools_sha] }/artifacts/#{variant}" package_name << "-#{opts[:pe_client_tools_version]}-x#{arch}.msi" generic_install_msi_on(host, File.join(release_path, package_name), {}, {:debug => true}) when /osx/ release_path = "#{opts[:dev_builds_url]}/#{product}/#{ opts[:pe_client_tools_sha] }/artifacts/apple/#{version}/#{opts[:puppet_collection]}/#{arch}" package_base = product.dup package_base << "-#{opts[:pe_client_tools_version]}" package_name = package_base.dup package_name << '-1' if opts[:pe_client_tools_version] installer = package_name + '-installer.pkg' package_name << ".#{variant}#{version}.dmg" copy_dir_local = File.join('tmp', 'repo_configs') fetch_http_file(release_path, package_name, copy_dir_local) scp_to host, File.join(copy_dir_local, package_name), host.external_copy_base host.generic_install_dmg(package_name, package_base, installer) else install_dev_repos_on(product, host, opts[:pe_client_tools_sha], '/tmp/repo_configs',{:dev_builds_url => opts[:dev_builds_url]}) host.install_package('pe-client-tools') end end end # Taken from puppet acceptance lib # Install development repos def install_dev_repos_on(package, host, sha, repo_configs_dir, opts={}) platform = host['platform'] =~ /^(debian|ubuntu)/ ? host['platform'].with_version_codename : host['platform'] platform_configs_dir = File.join(repo_configs_dir, platform) case platform when /^(fedora|el|centos|sles)-(\d+)-(.+)$/ variant = (($1 == 'centos') ? 'el' : $1) fedora_prefix = ((variant == 'fedora') ? 'f' : '') version = $2 arch = $3 pattern = 'pl-%s-%s-%s-%s%s-%s.repo' repo_filename = pattern % [ package, sha, variant, fedora_prefix, version, arch ] repo = fetch_http_file( "%s/%s/%s/repo_configs/rpm/" % [opts[:dev_builds_url],package, sha], repo_filename, platform_configs_dir ) if /sles/i.match(platform) scp_to(host, repo, '/etc/zypp/repos.d/') else scp_to(host, repo, '/etc/yum.repos.d/') end when /^(debian|ubuntu)-([^-]+)-(.+)$/ variant = $1 version = $2 arch = $3 list = fetch_http_file( "%s/%s/%s/repo_configs/deb/" % [opts[:dev_builds_url],package, sha], "pl-%s-%s-%s.list" % [package, sha, version], platform_configs_dir ) scp_to host, list, '/etc/apt/sources.list.d' on host, 'apt-get update' else host.logger.notify("No repository installation step for #{platform} yet...") end end end end end end