Don't load container models from app folder in migration

db/migrate/20160119125950_add_created_on_for_container_entities.rb

@@ -1,15 +1,43 @@
 class AddCreatedOnForContainerEntities < ActiveRecord::Migration
-  CONTAINER_TABLES = [:container_nodes, :container_projects, :container_services, :container_routes, :container_groups,
-                      :container_replicators, :container_quotas, :container_builds, :container_build_pods,
-                      :container_limits, :container_volumes]
+  class ContainerNode < ActiveRecord::Base
+    self.inheritance_column = :_type_disabled # disable STI
+  end
+
+  class ContainerProject < ActiveRecord::Base; end
+
+  class ContainerService < ActiveRecord::Base; end
+
+  class ContainerRoute < ActiveRecord::Base; end
+
+  class ContainerGroup < ActiveRecord::Base
+    self.inheritance_column = :_type_disabled # disable STI
+  end
+
+  class ContainerReplicator < ActiveRecord::Base; end
+
+  class ContainerQuota < ActiveRecord::Base; end
+
+  class ContainerBuild < ActiveRecord::Base; end
+
+  class ContainerBuildPod < ActiveRecord::Base; end
+
+  class ContainerLimit < ActiveRecord::Base; end
+
+  class ContainerVolume < ActiveRecord::Base
+    self.inheritance_column = :_type_disabled # disable STI
+  end
+
+  CONTAINER_MODELS = [ContainerNode, ContainerProject, ContainerService, ContainerRoute, ContainerGroup,
+                      ContainerReplicator, ContainerQuota, ContainerBuild, ContainerBuildPod, ContainerLimit,
+                      ContainerVolume].freeze
 
   def change
-    CONTAINER_TABLES.each do |t|
-      add_column t, :created_on, :datetime
-      rename_column t, :creation_timestamp, :ems_created_on
+    CONTAINER_MODELS.each do |model|
+      add_column model.table_name, :created_on, :datetime
+      rename_column model.table_name, :creation_timestamp, :ems_created_on
 
-      say_with_time("adding created_on datetime to all existing #{t.to_s.tr("_", " ")}") do
-        t.to_s.classify.constantize.update_all("created_on=ems_created_on")
+      say_with_time("adding created_on datetime to all existing #{model}") do
+        model.update_all("created_on=ems_created_on")
       end
     end
   end

spec/migrations/20160119125950_add_created_on_for_container_entities_spec.rb

@@ -0,0 +1,43 @@
+require_migration
+
+describe AddCreatedOnForContainerEntities do
+  CONTAINER_TABLES = described_class::CONTAINER_MODELS.collect { |m| m.table_name.to_sym }
+
+  CONTAINER_TABLES.each { |table| let("#{table}_stub") { migration_stub(table.to_s.classify.to_sym) } }
+
+  let(:mock_timestamp) { Time.zone.parse('2016-03-09 12:00:38.711120') }
+  let(:mock_name)      { "Name_1" }
+
+  migration_context :up do
+    it "populates new column created_on and ems_created_on with value from creation_timestamp" do
+      records = {}
+
+      CONTAINER_TABLES.each do |table|
+        records[table] = send("#{table}_stub").create!(:name => mock_name, :creation_timestamp => mock_timestamp)
+      end
+
+      migrate
+
+      CONTAINER_TABLES.each do |table|
+        expect(records[table].reload.created_on).to eq(mock_timestamp)
+        expect(records[table].reload.ems_created_on).to eq(mock_timestamp)
+      end
+    end
+  end
+
+  migration_context :down do
+    it "renames ems_created_on back to creation_timestamp and it has same value as ems_created_on" do
+      records = {}
+
+      CONTAINER_TABLES.each do |table|
+        records[table] = send("#{table}_stub").create!(:name => mock_name, :ems_created_on => mock_timestamp)
+      end
+
+      migrate
+
+      CONTAINER_TABLES.each do |table|
+        expect(records[table].reload.creation_timestamp).to eq(mock_timestamp)
+      end
+    end
+  end
+end