Allow to remove discount from customer

Author: mnin

lib/stripe_mock/request_handlers/customers.rb

@@ -87,10 +87,14 @@ def update_customer(route, method_url, params, headers)
         end
 
         if params[:coupon]
-          coupon = coupons[ params[:coupon] ]
-          assert_existence :coupon, params[:coupon], coupon
+          if params[:coupon] == ''
+            delete_coupon_from_object(cus)
+          else
+            coupon = coupons[params[:coupon]]
+            assert_existence :coupon, params[:coupon], coupon
 
-          add_coupon_to_object(cus, coupon)
+            add_coupon_to_object(cus, coupon)
+          end
         end
 
         cus

lib/stripe_mock/request_handlers/helpers/coupon_helpers.rb

@@ -12,6 +12,11 @@ def add_coupon_to_object(object, coupon)
         object[:discount] = Stripe::Discount.construct_from(discount_attrs)
         object
       end
+
+      def delete_coupon_from_object(object)
+        object[:discount] = nil
+        object
+      end
     end
   end
 end

spec/shared_stripe_examples/customer_examples.rb

@@ -270,6 +270,25 @@ def gen_card_tk
     }
   end
 
+  context 'with coupon on customer' do
+    before do
+      Stripe::Coupon.create(id: '10PERCENT', duration: 'once')
+      Stripe::Customer.create(id: 'test_cus_coupon', coupon: '10PERCENT')
+    end
+
+    it 'remove the coupon from customer' do
+      customer = Stripe::Customer.retrieve('test_cus_coupon')
+      expect(customer.discount).to_not be_nil
+      expect(customer.discount.coupon).to_not be_nil
+      expect(customer.discount.customer).to eq customer.id
+      expect(customer.discount.start).to be_within(1).of Time.now.to_i
+
+      Stripe::Customer.update('test_cus_coupon', coupon: '')
+      customer = Stripe::Customer.retrieve('test_cus_coupon')
+      expect(customer.discount).to be_nil
+    end
+  end
+
   it "stores a created stripe customer in memory" do
     customer = Stripe::Customer.create({
       email: '[email protected]',