diff --git a/spec/beef/core/main/command_spec.rb b/spec/beef/core/main/command_spec.rb index 0b7d8bc2b..e03b4336a 100644 --- a/spec/beef/core/main/command_spec.rb +++ b/spec/beef/core/main/command_spec.rb @@ -1,11 +1,13 @@ RSpec.describe 'BeEF Command class testing' do before(:each) do # Reset or re-initialise the configuration to a default state - @config_instance = BeEF::Core::Configuration.instance + # @config_instance = BeEF::Core::Configuration.instance end it 'should return a beef configuration variable' do - BeEF::Modules.load + expect { + BeEF::Modules.load if BeEF::Core::Configuration.instance.get('beef.module').nil? + }.to_not raise_error command_mock = BeEF::Core::Command.new('test_get_variable') expect(command_mock.config.beef_host).to eq('0.0.0.0') diff --git a/spec/beef/core/modules_spec.rb b/spec/beef/core/modules_spec.rb index f02b7a805..f9cb31937 100644 --- a/spec/beef/core/modules_spec.rb +++ b/spec/beef/core/modules_spec.rb @@ -2,7 +2,7 @@ RSpec.describe 'BeEF Modules' do it 'loaded successfully' do expect { - BeEF::Modules.load + BeEF::Modules.load if BeEF::Core::Configuration.instance.get('beef.module').nil? }.to_not raise_error modules = BeEF::Core::Configuration.instance.get('beef.module').select do |k,v|