modify tests for changed payment flow

spec/models/ticket_spec.rb

@@ -82,31 +82,72 @@ describe Ticket do
     end
   end
 
-  describe '#quantity_bought_by' do
-    it 'returns the correct value if the user has bought this ticket' do
-      create(:ticket_purchase,
-             user: user,
-             ticket: ticket,
-             quantity: 20)
-      expect(ticket.quantity_bought_by(user)).to eq(20)
+  describe '#unpaid?' do
+    let!(:ticket_purchase) { create(:ticket_purchase, user: user, ticket: ticket) }
+
+    context 'user has not paid' do
+
+      it 'returns true' do
+        expect(ticket.unpaid?(user)).to eq(true)
+      end
     end
 
-    it 'returns zero if the user has not bought this ticket' do
-      expect(ticket.quantity_bought_by(user)).to eq(0)
+    context 'user has paid' do
+      before { ticket_purchase.update_attributes(paid: true) }
+
+      it 'returns false' do
+        expect(ticket.unpaid?(user)).to eq(false)
+      end
+    end
+  end
+
+  describe '#quantity_bought_by' do
+    context 'user has not paid' do
+      it 'returns the correct value if the user has bought this ticket' do
+        create(:ticket_purchase,
+               user: user,
+               ticket: ticket,
+               quantity: 20)
+        expect(ticket.quantity_bought_by(user, paid: false)).to eq(20)
+      end
+
+      it 'returns zero if the user has not bought this ticket' do
+        expect(ticket.quantity_bought_by(user, paid: false)).to eq(0)
+      end
+    end
+
+    context 'user has paid' do
+      let!(:ticket_purchase) { create(:ticket_purchase, user: user, ticket: ticket, quantity: 20) }
+      before { ticket_purchase.update_attributes(paid: true) }
+
+      it 'returns the correct value if the user has bought and paid for this ticket' do
+        expect(ticket.quantity_bought_by(user, paid: true)).to eq(20)
+      end
     end
   end
 
   describe '#total_price' do
-    it 'returns the correct value if the user has bought this ticket' do
-      create(:ticket_purchase,
-             user: user,
-             ticket: ticket,
-             quantity: 20)
-      expect(ticket.total_price(user)).to eq(Money.new(100000, 'USD'))
+    context 'user has not paid' do
+      it 'returns the correct value if the user has bought this ticket' do
+        create(:ticket_purchase,
+               user: user,
+               ticket: ticket,
+               quantity: 20)
+        expect(ticket.total_price(user, paid: false)).to eq(Money.new(100000, 'USD'))
+      end
+
+      it 'returns zero if the user has not bought this ticket' do
+        expect(ticket.total_price(user, paid: false)).to eq(Money.new(0, 'USD'))
+      end
     end
 
-    it 'returns zero if the user has not bought this ticket' do
-      expect(ticket.total_price(user)).to eq(Money.new(0, 'USD'))
+    context 'user has paid' do
+      let!(:ticket_purchase) { create(:ticket_purchase, user: user, ticket: ticket, quantity: 20) }
+      before { ticket_purchase.update_attributes(paid: true) }
+
+      it 'returns the correct value if the user has bought this ticket' do
+        expect(ticket.total_price(user, paid: true)).to eq(Money.new(100000, 'USD'))
+      end
     end
   end
 
@@ -116,7 +157,7 @@ describe Ticket do
     describe 'user has bought' do
       context 'no tickets' do
         it 'returns zero' do
-          expect(Ticket.total_price(conference, user)).to eq(Money.new(0, 'USD'))
+          expect(Ticket.total_price(conference, user, paid: false)).to eq(Money.new(0, 'USD'))
         end
       end
 
@@ -126,7 +167,7 @@ describe Ticket do
         end
 
         it 'returns the correct total price' do
-          expect(Ticket.total_price(conference, user)).to eq(Money.new(100000, 'USD'))
+          expect(Ticket.total_price(conference, user, paid: false)).to eq(Money.new(100000, 'USD'))
         end
       end
 
@@ -138,7 +179,7 @@ describe Ticket do
 
         it 'returns the correct total price' do
           total_price = Money.new(200000, 'USD')
-          expect(Ticket.total_price(conference, user)).to eq(total_price)
+          expect(Ticket.total_price(conference, user, paid: false)).to eq(total_price)
         end
       end
     end