# Shared functions
def filter_nodes(hash, name, value)
  hash.select do |it|
    it[name] == value
  end
end

def nodes_to_hash(hash, name, value)
  result = {}
  hash.each do |element|
    result[element[name]] = element[value]
  end
  result
end

def ipsort (ips)
  require 'rubygems'
  require 'ipaddr'
  ips.sort { |a,b| IPAddr.new( a ) <=> IPAddr.new( b ) }
end

def test_ubuntu_and_centos(manifest, force_manifest = false)
  # check if task is present in the task list
  unless force_manifest or Noop.manifest_present? manifest
    # puts "Manifest '#{manifest}' is not enabled on the node '#{Noop.hostname}'. Skipping tests."
    return
  end

  # set manifest file
  before(:all) do
    Noop.manifest = manifest
  end

  let(:os) do
    os = facts[:operatingsystem]
    os = os.downcase if os
    os
  end

  shared_examples 'compile' do
    it do
      File.stubs(:exists?).with('/var/lib/astute/ceph/ceph').returns(true)
      File.stubs(:exists?).with('/var/lib/astute/mongodb/mongodb.key').returns(true)
      File.stubs(:exists?).with('/var/lib/astute/mongodb/mongodb.key').returns(true)
      File.stubs(:exists?).with('/var/lib/astute/ceph/ceph').returns(true)
      File.stubs(:exists?).with('/var/lib/astute/nova/nova').returns(true)
      File.stubs(:exists?).with('/var/lib/astute/ceph/ceph').returns(true)
      File.stubs(:exists?).returns(false)
      should compile.with_all_deps
    end
  end

  shared_examples 'save_files_list' do
    it 'should save the list of file resources' do
      catalog = subject
      catalog = subject.call if subject.is_a? Proc
      file_resources = {}
      catalog.resources.each do |resource|
        next unless resource.type == 'File'
        next unless %w(present file directory).include? resource[:ensure] or not resource[:ensure]

        if resource[:source]
          content = resource[:source]
        elsif resource[:content]
          content = 'TEMPLATE'
        else
          content = nil
        end
        next unless content
        file_resources[resource[:path]] = content
      end
      if file_resources.any?
        Noop.save_file_resources_list file_resources, manifest, os
      end
    end
  end

  shared_examples 'save_packages_list' do
    it 'should save the list of file resources' do
      catalog = subject
      catalog = subject.call if subject.is_a? Proc
      package_resources = {}
      catalog.resources.each do |resource|
        next unless resource.type == 'Package'
        next if %w(absent purged).include? resource[:ensure] or not resource[:ensure]
        package_resources[resource[:name]] = resource[:ensure]
      end
      if package_resources.any?
        Noop.save_package_resources_list package_resources, manifest, os
      end
    end
  end

  shared_examples 'debug' do
    it 'shows catalog contents' do
      Noop.show_catalog subject
    end
  end

  shared_examples 'generate' do
    it 'shows catalog contents' do
      Noop.catalog_to_spec subject
    end
  end

  shared_examples 'status' do
    it 'shows status' do
      puts <<-eos
      =============================================
      OS:       #{os}
      YAML:     #{Noop.astute_yaml_base}
      Manifest: #{Noop.manifest}
      Node:     #{Noop.fqdn}
      Role:     #{Noop.hiera 'role'}
      =============================================
      eos
    end
  end

  #######################################
  # Testing on different operating systems

  if Noop.test_ubuntu?
    context 'on Ubuntu platforms' do
      let(:facts) { Noop.ubuntu_facts }

      it_behaves_like 'compile'

      it_behaves_like 'status' if ENV['SPEC_SHOW_STATUS']
      it_behaves_like 'debug' if ENV['SPEC_CATALOG_DEBUG']
      it_behaves_like 'generate' if ENV['SPEC_SPEC_GENERATE']
      it_behaves_like 'save_files_list' if ENV['SPEC_SAVE_FILE_RESOURCES']
      it_behaves_like 'save_packages_list'if ENV['SPEC_SAVE_PACKAGE_RESOURCES']

      begin
        it_behaves_like 'catalog'
      rescue ArgumentError
        true
      end

      at_exit { RSpec::Puppet::Coverage.report! } if ENV['SPEC_COVERAGE']
    end
  end

  if Noop.test_centos?
    context 'on CentOS platforms' do
      let(:facts) { Noop.centos_facts }

      it_behaves_like 'compile'

      it_behaves_like 'status' if ENV['SPEC_SHOW_STATUS']
      it_behaves_like 'debug' if ENV['SPEC_CATALOG_DEBUG']
      it_behaves_like 'generate' if ENV['SPEC_SPEC_GENERATE']
      it_behaves_like 'save_files_list' if ENV['SPEC_SAVE_FILE_RESOURCES']
      it_behaves_like 'save_packages_list'if ENV['SPEC_SAVE_PACKAGE_RESOURCES']

      begin
        it_behaves_like 'catalog'
      rescue ArgumentError
        true
      end

      at_exit { RSpec::Puppet::Coverage.report! } if ENV['SPEC_COVERAGE']
    end
  end

end