Allow a hit count to be passed to punch

lib/punching_bag.rb

@@ -3,12 +3,13 @@ module PunchingBag
   require 'punching_bag/acts_as_punchable'
   require 'voight_kampff'
 
-  def self.punch(punchable, request=nil)
+  def self.punch(punchable, request = nil, count = 1)
     if request.try(:bot?)
       false
     else
       p = Punch.new
       p.punchable = punchable
+      p.hits = count
       p.save ? p : false
     end
   end
@@ -22,4 +23,4 @@ def self.average_time(*punches)
     end
     Time.zone.at(total_time / hits)
   end
-end
+end

lib/punching_bag/acts_as_punchable.rb

@@ -23,8 +23,9 @@ def hits(since=nil)
         punches.after(since).sum(:hits)
       end
 
-      def punch(request=nil)
-        PunchingBag.punch(self, request)
+      def punch(request=nil, options = {})
+        count = options[:count] || 1
+        PunchingBag.punch(self, request, count)
       end
     end
 

spec/lib/punching_bag_spec.rb

@@ -2,22 +2,38 @@
 
 describe PunchingBag do
   let(:article) { Article.create title: 'Hector', content: 'Ding, ding ding... ding. Ding. DING. DING! ' }
-  let(:human_request) { OpenStruct.new(bot?: false) }
-  let(:bot_request) { OpenStruct.new(bot?: true) }
 
   subject { PunchingBag }
 
   describe '.punch' do
-    it 'does nothing when the request is from a bot' do
-      expect(PunchingBag.punch(article, bot_request)).to be false
+    let(:request) { nil }
+
+    context 'when request is from a bot' do
+      let(:request) { instance_double(ActionDispatch::Request, bot?: true) }
+
+      it 'does nothing' do
+        expect(PunchingBag.punch(article, request)).to be false
+      end
+    end
+
+    context 'when the request is valid' do
+      let(:request) { instance_double(ActionDispatch::Request, bot?: false) }
+
+      it 'creates a new punch' do
+        expect { PunchingBag.punch(article, request) }.to change { Punch.count }.by 1
+      end
     end
 
-    it 'creates a new punch when the request is valid' do
-      expect { PunchingBag.punch(article, human_request) }.to change { Punch.count }.by 1
+    context 'when there is no request' do
+      it 'creates a new punch' do
+        expect { PunchingBag.punch(article) }.to change { Punch.count }.by 1
+      end
     end
 
-    it 'creates a new punch when there is no request' do
-      expect { PunchingBag.punch(article) }.to change { Punch.count }.by 1
+    context 'when count is more than one' do
+      it 'creates a new punch with a higher count' do
+        expect { PunchingBag.punch(article, nil, 2) }.to change { Punch.sum(:hits) }.by 2
+      end
     end
   end
 

spec/models/punchable_spec.rb

@@ -24,6 +24,12 @@
       it 'incleases hits by one' do
         expect { subject.punch }.to change { subject.hits }.by 1
       end
+
+      context 'when count is set to two' do
+        it 'increases hits by two' do
+          expect { subject.punch(nil, count: 2) }.to change { subject.hits }.by 2
+        end
+      end
     end
   end
 
@@ -54,4 +60,4 @@
       its('sort_by_popularity.second') { should eql article_1 }
     end
   end
-end
+end