Make sure virtual and rails query use tags when tag are searched

Up to now, when trying to match with tags:
File<<| tag == 'value' |>>
in fact we were querying parameters. Hopefully all the user tags
are stored in parameters so it was working.
But it wasn't possible to search on auto-tags (like class name).
This patch makes sure searching by tag is done on tags both on the
rails side and the resource side.

Signed-off-by: Brice Figureau <brice-puppet@daysofwonder.com>

2 files changed, 23 insertions, 2 deletions

diff --git a/spec/unit/parser/ast/collexpr.rb b/spec/unit/parser/ast/collexpr.rb
index e45203492..4dfc1e97a 100755
--- a/spec/unit/parser/ast/collexpr.rb
+++ b/spec/unit/parser/ast/collexpr.rb
@@ -74,6 +74,27 @@ describe Puppet::Parser::AST::CollExpr do
         end
     end
 
+    describe "when evaluating with tags" do
+        before :each do
+            @tag = stub 'tag', :safeevaluate => 'tag'
+            @value = stub 'value', :safeevaluate => 'value'
+
+            @resource = stub 'resource'
+            @resource.stubs(:tagged?).with("value").returns(true)
+        end
+
+        it "should produce a textual representation of the expression" do
+            collexpr = ast::CollExpr.new(:test1 => @tag, :test2 => @value, :oper=>"==")
+            result = collexpr.evaluate(@scope)
+            result[0].should == "puppet_tags.name = 'value'"
+        end
+
+        it "should inspect resource tags if the query term is on tags" do
+            collexpr = ast::CollExpr.new(:test1 => @tag, :test2 => @value, :oper => "==")
+            collexpr.evaluate(@scope)[1].call(@resource).should be_true
+        end
+    end
+
     [:exported,:virtual].each do |mode|
     it "should check for array member equality if resource parameter is an array for == in mode #{mode}" do
         array = mock 'array', :safeevaluate => "array"
diff --git a/spec/unit/parser/collector.rb b/spec/unit/parser/collector.rb
index 813fbb8c0..3bca564dc 100755
--- a/spec/unit/parser/collector.rb
+++ b/spec/unit/parser/collector.rb
@@ -502,10 +502,10 @@ describe Puppet::Parser::Collector, "when building its ActiveRecord query for co
         @collector.evaluate
     end
 
-    it "should return parameter names and parameter values when querying ActiveRecord" do
+    it "should return parameter names, parameter values and tags when querying ActiveRecord" do
         Puppet::Rails::Resource.stubs(:find).with { |*arguments|
             options = arguments[3]
-            options[:include] == {:param_values => :param_name}
+            options[:include] == {:param_values => :param_name, :puppet_tags => :resource_tags}
         }.returns([])
 
         @collector.evaluate