Adding new methods to Puppet::Module.

There are now boolean methods to test whether a given kind
of file is present in a given module.  E.g, you can do:

    Puppet::Module.new("mod", "/my/path").plugins?

There are also accessor-style methods that return the
full path for a given kind of file.

Signed-off-by: Luke Kanies <luke@madstop.com>

1 files changed, 37 insertions, 0 deletions

diff --git a/spec/unit/module.rb b/spec/unit/module.rb
index 70dd825d5..fd5f4fc19 100755
--- a/spec/unit/module.rb
+++ b/spec/unit/module.rb
@@ -2,6 +2,39 @@
 
 require File.dirname(__FILE__) + '/../spec_helper'
 
+describe Puppet::Module do
+    [:plugins, :templates, :files, :manifests].each do |filetype|
+        it "should be able to indicate whether it has #{filetype}" do
+            Puppet::Module.new("foo", "/foo/bar").should respond_to(filetype.to_s + "?")
+        end
+
+        it "should correctly detect when it has #{filetype}" do
+            FileTest.expects(:exist?).with("/foo/bar/#{filetype}").returns true
+            Puppet::Module.new("foo", "/foo/bar").send(filetype.to_s + "?").should be_true
+        end
+
+        it "should correctly detect when it does not have #{filetype}" do
+            FileTest.expects(:exist?).with("/foo/bar/#{filetype}").returns false
+            Puppet::Module.new("foo", "/foo/bar").send(filetype.to_s + "?").should be_false
+        end
+
+        it "should have a method for returning the full path to the #{filetype}" do
+            Puppet::Module.new("foo", "/foo/bar").send(filetype.to_s).should == File.join("/foo/bar", filetype.to_s)
+        end
+
+        it "should be able to return individual #{filetype}" do
+            path = File.join("/foo/bar", filetype.to_s, "my/file")
+            FileTest.expects(:exist?).with(path).returns true
+            Puppet::Module.new("foo", "/foo/bar").send(filetype.to_s.sub(/s$/, ''), "my/file").should == path
+        end
+
+        it "should return nil if asked to return individual #{filetype} that don't exist" do
+            FileTest.expects(:exist?).with(File.join("/foo/bar", filetype.to_s, "my/file")).returns false
+            Puppet::Module.new("foo", "/foo/bar").send(filetype.to_s.sub(/s$/, ''), "my/file").should be_nil
+        end
+    end
+end
+
 describe Puppet::Module, " when building its search path" do
     include PuppetTest
 
@@ -78,6 +111,8 @@ describe Puppet::Module, " when searching for templates" do
     it "should return the template from the first found module" do
         Puppet[:modulepath] = "/one:/two"
         File.stubs(:directory?).returns(true)
+        FileTest.stubs(:exist?).returns false
+        FileTest.expects(:exist?).with("/one/mymod/templates/mytemplate").returns true
         Puppet::Module.find_template("mymod/mytemplate").should == "/one/mymod/templates/mytemplate"
     end
     
@@ -151,6 +186,8 @@ describe Puppet::Module, " when searching for templates" do
         Puppet.settings.stubs(:value).returns.returns("/my/directory")
         Puppet.settings.expects(:value).with(:modulepath, "myenv").returns("/my/modules")
         File.stubs(:directory?).returns(true)
+        FileTest.stubs(:exist?).returns false
+        FileTest.expects(:exist?).with("/my/modules/mymod/templates/envtemplate").returns true
         Puppet::Module.find_template("mymod/envtemplate", "myenv").should == "/my/modules/mymod/templates/envtemplate"
     end