Add codeclimate badge

Refactoring at stock_item notification.
DynamoMTL · Dec 12, 2014 · 2d6f50f · 2d6f50f
1 parent 0236589
commit 2d6f50f
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 3 deletions.
diff --git a/Gemfile.lock b/Gemfile.lock
@@ -145,7 +145,7 @@ GEM
       coffee-script-source
       execjs
     coffee-script-source (1.8.0)
-    colorize (0.7.3)
+    colorize (0.7.4)
     database_cleaner (1.3.0)
     deface (1.0.1)
       colorize (>= 0.5.8)

diff --git a/README.md b/README.md
@@ -1,5 +1,6 @@
 Spree Waiting List
 ================
+[![Code Climate](https://codeclimate.com/github/ddomingues/spree_waiting_list/badges/gpa.svg)](https://codeclimate.com/github/ddomingues/spree_waiting_list)
 
 A waiting list extension for Spree.
 

diff --git a/app/models/spree/product_decorator.rb b/app/models/spree/product_decorator.rb
@@ -1,3 +1,7 @@
 Spree::Product.class_eval do
   has_many :stock_requests
+
+  def notify_waiting_list
+    stock_requests.without_variant.notified(false).each &:notify!
+  end
 end
diff --git a/app/models/spree/stock_item_decorator.rb b/app/models/spree/stock_item_decorator.rb
@@ -11,8 +11,7 @@ def notify_availability
       yield
 
       if available
-        variant.stock_requests.notified(false).each &:notify!
-        variant.product.stock_requests.without_variant.notified(false).each &:notify!
+        variant.notify_waiting_list
       end
     end
 

diff --git a/app/models/spree/variant_decorator.rb b/app/models/spree/variant_decorator.rb
@@ -2,6 +2,11 @@ module Spree
 
   Variant.class_eval do
     has_many :stock_requests
+
+    def notify_waiting_list
+      stock_requests.notified(false).each &:notify!
+      product.notify_waiting_list
+    end
   end
 
 end