# NOTE: This suite starts off with 2016.2.0 as the installation workflow
# changed in that release.
shared_examples 'provider/provisioner/pe_bootstrap/2016x' do |provider, options|
  if options[:boxes].empty?
    raise ArgumentError,
      "Box files must be downloaded for provider: #{provider}. Try: rake acceptance:setup"
  end

  include_context 'acceptance'

  let(:extra_env) do
    vars = options[:env_vars].dup
    vars['PE_BUILD_DOWNLOAD_ROOT'] = options[:archive_path]

    vars
  end

  before(:each) do
    # The skelton sets up a Vagrantfile which expects the OS under test to be
    # available as `box`.
    environment.skeleton('2016x_acceptance')
    options[:boxes].each do |box|
      name = File.basename(box).split('-').first
      assert_execute('vagrant', 'box', 'add', name, box)
    end
  end

  after(:each) do
    # Ensure any VMs that survived tests are cleaned up.
    assert_execute('vagrant', 'destroy', '--force', log: false)
  end

  # TODO: Refactor into a shared example so tha this testcase can be run on
  # multiple versions.
  context 'when installing PE 2016.2.x' do

    it 'provisions masters with pe_bootstrap and agents with pe_agent' do
      status('Test: pe_bootstrap master install')
      assert_execute('vagrant', 'up', "--provider=#{provider}", 'pe-20162-master')

      status('Test: pe_bootstrap master running after install')
      result = execute('vagrant', 'ssh',
        'pe-20162-master',
        '-c', 'sudo /opt/puppetlabs/bin/puppet status --terminus=rest')
      expect(result).to exit_with(0)
      expect(result.stdout).to match('"is_alive": true')

      status('Test: pe_agent install')
      result = assert_execute('vagrant', 'up', "--provider=#{provider}", 'pe-20162-agent')

      status('Test: pe_agent signed cert during install')
      result = execute('vagrant', 'ssh',
        'pe-20162-master',
        '-c', 'sudo /opt/puppetlabs/bin/puppet cert list pe-20162-agent.pe-bootstrap.vlan')
      expect(result).to exit_with(0)

      status('Test: pe_agent cert purged when vm destroyed')
      result = assert_execute('vagrant', 'destroy', '-f', 'pe-20162-agent')
      result = execute('vagrant', 'ssh',
        'pe-20162-master',
        '-c', 'sudo /opt/puppetlabs/bin/puppet cert list pe-20162-agent.pe-bootstrap.vlan')
      expect(result.stderr).to match(/Could not find a certificate/)
    end
  end

  context 'when installing PE 2016.latest' do
    it 'provisions masters with pe_bootstrap and agents with pe_agent' do
      status('Test: pe_bootstrap master install')
      assert_execute('vagrant', 'up', "--provider=#{provider}", 'pe-2016latest-master')

      status('Test: pe_bootstrap master running after install')
      result = execute('vagrant', 'ssh',
        'pe-2016latest-master',
        '-c', 'sudo /opt/puppetlabs/bin/puppet status --terminus=rest')
      expect(result).to exit_with(0)
      expect(result.stdout).to match('"is_alive": true')

      status('Test: pe_agent install')
      result = assert_execute('vagrant', 'up', "--provider=#{provider}", 'pe-2016latest-agent')

      status('Test: pe_agent signed cert during install')
      result = execute('vagrant', 'ssh',
        'pe-2016latest-master',
        '-c', 'sudo /opt/puppetlabs/bin/puppet cert list pe-2016latest-agent.pe-bootstrap.vlan')
      expect(result).to exit_with(0)

      status('Test: pe_agent cert purged when vm destroyed')
      result = assert_execute('vagrant', 'destroy', '-f', 'pe-2016latest-agent')
      result = execute('vagrant', 'ssh',
        'pe-2016latest-master',
        '-c', 'sudo /opt/puppetlabs/bin/puppet cert list pe-2016latest-agent.pe-bootstrap.vlan')
      expect(result.stderr).to match(/Could not find a certificate/)
    end
  end

end