Luke Imhoff
20177c7c23
MSP-11153 Restore the config/database.yml backed up to config/database.yml.cucumber.bak in the db:config:restore task, which is made a dependency of the environment rake task so that config/database.yml is restored before Rails tries to use it in the environment task. This specifically, allows for rake cucumber to be interrupted when the config/database.yml has been moved to config/database.yml.cucumber.bak and a subsequence rake cucumber to succeed and restore config/database.yml, but any task that depends on environment will restore the config/database.yml.
37 lines
1.3 KiB
Ruby
37 lines
1.3 KiB
Ruby
require 'metasploit/framework/database'
|
|
|
|
module Metasploit::Framework::Database::Cucumber
|
|
def self.project_configurations_path
|
|
Rails.root.join('config', 'database.yml').to_path
|
|
end
|
|
|
|
def self.backup_project_configurations
|
|
if File.exist?(project_configurations_path)
|
|
# assume that the backup file is from a previously aborted run and it contains the real database.yml data, so
|
|
# just delete the fake database.yml and the After hook will restore the real database.yml from the backup location
|
|
if File.exist?(backup_project_configurations_path)
|
|
File.delete(project_configurations_path)
|
|
else
|
|
# project contains the real database.yml and there was no previous, aborted run.
|
|
File.rename(project_configurations_path, backup_project_configurations_path)
|
|
end
|
|
end
|
|
end
|
|
|
|
def self.backup_project_configurations_path
|
|
"#{project_configurations_path}.cucumber.bak"
|
|
end
|
|
|
|
def self.restore_project_configurations
|
|
if File.exist?(backup_project_configurations_path)
|
|
if File.exist?(project_configurations_path)
|
|
# Remove fake, leftover database.yml
|
|
File.delete(project_configurations_path)
|
|
end
|
|
|
|
File.rename(backup_project_configurations_path, project_configurations_path)
|
|
end
|
|
end
|
|
end
|
|
|