path: root/spec/unit/parser/collector.rb

#!/usr/bin/env ruby

require File.dirname(__FILE__) + '/../../spec_helper'

require 'puppet/parser/collector'

describe Puppet::Parser::Collector, "when initializing" do
    before do
        @scope = mock 'scope'
        @resource_type = 'resource_type'
        @form = :exported
        @vquery = mock 'vquery'
        @equery = mock 'equery'

        @collector = Puppet::Parser::Collector.new(@scope, @resource_type, @equery, @vquery, @form)
    end

    it "should require a scope" do
        @collector.scope.should equal(@scope)
    end

    it "should require a resource type" do
        @collector.type.should == 'Resource_type'
    end

    it "should only accept :virtual or :exported as the collector form" do
        proc { @collector = Puppet::Parser::Collector.new(@scope, @resource_type, @vquery, @equery, :other) }.should raise_error(ArgumentError)
    end

    it "should accept an optional virtual query" do
        @collector.vquery.should equal(@vquery)
    end

    it "should accept an optional exported query" do
        @collector.equery.should equal(@equery)
    end

    it "should canonize the type name" do
        @collector = Puppet::Parser::Collector.new(@scope, "resource::type", @equery, @vquery, @form)
        @collector.type.should == "Resource::Type"
    end
end

describe Puppet::Parser::Collector, "when collecting specific virtual resources" do
    before do
        @scope = mock 'scope'
        @resource_type = mock 'resource_type'
        @vquery = mock 'vquery'
        @equery = mock 'equery'

        @collector = Puppet::Parser::Collector.new(@scope, @resource_type, @equery, @vquery, :virtual)
    end

    it "should not fail when it does not find any resources to collect" do
        @collector.resources = ["File[virtual1]", "File[virtual2]"]
        @scope.stubs(:findresource).returns(false)
        proc { @collector.evaluate }.should_not raise_error
    end

    it "should mark matched resources as non-virtual" do
        @collector.resources = ["File[virtual1]", "File[virtual2]"]
        one = mock 'one'
        one.expects(:virtual=).with(false)
        @scope.stubs(:findresource).with("File[virtual1]").returns(one)
        @scope.stubs(:findresource).with("File[virtual2]").returns(nil)
        @collector.evaluate
    end

    it "should return matched resources" do
        @collector.resources = ["File[virtual1]", "File[virtual2]"]
        one = mock 'one'
        one.stubs(:virtual=)
        @scope.stubs(:findresource).with("File[virtual1]").returns(one)
        @scope.stubs(:findresource).with("File[virtual2]").returns(nil)
        @collector.evaluate.should == [one]
    end

    it "should delete itself from the compile's collection list if it has found all of its resources" do
        @collector.resources = ["File[virtual1]"]
        one = mock 'one'
        one.stubs(:virtual=)
        @compiler.expects(:delete_collection).with(@collector)
        @scope.expects(:compiler).returns(@compiler)
        @scope.stubs(:findresource).with("File[virtual1]").returns(one)
        @collector.evaluate
    end

    it "should not delete itself from the compile's collection list if it has unfound resources" do
        @collector.resources = ["File[virtual1]"]
        one = mock 'one'
        one.stubs(:virtual=)
        @compiler.expects(:delete_collection).never
        @scope.stubs(:findresource).with("File[virtual1]").returns(nil)
        @collector.evaluate
    end
end

describe Puppet::Parser::Collector, "when collecting virtual resources" do
    before do
        @scope = mock 'scope'
        @compiler = mock 'compile'
        @scope.stubs(:compiler).returns(@compiler)
        @resource_type = "Mytype"
        @vquery = proc { |res| true }

        @collector = Puppet::Parser::Collector.new(@scope, @resource_type, nil, @vquery, :virtual)
    end

    it "should find all resources matching the vquery" do
        one = stub 'one', :type => "Mytype", :virtual? => true
        two = stub 'two', :type => "Mytype", :virtual? => true

        one.stubs(:virtual=)
        two.stubs(:virtual=)

        @compiler.expects(:resources).returns([one, two])

        @collector.evaluate.should == [one, two]
    end

    it "should mark all matched resources as non-virtual" do
        one = stub 'one', :type => "Mytype", :virtual? => true

        one.expects(:virtual=).with(false)

        @compiler.expects(:resources).returns([one])

        @collector.evaluate
    end

    it "should return matched resources" do
        one = stub 'one', :type => "Mytype", :virtual? => true
        two = stub 'two', :type => "Mytype", :virtual? => true

        one.stubs(:virtual=)
        two.stubs(:virtual=)

        @compiler.expects(:resources).returns([one, two])

        @collector.evaluate.should == [one, two]
    end

    it "should return all resources of the correct type if there is no virtual query" do
        one = stub 'one', :type => "Mytype", :virtual? => true
        two = stub 'two', :type => "Mytype", :virtual? => true

        one.expects(:virtual=).with(false)
        two.expects(:virtual=).with(false)

        @compiler.expects(:resources).returns([one, two])

        @collector = Puppet::Parser::Collector.new(@scope, @resource_type, nil, nil, :virtual)

        @collector.evaluate.should == [one, two]
    end

    it "should not return or mark resources of a different type" do
        one = stub 'one', :type => "Mytype", :virtual? => true
        two = stub 'two', :type => :other, :virtual? => true

        one.expects(:virtual=).with(false)
        two.expects(:virtual=).never

        @compiler.expects(:resources).returns([one, two])

        @collector.evaluate.should == [one]
    end

    it "should not return or mark non-virtual resources" do
        one = stub 'one', :type => "Mytype", :virtual? => false
        two = stub 'two', :type => :other, :virtual? => false

        one.expects(:virtual=).never
        two.expects(:virtual=).never

        @compiler.expects(:resources).returns([one, two])

        @collector.evaluate.should be_false
    end

    it "should not return or mark non-matching resources" do
        @collector.vquery = proc { |res| res.name == :one }

        one = stub 'one', :name => :one, :type => "Mytype", :virtual? => true
        two = stub 'two', :name => :two, :type => "Mytype", :virtual? => true

        one.expects(:virtual=).with(false)
        two.expects(:virtual=).never

        @compiler.expects(:resources).returns([one, two])

        @collector.evaluate.should == [one]
    end
end

describe Puppet::Parser::Collector, "when collecting exported resources" do
    confine Puppet.features.rails? => "Cannot test Rails integration without ActiveRecord"

    before do
        @scope = stub 'scope', :host => "myhost", :debug => nil
        @compiler = mock 'compile'
        @scope.stubs(:compiler).returns(@compiler)
        @resource_type = "Mytype"
        @equery = "test = true"
        @vquery = proc { |r| true }

        Puppet.settings.stubs(:value).with(:storeconfigs).returns true

        @collector = Puppet::Parser::Collector.new(@scope, @resource_type, @equery, @vquery, :exported)
    end

    # Stub most of our interface to Rails.
    def stub_rails(everything = false)
        ActiveRecord::Base.stubs(:connected?).returns(false)
        Puppet::Rails.stubs(:init)
        if everything
            Puppet::Rails::Host.stubs(:find_by_name).returns(nil)
            Puppet::Rails::Resource.stubs(:find).returns([])
        end
    end

    it "should just return false if :storeconfigs is not enabled" do
        Puppet.settings.expects(:value).with(:storeconfigs).returns false
        @collector.evaluate.should be_false
    end

    it "should use initialize the Rails support if ActiveRecord is not connected" do
        @compiler.stubs(:resources).returns([])
        ActiveRecord::Base.expects(:connected?).returns(false)
        Puppet::Rails.expects(:init)
        Puppet::Rails::Host.stubs(:find_by_name).returns(nil)
        Puppet::Rails::Resource.stubs(:find).returns([])

        @collector.evaluate
    end

    it "should return all matching resources from the current compile and mark them non-virtual and non-exported" do
        stub_rails(true)

        one = stub 'one', :type => "Mytype", :virtual? => true, :exported? => true
        two = stub 'two', :type => "Mytype", :virtual? => true, :exported? => true

        one.expects(:exported=).with(false)
        one.expects(:virtual=).with(false)
        two.expects(:exported=).with(false)
        two.expects(:virtual=).with(false)

        @compiler.expects(:resources).returns([one, two])

        @collector.evaluate.should == [one, two]
    end

    it "should mark all returned resources as not virtual" do
        stub_rails(true)

        one = stub 'one', :type => "Mytype", :virtual? => true, :exported? => true

        one.stubs(:exported=)
        one.expects(:virtual=).with(false)

        @compiler.expects(:resources).returns([one])

        @collector.evaluate.should == [one]
    end

    it "should convert all found resources into parser resources" do
        stub_rails()
        Puppet::Rails::Host.stubs(:find_by_name).returns(nil)

        one = stub 'one', :restype => "Mytype", :title => "one", :virtual? => true, :exported? => true
        Puppet::Rails::Resource.stubs(:find).returns([one])

        resource = mock 'resource'
        one.expects(:to_resource).with(@scope).returns(resource)
        resource.stubs(:exported=)
        resource.stubs(:virtual=)

        @compiler.stubs(:resources).returns([])
        @scope.stubs(:findresource).returns(nil)

        @compiler.stubs(:add_resource)

        @collector.evaluate.should == [resource]
    end

    it "should store converted resources in the compile's resource list" do
        stub_rails()
        Puppet::Rails::Host.stubs(:find_by_name).returns(nil)

        one = stub 'one', :restype => "Mytype", :title => "one", :virtual? => true, :exported? => true
        Puppet::Rails::Resource.stubs(:find).returns([one])

        resource = mock 'resource'
        one.expects(:to_resource).with(@scope).returns(resource)
        resource.stubs(:exported=)
        resource.stubs(:virtual=)

        @compiler.stubs(:resources).returns([])
        @scope.stubs(:findresource).returns(nil)

        @compiler.expects(:add_resource).with(@scope, resource)

        @collector.evaluate.should == [resource]
    end

    # This way one host doesn't store another host's resources as exported.
    it "should mark resources collected from the database as not exported" do
        stub_rails()
        Puppet::Rails::Host.stubs(:find_by_name).returns(nil)

        one = stub 'one', :restype => "Mytype", :title => "one", :virtual? => true, :exported? => true
        Puppet::Rails::Resource.stubs(:find).returns([one])

        resource = mock 'resource'
        one.expects(:to_resource).with(@scope).returns(resource)
        resource.expects(:exported=).with(false)
        resource.stubs(:virtual=)

        @compiler.stubs(:resources).returns([])
        @scope.stubs(:findresource).returns(nil)

        @compiler.stubs(:add_resource)

        @collector.evaluate
    end

    it "should fail if an equivalent resource already exists in the compile" do
        stub_rails()
        Puppet::Rails::Host.stubs(:find_by_name).returns(nil)

        rails = stub 'one', :restype => "Mytype", :title => "one", :virtual? => true, :exported? => true, :id => 1, :ref => "yay"
        inmemory = stub 'one', :type => "Mytype", :virtual? => true, :exported? => true, :rails_id => 2

        Puppet::Rails::Resource.stubs(:find).returns([rails])

        resource = mock 'resource'

        @compiler.stubs(:resources).returns([])
        @scope.stubs(:findresource).returns(inmemory)

        @compiler.stubs(:add_resource)

        proc { @collector.evaluate }.should raise_error(Puppet::ParseError)
    end

    it "should ignore exported resources that match already-collected resources" do
        stub_rails()
        Puppet::Rails::Host.stubs(:find_by_name).returns(nil)

        rails = stub 'one', :restype => "Mytype", :title => "one", :virtual? => true, :exported? => true, :id => 1, :ref => "yay"
        inmemory = stub 'one', :type => "Mytype", :virtual? => true, :exported? => true, :rails_id => 1

        Puppet::Rails::Resource.stubs(:find).returns([rails])

        resource = mock 'resource'

        @compiler.stubs(:resources).returns([])
        @scope.stubs(:findresource).returns(inmemory)

        @compiler.stubs(:add_resource)

        proc { @collector.evaluate }.should_not raise_error(Puppet::ParseError)
    end
end

describe Puppet::Parser::Collector, "when building its ActiveRecord query for collecting exported resources" do
    confine Puppet.features.rails? => "Cannot test Rails integration without ActiveRecord"

    before do
        @scope = stub 'scope', :host => "myhost", :debug => nil
        @compiler = mock 'compile'
        @scope.stubs(:compiler).returns(@compiler)
        @resource_type = "Mytype"
        @equery = nil
        @vquery = proc { |r| true }

        @collector = Puppet::Parser::Collector.new(@scope, @resource_type, @equery, @vquery, :exported)
        @compiler.stubs(:resources).returns([])

        ActiveRecord::Base.stubs(:connected?).returns(false)

        Puppet::Rails.stubs(:init)
        Puppet::Rails::Host.stubs(:find_by_name).returns(nil)
        Puppet::Rails::Resource.stubs(:find).returns([])

        Puppet.settings.stubs(:value).with(:storeconfigs).returns true
    end

    it "should exclude all resources from the host if ActiveRecord contains information for this host" do
        @host = mock 'host'
        @host.stubs(:id).returns 5

        Puppet::Rails::Host.expects(:find_by_name).with(@scope.host).returns(@host)

        Puppet::Rails::Resource.stubs(:find).with { |*arguments|
            options = arguments[3]
            options[:conditions][0] =~ /^host_id != \?/ and options[:conditions][1] == 5 
        }.returns([])

        @collector.evaluate
    end

    it "should return parameter names and parameter values when querying ActiveRecord" do
        Puppet::Rails::Resource.stubs(:find).with { |*arguments|
            options = arguments[3]
            options[:include] == {:param_values => :param_name}
        }.returns([])

        @collector.evaluate
    end

    it "should only search for exported resources with the matching type" do
        Puppet::Rails::Resource.stubs(:find).with { |*arguments|
            options = arguments[3]
            options[:conditions][0].include?("(exported=? AND restype=?)") and options[:conditions][1] == true and options[:conditions][2] == "Mytype"
        }.returns([])
    end

    it "should include the export query if one is provided" do
        @collector = Puppet::Parser::Collector.new(@scope, @resource_type, "test = true", @vquery, :exported)
        Puppet::Rails::Resource.stubs(:find).with { |*arguments|
            options = arguments[3]
            options[:conditions][0].include?("test = true")
        }.returns([])
    end
end