From 5fcaa4949526386e12a75e569501098ab41be45a Mon Sep 17 00:00:00 2001
From: Ross Boucher <rboucher@gmail.com>
Date: Thu, 15 Sep 2011 14:09:03 -0700
Subject: [PATCH] Update stripe tests for newer API format.

---
 test/test_helper.rb | 4 ++--
 test/test_stripe.rb | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/test/test_helper.rb b/test/test_helper.rb
index 7d3e79e5..65f4cb93 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -57,7 +57,7 @@ def test_customer(params={})
 end
 
 def test_customer_array
-  [test_customer, test_customer, test_customer]
+  {:data => [test_customer, test_customer, test_customer]}
 end
 
 def test_charge(params={})
@@ -84,7 +84,7 @@ def test_charge(params={})
 end
 
 def test_charge_array
-  [test_charge, test_charge, test_charge]
+  {:data => [test_charge, test_charge, test_charge]}
 end
 
 def test_card(params={})
diff --git a/test/test_stripe.rb b/test/test_stripe.rb
index fd899b51..3b837c6a 100644
--- a/test/test_stripe.rb
+++ b/test/test_stripe.rb
@@ -153,7 +153,7 @@ class TestStripeRuby < Test::Unit::TestCase
 
       should "loading all of an APIResource should return an array of recursively instantiated objects" do
         @mock.expects(:get).once.returns(test_response(test_charge_array))
-        c = Stripe::Charge.all
+        c = Stripe::Charge.all.data
         assert c.kind_of? Array
         assert c[0].kind_of? Stripe::Charge
         assert c[0].card.kind_of?(Stripe::StripeObject) && c[0].card.object == 'card'
@@ -163,7 +163,7 @@ class TestStripeRuby < Test::Unit::TestCase
 
         should "charges should be listable" do
           @mock.expects(:get).once.returns(test_response(test_charge_array))
-          c = Stripe::Charge.all
+          c = Stripe::Charge.all.data
           assert c.kind_of? Array
         end
 
@@ -223,7 +223,7 @@ class TestStripeRuby < Test::Unit::TestCase
 
         should "customers should be listable" do
           @mock.expects(:get).once.returns(test_response(test_customer_array))
-          c = Stripe::Customer.all
+          c = Stripe::Customer.all.data
           assert c.kind_of? Array
           assert c[0].kind_of? Stripe::Customer
         end