require 'spec_helper'

class ClassMixedWithDSLInstallUtils
  include Beaker::DSL::Wrappers
  include Beaker::DSL::Helpers
  include Beaker::DSL::Structure
  include Beaker::DSL::Roles
  include Beaker::DSL::Patterns
  include Beaker::DSL::InstallUtils

  def logger
    @logger ||= RSpec::Mocks::Double.new('logger').as_null_object
  end
end

describe ClassMixedWithDSLInstallUtils do
  let(:metadata)      { @metadata ||= {} }
  let(:presets)       { Beaker::Options::Presets.new }
  let(:opts)          { presets.presets.merge(presets.env_vars) }
  let(:basic_hosts)   do
    make_hosts({ pe_ver: '3.0',
                 platform: 'linux',
                 roles: ['agent'], }, 4)
  end
  let(:hosts) do
    basic_hosts[0][:roles] = %w[master database dashboard]
    basic_hosts[1][:platform] = 'windows'
    basic_hosts[2][:platform] = 'osx-10.9-x86_64'
    basic_hosts[3][:platform] = 'eos'
    basic_hosts
  end
  let(:hosts_sorted)  { [hosts[1], hosts[0], hosts[2], hosts[3]] }
  let(:winhost)       do
    make_host('winhost', { platform: 'windows',
                           pe_ver: '3.0',
                           working_dir: '/tmp',
                           is_cygwin: true, })
  end
  let(:winhost_non_cygwin) do
    make_host('winhost_non_cygwin', { platform: 'windows',
                                      pe_ver: '3.0',
                                      working_dir: '/tmp',
                                      is_cygwin: 'false', })
  end
  let(:machost) do
    make_host('machost', { platform: 'osx-10.9-x86_64',
                           pe_ver: '3.0',
                           working_dir: '/tmp', })
  end
  let(:unixhost) do
    make_host('unixhost', { platform: 'linux',
                            pe_ver: '3.0',
                            working_dir: '/tmp',
                            dist: 'puppet-enterprise-3.1.0-rc0-230-g36c9e5c-debian-7-i386', })
  end
  let(:eoshost) do
    make_host('eoshost', { platform: 'eos',
                           pe_ver: '3.0',
                           working_dir: '/tmp',
                           dist: 'puppet-enterprise-3.7.1-rc0-78-gffc958f-eos-4-i386', })
  end

  describe '#configure_defaults_on' do
    it 'can set foss defaults' do
      expect(subject).to receive(:add_foss_defaults_on).exactly(hosts.length).times
      subject.configure_defaults_on(hosts, 'foss')
    end

    it 'can set aio defaults' do
      expect(subject).to receive(:add_aio_defaults_on).exactly(hosts.length).times
      subject.configure_defaults_on(hosts, 'aio')
    end

    it 'can set pe defaults' do
      expect(subject).to receive(:add_pe_defaults_on).exactly(hosts.length).times
      subject.configure_defaults_on(hosts, 'pe')
    end

    it 'can remove old defaults ands replace with new' do
      # fake the results of calling configure_pe_defaults_on
      hosts.each do |host|
        host['type'] = 'pe'
      end
      expect(subject).to receive(:remove_pe_defaults_on).exactly(hosts.length).times
      expect(subject).to receive(:add_foss_defaults_on).exactly(hosts.length).times
      subject.configure_defaults_on(hosts, 'foss')
    end
  end

  describe '#configure_type_defaults_on' do
    it 'can set foss defaults for foss type' do
      hosts.each do |host|
        host['type'] = 'foss'
      end
      expect(subject).to receive(:add_foss_defaults_on).exactly(hosts.length).times
      subject.configure_type_defaults_on(hosts)
    end

    it 'adds aio defaults to foss hosts when they have an aio foss puppet version' do
      hosts.each do |host|
        host[:pe_ver] = nil
        host[:version] = nil
        host['type'] = 'foss'
        host['version'] = '4.0'
      end
      expect(subject).to receive(:add_foss_defaults_on).exactly(hosts.length).times
      expect(subject).to receive(:add_aio_defaults_on).exactly(hosts.length).times
      subject.configure_type_defaults_on(hosts)
    end

    it 'adds aio defaults to foss hosts when they have type foss-aio' do
      hosts.each do |host|
        host[:pe_ver] = nil
        host[:version] = nil
        host['type'] = 'foss-aio'
      end
      expect(subject).to receive(:add_foss_defaults_on).exactly(hosts.length).times
      expect(subject).to receive(:add_aio_defaults_on).exactly(hosts.length).times
      subject.configure_type_defaults_on(hosts)
    end

    it 'can set aio defaults for aio type (backwards compatability)' do
      hosts.each do |host|
        host[:pe_ver] = nil
        host[:version] = nil
        host['type'] = 'aio'
      end
      expect(subject).to receive(:add_aio_defaults_on).exactly(hosts.length).times
      subject.configure_type_defaults_on(hosts)
    end

    it 'can set pe defaults for pe type' do
      hosts.each do |host|
        host['type'] = 'pe'
      end
      expect(subject).to receive(:add_pe_defaults_on).exactly(hosts.length).times
      subject.configure_type_defaults_on(hosts)
    end

    it 'adds aio defaults to pe hosts when they an aio pe version' do
      hosts.each do |host|
        host['type'] = 'pe'
        host['pe_ver'] = '4.0'
      end
      expect(subject).to receive(:add_pe_defaults_on).exactly(hosts.length).times
      expect(subject).to receive(:add_aio_defaults_on).exactly(hosts.length).times
      subject.configure_type_defaults_on(hosts)
    end
  end

  describe '#puppetserver_version_on' do
    it 'returns the tag on a released version' do
      result = object_double(Beaker::Result.new({}, 'puppetserver --version'), stdout: 'puppetserver version: 6.13.0',
                                                                               exit_code: 0)
      expect(subject).to receive(:on).with(hosts.first, 'puppetserver --version',
                                           accept_all_exit_codes: true).and_return(result)
      expect(subject.puppetserver_version_on(hosts.first)).to eq('6.13.0')
    end

    it 'returns the tag on a nightly version' do
      result = object_double(Beaker::Result.new({}, 'puppetserver --version'),
                             stdout: 'puppetserver version: 7.0.0.SNAPSHOT.2020.10.14T0512', exit_code: 0)
      expect(subject).to receive(:on).with(hosts.first, 'puppetserver --version',
                                           accept_all_exit_codes: true).and_return(result)
      expect(subject.puppetserver_version_on(hosts.first)).to eq('7.0.0')
    end
  end

  describe '#puppet_collection_for' do
    it 'raises an error when given an invalid package' do
      expect do
        subject.puppet_collection_for(:foo, '5.5.4')
      end.to raise_error(RuntimeError, /package must be one of puppet_agent, puppet, puppetserver/)
    end

    context 'when the :puppet_agent package is passed in' do
      context 'given a valid version' do
        {
          '1.10.14' => 'pc1',
          '1.10.x' => 'pc1',
          '5.3.1' => 'puppet5',
          '5.3.x' => 'puppet5',
          '5.99.0' => 'puppet6',
          '6.1.99-foo' => 'puppet6',
          '6.99.99' => 'puppet7',
          '7.0.0' => 'puppet7',
        }.each do |version, collection|
          it "returns collection '#{collection}' for version '#{version}'" do
            expect(subject.puppet_collection_for(:puppet_agent, version)).to eq(collection)
          end
        end
      end

      it "returns the default, latest puppet collection given the version 'latest'" do
        expect(subject.puppet_collection_for(:puppet_agent, 'latest')).to eq('puppet')
      end

      context 'given an invalid version' do
        [nil, '', '0.1.0', '3.8.1', '', 'not-semver', 'not.semver.either'].each do |version|
          it "returns a nil collection value for version '#{version}'" do
            expect(subject.puppet_collection_for(:puppet_agent, version)).to be_nil
          end
        end
      end
    end

    context 'when the :puppet package is passed-in' do
      context 'given a valid version' do
        {
          '4.9.0' => 'pc1',
          '4.10.x' => 'pc1',
          '5.3.1' => 'puppet5',
          '5.3.x' => 'puppet5',
          '5.99.0' => 'puppet6',
          '6.1.99-foo' => 'puppet6',
          '6.99.99' => 'puppet7',
          '7.0.0' => 'puppet7',
        }.each do |version, collection|
          it "returns collection '#{collection}' for version '#{version}'" do
            expect(subject.puppet_collection_for(:puppet, version)).to eq(collection)
          end
        end
      end

      it "returns the default, latest puppet collection given the version 'latest'" do
        expect(subject.puppet_collection_for(:puppet, 'latest')).to eq('puppet')
      end

      context 'given an invalid version' do
        [nil, '', '0.1.0', '3.8.1', '', 'not-semver', 'not.semver.either'].each do |version|
          it "returns a nil collection value for version '#{version}'" do
            expect(subject.puppet_collection_for(:puppet, version)).to be_nil
          end
        end
      end
    end

    context 'when the :puppetserver package is passed in' do
      context 'given a valid version' do
        {
          '2.0.0' => 'pc1',
          '2.0.x' => 'pc1',
          '5.3.1' => 'puppet5',
          '5.3.x' => 'puppet5',
          '5.99.0' => 'puppet6',
          '6.1.99-foo' => 'puppet6',
          '6.99.99' => 'puppet7',
          '7.0.0' => 'puppet7',
        }.each do |version, collection|
          it "returns collection '#{collection}' for version '#{version}'" do
            expect(subject.puppet_collection_for(:puppetserver, version)).to eq(collection)
          end
        end
      end

      it "returns the default, latest puppet collection given the version 'latest'" do
        expect(subject.puppet_collection_for(:puppetserver, 'latest')).to eq('puppet')
      end

      context 'given an invalid version' do
        [nil, '', '0.1.0', '3.8.1', '', 'not-semver', 'not.semver.either'].each do |version|
          it "returns a nil collection value for version '#{version}'" do
            expect(subject.puppet_collection_for(:puppetserver, version)).to be_nil
          end
        end
      end
    end
  end
end