#! /usr/bin/env ruby
require 'spec_helper'

require 'puppet/reports'

describe Puppet::Reports do
  it "should instance-load report types" do
    expect(Puppet::Reports.instance_loader(:report)).to be_instance_of(Puppet::Util::Autoload)
  end

  it "should have a method for registering report types" do
    expect(Puppet::Reports).to respond_to(:register_report)
  end

  it "should have a method for retrieving report types by name" do
    expect(Puppet::Reports).to respond_to(:report)
  end

  it "should provide a method for returning documentation for all reports" do
    Puppet::Reports.expects(:loaded_instances).with(:report).returns([:one, :two])
    one = mock 'one', :doc => "onedoc"
    two = mock 'two', :doc => "twodoc"
    Puppet::Reports.expects(:report).with(:one).returns(one)
    Puppet::Reports.expects(:report).with(:two).returns(two)

    doc = Puppet::Reports.reportdocs
    expect(doc.include?("onedoc")).to be_truthy
    expect(doc.include?("twodoc")).to be_truthy
  end
end


describe Puppet::Reports, " when loading report types" do
  it "should use the instance loader to retrieve report types" do
    Puppet::Reports.expects(:loaded_instance).with(:report, :myreporttype)
    Puppet::Reports.report(:myreporttype)
  end
end

describe Puppet::Reports, " when registering report types" do
  it "should evaluate the supplied block as code for a module" do
    Puppet::Reports.expects(:genmodule).returns(Module.new)
    Puppet::Reports.register_report(:testing) { }
  end

  it "should allow a successful report to be reloaded" do
    Puppet::Reports.register_report(:testing) { }
    Puppet::Reports.register_report(:testing) { }
  end

  it "should allow a failed report to be reloaded and show the correct exception both times" do
    expect { Puppet::Reports.register_report(:testing) { raise TypeError, 'failed report' } }.to raise_error(TypeError)
    expect { Puppet::Reports.register_report(:testing) { raise TypeError, 'failed report' } }.to raise_error(TypeError)
  end

  it "should extend the report type with the Puppet::Util::Docs module" do
    mod = stub 'module', :define_method => true

    Puppet::Reports.expects(:genmodule).with { |name, options, block| options[:extend] == Puppet::Util::Docs }.returns(mod)
    Puppet::Reports.register_report(:testing) { }
  end

  it "should define a :report_name method in the module that returns the name of the report" do
    mod = mock 'module'
    mod.expects(:define_method).with(:report_name)

    Puppet::Reports.expects(:genmodule).returns(mod)
    Puppet::Reports.register_report(:testing) { }
  end
end