Added spec to new methods in Jboss mixin
parent
7125a9f047
commit
f9b6519e49
|
@ -58,4 +58,72 @@ describe Msf::HTTP::JBoss::Base do
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
describe "#query_serverinfo" do
|
||||||
|
before :each do
|
||||||
|
allow(subject).to receive(:send_request_cgi) do
|
||||||
|
if res_code.nil?
|
||||||
|
res = nil
|
||||||
|
else
|
||||||
|
res = Rex::Proto::Http::Response.new
|
||||||
|
res.code = res_code
|
||||||
|
end
|
||||||
|
|
||||||
|
res
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
context 'when server timeouts' do
|
||||||
|
let(:res_code) { nil }
|
||||||
|
it { expect(subject.query_serverinfo()).to be_nil }
|
||||||
|
end
|
||||||
|
|
||||||
|
context 'when server returns 200' do
|
||||||
|
let(:res_code) { 200 }
|
||||||
|
it { expect(subject.query_serverinfo()).to be_kind_of Rex::Proto::Http::Response }
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
describe "#detect_plateform" do
|
||||||
|
context "when server arch is Linux" do
|
||||||
|
res = Rex::Proto::Http::Response.new
|
||||||
|
res.body = "<td>OSName: Linux</td>"
|
||||||
|
it { expect(subject.detect_platform(res)).to eq "linux" }
|
||||||
|
end
|
||||||
|
|
||||||
|
context "when server arch is Windows" do
|
||||||
|
res = Rex::Proto::Http::Response.new
|
||||||
|
res.body = "<td>OSName: Windows</td>"
|
||||||
|
it { expect(subject.detect_platform(res)).to eq "win" }
|
||||||
|
end
|
||||||
|
|
||||||
|
context "return nil if no OS match" do
|
||||||
|
res = Rex::Proto::Http::Response.new
|
||||||
|
res.body = "<td>OSName: Blah</td>"
|
||||||
|
it { expect(subject.detect_platform(res)).to be_nil }
|
||||||
|
end
|
||||||
|
|
||||||
|
context "return nil res is nil" do
|
||||||
|
res = nil
|
||||||
|
it { expect(subject.detect_platform(res)).to be_nil }
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
describe "#detect_architecture" do
|
||||||
|
context "when server arch is x86" do
|
||||||
|
res = Rex::Proto::Http::Response.new
|
||||||
|
res.body = "<td>OSArch: i386</td>"
|
||||||
|
it { expect(subject.detect_architecture(res)).to eq ARCH_X86 }
|
||||||
|
end
|
||||||
|
|
||||||
|
context "return nil if no architecture match" do
|
||||||
|
res = Rex::Proto::Http::Response.new
|
||||||
|
res.body = "<td>OSArch: Blah</td>"
|
||||||
|
it { expect(subject.detect_architecture(res)).to be_nil }
|
||||||
|
end
|
||||||
|
|
||||||
|
context "return nil res is nil" do
|
||||||
|
res = nil
|
||||||
|
it { expect(subject.detect_architecture(res)).to be_nil }
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in New Issue