Merge pull request abronte#37 from kysnm/add-method-table-raw-data

Add method table_raw_data

Author: abronte

lib/big_query/client/tables.rb

@@ -31,18 +31,28 @@ def tables_formatted(dataset = @dataset)
         tables(dataset).map { |t| t['tableReference']['tableId'] }
       end
 
-      # Returns all rows of table data
+      # Returns entire response of table data
       #
       # @param tableId [String] id of the table to look for
       # @param dataset [String] dataset to look for
       # @param options [Hash] hash of optional query parameters (maxResults, startIndex)
       # @return [Hash] json api response
-      def table_data(tableId, dataset = @dataset, options = {})
+      def table_raw_data(tableId, dataset = @dataset, options = {})
         parameters = { 'datasetId' => dataset, 'tableId' => tableId }
         parameters['maxResults'] = options[:maxResults] if options[:maxResults]
         parameters['startIndex'] = options[:startIndex] if options[:startIndex]
-        response = api(api_method: @bq.tabledata.list,
+        api(api_method: @bq.tabledata.list,
                        parameters: parameters)
+      end
+
+      # Returns all rows of table data
+      #
+      # @param tableId [String] id of the table to look for
+      # @param dataset [String] dataset to look for
+      # @param options [Hash] hash of optional query parameters (maxResults, startIndex)
+      # @return [Hash] json api response
+      def table_data(tableId, dataset = @dataset, options = {})
+        response = table_raw_data(tableId, dataset, options)
         response['rows'] || []
       end
 

test/bigquery.rb

@@ -42,10 +42,11 @@ def test_for_tables_formatted
     assert_includes result, 'test'
   end
 
-  def test_for_table_data
-    result = @bq.table_data('test')
+  def test_for_table_raw_data
+    result = @bq.table_raw_data('test')
 
-    assert_kind_of Array, result
+    assert_kind_of Hash, result
+    assert_equal result['kind'], "bigquery#tableDataList"
   end
 
   def test_for_table_data_maxResults