Merge pull request #10 from eric-hu/master

Add find_last_flag_for and find_all_flags_for methods to Flagger

Rakefile

@@ -4,4 +4,8 @@ Bundler::GemHelper.install_tasks
 require 'rspec/core/rake_task'
 RSpec::Core::RakeTask.new(:spec)
 
+# Allows for 'rake' and 'rake spec' to execute specs.  Fixes bug with message:
+#   undefined method `shellescape' ...
+require 'shellwords'
+
 task :default => :spec

lib/make_flaggable/flagger.rb

@@ -68,6 +68,16 @@ def flagged?(flaggable, flag = nil)
       fetch_flaggings(flaggable, flag).try(:first) ? true : false
     end
 
+    # Returns the most recently created flag
+    def find_last_flag_for(flaggable, flag=nil)
+      find_all_flags_for(flaggable, flag).first
+    end
+
+    # Returns all flags created by the user for a given flag
+    def find_all_flags_for(flaggable, flag=nil)
+      fetch_flaggings(flaggable, flag).order('created_at desc')
+    end
+
     private
 
     def fetch_flaggings(flaggable, flag)

spec/lib/make_flaggable_spec.rb

@@ -151,6 +151,39 @@
         @flagger.flagged?(@flaggable).should == false
       end
     end
+
+    describe "find_last_flag_for" do
+      it "should fetch the most recent flag the flagger has made on the flaggable" do
+        @flagger.flagged?(@flaggable, :favorite).should == false
+        @flagger.flagged?(@flaggable, :inappropriate).should == false
+        @flagger.flag!(@flaggable, :favorite)
+        @flagger.flag!(@flaggable, :inappropriate)
+        @flagger.find_last_flag_for(@flaggable).flag.should == "inappropriate"
+      end
+
+      it "should return nil when the flagger has made no flags on the flaggable" do
+        @flagger.flagged?(@flaggable).should == false
+
+        @flagger.find_last_flag_for(@flagger).should == nil
+      end
+    end
+
+    describe "find_all_flags_for" do
+      it "should fetch all flags the flagger has made on the flaggable" do
+        @flagger.flagged?(@flaggable, :favorite).should == false
+        @flagger.flagged?(@flaggable, :inappropriate).should == false
+        @flagger.flag!(@flaggable, :favorite)
+        @flagger.flag!(@flaggable, :inappropriate)
+        flag_types = @flagger.find_all_flags_for(@flaggable).map(&:flag)
+        flag_types.should include("inappropriate")
+        flag_types.should include("favorite")
+      end
+
+      it "should return an empty ActiveRecord relation when the flagger has made no flags on the flaggable" do
+        @flagger.flagged?(@flaggable).should == false
+        @flagger.find_all_flags_for(@flaggable).should == []
+      end
+    end
   end
 
   describe "flaggable" do