lib/conekta/customer.rb in conekta-0.3.1 vs lib/conekta/customer.rb in conekta-0.3.3

@@ -4,14 +4,31 @@
     include Conekta::APIOperations::Delete
     include Conekta::APIOperations::Update
     include Conekta::APIOperations::List
     include Conekta::APIOperations::CreateMember
 
+    def refresh_from(values, api_key, partial=false)
+      super
+      customer = self
+      for i in 0..(customer.cards.count - 1)
+        customer.cards[i].customer = customer
+      end
+      if customer.subscription
+        customer.subscription.customer = customer
+      end
+    end
+
     def create_subscription(params={})
-      self.create_member('subscription', params)
+      subscription = create_member('subscription', params)
+      subscription.customer = self
+      self.subscription = subscription
+      subscription
     end
 
     def create_card(params={})
-      self.create_member('cards', params)
+      card = create_member('cards', params)
+      card.customer = self
+      self.cards << card
+      card
     end
   end
 end