Fix migrate failures on vms with missing spec folder

cloudfoundry · Jan 2, 2024 · b7f3277 · b7f3277
1 parent 596ded9
commit b7f3277
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/lib/tasks/db.rake b/lib/tasks/db.rake
@@ -244,16 +244,24 @@ namespace :db do
   def migrate
     Steno.init(Steno::Config.new(sinks: [Steno::Sink::IO.new($stdout)]))
     db_logger = Steno.logger('cc.db.migrations')
-    require_relative '../../spec/support/bootstrap/db_config'
-    DbConfig.new
+    begin
+      require_relative '../../spec/support/bootstrap/db_config'
+      DbConfig.new
+    rescue LoadError
+      # Only needed when running tests
+    end
     DBMigrator.from_config(RakeConfig.config, db_logger).apply_migrations
   end
 
   def rollback(number_to_rollback)
     Steno.init(Steno::Config.new(sinks: [Steno::Sink::IO.new($stdout)]))
     db_logger = Steno.logger('cc.db.migrations')
-    require_relative '../../spec/support/bootstrap/db_config'
-    DbConfig.new
+    begin
+      require_relative '../../spec/support/bootstrap/db_config'
+      DbConfig.new
+    rescue LoadError
+      # Only needed when running tests
+    end
     DBMigrator.from_config(RakeConfig.config, db_logger).rollback(number_to_rollback)
   end