From b9839463cd8c9c95487d5e4263a667bc7663cca5 Mon Sep 17 00:00:00 2001
From: Jeff Zellman <jzellman@gmail.com>
Date: Wed, 14 Sep 2011 16:58:23 -0500
Subject: [PATCH] Add testing of data field on payment method

---
 lib/spreedly_core/test_extensions.rb | 9 +++++++--
 test/spreedly_core_test.rb           | 6 ++++--
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/lib/spreedly_core/test_extensions.rb b/lib/spreedly_core/test_extensions.rb
index 84ceb81..5c82f91 100644
--- a/lib/spreedly_core/test_extensions.rb
+++ b/lib/spreedly_core/test_extensions.rb
@@ -28,9 +28,14 @@ def cc_data(cc_type, options={})
 
 module SpreedlyCore
   class PaymentMethod
-    def self.create_test_token(cc_data={})
+
+    # Call spreedly core to create a test token.
+    # pass_through_data will be added as the "data" field.
+    # 
+    def self.create_test_token(cc_data={}, pass_through_data=nil)
       data = cc_data.merge(:redirect_url => "http://example.com",
-                           :api_login => SpreedlyCore::Base.login)
+                           :api_login => SpreedlyCore::Base.login,
+                           :data => pass_through_data)
       
       response = self.post("/payment_methods", :body => data, :no_follow => true)
     rescue HTTParty::RedirectionTooDeep => e
diff --git a/test/spreedly_core_test.rb b/test/spreedly_core_test.rb
index a310f1d..34c58ea 100644
--- a/test/spreedly_core_test.rb
+++ b/test/spreedly_core_test.rb
@@ -12,9 +12,11 @@ def setup
   end
 
   def given_a_payment_method(cc_card=:master, card_options={})
-    token = SpreedlyCore::PaymentMethod.create_test_token(cc_data(cc_card, card_options))
+    token = SpreedlyCore::PaymentMethod.
+      create_test_token(cc_data(cc_card, card_options), "customer-42")
     assert payment_method = SpreedlyCore::PaymentMethod.find(token)
-    assert token, payment_method.token
+    assert_equal "customer-42", payment_method.data
+    assert_equal token, payment_method.token
     payment_method
   end