---
 .../database/001_add_created_at_to_all_tables.rb   |    2 +-
 .../rails/database/002_add_environment_to_host.rb  |    4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/lib/puppet/rails/database/001_add_created_at_to_all_tables.rb 
b/lib/puppet/rails/database/001_add_created_at_to_all_tables.rb
index 71ee6ae..b1d3e20 100644
--- a/lib/puppet/rails/database/001_add_created_at_to_all_tables.rb
+++ b/lib/puppet/rails/database/001_add_created_at_to_all_tables.rb
@@ -9,7 +9,7 @@ class AddCreatedAtToAllTables < ActiveRecord::Migration
 
   def self.down
     ActiveRecord::Base.connection.tables.each do |t|
-      unless ActiveRecord::Base.connection.columns(t).collect {|c| 
c.name}.include?("created_at")
+      if ActiveRecord::Base.connection.columns(t).collect {|c| 
c.name}.include?("created_at")
         remove_column t.to_s, :created_at
       end  
     end  
diff --git a/lib/puppet/rails/database/002_add_environment_to_host.rb 
b/lib/puppet/rails/database/002_add_environment_to_host.rb
index 712265a..d01cb83 100644
--- a/lib/puppet/rails/database/002_add_environment_to_host.rb
+++ b/lib/puppet/rails/database/002_add_environment_to_host.rb
@@ -1,12 +1,12 @@
 class AddEnvironmentToHost < ActiveRecord::Migration
     def self.up
-        unless ActiveRecord::Base.connection.columns("hosts").collect {|c| 
c.name}.include?("created_at")
+        unless ActiveRecord::Base.connection.columns("hosts").collect {|c| 
c.name}.include?("environment")
             add_column :hosts, :environment, :string 
         end    
     end
 
     def self.down
-        unless ActiveRecord::Base.connection.columns("hosts").collect {|c| 
c.name}.include?("created_at")
+        if ActiveRecord::Base.connection.columns("hosts").collect {|c| 
c.name}.include?("environment")
             remove_column :hosts, :environment
         end    
     end
-- 
1.5.4.3


--~--~---------~--~----~------------~-------~--~----~
You received this message because you are subscribed to the Google Groups 
"Puppet Developers" group.
To post to this group, send email to [email protected]
To unsubscribe from this group, send email to [EMAIL PROTECTED]
For more options, visit this group at 
http://groups.google.com/group/puppet-dev?hl=en
-~----------~----~----~----~------~----~------~--~---

Reply via email to