lib/fake_braintree/customer.rb in fake_braintree-0.0.5 vs lib/fake_braintree/customer.rb in fake_braintree-0.0.6

@@ -1,34 +1,47 @@
 module FakeBraintree
   class Customer
     include Helpers
 
-    def initialize(request, merchant_id)
-      @request_hash = Hash.from_xml(request.body).delete("customer")
-      @merchant_id  = merchant_id
+    def initialize(customer_hash, options)
+      @customer_hash = customer_hash.merge("merchant_id" => options[:merchant_id],
+                                           "id" => options[:id])
     end
 
-    def invalid?
-      credit_card_is_failure? || invalid_credit_card?
+    def create
+      if invalid?
+        failure_response
+      else
+        hash = customer_hash
+        FakeBraintree.customers[hash["id"]] = hash
+        gzipped_response(201, hash.to_xml(:root => 'customer'))
+      end
     end
 
-    def failure_response
-      gzipped_response(422, FakeBraintree.failure_response(@request_hash["credit_card"]["number"]).to_xml(:root => 'api_error_response'))
+    def update
+      if existing_customer_hash
+        hash = update_existing_customer!
+        gzipped_response(200, hash.to_xml(:root => 'customer'))
+      else
+        failure_response(404)
+      end
     end
 
+    def delete
+      FakeBraintree.customers[existing_customer_id] = nil
+      gzipped_response(200, '')
+    end
+
     def customer_hash
-      hash = @request_hash.dup
+      hash = @customer_hash.dup
       hash["id"] ||= create_id
-      hash["merchant-id"] = @merchant_id
+
       if hash["credit_card"] && hash["credit_card"].is_a?(Hash)
-        hash["credit_card"].delete("__content__")
         if !hash["credit_card"].empty?
-          hash["credit_card"]["last_4"]           = hash["credit_card"].delete("number")[-4..-1]
-          hash["credit_card"]["token"]            = md5("#{hash['merchant_id']}#{hash['id']}")
-          expiration_date = hash["credit_card"].delete("expiration_date")
-          hash["credit_card"]["expiration_month"] = expiration_date.split('/')[0]
-          hash["credit_card"]["expiration_year"]  = expiration_date.split('/')[1]
+          hash["credit_card"]["last_4"] = last_four(hash)
+          hash["credit_card"]["token"]  = credit_card_token(hash)
+          split_expiration_date_into_month_and_year!(hash)
 
           credit_card = hash.delete("credit_card")
           hash["credit_cards"] = [credit_card]
         end
       end
@@ -36,32 +49,75 @@
       hash
     end
 
     private
 
-    def create_id
-      md5("#{@merchant_id}#{Time.now.to_f}")
+    def invalid?
+      credit_card_is_failure? || invalid_credit_card?
     end
 
+    def split_expiration_date_into_month_and_year!(hash)
+      if expiration_date = hash["credit_card"].delete("expiration_date")
+        hash["credit_card"]["expiration_month"] = expiration_date.split('/')[0]
+        hash["credit_card"]["expiration_year"]  = expiration_date.split('/')[1]
+      end
+    end
+
+    def existing_customer_hash
+      existing_customer_id && FakeBraintree.customers[existing_customer_id]
+    end
+
+    def update_existing_customer!
+      existing_customer_hash.merge!(customer_hash)
+    end
+
+    def credit_card_token(hash)
+      md5("#{hash['merchant_id']}#{hash['id']}")
+    end
+
+    def last_four(hash)
+      hash["credit_card"].delete("number")[-4..-1]
+    end
+
+    def failure_response(code = 422)
+      gzipped_response(code, FakeBraintree.failure_response(credit_card_number).to_xml(:root => 'api_error_response'))
+    end
+
     def credit_card_is_failure?
-      @request_hash.key?('credit_card') &&
-        FakeBraintree.failure?(@request_hash["credit_card"]["number"])
+      @customer_hash.key?('credit_card') &&
+        FakeBraintree.failure?(@customer_hash["credit_card"]["number"])
     end
 
     def invalid_credit_card?
-      verify_credit_card?(@request_hash) && has_invalid_credit_card?(@request_hash)
+      verify_credit_card?(@customer_hash) && has_invalid_credit_card?(@customer_hash)
     end
 
     def verify_credit_card?(customer_hash)
       return true if FakeBraintree.verify_all_cards
 
-      @request_hash.key?("credit_card") &&
-        @request_hash["credit_card"].key?("options") &&
-        @request_hash["credit_card"]["options"].is_a?(Hash) &&
-        @request_hash["credit_card"]["options"]["verify_card"] == true
+      @customer_hash.key?("credit_card") &&
+        @customer_hash["credit_card"].is_a?(Hash) &&
+        @customer_hash["credit_card"].key?("options") &&
+        @customer_hash["credit_card"]["options"].is_a?(Hash) &&
+        @customer_hash["credit_card"]["options"]["verify_card"] == true
     end
 
     def has_invalid_credit_card?(customer_hash)
-      ! FakeBraintree::VALID_CREDIT_CARDS.include?(@request_hash["credit_card"]["number"])
+      has_credit_card_number? &&
+        ! FakeBraintree::VALID_CREDIT_CARDS.include?(@customer_hash["credit_card"]["number"])
+    end
+
+    def has_credit_card_number?
+      @customer_hash.key?("credit_card") &&
+        @customer_hash["credit_card"].is_a?(Hash) &&
+        @customer_hash["credit_card"].key?("number")
+    end
+
+    def credit_card_number
+      has_credit_card_number? && @customer_hash["credit_card"]["number"]
+    end
+
+    def existing_customer_id
+      @customer_hash['id']
     end
   end
 end