spec/postgres/lib/table_reconcile_spec.rb in volt-sql-0.0.3 vs spec/postgres/lib/table_reconcile_spec.rb in volt-sql-0.0.4

@@ -106,11 +106,11 @@
     reconcile = Volt::Sql::TableReconcile.new(db_adaptor, db, SampleModel3)
 
     allow(reconcile.field_updater).to receive(:generate_and_run)
     .with(
       "column_change_sample_model3s_some_num",
-      "set_column_type :sample_model3s, :some_num, String, {:allow_null=>true, :text=>true}",
+      "if column_exists?(:sample_model3s, :some_num)\n  set_column_type :sample_model3s, :some_num, String, {:allow_null=>true, :text=>true}\nend",
       "set_column_type :sample_model3s, :some_num, Fixnum, {:allow_null=>true}"
     ).and_return(nil)
 
     reconcile.run
   end
@@ -136,11 +136,11 @@
     reconcile = Volt::Sql::TableReconcile.new(db_adaptor, db, SampleModel4)
 
     expect(reconcile.field_updater).to receive(:generate_and_run)
     .with(
       "remove_sample_model4s_some_num",
-      "drop_column :sample_model4s, :some_num",
+      "if column_exists?(:sample_model4s, :some_num)\n  drop_column :sample_model4s, :some_num\nend",
       "add_column :sample_model4s, :some_num, Fixnum, {:allow_null=>true}"
     )
 
     reconcile.run
   end
@@ -167,10 +167,10 @@
     reconcile = Volt::Sql::TableReconcile.new(db_adaptor, db, SampleModel5)
 
     expect(reconcile.field_updater).to receive(:generate_and_run)
     .with(
       "column_change_sample_model5s_some_num",
-      "set_column_not_null :sample_model5s, :some_num",
+      "if column_exists?(:sample_model5s, :some_num)\n  set_column_not_null :sample_model5s, :some_num\nend",
       "set_column_allow_null :sample_model5s, :some_num"
     )
 
     reconcile.run
   end