Add Msf::HTTP::JBoss::BeanShellScripts specs

Author: jvazquez-r7

spec/lib/msf/http/jboss/bean_shell_scripts_spec.rb

@@ -0,0 +1,67 @@
+#-*- coding:binary -*-
+require 'spec_helper'
+
+require 'msf/core'
+require 'msf/http/jboss'
+
+describe Msf::HTTP::JBoss::BeanShellScripts do
+  subject do
+    mod = ::Msf::Exploit.new
+    mod.extend Msf::HTTP::JBoss
+    mod.send(:initialize)
+    mod
+  end
+
+  describe "#generate_bsh" do
+
+  end
+
+  describe "#stager_jsp" do
+    it "returns the JSP stager" do
+      expect(subject.stager_jsp('metasploit')).to include('System.getProperty("jboss.server.home.dir");')
+    end
+
+    it "uses the provided application name" do
+      expect(subject.stager_jsp('metasploit')).to include('"/deploy/" + "metasploit.war";')
+    end
+  end
+
+  describe "#create_file_bsh" do
+    it "returns the Bean Shell script" do
+      expect(subject.create_file_bsh({})).to include('String jboss_home = System.getProperty("jboss.server.home.dir");')
+    end
+
+    context "when options are provided" do
+      let(:opts) do
+        {
+          :file     => 'file',
+          :dir      => 'dir',
+          :contents => 'contents'
+        }
+      end
+
+      it { expect(subject.create_file_bsh(opts)).to include('String location = jboss_home + "/deploy/file";')}
+      it { expect(subject.create_file_bsh(opts)).to include('"/deploy/dir").mkdir()')}
+      it { expect(subject.create_file_bsh(opts)).to include('String val = "contents";')}
+    end
+  end
+
+  describe "#delete_files_bsh" do
+    it "returns the Bean Shell script" do
+      expect(subject.delete_files_bsh({})).to include('String jboss_home = System.getProperty("jboss.server.home.dir");')
+    end
+
+    context "when filenames are provided" do
+      let(:opts) do
+        {
+          'one' => '/tmp/one',
+          'two' => '/tmp/two'
+        }
+      end
+
+      it { expect(subject.delete_files_bsh(opts)).to include('new File(jboss_home + "/deploy//tmp/one").delete();')}
+      it { expect(subject.delete_files_bsh(opts)).to include('new File(jboss_home + "/deploy//tmp/two").delete();')}
+    end
+  end
+
+end