Fix integration tests due to new paginated responses

Author: jsklan

lib/square/inventory/client.rb

@@ -170,6 +170,11 @@ def batch_create_changes(request_options: {}, **params)
       #
       # @return [Square::Types::BatchGetInventoryChangesResponse]
       def batch_get_changes(request_options: {}, **params)
+        params = Square::Internal::Types::Utils.symbolize_keys(params)
+        _query_param_names = %i[cursor]
+        _query = params.slice(*_query_param_names)
+        params = params.except(*_query_param_names)
+
         Square::Internal::CursorItemIterator.new(
           cursor_field: :cursor,
           item_field: :changes,
@@ -211,6 +216,11 @@ def batch_get_changes(request_options: {}, **params)
       #
       # @return [Square::Types::BatchGetInventoryCountsResponse]
       def batch_get_counts(request_options: {}, **params)
+        params = Square::Internal::Types::Utils.symbolize_keys(params)
+        _query_param_names = %i[cursor]
+        _query = params.slice(*_query_param_names)
+        params = params.except(*_query_param_names)
+
         Square::Internal::CursorItemIterator.new(
           cursor_field: :cursor,
           item_field: :counts,

test/square/integration/client_tests/test_customers_groups.rb

@@ -42,11 +42,12 @@ def delete_test_customer_group(group_id)
       # list
       list_response = client.customers.groups.list
       refute_nil list_response
-      assert_equal list_response.class, Square::Types::ListCustomerGroupsResponse
-      refute_nil list_response.groups
-      assert list_response.groups.length > 0
+      assert_equal Square::Internal::CursorItemIterator, list_response.class
+      groups = list_response.to_a
+      refute_nil groups
+      assert groups.length > 0
 
-      puts "list_response #{list_response.to_h}" if verbose?
+      puts "list_response #{groups}" if verbose?
 
       # Cleanup
       delete_test_customer_group(response.group.id)

test/square/integration/client_tests/test_customers_segments.rb

@@ -8,17 +8,19 @@
 
       response = client.customers.segments.list
       refute_nil response
-      assert_equal response.class, Square::Types::ListCustomerSegmentsResponse
-      refute_nil response.segments
+      assert_equal Square::Internal::CursorItemIterator, response.class
+      # Iterator wraps the response, we need to access the first page to get segments
+      refute_nil response.to_a
 
-      puts "response #{response.to_h}" if verbose?
+      puts "response #{response.to_a}" if verbose?
     end
   end
 
   describe "#get" do
     it "should retrieve a customer segment" do
       list_response = client.customers.segments.list
-      segment_id = list_response.segments.first.id
+      segments = list_response.to_a
+      segment_id = segments.first.id
 
       _request = Square::Customers::Segments::Types::GetSegmentsRequest.new(
         segment_id: segment_id

test/square/integration/client_tests/test_merchants.rb

@@ -6,19 +6,21 @@
   before do
     # Get first merchant ID
     merchant_response = client.merchants.list
-    @merchant_id = merchant_response.merchant.first.id
+    merchants = merchant_response.to_a
+    @merchant_id = merchants.first.id
   end
 
   describe "#list" do
     it "should list merchants" do
 
       response = client.merchants.list
       refute_nil response
-      assert_equal response.class, Square::Types::ListMerchantsResponse
-      refute_nil response.merchant
-      assert response.merchant.length > 0
+      assert_equal Square::Internal::CursorItemIterator, response.class
+      merchants = response.to_a
+      refute_nil merchants
+      assert merchants.length > 0
 
-      puts "response #{response.to_h}" if verbose?
+      puts "response #{merchants}" if verbose?
     end
   end
 

test/square/integration/client_tests/test_payments.rb

@@ -31,11 +31,12 @@ def create_sample_payment
 
       response = client.payments.list
       refute_nil response
-      assert_equal response.class, Square::Types::ListPaymentsResponse
-      refute_nil response.payments
-      assert response.payments.length > 0
+      assert_equal Square::Internal::CursorItemIterator, response.class
+      payments = response.to_a
+      refute_nil payments
+      assert payments.length > 0
 
-      puts "response #{response.to_h}" if verbose?
+      puts "response #{payments}" if verbose?
     end
   end
 

test/square/integration/client_tests/test_refunds.rb

@@ -47,10 +47,11 @@
       puts "request #{_request.to_h}" if verbose?
 
       response = client.refunds.list
-      refute_nil response.refunds
-      assert response.refunds.length > 0
+      refunds = response.to_a
+      refute_nil refunds
+      assert refunds.length > 0
 
-      puts "response #{response.to_h}" if verbose?
+      puts "response #{refunds}" if verbose?
     end
   end
 

test/square/internal/multipart/test_form_data.rb

@@ -4,6 +4,7 @@
 require "stringio"
 require "json"
 require "test_helper"
+require "square/file_param"
 
 class MockFile
   attr_reader :name, :content, :content_type