require_relative '../../spec_helper'

require 'wright/provider/package/yum'

describe Wright::Provider::Package::Yum do
  def rpm_q(pkg_name)
    version_format = '%{VERSION}-%{RELEASE}'
    %W(rpm -q #{pkg_name} --qf #{version_format})
  end

  def package_provider(pkg_name, args = {})
    properties = { name: pkg_name,
                   version: args[:version],
                   options: args[:options] }
    pkg_resource = OpenStruct.new(properties)
    Wright::Provider::Package::Yum.new(pkg_resource)
  end

  def yum(action, pkg_name, args = {})
    version = args[:version].nil? ? '' : "-#{args[:version]}"
    ['yum', '-y', *args[:options], action.to_s, pkg_name + version]
  end

  before :each do
    yum_dir = File.join(File.dirname(__FILE__), 'yum')
    @fake_capture3 = FakeCapture3.new(yum_dir)
  end

  describe '#installed_versions' do
    it 'should return the installed package version via brew info' do
      pkg_name = 'zsh'
      pkg_versions = ['5.0.2-7.el7_1.1']
      pkg_provider = package_provider(pkg_name)
      rpm_q_cmd = rpm_q(pkg_name)

      @fake_capture3.expect(rpm_q_cmd, 'rpm_-q_zsh')
      @fake_capture3.stub do
        pkg_provider.installed_versions.must_equal pkg_versions
      end
    end

    it 'should return an empty array for uninstalled packages' do
      pkg_name = 'httpd'
      pkg_versions = []
      pkg_provider = package_provider(pkg_name)
      rpm_q_cmd = rpm_q(pkg_name)

      @fake_capture3.expect(rpm_q_cmd, 'rpm_-q_httpd')
      @fake_capture3.stub do
        pkg_provider.installed_versions.must_equal pkg_versions
      end
    end
  end

  describe '#install_package' do
    it 'should install packages' do
      pkg_name = 'nano'
      pkg_provider = package_provider(pkg_name)
      yum_cmd = yum(:install, pkg_name)

      @fake_capture3.expect(yum_cmd)
      @fake_capture3.stub do
        pkg_provider.send(:install_package)
      end
    end

    it 'should install specific package versions' do
      pkg_name = 'mc'
      pkg_version = '4.8.7-8.el7'
      pkg_provider = package_provider(pkg_name, version: pkg_version)
      yum_cmd = yum(:install, pkg_name, version: pkg_version)

      @fake_capture3.expect(yum_cmd)
      @fake_capture3.stub do
        pkg_provider.send(:install_package)
      end
    end

    it 'should pass options to the package manager' do
      pkg_name = 'postfix'
      pkg_options = '--enablerepo=centosplus'
      pkg_provider = package_provider(pkg_name, options: pkg_options)
      yum_cmd = yum(:install, pkg_name, options: pkg_options)

      @fake_capture3.expect(yum_cmd, 'yum_install_with_options')
      @fake_capture3.stub do
        pkg_provider.send(:install_package)
      end
    end

    it 'should raise exceptions for unknown packages' do
      pkg_name = 'not-a-real-package'
      pkg_provider = package_provider(pkg_name)
      yum_cmd = yum(:install, pkg_name)

      @fake_capture3.expect(yum_cmd)
      @fake_capture3.stub do
        e = -> { pkg_provider.send(:install_package) }.must_raise RuntimeError
        wright_error = "cannot install package '#{pkg_name}'"
        yum_error = 'Error: Nothing to do'
        e.message.must_equal %(#{wright_error}: "#{yum_error}")
      end
    end
  end

  describe '#remove_package' do
    it 'should remove packages' do
      pkg_name = 'screen'
      pkg_provider = package_provider(pkg_name)
      yum_cmd = yum(:remove, pkg_name)

      @fake_capture3.expect(yum_cmd)
      @fake_capture3.stub do
        pkg_provider.send(:remove_package)
      end
    end

    it 'should pass options to the package manager' do
      pkg_name = 'vim'
      pkg_options = ['--noplugins']
      pkg_provider = package_provider(pkg_name, options: pkg_options)
      yum_cmd = yum(:remove, pkg_name, options: pkg_options)

      @fake_capture3.expect(yum_cmd, 'yum_remove_with_options')
      @fake_capture3.stub do
        pkg_provider.send(:remove_package)
      end
    end
  end
end