require 'spec_helper'

class ClassMixedWithDSLRoles
  include Beaker::DSL::Roles
  include Beaker::DSL::Outcomes
end

describe ClassMixedWithDSLRoles do
  let(:hosts)      { @hosts || {} }
  let(:options)    { @options || {} }
  let(:agent1)     { make_host('agent1',     { :roles => ['agent'] }) }
  let(:agent2)     { make_host('agent2',     { :roles => ['agent'] }) }
  let(:a_and_dash) { make_host('a_and_dash', { :roles => %w[agent dashboard] }) }
  let(:custom)     { make_host('custom',     { :roles => ['custom_role'] }) }
  let(:db)         { make_host('db',         { :roles => ['database'] }) }
  let(:master)     { make_host('master',     { :roles => %w[master agent] }) }
  let(:default)    { make_host('default',    { :roles => ['default'] }) }
  let(:monolith)   { make_host('monolith',   { :roles => %w[agent dashboard database master custom_role] }) }

  describe '#agents' do
    it 'returns an array of hosts that are agents' do
      @hosts = [agent1, agent2, master]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject.agents).to be == [agent1, agent2, master]
    end

    it 'and an empty array when none match' do
      @hosts = [db, custom]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject.agents).to be == []
    end
  end

  describe '#master' do
    it 'returns the master if there is one' do
      @hosts = [master, agent1]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject.master).to be == master
    end

    it 'raises an error if there is more than one master' do
      @hosts = [master, monolith]
      expect(subject).to receive(:hosts).once.and_return(hosts)
      expect { subject.master }.to raise_error Beaker::DSL::FailTest
    end

    it 'returns nil if no master and masterless is set' do
      @options = { :masterless => true }
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject).to receive(:options).and_return(options)
      expect(subject.master).to be_nil
    end
  end

  describe '#dashboard' do
    it 'returns the dashboard if there is one' do
      @hosts = [a_and_dash, agent1]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject.dashboard).to be == a_and_dash
    end

    it 'raises an error if there is more than one dashboard' do
      @hosts = [a_and_dash, monolith]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect { subject.dashboard }.to raise_error Beaker::DSL::FailTest
    end

    it 'and raises an error if there is no dashboard' do
      @hosts = [agent1, agent2, custom]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect { subject.dashboard }.to raise_error Beaker::DSL::FailTest
    end

    it 'returns nil if no dashboard and masterless is set' do
      @options = { :masterless => true }
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject).to receive(:options).and_return(options)
      expect(subject.dashboard).to be_nil
    end
  end

  describe '#database' do
    it 'returns the database if there is one' do
      @hosts = [db, agent1]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject.database).to be == db
    end

    it 'raises an error if there is more than one database' do
      @hosts = [db, monolith]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect { subject.database }.to raise_error Beaker::DSL::FailTest
    end

    it 'and raises an error if there is no database' do
      @hosts = [agent1, agent2, custom]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect { subject.database }.to raise_error Beaker::DSL::FailTest
    end

    it 'returns nil if no database and masterless is set' do
      @options = { :masterless => true }
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject).to receive(:options).and_return(options)
      expect(subject.database).to be_nil
    end
  end

  describe '#not_controller' do
    it 'returns true when a host does not have the roles master/database/dashboard' do
      expect(subject.not_controller(agent1)).to be == true
    end

    it 'returns false when a host has one of the roles master/database/dashboard' do
      expect(subject.not_controller(a_and_dash)).to be == false
    end
  end

  describe '#agent_only' do
    it 'returns true when a host has the single role agent' do
      expect(subject.agent_only(agent1)).to be == true
    end

    it 'returns false when a host has more than a single role' do
      expect(subject.agent_only(a_and_dash)).to be == false
    end

    it 'returns false when a host has the role master' do
      expect(subject.agent_only(master)).to be == false
    end
  end

  describe '#aio_version?' do
    it 'returns false if the host doesn\'t have a :pe_ver or :version' do
      agent1[:pe_ver] = nil
      agent1[:version] = nil
      expect(subject.aio_version?(agent1)).to be === false
    end

    it 'returns false if :version < 4.0 and pe_ver is nil, type foss' do
      agent1[:pe_ver] = nil
      agent1[:version] = '3.8'
      agent1[:type] = 'foss'
      expect(subject.aio_version?(agent1)).to be === false
    end

    it 'returns false if the host :pe_ver is set < 4.0' do
      agent1[:pe_ver] = '3.8'
      expect(subject.aio_version?(agent1)).to be === false
    end

    it 'returns false if the host :version is set < 4.0' do
      agent1[:version] = '3.8'
      expect(subject.aio_version?(agent1)).to be === false
    end

    it 'returns true if the host :pe_ver is 4.0' do
      agent1[:pe_ver] = '4.0'
      expect(subject.aio_version?(agent1)).to be === true
    end

    it 'returns true if the host :version is 4.0' do
      agent1[:version] = '4.0'
      expect(subject.aio_version?(agent1)).to be === true
    end

    it 'returns true if the host :pe_ver is 2015.5' do
      agent1[:pe_ver] = '2015.5'
      expect(subject.aio_version?(agent1)).to be === true
    end

    it 'returns true if the host has role aio' do
      agent1[:roles] = agent1[:roles] | ['aio']
      expect(subject.aio_version?(agent1)).to be === true
    end

    it 'returns true if the host is type aio' do
      agent1[:type] = 'aio'
      expect(subject.aio_version?(agent1)).to be === true
    end

    it 'returns true if the host is type aio-foss' do
      agent1[:type] = 'aio-foss'
      expect(subject.aio_version?(agent1)).to be === true
    end

    it 'returns true if the host is type foss-aio' do
      agent1[:type] = 'aio-foss'
      expect(subject.aio_version?(agent1)).to be === true
    end

    it 'can take an empty string for pe_ver' do
      agent1[:pe_ver] = ''
      expect { subject.aio_version?(agent1) }.not_to raise_error
    end

    it 'can take an empty string for FOSS version' do
      agent1[:version] = ''
      expect { subject.aio_version?(agent1) }.not_to raise_error
    end

    context 'truth table-type testing' do
      before do
        @old_pe_ver  = agent1[:pe_ver]
        @old_version = agent1[:version]
        @old_roles   = agent1[:roles]
        @old_type    = agent1[:type]
      end

      after do
        agent1[:pe_ver]   = @old_pe_ver
        agent1[:version]  = @old_version
        agent1[:roles]    = @old_roles
        agent1[:type]     = @old_type
      end

      context 'version values table' do
        # pe_ver, version, answer
        versions_table = [
          [nil, nil, false],
          [nil, '', false],
          [nil,    '3.9', false],
          [nil,    '4.0', true],
          [nil, '2015.1', true],
 \
          ['', nil, false],
          ['', '', false],
          ['',    '3.9', false],
          ['',    '4.0', true],
          ['', '2015.1', true],
 \
          ['3.9', nil, false],
          ['3.9', '', false],
          ['3.9',    '3.9', false],
          ['3.9',    '4.0', false],
          ['3.9', '2015.1', false],
 \
          ['4.0', nil, true],
          ['4.0', '', true],
          ['4.0',    '3.9', true],
          ['4.0',    '4.0', true],
          ['4.0', '2015.1', true],
 \
          ['2015.1', nil, true],
          ['2015.1', '', true],
          ['2015.1',    '3.9', true],
          ['2015.1',    '4.0', true],
          ['2015.1', '2015.1', true],
        ]

        versions_table.each do |answers_row|
          it "acts with values #{answers_row} correctly" do
            agent1[:pe_ver]   = answers_row[0]
            agent1[:version]  = answers_row[1]
            agent1[:roles]    = nil
            agent1[:type]     = nil
            expect(subject.aio_version?(agent1)).to be === answers_row[2]
          end
        end
      end

      context 'roles values table' do
        roles_table = [
          [nil,           false],
          [[],            false],
          [['aio'],       true],
          [['gun'],       false],
          [%w[a b],    false],
          [%w[c aio],  true],
        ]

        roles_table.each do |answers_row|
          it "acts with values #{answers_row} correctly" do
            agent1[:pe_ver]   = nil
            agent1[:version]  = nil
            agent1[:roles]    = answers_row[0]
            agent1[:type]     = nil
            expect(subject.aio_version?(agent1)).to be === answers_row[1]
          end
        end
      end

      context 'type values table' do
        type_table = [
          [nil,         false],
          ['',          false],
          ['cheese',    false],
          ['paionts',   false],
          ['aioch',     false],
          ['chaio',     false],
          ['aio',       true],
          ['aio-',      true],
          ['ew-aio-ji', true],
          ['id-aiot',   false],
        ]

        type_table.each do |answers_row|
          it "acts with values #{answers_row} correctly" do
            agent1[:pe_ver]   = nil
            agent1[:version]  = nil
            agent1[:roles]    = nil
            agent1[:type]     = answers_row[0]
            expect(subject.aio_version?(agent1)).to be === answers_row[1]
          end
        end
      end
    end
  end

  describe '#aio_agent?' do
    it 'returns false if agent_only check doesn\'t pass' do
      agent1[:roles] = %w[agent headless]
      expect(subject.aio_agent?(agent1)).to be === false
    end

    it 'returns false if aio_capable? check doesn\'t pass' do
      agent1[:pe_ver] = '3.8'
      expect(subject.aio_agent?(agent1)).to be === false
    end

    it 'returns true if both checks pass' do
      agent1[:pe_ver] = '4.0'
      expect(subject.aio_agent?(agent1)).to be === true
    end
  end

  describe '#default' do
    it 'returns the default host when one is specified' do
      @hosts = [db, agent1, agent2, default, master]
      expect(subject).to receive(:hosts).once.and_return(hosts)
      expect(subject.default).to be == default
    end

    it 'raises an error if there is more than one default' do
      @hosts = [db, monolith, default, default]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect { subject.default }.to raise_error Beaker::DSL::FailTest
    end

    it 'and raises an error if there is no default' do
      @hosts = [agent1, agent2, custom]
      expect(subject).to receive(:hosts).and_return(hosts)
      expect { subject.default }.to raise_error Beaker::DSL::FailTest
    end

    it 'returns nil if no default and masterless is set' do
      @options = { :masterless => true }
      expect(subject).to receive(:hosts).and_return(hosts)
      expect(subject).to receive(:options).and_return(options)
      expect(subject.default).to be_nil
    end
  end

  describe '#add_role_def' do
    it 'raises an error on unsupported role format "1role"' do
      expect { subject.add_role_def("1role") }.to raise_error ArgumentError
    end

    it 'raises an error on unsupported role format "role_!a"' do
      expect { subject.add_role_def("role_!a") }.to raise_error ArgumentError
    end

    it 'raises an error on unsupported role format "role=="' do
      expect { subject.add_role_def("role==") }.to raise_error ArgumentError
    end

    it 'creates new method for role "role_correct!"' do
      test_role = "role_correct!"
      subject.add_role_def(test_role)
      expect(subject).to respond_to test_role
      subject.class.send(:undef_method, test_role)
    end

    it 'returns a single node for a new method for a role defined in a single node' do
      @hosts = [agent1, agent2, monolith]
      expect(subject).to receive(:hosts).and_return(hosts)
      test_role = "custom_role"
      subject.add_role_def(test_role)
      expect(subject).to respond_to test_role
      expect(subject.send(test_role)).to be == @hosts[2]
      subject.class.send(:undef_method, test_role)
    end

    it 'returns an array of nodes for a new method for a role defined in multiple nodes' do
      @hosts = [agent1, agent2, monolith, custom]
      expect(subject).to receive(:hosts).and_return(hosts)
      test_role = "custom_role"
      subject.add_role_def(test_role)
      expect(subject).to respond_to test_role
      expect(subject.send(test_role)).to be == [@hosts[2], @hosts[3]]
      subject.class.send(:undef_method, test_role)
    end
  end

  describe '#any_hosts_as?' do
    it 'returns true if a host exists, false otherwise' do
      @hosts = [agent1, agent2]
      # expect( subject ).to receive( :hosts ).and_return( hosts )
      expect(subject).to receive(:hosts).twice.and_return(hosts)
      expect(subject.any_hosts_as?("agent")).to be == true
      expect(subject.any_hosts_as?("custom_role")).to be == false
    end
  end
end