lib/tickethub/supplier/payment.rb in tickethub-0.3.20 vs lib/tickethub/supplier/payment.rb in tickethub-0.3.21

@@ -3,10 +3,11 @@
 module Tickethub
   class Supplier::Payment < Resource
     path '/supplier/payments'
 
     require_relative 'charge'
+    require_relative 'refund'
 
     require_relative 'payment/cash'
     require_relative 'payment/credit'
     require_relative 'payment/direct'
     require_relative 'payment/stripe'
@@ -16,22 +17,18 @@
     require_relative 'user'
 
     association :order, Supplier::Order
     association :user, Supplier::User
 
+    collection :refunds, Supplier::Refund
+
+    attribute :balance, type: :money
     attribute :amount, type: :money
     attribute :refunded, type: :money
     attribute :currency, type: :currency
     attribute :refunded_at, type: :datetime
     attribute :confirmed_at, type: :datetime
+
     attribute :created_at, type: :datetime
     attribute :updated_at, type: :datetime
-
-    def refund(attributes)
-      self.load @endpoint[:refund].post(attributes).decoded
-      return true
-    rescue Tickethub::ResourceInvalid => err
-      self.load Tickethub::Response.new(err.response).decoded
-      return false
-    end
   end
 end
\ No newline at end of file