Added a 'tagged?' method to the Tagging module.

2 files changed, 14 insertions, 0 deletions

diff --git a/lib/puppet/util/tagging.rb b/lib/puppet/util/tagging.rb
index 25d74c420..9abb3fb2b 100644
--- a/lib/puppet/util/tagging.rb
+++ b/lib/puppet/util/tagging.rb
@@ -19,6 +19,11 @@ module Puppet::Util::Tagging
         qualified.collect { |name| name.split("::") }.flatten.each { |tag| @tags << tag unless @tags.include?(tag) }
     end
 
+    # Are we tagged with the provided tag?
+    def tagged?(tag)
+        defined?(@tags) and @tags.include?(tag.to_s)
+    end
+
     # Return a copy of the tag list, so someone can't ask for our tags
     # and then modify them.
     def tags
diff --git a/spec/unit/util/tagging.rb b/spec/unit/util/tagging.rb
index 51b69a63c..91cbb213d 100755
--- a/spec/unit/util/tagging.rb
+++ b/spec/unit/util/tagging.rb
@@ -76,4 +76,13 @@ describe Puppet::Util::Tagging, "when adding tags" do
         @tagger.tags.should be_include("two")
         @tagger.tags.should be_include("three")
     end
+
+    it "should indicate when the object is tagged with a provided tag" do
+        @tagger.tag("one")
+        @tagger.should be_tagged("one")
+    end
+
+    it "should indicate when the object is not tagged with a provided tag" do
+        @tagger.should_not be_tagged("one")
+    end
 end