bigquery-client 0.3.4 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: dcdf35c7bc17562e5cdd590bdbac3d6ae9df055d
-  data.tar.gz: 206728dd2bd10c4bda55ce79857e73b6c6f9d85b
+  metadata.gz: e7f17a75e10d343f606ac5bcf6fdf6f4380c8400
+  data.tar.gz: 287c065f3fb531df155dd1748bd075771174644b
 SHA512:
-  metadata.gz: 8a23eff48fd905f4de17ce835e2d7c7c3b85a8e31be65c23949b41640c846aa4061e1ecdb1b9661c83aba6e2e052323c41ac1117a39f87d1fa305b2bf676f6fd
-  data.tar.gz: c944d45ce22b57b5566b2218aa85baa4ca03e5541d8b10f6bcb9d50d5171a7c54e8d4b2afe40e649067996194f82fe2d6cfa812c5617dba989635c6ab4f35660
+  metadata.gz: 6e14f2c77368f4849a7eb5603e8f649ff804307742d59d45e6a59c0c74b5e61297588631f7259522592702466a95d63017834e5cce65c2511e1c60b3a58fc27e
+  data.tar.gz: d11a77c7a18345800ba7e5b8d6ff661d62381e0ff58c77584c81271bd56bbccb5451a13ece01c2d4855d5b757bfb5382d64f15bac0e71403af87dd6ca6ce7259

data/.gitignore

@@ -15,3 +15,4 @@
 *.o
 *.a
 mkmf.log
+/vcr_cassettes

data/Gemfile

@@ -10,6 +10,8 @@ end
 
 group :test do
   gem 'test-unit', '~> 3.0.0'
+  gem "webmock"
+  gem 'vcr'
 end
 
 gemspec

data/README.md

@@ -54,7 +54,7 @@ https://cloud.google.com/bigquery/docs/reference/v2/
 | Datasets      | delete          | `delete_dataset`              | :white_check_mark:    |
 | Projects      | list            | `projects`, `list_projects`   | :white_check_mark:    |
 
-## Usage
+## Basic Usage
 
 ```ruby
 # insert
@@ -74,26 +74,43 @@ schema = [
 ]
 client.create_table("new_table", schema)
 
-# SQL
+# sql
 client.sql "SELECT * FROM your_dataset.your_table LIMIT 100"
 
-# SQL (public data)
-client.sql "SELECT * FROM publicdata:samples.wikipedia LIMIT 10"
+# sql (public data)
+client.sql "SELECT born_alive_alive,is_male,weight_pounds FROM publicdata:samples.natality LIMIT 3"
+#=> [{"born_alive_alive"=>0, "is_male"=>true, "weight_pounds"=>8.437090766739999}, {"born_alive_alive"=>2, "is_male"=>true, "weight_pounds"=>6.8122838958}, {"born_alive_alive"=>4, "is_male"=>false, "weight_pounds"=>6.9996768185}]
+
+# query
+client.query "SELECT born_alive_alive,is_male,weight_pounds FROM publicdata:samples.natality LIMIT 3"
+#=> #<struct BigQuery::ResultSet job_id="job_wNWRgrTUJKIi-IUFf9bIqe1mpU8", column_names=["born_alive_alive", "is_male", "weight_pounds"], column_types=["INTEGER", "BOOLEAN", "FLOAT"], records=[["0", "true", "8.437090766739999"], ["2", "true", "6.8122838958"], ["4", "false", "6.9996768185"]]>
 
 # tables
 client.tables
 #=> ["your_table", "your_table2", "your_table3"]
 
-# datasets
-client.datasets
-#=> ["your_dataset", "your_dataset2"]
-
 # fetch schema
 client.fetch_schema("your_table")
 #=> [{"name"=>"nickname", "type"=>"STRING"}, {"name"=>"age", "type"=>"INTEGER"}]
 
 # delete table
 client.delete_table('your_table')
+```
+
+## Datasets API
+
+```ruby
+# No need to specify `:dataset`
+client = BigQuery::Client.new(
+  project:                "your-project-42",
+  email:                  "1234567890@developer.gserviceaccount.com",
+  private_key_path:       "/path/to/keyfile.p12",
+  private_key_passphrase: "notasecret",
+  auth_method:            "private_key"
+)
+
+client.datasets
+#=> ["your_dataset", "your_dataset2"]
 
 # create dataset
 client.create_dataset('your_dataset')
@@ -102,6 +119,21 @@ client.create_dataset('your_dataset')
 client.delete_dataset('your_dataset')
 ```
 
+## Projects API
+
+```ruby
+# No need to specify `:project` and `:dataset`
+client = BigQuery::Client.new(
+  email:                  "1234567890@developer.gserviceaccount.com",
+  private_key_path:       "/path/to/keyfile.p12",
+  private_key_passphrase: "notasecret",
+  auth_method:            "private_key"
+)
+
+client.projects
+#=> ["your_project"]
+```
+
 ## TODO
 
 - [ ] Support all API methods

data/bigquery-client.gemspec

@@ -16,7 +16,7 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ['lib']
 
-  spec.required_ruby_version = '>= 1.9.3'
+  spec.required_ruby_version = '>= 2.0.0'
 
   spec.add_runtime_dependency 'google-api-client', '~> 0.8.0'
 

data/bin/console

@@ -0,0 +1,30 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "bigquery-client"
+
+require 'awesome_print'
+require 'pry'
+require 'pry-byebug'
+require 'vcr'
+
+require 'dotenv'
+Dotenv.load
+
+if defined?(PryByebug)
+  Pry.commands.alias_command 'c', 'continue'
+  Pry.commands.alias_command 's', 'step'
+  Pry.commands.alias_command 'n', 'next'
+  Pry.commands.alias_command 'f', 'finish'
+end
+
+client = BigQuery::Client.new(
+  project:                ENV['BIGQUERY_PROJECT'],
+  email:                  ENV['BIGQUERY_EMAIL'],
+  private_key_path:       ENV['BIGQUERY_PRIVATE_KEY_PATH'],
+  dataset:                'test_bigquery_client_default',
+  private_key_passphrase: 'notasecret',
+  auth_method:            'private_key'
+)
+
+Pry.start

data/lib/bigquery-client.rb

@@ -1,3 +1,13 @@
+require 'bigquery-client/attribute'
+require 'bigquery-client/attribute/base'
+require 'bigquery-client/attribute/boolean'
+require 'bigquery-client/attribute/float'
+require 'bigquery-client/attribute/integer'
+require 'bigquery-client/attribute/string'
+require 'bigquery-client/attribute/timestamp'
+require 'bigquery-client/query_result'
+require 'bigquery-client/run_query'
+require 'bigquery-client/insert_rows'
 require 'bigquery-client/datasets'
 require 'bigquery-client/jobs'
 require 'bigquery-client/projects'

data/lib/bigquery-client/attribute.rb

@@ -0,0 +1,15 @@
+module BigQuery
+  module Attribute
+    class UnknownType < StandardError
+    end
+
+    def self.new(name: nil, type: nil, value: nil)
+      class_name = (type[0] || '').upcase + (type[1..-1] || '').downcase
+      if klass = BigQuery::Attribute.const_get(class_name)
+        klass.new(name, value)
+      else
+        fail UnknownType, "unknown type: #{type}"
+      end
+    end
+  end
+end

data/lib/bigquery-client/attribute/base.rb

@@ -0,0 +1,15 @@
+module BigQuery
+  module Attribute
+    class Base
+      attr_reader :name, :type, :value
+
+      def initialize(name, value)
+        @name, @value = name, value
+        @type = self.class.name.split('::').last.downcase.to_sym
+      end
+
+      def parse
+      end
+    end
+  end
+end

data/lib/bigquery-client/attribute/boolean.rb

@@ -0,0 +1,13 @@
+module BigQuery
+  module Attribute
+    class Boolean < Base
+      def parse
+        case @value
+        when nil     then nil
+        when 'true'  then true
+        when 'false' then false
+        end
+      end
+    end
+  end
+end

data/lib/bigquery-client/attribute/float.rb

@@ -0,0 +1,10 @@
+module BigQuery
+  module Attribute
+    class Float < Base
+      def parse
+        return nil if @value.nil?
+        @value.to_f
+      end
+    end
+  end
+end

data/lib/bigquery-client/attribute/integer.rb

@@ -0,0 +1,10 @@
+module BigQuery
+  module Attribute
+    class Integer < Base
+      def parse
+        return nil if @value.nil?
+        @value.to_i
+      end
+    end
+  end
+end

data/lib/bigquery-client/attribute/string.rb

@@ -0,0 +1,9 @@
+module BigQuery
+  module Attribute
+    class String < Base
+      def parse
+        @value
+      end
+    end
+  end
+end

data/lib/bigquery-client/attribute/timestamp.rb

@@ -0,0 +1,10 @@
+module BigQuery
+  module Attribute
+    class Timestamp < Base
+      def parse
+        return nil if @value.nil?
+        Time.at(@value.to_f)
+      end
+    end
+  end
+end

data/lib/bigquery-client/client.rb

@@ -12,12 +12,12 @@ module BigQuery
     include BigQuery::Tabledata
     include BigQuery::Tables
 
+    private
+
     def initialize(attributes = {})
       attributes.each { |name, value| instance_variable_set("@#{name}", value) }
     end
 
-    private
-
     def access_api(params = {})
       params[:parameters] ||= {}
       params[:parameters][:projectId] ||= @project

data/lib/bigquery-client/insert_rows.rb

@@ -0,0 +1,23 @@
+module BigQuery
+  module Tabledata
+    class InsertRows
+      class InsertError < StandardError
+      end
+
+      def initialize(client, table, arg)
+        @client, @table = client, table
+        @rows = arg.is_a?(Array) ? arg : [arg]
+      end
+
+      def call
+        result = @client.insert_all(@table, @rows)
+        handle_errors(result['insertErrors']) if result['insertErrors']
+        result
+      end
+
+      def handle_errors(errors)
+        fail InsertError, errors
+      end
+    end
+  end
+end

data/lib/bigquery-client/jobs.rb

@@ -3,21 +3,15 @@
 module BigQuery
   module Jobs
     def sql(query, options = {})
-      jobs_query_response = jobs_query(query, options)
-      fields = jobs_query_response['schema']['fields']
-      names = fields.map {|field| field['name'] }
-      types = fields.map {|field| field['type'] }
-      records = extract_records(jobs_query_response)
-      job_id = jobs_query_response['jobReference']['jobId']
-      page_token = jobs_query_response['pageToken']
+      query(query, options).to_a
+    end
 
-      while page_token
-        query_results_response = query_results(job_id, { pageToken: page_token }.merge(options))
-        records += extract_records(query_results_response)
-        page_token = query_results_response['pageToken']
-      end
+    def find_all(query, options = {})
+      raise NotImplementedError
+    end
 
-      convert(records, types).map { |values| [names, values].transpose.to_h }
+    def query(query, options = {})
+      RunQuery.new(self, query, options).call
     end
 
     def jobs_query(query, options = {})
@@ -59,26 +53,5 @@ module BigQuery
         parameters: { jobId: id }.merge(options)
       )
     end
-
-    private
-
-    def extract_records(response)
-      (response['rows'] || []).map {|row| row['f'].map {|record| record['v'] } }
-    end
-
-    def convert(records, types)
-      records.map do |values|
-        values.map.with_index do |value, index|
-          case types[index]
-          when 'INTEGER'
-            value.to_i
-          when 'BOOLEAN'
-            value == 'true'
-          else
-            value
-          end
-        end
-      end
-    end
   end
 end

data/lib/bigquery-client/query_result.rb

@@ -0,0 +1,12 @@
+module BigQuery
+  class ResultSet < Struct.new(:job_id, :column_names, :column_types, :records)
+    def to_a
+      records.map {|record|
+        values = record.map.with_index do |value, index|
+          Attribute.new(value: value, type: column_types[index]).parse
+        end
+        [column_names, values].transpose.to_h
+      }
+    end
+  end
+end

data/lib/bigquery-client/run_query.rb

@@ -0,0 +1,40 @@
+module BigQuery
+  module Jobs
+    class RunQuery
+      def initialize(client, query, options)
+        @client, @query, @options = client, query, options
+      end
+
+      def call
+        @result = ResultSet.new
+        execute_query
+        fetch_pagenated_result
+        @result
+      end
+
+      private
+
+      def execute_query
+        response = @client.jobs_query(@query, @options)
+        @page_token = response['pageToken']
+        fields = response['schema']['fields']
+        @result.job_id = response['jobReference']['jobId']
+        @result.column_names = fields.map {|field| field['name'] }
+        @result.column_types = fields.map {|field| field['type'] }
+        @result.records = extract_records(response)
+      end
+
+      def fetch_pagenated_result
+        while @page_token
+          response = @client.query_results(@result.job_id, { pageToken: @page_token }.merge(@options))
+          @result.records += extract_records(response)
+          @page_token = response['pageToken']
+        end
+      end
+
+      def extract_records(response)
+        (response['rows'] || []).map {|row| row['f'].map {|record| record['v'] } }
+      end
+    end
+  end
+end

data/lib/bigquery-client/tabledata.rb

@@ -3,7 +3,10 @@
 module BigQuery
   module Tabledata
     def insert(table, arg)
-      rows = arg.is_a?(Array) ? arg : [arg]
+      InsertRows.new(self, table, arg).call
+    end
+
+    def insert_all(table, rows)
       access_api(
         api_method: bigquery.tabledata.insert_all,
         parameters: {
@@ -14,7 +17,6 @@ module BigQuery
         }
       )
     end
-    alias_method :insert_all, :insert
 
     def list_tabledata(table)
       access_api(

data/lib/bigquery-client/version.rb

@@ -1,5 +1,5 @@
 module BigQuery
   class Client
-    VERSION = '0.3.4'
+    VERSION = '0.4.0'
   end
 end

data/test/helper.rb

@@ -5,6 +5,9 @@ require 'bigquery-client'
 require 'dotenv'
 require 'awesome_print'
 require 'pry-byebug'
+require 'vcr'
+
+Dotenv.load
 
 if defined?(PryByebug)
   Pry.commands.alias_command 'c', 'continue'
@@ -13,9 +16,13 @@ if defined?(PryByebug)
   Pry.commands.alias_command 'f', 'finish'
 end
 
-Dotenv.load
+VCR.configure do |config|
+  config.cassette_library_dir = "vcr_cassettes"
+  config.hook_into :webmock
+end
 
-$dataset = 'test_bigquery_client'
+$prefix  = 'test_bigquery_client'
+$dataset = "#{$prefix}_default"
 
 $client = BigQuery::Client.new(
   project:                ENV['BIGQUERY_PROJECT'],
@@ -26,10 +33,19 @@ $client = BigQuery::Client.new(
   auth_method:            'private_key'
 )
 
-$client.datasets.select { |dataset|
-  Regexp.new($dataset) =~ dataset
-}.each { |dataset|
-  $client.delete_dataset(dataset)
-}
+VCR.use_cassette("helper") do
+  $client.datasets.each do |dataset|
+    $client.delete_dataset(dataset) if Regexp.new($prefix) =~ dataset
+  end
+  $client.create_dataset($dataset)
+end
+
+class ApiTest < Test::Unit::TestCase
+  def setup
+    VCR.insert_cassette("#{self.class}_#{@method_name}")
+  end
 
-$client.create_dataset($dataset)
+  def cleanup
+    VCR.eject_cassette
+  end
+end

data/test/test_attribute.rb

@@ -0,0 +1,66 @@
+require 'helper'
+
+class AttributeTest < Test::Unit::TestCase
+  def test_attribute
+    attribute = BigQuery::Attribute.new(name: 'is_admin', type: 'BOOLEAN', value: 'true')
+    assert { attribute.parse == true }
+    assert { attribute.name  == 'is_admin' }
+    assert { attribute.value == 'true' }
+    assert { attribute.type  == :boolean }
+  end
+
+  def test_string
+    attribute = BigQuery::Attribute::String.new(nil, 'foo')
+    assert { attribute.parse == 'foo' }
+    assert { attribute.type  == :string }
+  end
+
+  def test_string_nil
+    attribute = BigQuery::Attribute::String.new(nil, nil)
+    assert { attribute.parse.nil? }
+  end
+
+  def test_integer
+    attribute = BigQuery::Attribute::Integer.new(nil, '42')
+    assert { attribute.parse == 42 }
+    assert { attribute.type  == :integer }
+  end
+
+  def test_integer_nil
+    attribute = BigQuery::Attribute::Integer.new(nil, nil)
+    assert { attribute.parse.nil? }
+  end
+
+  def test_float
+    attribute = BigQuery::Attribute::Float.new(nil, '6.4992274837599995')
+    assert { attribute.parse == 6.4992274837599995 }
+    assert { attribute.type  == :float }
+  end
+
+  def test_float_nil
+    attribute = BigQuery::Attribute::Float.new(nil, nil)
+    assert { attribute.parse.nil? }
+  end
+
+  def test_boolean
+    attribute = BigQuery::Attribute::Boolean.new(nil, 'false')
+    assert { attribute.parse == false }
+    assert { attribute.type  == :boolean }
+  end
+
+  def test_boolean_nil
+    attribute = BigQuery::Attribute::Boolean.new(nil, nil)
+    assert { attribute.parse.nil? }
+  end
+
+  def test_timestamp
+    attribute = BigQuery::Attribute::Timestamp.new(nil, "-5.51952E7")
+    assert { attribute.parse == Time.new(1968, 4, 2, 13, 0, 0) }
+    assert { attribute.type  == :timestamp }
+  end
+
+  def test_timestamp_nil
+    attribute = BigQuery::Attribute::Timestamp.new(nil, nil)
+    assert { attribute.parse.nil? }
+  end
+end

data/test/test_client.rb

@@ -1,6 +1,6 @@
 require 'helper'
 
-class ClientTest < Test::Unit::TestCase
+class ClientTest < ApiTest
   def test_initialize
     actual  = $client.instance_variables
     minimal = [:@project, :@dataset, :@email, :@private_key_path, :@private_key_passphrase, :@auth_method]

data/test/test_datasets.rb

@@ -1,25 +1,31 @@
 require 'helper'
 
-class DatasetsTest < Test::Unit::TestCase
+class DatasetsTest < ApiTest
   def test_datasets
-    dataset_name = "#{$dataset}_test_datasets"
-    $client.create_dataset(dataset_name)
-    $client.datasets.include?(dataset_name)
+    dataset_name = "#{$prefix}_#{__method__}"
+    unless $client.datasets.include?(dataset_name)
+      $client.create_dataset(dataset_name)
+    end
+    assert { $client.datasets.include?(dataset_name) }
   end
 
   def test_create_dataset
-    dataset_name = "#{$dataset}_create_dataset"
+    dataset_name = "#{$prefix}_#{__method__}"
+    if $client.datasets.include?(dataset_name)
+      $client.delete_dataset(dataset_name)
+    end
+    before = $client.datasets
     $client.create_dataset(dataset_name)
+    after = $client.datasets
+    assert { (after - before) == [dataset_name] }
   end
 
   def test_delete_dataset
-    dataset_name = "#{$dataset}_delete_dataset"
+    dataset_name = "#{$prefix}_#{__method__}"
     $client.create_dataset(dataset_name)
     before = $client.datasets
     $client.delete_dataset(dataset_name)
     after = $client.datasets
-    actual = before - after
-    expected = [dataset_name]
-    assert { actual == expected }
+    assert { (before - after) == [dataset_name] }
   end
 end

data/test/test_jobs.rb

@@ -1,9 +1,9 @@
 require 'helper'
 
-class JobsTest < Test::Unit::TestCase
-  @@normal_query = <<-"EOS"
+class JobsTest < ApiTest
+  NORMAL_QUERY = <<-"EOS"
     SELECT
-      born_alive_alive, mother_residence_state, is_male
+      born_alive_alive, mother_residence_state, is_male, weight_pounds
     FROM
       publicdata:samples.natality
     ORDER BY
@@ -12,41 +12,41 @@ class JobsTest < Test::Unit::TestCase
       100
   EOS
 
-  @@no_rows_query = <<-"EOS"
-    SELECT
-      born_alive_alive, mother_residence_state, is_male
-    FROM
-      publicdata:samples.natality
-    ORDER BY
-      day
-    LIMIT
-      0
-  EOS
-
-  @@huge_result_query = <<-"EOS"
-    SELECT
-      title
-    FROM
-      publicdata:samples.wikipedia
-    LIMIT
-      1234567
-  EOS
-
   def test_sql
-    result = $client.sql(@@normal_query)
+    result = $client.sql NORMAL_QUERY
     assert { result.size == 100 }
     assert { result.sample["born_alive_alive"].is_a? Fixnum }
     assert { result.sample["mother_residence_state"].is_a? String }
+    assert { result.sample["weight_pounds"].is_a? Float }
     assert { result.first["is_male"] == true || result.first["is_male"] == false }
   end
 
+  def test_sql_timestamp_columns
+    table_name = 'test_timestamp'
+    $client.create_table(table_name, [{ name: 'time', type: 'timestamp' }])
+    $client.insert(table_name, time: "2015-10-13 20:05:43")
+    result = $client.sql('SELECT * FROM test_bigquery_client_default.test_timestamp LIMIT 1')
+    assert { result.last["time"] == Time.parse("2015-10-13 20:05:43 UTC") }
+  end
+
+  def test_query
+    result = $client.query NORMAL_QUERY
+    assert { result.column_names == %w(born_alive_alive mother_residence_state is_male weight_pounds) }
+    assert { result.column_types == %w(INTEGER STRING BOOLEAN FLOAT) }
+    records = result.records
+    assert { records.size == 100 }
+    assert { records.is_a?(Array) && records.sample.is_a?(Array) }
+    assert { records.sample.size == 4 }
+    assert { records.sample.all? {|value| value.is_a?(String) } }
+  end
+
   def test_sql_when_no_rows
-    result = $client.sql(@@no_rows_query)
-    assert { result == [] }
+    no_rows_query = 'SELECT * FROM publicdata:samples.natality LIMIT 0'
+    assert { $client.sql(no_rows_query) == [] }
   end
 
   def test_sql_pagination
-    record_size = $client.sql(@@huge_result_query, maxResults: 10000).size
-    assert { record_size == 1234567 }
+    pagination_query = 'SELECT title FROM publicdata:samples.wikipedia LIMIT 123'
+    assert { $client.sql(pagination_query, maxResults: 100).size == 123 }
   end
 end

data/test/test_projects.rb

@@ -1,15 +1,7 @@
 require 'helper'
 
-class ProjectsTest < Test::Unit::TestCase
+class ProjectsTest < ApiTest
   def test_projects
-    actual = $client.projects.first
-    expected = ENV['BIGQUERY_PROJECT']
-    assert { actual == expected }
-  end
-
-  def test_list_projects
-    actual = $client.list_projects['projects'].first['id']
-    expected = ENV['BIGQUERY_PROJECT']
-    assert { actual == expected }
+    assert { $client.projects.include? ENV['BIGQUERY_PROJECT'] }
   end
 end

data/test/test_tabledata.rb

@@ -1,12 +1,20 @@
 require 'helper'
 
-class TablesTest < Test::Unit::TestCase
+class TabledataTest < ApiTest
   def test_insert
     table_name = __method__.to_s
     schema = [{ name: 'bar', type: 'string' }]
     $client.create_table(table_name, schema)
-    result = $client.insert(table_name, bar: "foo")
-    assert { result['kind'] == 'bigquery#tableDataInsertAllResponse' }
+    assert { $client.insert(table_name, bar: "foo") == { 'kind' => 'bigquery#tableDataInsertAllResponse' } }
+  end
+
+  def test_insert_invalid_timestamp
+    table_name = __method__.to_s
+    schema = [{ name: 'time', type: 'timestamp' }]
+    $client.create_table(table_name, schema)
+    assert_raise(BigQuery::Tabledata::InsertRows::InsertError) do
+      $client.insert(table_name, time: "invalid_timestamp")
+    end
   end
 
   def test_insert_with_array
@@ -15,8 +23,7 @@ class TablesTest < Test::Unit::TestCase
     $client.create_table(table_name, schema)
     result = $client.list_tabledata(table_name)
     rows = [{ bar: "foo" },  { bar: "foo2" },  { bar: "foo3" }]
-    result = $client.insert(table_name, rows)
-    assert { result['kind'] == 'bigquery#tableDataInsertAllResponse' }
+    assert { $client.insert(table_name, rows) == { 'kind' => 'bigquery#tableDataInsertAllResponse' } }
   end
 
   def test_list_tabledata

data/test/test_tables.rb

@@ -1,6 +1,6 @@
 require 'helper'
 
-class TablesTest < Test::Unit::TestCase
+class TablesTest < ApiTest
   def test_tables
     table_name = __method__.to_s
     schema = [{ name: 'bar', type: 'string' }]
@@ -8,22 +8,11 @@ class TablesTest < Test::Unit::TestCase
     assert { $client.tables.include?(table_name) }
   end
 
-  def test_table_pagination
-    $client.tables.each {|t| $client.delete_table(t) }
-    3.times do |i|
-      table_name = __method__.to_s + "_#{i.to_s}"
-      schema = [{ name: 'bar', type: 'string' }]
-      $client.create_table(table_name, schema)
-    end
-    assert { $client.tables(maxResults: 1).count == 3 }
-  end
-
   def test_fetch_schema
     table_name = __method__.to_s
     schema = [{ name: 'bar', type: 'string' }]
     $client.create_table(table_name, schema)
-    result = $client.fetch_schema(table_name)
-    assert { result == [{"name"=>"bar", "type"=>"STRING"}] }
+    assert { $client.fetch_schema(table_name) == [{"name"=>"bar", "type"=>"STRING"}] }
   end
 
   def test_create_table
@@ -31,10 +20,8 @@ class TablesTest < Test::Unit::TestCase
     schema = [{ name: 'bar', type: 'string' }]
     before = $client.tables
     $client.create_table(table_name, schema)
-    after  = $client.tables
-    actual = after - before
-    expected = [table_name]
-    assert { expected == actual }
+    after = $client.tables
+    assert { (after - before) == [table_name] }
   end
 
   def test_delete_table
@@ -44,8 +31,16 @@ class TablesTest < Test::Unit::TestCase
     before = $client.tables
     $client.delete_table(table_name)
     after  = $client.tables
-    actual = before - after
-    expected = [table_name]
-    assert { actual == expected }
+    assert { (before - after) == [table_name] }
+  end
+
+  def test_table_pagination
+    $client.tables.each {|t| $client.delete_table(t) }
+    3.times do |i|
+      table_name = __method__.to_s + "_#{i.to_s}"
+      schema = [{ name: 'bar', type: 'string' }]
+      $client.create_table(table_name, schema)
+    end
+    assert { $client.tables(maxResults: 1).count == 3 }
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bigquery-client
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  version: 0.4.0
 platform: ruby
 authors:
 - Tsukuru Tanimichi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-10 00:00:00.000000000 Z
+date: 2015-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: google-api-client
@@ -55,7 +55,8 @@ dependencies:
 description: 
 email:
 - ttanimichi@hotmail.com
-executables: []
+executables:
+- console
 extensions: []
 extra_rdoc_files: []
 files:
@@ -66,16 +67,28 @@ files:
 - README.md
 - Rakefile
 - bigquery-client.gemspec
+- bin/console
 - lib/bigquery-client.rb
+- lib/bigquery-client/attribute.rb
+- lib/bigquery-client/attribute/base.rb
+- lib/bigquery-client/attribute/boolean.rb
+- lib/bigquery-client/attribute/float.rb
+- lib/bigquery-client/attribute/integer.rb
+- lib/bigquery-client/attribute/string.rb
+- lib/bigquery-client/attribute/timestamp.rb
 - lib/bigquery-client/client.rb
 - lib/bigquery-client/datasets.rb
 - lib/bigquery-client/errors.rb
+- lib/bigquery-client/insert_rows.rb
 - lib/bigquery-client/jobs.rb
 - lib/bigquery-client/projects.rb
+- lib/bigquery-client/query_result.rb
+- lib/bigquery-client/run_query.rb
 - lib/bigquery-client/tabledata.rb
 - lib/bigquery-client/tables.rb
 - lib/bigquery-client/version.rb
 - test/helper.rb
+- test/test_attribute.rb
 - test/test_client.rb
 - test/test_datasets.rb
 - test/test_jobs.rb
@@ -94,7 +107,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 1.9.3
+      version: 2.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -102,12 +115,13 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.5
+rubygems_version: 2.4.5.1
 signing_key: 
 specification_version: 4
 summary: A Ruby interface to the BigQuery API.
 test_files:
 - test/helper.rb
+- test/test_attribute.rb
 - test/test_client.rb
 - test/test_datasets.rb
 - test/test_jobs.rb