This is so everyone doesn't have to define the same method everywhere. Signed-off-by: Luke Kanies <[email protected]> --- .../parser/resource_type_collection_helper.rb | 5 ++++ .../unit/parser/resource_type_collection_helper.rb | 25 ++++++++++++++++++++ 2 files changed, 30 insertions(+), 0 deletions(-) create mode 100644 lib/puppet/parser/resource_type_collection_helper.rb create mode 100644 spec/unit/parser/resource_type_collection_helper.rb
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 0000000..4f66c77 --- /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 0000000..c93c9f2 --- /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 -- 1.6.1
-- You received this message because you are subscribed to the Google Groups "Puppet Developers" group. To post to this group, send email to [email protected]. To unsubscribe from this group, send email to [email protected]. For more options, visit this group at http://groups.google.com/group/puppet-dev?hl=en.
