summaryrefslogtreecommitdiffstats
path: root/spec/unit/parser/ast/collexpr_spec.rb
blob: f5e340ce8a371b195f43fca36df0be0b1b244e7e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#!/usr/bin/env ruby

require 'spec_helper'

describe Puppet::Parser::AST::CollExpr do

  ast = Puppet::Parser::AST

  before :each do
    @scope = Puppet::Parser::Scope.new
  end

  describe "when evaluating with two operands" do
    before :each do
      @test1 = mock 'test1'
      @test1.expects(:safeevaluate).with(@scope).returns("test1")
      @test2 = mock 'test2'
      @test2.expects(:safeevaluate).with(@scope).returns("test2")
    end

    it "should evaluate both" do
      collexpr = ast::CollExpr.new(:test1 => @test1, :test2 => @test2, :oper=>"==")
      collexpr.evaluate(@scope)
    end

    it "should produce a textual representation and code of the expression" do
      collexpr = ast::CollExpr.new(:test1 => @test1, :test2 => @test2, :oper=>"==")
      result = collexpr.evaluate(@scope)
      result[0].should == "param_values.value = 'test2' and param_names.name = 'test1'"
      result[1].should be_an_instance_of(Proc)
    end

    it "should propagate expression type and form to child if expression themselves" do
      [@test1, @test2].each do |t|
        t.expects(:is_a?).returns(true)
        t.expects(:form).returns(false)
        t.expects(:type).returns(false)
        t.expects(:type=)
        t.expects(:form=)
      end

      collexpr = ast::CollExpr.new(:test1 => @test1, :test2 => @test2, :oper=>"==", :form => true, :type => true)
      result = collexpr.evaluate(@scope)
    end

    describe "and when evaluating the produced code" do
      before :each do
        @resource = mock 'resource'
        @resource.expects(:[]).with("test1").at_least(1).returns("test2")
      end

      it "should evaluate like the original expression for ==" do
        collexpr = ast::CollExpr.new(:test1 => @test1, :test2 => @test2, :oper => "==")
        collexpr.evaluate(@scope)[1].call(@resource).should === (@resource["test1"] == "test2")
      end

      it "should evaluate like the original expression for !=" do
        collexpr = ast::CollExpr.new(:test1 => @test1, :test2 => @test2, :oper => "!=")
        collexpr.evaluate(@scope)[1].call(@resource).should === (@resource["test1"] != "test2")
      end
    end

    it "should warn if this is an exported collection containing parenthesis (unsupported)" do
      collexpr = ast::CollExpr.new(:test1 => @test1, :test2 => @test2, :oper=>"==", :parens => true, :form => :exported)
      Puppet.expects(:warning)
      collexpr.evaluate(@scope)
    end

    %w{and or}.each do |op|
      it "should raise an error if this is an exported collection with #{op} operator (unsupported)" do
        collexpr = ast::CollExpr.new(:test1 => @test1, :test2 => @test2, :oper=> op, :form => :exported)
        lambda { collexpr.evaluate(@scope) }.should raise_error(Puppet::ParseError)
      end
    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"
    test1 = mock 'test1'
    test1.expects(:safeevaluate).with(@scope).returns("test1")

    resource = mock 'resource'
    resource.expects(:[]).with("array").at_least(1).returns(["test1","test2","test3"])
    collexpr = ast::CollExpr.new(:test1 => array, :test2 => test1, :oper => "==", :form => mode)
    collexpr.evaluate(@scope)[1].call(resource).should be_true
  end
  end

  it "should raise an error for invalid operator" do
    lambda { collexpr = ast::CollExpr.new(:oper=>">") }.should raise_error
  end

end