require "helper"
require "inspec/resource"
require "inspec/resources/service"
require "hashie"

describe "Inspec::Resources::Service" do
  let(:runlevels) { { 0 => false, 1 => false, 2 => true, 3 => true, 4 => true, 5 => true, 6 => false } }

  # windows
  it "verify service parsing" do
    resource = MockLoader.new(:windows).load_resource("service", "dhcp")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "windows"
    _(resource.name).must_equal "dhcp"
    _(resource.description).must_equal "DHCP Client"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.startmode). must_equal "Auto"
    _(resource.startname). must_equal "LocalSystem"
    _(resource.params).must_equal params
  end

  # ubuntu 14.04 with upstart
  it "verify ubuntu service parsing" do
    resource = MockLoader.new(:ubuntu1404).load_resource("service", "ssh")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "upstart"
    _(resource.name).must_equal "ssh"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify ubuntu service parsing with default upstart_service" do
    resource = MockLoader.new(:ubuntu1404).load_resource("upstart_service", "ssh")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "upstart"
    _(resource.name).must_equal "ssh"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.UnitFileState).must_be_nil
  end

  # ubuntu 15.04 with systemd
  it "verify ubuntu service parsing" do
    resource = MockLoader.new(:ubuntu1504).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.SubState).must_equal "running"
  end

  it "verify ubuntu service parsing with default systemd_service" do
    resource = MockLoader.new(:ubuntu1504).load_resource("systemd_service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # linux mint 17 with upstart
  it "verify mint service parsing" do
    resource = MockLoader.new(:mint17).load_resource("service", "ssh")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "upstart"
    _(resource.name).must_equal "ssh"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify mint service parsing with default upstart_service" do
    resource = MockLoader.new(:mint17).load_resource("upstart_service", "ssh")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "upstart"
    _(resource.name).must_equal "ssh"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.UnitFileState).must_be_nil
  end

  # mint 18 with systemd
  it "verify mint service parsing" do
    resource = MockLoader.new(:mint18).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.SubState).must_equal "running"
  end

  it "verify mint service parsing with default systemd_service" do
    resource = MockLoader.new(:mint18).load_resource("systemd_service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # Amazon Linux
  it "verify amazon linux service parsing" do
    resource = MockLoader.new(:amazon).load_resource("service", "ssh")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "upstart"
    _(resource.name).must_equal "ssh"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.UnitFileState).must_be_nil
  end

  # Amazon Linux 2
  it "verify amazon linux 2 service parsing" do
    resource = MockLoader.new(:amazon2).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # centos 6 with sysv
  it "verify centos 6 service parsing" do
    resource = MockLoader.new(:centos6).load_resource("service", "sshd")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "sysv"
    _(resource.name).must_equal "sshd"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.SubState).must_be_nil
  end

  it "verify centos 6 service parsing with default sysv_service" do
    resource = MockLoader.new(:centos6).load_resource("sysv_service", "sshd")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "sysv"
    _(resource.name).must_equal "sshd"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # centos 7 with systemd
  it "verify centos 7 service parsing" do
    resource = MockLoader.new(:centos7).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify centos 7 service parsing with systemd_service and service_ctl override" do
    resource = MockLoader.new(:centos7).load_resource("systemd_service", "sshd", "/path/to/systemctl")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "UnitFileState" => "enabled", "SubState" => "running" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify centos 7 service parsing with static loaded service" do
    resource = MockLoader.new(:centos7).load_resource("service", "dbus")
    params = Hashie::Mash.new({ "Description" => "D-Bus System Message Bus", "Id" => "dbus.service", "LoadState" => "loaded", "Names" => "messagebus.service dbus.service", "SubState" => "running", "UnitFileState" => "static" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "dbus.service"
    _(resource.description).must_equal "D-Bus System Message Bus"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.UnitFileState).must_equal "static"
  end

  # cloudlinux 7 with systemd
  it "verify cloudlinux 7 service parsing" do
    resource = MockLoader.new(:cloudlinux).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify cloudlinux 7 service parsing with systemd_service and service_ctl override" do
    resource = MockLoader.new(:cloudlinux).load_resource("systemd_service", "sshd", "/path/to/systemctl")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "UnitFileState" => "enabled", "SubState" => "running" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify cloudlinux 7 service parsing with static loaded service" do
    resource = MockLoader.new(:cloudlinux).load_resource("service", "dbus")
    params = Hashie::Mash.new({ "Description" => "D-Bus System Message Bus", "Id" => "dbus.service", "LoadState" => "loaded", "Names" => "messagebus.service dbus.service", "SubState" => "running", "UnitFileState" => "static" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "dbus.service"
    _(resource.description).must_equal "D-Bus System Message Bus"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
    _(resource.params.UnitFileState).must_equal "static"
  end

  # freebsd
  it "verify freebsd10 service parsing" do
    resource = MockLoader.new(:freebsd10).load_resource("service", "sendmail")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "bsd-init"
    _(resource.name).must_equal "sendmail"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify freebsd10 service parsing with default bsd_service" do
    resource = MockLoader.new(:freebsd10).load_resource("bsd_service", "sendmail")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "bsd-init"
    _(resource.name).must_equal "sendmail"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # arch linux with systemd
  it "verify arch linux service parsing" do
    resource = MockLoader.new(:arch).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # coreos linux with systemd
  it "verify coreos linux service parsing" do
    resource = MockLoader.new(:coreos).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # debian 7 with systemv
  it "verify debian 7 service parsing" do
    resource = MockLoader.new(:debian7).load_resource("service", "sshd")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "sysv"
    _(resource.name).must_equal "sshd"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # debian 8 with systemd
  it "verify debian 8 service parsing" do
    resource = MockLoader.new(:debian8).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # debian 10 with systemd
  it "verify debian 10 service parsing" do
    resource = MockLoader.new(:debian10).load_resource("service", "sshd")
    params = Hashie::Mash.new({ "ActiveState" => "active", "Description" => "OpenSSH server daemon", "Id" => "sshd.service", "LoadState" => "loaded", "Names" => "sshd.service", "SubState" => "running", "UnitFileState" => "enabled" })
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "sshd.service"
    _(resource.description).must_equal "OpenSSH server daemon"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # debian 8 with systemd but no service file
  it "gets the correct service info when the `.service` file is missing" do
    resource = MockLoader.new(:debian8).load_resource("service", "apache2")
    params = Hashie::Mash.new(
      "ActiveState"   => "active",
      "Description"   => "LSB: Apache2 web server",
      "Id"            => "apache2.service",
      "LoadState"     => "loaded",
      "Names"         => "apache2.service",
      "SubState"      => "running",
      "UnitFileState" => ""
    )
    _(resource.type).must_equal "systemd"
    _(resource.name).must_equal "apache2.service"
    _(resource.description).must_equal "LSB: Apache2 web server"
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # macos test
  it "verify mac osx service parsing" do
    resource = MockLoader.new(:osx104).load_resource("service", "ssh")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "darwin"
    _(resource.name).must_equal "org.openbsd.ssh-agent"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  it "verify mac osx service parsing with not-running service" do
    resource = MockLoader.new(:osx104).load_resource("service", "FilesystemUI")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "darwin"
    _(resource.name).must_equal "com.apple.FilesystemUI"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal false
    _(resource.params).must_equal params
  end

  it "verify mac osx service parsing with default launchd_service" do
    resource = MockLoader.new(:osx104).load_resource("launchd_service", "ssh")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "darwin"
    _(resource.name).must_equal "org.openbsd.ssh-agent"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # wrlinux
  it "verify wrlinux service parsing" do
    resource = MockLoader.new(:wrlinux).load_resource("service", "sshd")
    params = Hashie::Mash.new({})
    _(resource.type).must_equal "sysv"
    _(resource.name).must_equal "sshd"
    _(resource.description).must_be_nil
    _(resource.installed?).must_equal true
    _(resource.enabled?).must_equal true
    _(resource.running?).must_equal true
    _(resource.params).must_equal params
  end

  # unknown OS
  it "verify service handling on unsupported os" do
    resource = MockLoader.new(:undefined).load_resource("service", "dhcp")
    params = Hashie::Mash.new({})
    _(resource.installed?).must_equal false
    _(resource.description).must_be_nil
    _(resource.params).must_equal params
  end

  # runlevel detection
  describe "runlevels on centos 6 (system V)" do
    let(:service) { MockLoader.new(:centos6).load_resource("service", "sshd") }

    it "grabs all runlevels" do
      _(service.runlevels.keys).must_equal [0, 1, 2, 3, 4, 5, 6]
    end

    it "grabs runlevels via filter nil" do
      _(service.runlevels(nil).keys).must_equal [0, 1, 2, 3, 4, 5, 6]
    end

    it "grabs runlevels by number" do
      _(service.runlevels(3).keys).must_equal [3]
    end

    it "grabs runlevels by multiple numbers" do
      _(service.runlevels(3, 4, 8).keys).must_equal [3, 4]
    end

    it "grabs runlevels via regex" do
      _(service.runlevels(/[5-9]/).keys).must_equal [5, 6]
    end

    it "checks enabled true if all services are enabled" do
      _(service.runlevels(2, 4).enabled?).must_equal true
    end

    it "checks enabled false if some services are not enabled" do
      _(service.runlevels(1, 4).enabled?).must_equal false
    end

    it "checks disabled true if all services are disabled" do
      _(service.runlevels(0, 1).disabled?).must_equal true
    end

    it "checks disabled false if some services are not disabled" do
      _(service.runlevels(0, 4).enabled?).must_equal false
    end
  end
end