require "spec_helper"

module Beaker
  module Options

    describe Parser do
      let(:parser)           { Parser.new }
      let(:opts_path)        { File.join(File.expand_path(File.dirname(__FILE__)), "data", "opts.txt") }
      let(:hosts_path)       { File.join(File.expand_path(File.dirname(__FILE__)), "data", "hosts.cfg") }
      let(:badyaml_path)     { File.join(File.expand_path(File.dirname(__FILE__)), "data", "badyaml.cfg") }
      let(:home)             { ENV['HOME'] }

      it "supports usage function" do
        expect{parser.usage}.to_not raise_error
      end

      describe 'parse_git_repos' do

        it "transforms arguments of <PROJECT_NAME>/<REF> to <GIT_BASE_URL>/<lowercased_project_name>#<REF>" do
          opts = ["PUPPET/3.1"]
          expect(parser.parse_git_repos(opts)).to be === ["#{parser.repo}/puppet.git#3.1"]
        end

        it "recognizes PROJECT_NAMEs of PUPPET, FACTER, HIERA, and HIERA-PUPPET" do
          projects = [ ['puppet', 'my_branch', 'PUPPET/my_branch'],
                       ['facter', 'my_branch', 'FACTER/my_branch'],
                       ['hiera', 'my_branch', 'HIERA/my_branch'],
                       ['hiera-puppet', 'my_branch', 'HIERA-PUPPET/my_branch'] ]
          projects.each do |project, ref, input|
            expect(parser.parse_git_repos([input])).to be === ["#{parser.repo}/#{project}.git##{ref}"]
          end
        end
      end

      describe 'split_arg' do

        it "can split comma separated list into an array" do
          arg = "file1,file2,file3"
          expect(parser.split_arg(arg)).to be === ["file1", "file2", "file3"]
        end

        it "can use an existing Array as an acceptable argument" do
          arg = ["file1", "file2", "file3"]
          expect(parser.split_arg(arg)).to be === ["file1", "file2", "file3"]
        end

        it "can generate an array from a single value" do
          arg = "i'mjustastring"
          expect(parser.split_arg(arg)).to be === ["i'mjustastring"]
        end
      end

      context 'testing path traversing', :use_fakefs => true do

        let(:test_dir) { 'tmp/tests' }
        let(:rb_test)  { File.expand_path(test_dir + '/my_ruby_file.rb')    }
        let(:pl_test)  { File.expand_path(test_dir + '/my_perl_file.pl')    }
        let(:sh_test)  { File.expand_path(test_dir + '/my_shell_file.sh')   }
        let(:rb_other)  { File.expand_path(test_dir + '/other/my_other_ruby_file.rb')   }

        it 'only collects ruby files as test files' do
          files = [ rb_test, pl_test, sh_test, rb_other ]
          create_files( files )
          expect(parser.file_list([File.expand_path(test_dir)])).to be === [rb_test, rb_other]
        end

        it 'raises an error when no ruby files are found' do
          files = [ pl_test, sh_test ]
          create_files( files )
          expect{parser.file_list([File.expand_path(test_dir)])}.to raise_error(ArgumentError)
        end

        it 'raises an error when no paths are specified for searching' do
          @files = ''
          expect{parser.file_list('')}.to raise_error(ArgumentError)
        end
      end

      context 'combining split_arg and file_list maintain test file ordering', :use_fakefs => true do
        let(:test_dir) { 'tmp/tests' }
        let(:other_test_dir) {'tmp/tests2' }

        before :each do
          files = [
            '00_EnvSetup.rb', '035_StopFirewall.rb', '05_HieraSetup.rb',
            '01_TestSetup.rb', '03_PuppetMasterSanity.rb',
            '06_InstallModules.rb','02_PuppetUserAndGroup.rb',
            '04_ValidateSignCert.rb', '07_InstallCACerts.rb'              ]

          @lone_file = '08_foss.rb'

          @fileset1 = files.shuffle.map {|file| test_dir + '/' + file }
          @fileset2 = files.shuffle.map {|file| other_test_dir + '/' + file }

          @sorted_expanded_fileset1 = @fileset1.map {|f| File.expand_path(f) }.sort
          @sorted_expanded_fileset2 = @fileset2.map {|f| File.expand_path(f) }.sort

          create_files( @fileset1 )
          create_files( @fileset2 )
          create_files( [@lone_file] )
        end

        it "when provided a file followed by dir, runs the file first" do
          arg = "#{@lone_file},#{test_dir}"
          output = parser.file_list( parser.split_arg( arg ))
          expect( output ).to be === [ @lone_file, @sorted_expanded_fileset1 ].flatten
        end

        it "when provided a dir followed by a file, runs the file last" do
          arg = "#{test_dir},#{@lone_file}"
          output = parser.file_list( parser.split_arg( arg ))
          expect( output ).to be === [ @sorted_expanded_fileset1, @lone_file ].flatten
        end

        it "correctly orders files in a directory" do
          arg = "#{test_dir}"
          output = parser.file_list( parser.split_arg( arg ))
          expect( output ).to be === @sorted_expanded_fileset1
        end

        it "when provided two directories orders each directory separately" do
          arg = "#{test_dir}/,#{other_test_dir}/"
          output = parser.file_list( parser.split_arg( arg ))
          expect( output ).to be === @sorted_expanded_fileset1 + @sorted_expanded_fileset2
        end
      end

      describe 'check_yaml_file' do
        it "raises error on improperly formatted yaml file" do
          FakeFS.deactivate!
          expect{parser.check_yaml_file(badyaml_path)}.to raise_error(ArgumentError)
        end

        it "raises an error when a yaml file is missing" do
          FakeFS.deactivate!
          expect{parser.check_yaml_file("not a path")}.to raise_error(ArgumentError)
        end
      end

      describe 'parse_args' do
        before { FakeFS.deactivate! }

        it 'pulls the args into key called :command_line' do
          my_args = [ '--log-level', 'debug', '-h', hosts_path]
          expect(parser.parse_args( my_args )[:command_line]).to include(my_args.join(' '))
        end

        it "can correctly combine arguments from different sources" do
          build_url = 'http://my.build.url/'
          type = 'git'
          log_level = 'debug'

          old_build_url = ENV["BUILD_URL"]
          ENV["BUILD_URL"] = build_url

          args = ["-h", hosts_path, "--log-level", log_level, "--type", type, "--install", "PUPPET/1.0,HIERA/hello"]
          output = parser.parse_args( args )
          expect( output[:hosts_file] ).to be == hosts_path
          expect( output[:jenkins_build_url] ).to be == build_url
          expect( output[:install] ).to include( 'git://github.com/puppetlabs/hiera.git#hello' )

          ENV["BUILD_URL"] = old_build_url
        end

        it "ensures that fail-mode is one of fast/slow" do
          args = ["-h", hosts_path, "--log-level", "debug", "--fail-mode", "nope"]
          expect{parser.parse_args(args)}.to raise_error(ArgumentError)
        end

        it "ensures that type is one of pe/git" do
          args = ["-h", hosts_path, "--log-level", "debug", "--type", "unkowns"]
          expect{parser.parse_args(args)}.to raise_error(ArgumentError)
        end
      end

      context "set_default_host!" do

        let(:roles) { @roles || [ [ "master", "agent", "database"], ["agent"]] }
        let(:node1) { { :node1 => { :roles => roles[0]}} }
        let(:node2) { { :node2 => { :roles => roles[1]}} }
        let(:hosts) { node1.merge(node2) }

        it "does nothing if the default host is already set" do
          @roles = [ ["master"], ["agent", "default"] ]
          parser.set_default_host!(hosts)
          expect( hosts[:node1][:roles].include?('default') ).to be === false
          expect( hosts[:node2][:roles].include?('default') ).to be === true
        end

        it "makes the master default" do
          @roles = [ ["master"], ["agent"] ]
          parser.set_default_host!(hosts)
          expect( hosts[:node1][:roles].include?('default') ).to be === true
          expect( hosts[:node2][:roles].include?('default') ).to be === false
        end

        it "makes a single node default" do
          @roles = [ ["master", "database", "dashboard", "agent"] ]
          parser.set_default_host!(node1)
          expect( hosts[:node1][:roles].include?('default') ).to be === true
        end

        it "makes a single non-master node default" do
          @roles = [ ["database", "dashboard", "agent"] ]
          parser.set_default_host!(node1)
          expect( hosts[:node1][:roles].include?('default') ).to be === true
        end

        it "raises an error if two nodes are defined as default" do
          @roles = [ ["master", "default"], ["default"] ]
          expect{ parser.set_default_host!(hosts) }.to raise_error(ArgumentError)
        end

      end

      describe "normalize_args" do
        let(:hosts) do
          {
            'HOSTS' => {
              :master => {
                :roles => ["master","agent","arbitrary_role"],
              },
              :agent => {
                :roles => ["agent","default","other_abitrary_role"],
              },
            }
          }
        end

        def fake_hosts_file_for_platform(hosts, platform)
          hosts['HOSTS'].values.each { |h| h[:platform] = platform }
          filename = "hosts_file_#{platform}"
          File.open(filename, "w") do |file|
            YAML.dump(hosts, file)
          end
          filename
        end

        shared_examples_for(:a_platform_supporting_only_agents) do |platform,type|
          let(:args) { ["--type", type] }

          it "restricts #{platform} hosts to agent for #{type}" do
            hosts_file = fake_hosts_file_for_platform(hosts, platform)
            args << "--hosts" << hosts_file
            expect { parser.parse_args(args) }.to raise_error(ArgumentError, /#{platform}.*may not have roles 'master', 'dashboard', or 'database'/)
          end
        end

        context "for pe" do
          it_should_behave_like(:a_platform_supporting_only_agents, 'solaris-version-arch', 'pe')
          it_should_behave_like(:a_platform_supporting_only_agents, 'windows-version-arch', 'pe')
          it_should_behave_like(:a_platform_supporting_only_agents, 'el-4-arch', 'pe')
        end

        context "for foss" do
          it_should_behave_like(:a_platform_supporting_only_agents, 'windows-version-arch', 'git')
          it_should_behave_like(:a_platform_supporting_only_agents, 'el-4-arch', 'git')

          it "allows master role for solaris" do
            hosts_file = fake_hosts_file_for_platform(hosts, 'solaris-version-arch')
            args = ["--type", "git", "--hosts", hosts_file]
            options_hash = parser.parse_args(args)
            expect(options_hash[:HOSTS][:master][:platform]).to match(/solaris/)
            expect(options_hash[:HOSTS][:master][:roles]).to include('master')
            expect(options_hash[:type]).to eq('git')
          end
        end
      end
    end
  end
end