Adding helper module for finding known resource types

This is so everyone doesn't have to define the same method
everywhere.

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

2 files changed, 30 insertions, 0 deletions

diff --git a/lib/puppet/parser/resource_type_collection_helper.rb b/lib/puppet/parser/resource_type_collection_helper.rb
new file mode 100644
index 000000000..4f66c773a
--- /dev/null
+++ b/lib/puppet/parser/resource_type_collection_helper.rb
@@ -0,0 +1,5 @@
+module Puppet::Parser::ResourceTypeCollectionHelper
+    def known_resource_types
+        environment.known_resource_types
+    end
+end
diff --git a/spec/unit/parser/resource_type_collection_helper.rb b/spec/unit/parser/resource_type_collection_helper.rb
new file mode 100644
index 000000000..c93c9f20e
--- /dev/null
+++ b/spec/unit/parser/resource_type_collection_helper.rb
@@ -0,0 +1,25 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+require 'puppet/parser/resource_type_collection_helper'
+
+class RTCHelperTester
+    include Puppet::Parser::ResourceTypeCollectionHelper
+end
+
+describe Puppet::Parser::ResourceTypeCollectionHelper do
+    before do
+        @helper = RTCHelperTester.new
+    end
+
+    it "should use its current environment to retrieve the known resource type collection" do
+        env = stub 'environment'
+        @helper.expects(:environment).returns env
+
+        rtc = stub 'known_resource_types'
+        env.expects(:known_resource_types).returns  rtc
+
+        @helper.known_resource_types.should equal(rtc)
+    end
+end