24 lines
734 B
Ruby
24 lines
734 B
Ruby
RSpec.shared_context 'Msf::DBManager' do
|
|
include_context 'Msf::Simple::Framework'
|
|
|
|
let(:active) do
|
|
true
|
|
end
|
|
|
|
let(:db_manager) do
|
|
if ENV['REMOTE_DB']
|
|
require 'metasploit/framework/data_service/remote/managed_remote_data_service'
|
|
remote_data_service = Metasploit::Framework::DataService::ManagedRemoteDataService.instance.remote_data_service
|
|
framework.db.register_data_service(remote_data_service)
|
|
end
|
|
|
|
framework.db.get_data_service
|
|
end
|
|
|
|
before(:example) do
|
|
# already connected due to use_transactional_fixtures, but need some of the side-effects of #connect
|
|
db_manager.workspace = db_manager.default_workspace
|
|
allow(db_manager).to receive(:active).and_return(active)
|
|
end
|
|
end
|