fix MSSQL test stubs

bug/bundler_fix
Greg Mikeska 2015-12-09 10:02:35 -06:00 committed by Brent Cook
parent 5c733a3b6d
commit b84bd0327d
1 changed files with 6 additions and 6 deletions

View File

@ -5,11 +5,11 @@ require 'msf/core/post/windows/mssql'
RSpec.describe Msf::Post::Windows::MSSQL do RSpec.describe Msf::Post::Windows::MSSQL do
let(:subject) do let(:subject) do
mod = Module.new mod = double(Module.new)
mod.extend described_class mod.extend described_class
stubs = [ :vprint_status, :print_status, :vprint_good, :print_good, :print_error, :print_warning ] stubs = [ :vprint_status, :print_status, :vprint_good, :print_good, :print_error, :print_warning ]
stubs.each { |meth| expect(mod).to receive(meth) } stubs.each { |meth| allow(mod).to receive(meth) }
expect(mod).to receive(:service_info).and_return({}) allow(mod).to receive(:service_info).and_return({})
mod mod
end end
@ -288,8 +288,8 @@ RSpec.describe Msf::Post::Windows::MSSQL do
it 'should return false if service is invalid or pid is invalid' do it 'should return false if service is invalid or pid is invalid' do
expect(subject.impersonate_sql_user(nil)).to be_falsey expect(subject.impersonate_sql_user(nil)).to be_falsey
subject.impersonate_sql_user(pid: expect(nil)).to be_falsey expect(subject.impersonate_sql_user(pid: nil)).to be_falsey
subject.impersonate_sql_user(pid: expect(0)).to be_falsey expect(subject.impersonate_sql_user(pid: 0)).to be_falsey
end end
context 'user has privs to impersonate' do context 'user has privs to impersonate' do
@ -361,7 +361,7 @@ RSpec.describe Msf::Post::Windows::MSSQL do
it 'should return a string' do it 'should return a string' do
p = double('process') p = double('process')
c = double('channel') c = double('channel')
expect(p).to receive(:channel).and_return(c) allow(p).to receive(:channel).and_return(c)
allow(subject).to receive_message_chain('session.sys.process.execute').and_return(p) allow(subject).to receive_message_chain('session.sys.process.execute').and_return(p)
expect(c).to receive(:read).and_return('hello') expect(c).to receive(:read).and_return('hello')
expect(c).to receive(:read).and_return(nil) expect(c).to receive(:read).and_return(nil)