#
#  Author:: Jennifer Marie Howard-Brown <jm.howardbrown@gmail.com>
#  License:: Apache License, Version 2.0
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.
#

require "spec_helper"

describe Ohai::System, "Linux Block Device Plugin" do
  DISKS = {
    "sda" => {
      "size" => "7814037168",
      "removable" => "0",
      "model" => "WDC WD4000F9YZ-0",
      "rev" => "1A01",
      "state" => "running",
      "timeout" => "30",
      "queue_depth" => "1",
      "vendor" => "ATA",
    },
    "nvme0n1" => {
      "size" => "500118192",
      "removable" => "0",
      "model" => "KXG50ZNV256G TOSHIBA",
      "state" => "live",
      "firmware_rev" => "AAGA4103",
    },
    "dm-0" => {
      "size" => "7806976",
      "removable" => "0",
    },
  }.freeze

  def file_double(value)
    tmp_double = double
    expect(tmp_double).to receive(:read_nonblock).and_return(value)
    tmp_double
  end

  before do
    @plugin = get_plugin("linux/block_device")
    allow(@plugin).to receive(:collect_os).and_return(:linux)

    allow(File).to receive(:exist?).with("/sys/block").and_return(true)
    allow(@plugin).to receive(:dir_glob).with("/sys/block/*") do
      DISKS.collect { |disk, _files| "/sys/block/#{disk}" }
    end

    DISKS.each do |disk, checks|
      checks.each do |check, value|
        allow(File).to receive(:open).with(Regexp.new("#{disk}.*#{check}")).and_yield(file_double(value))
      end

      allow(File).to receive(:exist?).with(Regexp.new(disk)) do |arg|
        filepath = arg.split("/")
        checks[filepath.last].nil? ? false : true
      end

      allow(File).to receive(:basename) do |arg|
        arg.split("/").last
      end
    end
  end

  it "collects all relevant data from disks" do
    @plugin.run
    DISKS.each do |disk, checks|
      expect(@plugin[:block_device][disk.to_sym]).to include(checks)
    end
  end
end
