citrusbyte-schemer 0.0.9 → 0.0.10

data/lib/schemer.rb

@@ -37,13 +37,20 @@ module Schemer
     def update_schema
       create_table unless table_exists?
 
+      columns.reject{ |column| protected_columns.include?(column.name) }.each do |column|
+        if !schema_columns.has_key?(column.name)
+          # remove any extraneous columns
+          ActiveRecord::Migration.remove_column(table_name, column.name)
+        elsif column.type != schema_columns[column.name]
+          # change any columns w/ wrong type
+          ActiveRecord::Migration.change_column(table_name, column.name, schema_columns[column.name])
+        end
+      end
+
+      # add any missing columns
       (schema_columns.keys - column_names).each do |column|
         ActiveRecord::Migration.add_column(table_name, column, schema_columns[column])
       end
-      
-      (column_names - protected_columns - schema_columns.keys).each do |column|
-        ActiveRecord::Migration.remove_column(table_name, column)
-      end
     end
   end
 end

data/test/schemer_test.rb

@@ -69,6 +69,12 @@ class FooTest < Test::Unit::TestCase
     should "create foo column using string datatype" do
       assert_equal '5', @foo.foo
     end
+    
+    should "recreate foo column as integer" do
+      Foo.schema({ :foo => :integer }, { :bar => :integer }, :baz)
+      foo = Foo.find(Foo.create!(:foo => '5', :bar => '5').id)
+      assert_equal 5, foo.foo
+    end
   end
   
   context "building a Rails migration" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: citrusbyte-schemer
 version: !ruby/object:Gem::Version 
-  version: 0.0.9
+  version: 0.0.10
 platform: ruby
 authors: 
 - Ben Alavi
@@ -27,9 +27,9 @@ files:
 - Rakefile
 - lib/schemer/migrator.rb
 - lib/schemer.rb
+- lib/schemer/tasks/schemer.rake
 - rails/init.rb
 - test/schemer_test.rb
-- tasks/schemer.rake
 has_rdoc: false
 homepage: http://github.com/citrusbyte/schemer
 post_install_message: