Accept symbol as a column name and type

tokumine · Mar 17, 2012 · 02c9e34 · 02c9e34
1 parent 8e97554
commit 02c9e34
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lib/fusion_tables/ext/fusion_tables.rb b/lib/fusion_tables/ext/fusion_tables.rb
@@ -50,6 +50,9 @@ def create_table(table_name, columns)
 
         # ensure all column types are valid
         columns.each do |col|
+          col[:name] = col[:name].to_s
+          col[:type] = col[:type].to_s
+
           if !DATATYPES.include? col[:type].downcase
             raise ArgumentError, "Ensure input types are: 'number', 'string', 'location' or 'datetime'"
           end

diff --git a/test/test_ext.rb b/test/test_ext.rb
@@ -24,6 +24,13 @@ class TestExt < Test::Unit::TestCase
       assert_equal GData::Client::FusionTables::Table, @table.class
     end
 
+    should "accept symbol for name and type" do
+      @table = @ft.create_table "test_table", [{:name => :test_col, :type => :string }]
+      first_column = @table.describe[0]
+      assert_equal 'test_col', first_column[:name]
+      assert_equal 'string', first_column[:type]
+    end
+
     should "correct your table name to a certain degree on create" do
       @table = @ft.create_table "test table", [{:name => "test col", :type => "string" }]
       assert_equal "test_table", @table.name