1 files changed, 47 insertions, 0 deletions
diff --git a/spec/integration/parser/functions/require.rb b/spec/integration/parser/functions/require.rb
index 68a1e0b54..15ff51b95 100755
--- a/spec/integration/parser/functions/require.rb
+++ b/spec/integration/parser/functions/require.rb
@@ -25,3 +25,50 @@ describe "the require function" do
     end
 
 end
+
+describe "the include function" do
+    require 'puppet_spec/files'
+    include PuppetSpec::Files
+
+    before :each do
+        @real_dir = Dir.getwd
+        @temp_dir = tmpfile('include_function_integration_test')
+        Dir.mkdir @temp_dir
+        Dir.chdir @temp_dir
+        @parser = Puppet::Parser::Parser.new :Code => ""
+        @node = Puppet::Node.new("mynode")
+        @compiler = Puppet::Parser::Compiler.new(@node, @parser)
+        @compiler.send(:evaluate_main)
+        @scope = @compiler.topscope
+        # preload our functions
+        Puppet::Parser::Functions.function(:include)
+        Puppet::Parser::Functions.function(:require)
+    end
+
+    after :each do
+        Dir.chdir @real_dir
+        Dir.rmdir @temp_dir
+    end
+
+    def with_file(filename,contents)
+        path = File.join(@temp_dir,filename)
+        File.open(path, "w") { |f|f.puts contents }
+        yield
+        File.delete(path)
+    end
+
+    it "should add a relationship between the 'included' class and our class" do
+        with_file('includedclass',"class includedclass {}") {
+            @scope.function_include("includedclass")
+            }
+        @compiler.catalog.edge?(@scope.resource,@compiler.findresource(:class,"includedclass")).should be_true
+    end
+
+    it "should find a file with an all lowercase name given a mixed case name" do
+        with_file('includedclass',"class includedclass {}") {
+            @scope.function_include("includedclass")
+            }
+        @compiler.catalog.edge?(@scope.resource,@compiler.findresource(:class,"IncludedClass")).should be_true
+    end
+
+end