Land rapid7#2120, specs for command_dispatcher

Author: wvu

spec/lib/msf/ui/command_dispatcher/auxiliary_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+require 'msf/ui'
+require 'msf/ui/console/command_dispatcher/auxiliary'
+
+describe Msf::Ui::Console::CommandDispatcher::Auxiliary do
+	include_context 'Msf::DBManager'
+	include_context 'Msf::UIDriver'
+
+	subject(:aux) do
+		described_class.new(driver)
+	end
+
+	describe "#cmd_run" do
+	end
+
+	describe "#cmd_rerun" do
+	end
+
+	describe "#cmd_exploit" do
+	end
+
+	describe "#cmd_reload" do
+	end
+end

spec/lib/msf/ui/command_dispatcher/core_spec.rb

@@ -5,103 +5,94 @@
 require 'msf/ui/console/command_dispatcher/core'
 
 describe Msf::Ui::Console::CommandDispatcher::Core do
-  include_context 'Msf::DBManager'
+	include_context 'Msf::DBManager'
+	include_context 'Msf::UIDriver'
 
-  let(:driver) do
-    mock(
-        'Driver',
-        :framework => framework
-    ).tap { |driver|
-      driver.stub(:on_command_proc=).with(kind_of(Proc))
-      driver.stub(:print_line).with(kind_of(String))
-    }
-  end
+	subject(:core) do
+		described_class.new(driver)
+	end
 
-  subject(:core) do
-    described_class.new(driver)
-  end
+	context '#search_modules_sql' do
+		def search_modules_sql
+			core.search_modules_sql(match)
+		end
 
-  context '#search_modules_sql' do
-    def search_modules_sql
-      core.search_modules_sql(match)
-    end
+		let(:match) do
+			''
+		end
 
-    let(:match) do
-      ''
-    end
+		it 'should generate Matching Modules table' do
+			core.should_receive(:generate_module_table).with('Matching Modules').and_call_original
 
-    it 'should generate Matching Modules table' do
-      core.should_receive(:generate_module_table).with('Matching Modules').and_call_original
+			search_modules_sql
+		end
 
-      search_modules_sql
-    end
+		it 'should call Msf::DBManager#search_modules' do
+			db_manager.should_receive(:search_modules).with(match).and_return([])
 
-    it 'should call Msf::DBManager#search_modules' do
-      db_manager.should_receive(:search_modules).with(match).and_return([])
+			search_modules_sql
+		end
 
-      search_modules_sql
-    end
+		context 'with matching Mdm::Module::Details' do
+			let(:match) do
+				module_detail.fullname
+			end
 
-    context 'with matching Mdm::Module::Details' do
-      let(:match) do
-        module_detail.fullname
-      end
+			let!(:module_detail) do
+				FactoryGirl.create(:mdm_module_detail)
+			end
 
-      let!(:module_detail) do
-        FactoryGirl.create(:mdm_module_detail)
-      end
+			context 'printed table' do
+				def cell(table, row, column)
+					row_line_number = 6 + row
+					line_number     = 0
 
-      context 'printed table' do
-        def cell(table, row, column)
-          row_line_number = 6 + row
-          line_number = 0
+					cell = nil
 
-          cell = nil
+					table.each_line do |line|
+						if line_number == row_line_number
+							# strip prefix and postfix
+							padded_cells = line[3...-1]
+							cells        = padded_cells.split(/\s{2,}/)
 
-          table.each_line do |line|
-            if line_number == row_line_number
-              # strip prefix and postfix
-              padded_cells = line[3...-1]
-              cells = padded_cells.split(/\s{2,}/)
+							cell = cells[column]
+							break
+						end
 
-              cell = cells[column]
-              break
-            end
+						line_number += 1
+					end
 
-            line_number += 1
-          end
+					cell
+				end
 
-          cell
-        end
+				let(:printed_table) do
+					table = ''
 
-        let(:printed_table) do
-          table = ''
+					core.stub(:print_line) do |string|
+						table = string
+					end
 
-          core.stub(:print_line) do |string|
-            table = string
-          end
+					search_modules_sql
 
-          search_modules_sql
+					table
+				end
 
-          table
-        end
+				it 'should have fullname in first column' do
+					cell(printed_table, 0, 0).should include(module_detail.fullname)
+				end
 
-        it 'should have fullname in first column' do
-          cell(printed_table, 0, 0).should include(module_detail.fullname)
-        end
+				it 'should have disclosure date in second column' do
+					cell(printed_table, 0, 1).should include(module_detail.disclosure_date.to_s)
+				end
 
-        it 'should have disclosure date in second column' do
-          cell(printed_table, 0, 1).should include(module_detail.disclosure_date.to_s)
-        end
+				it 'should have rank name in third column' do
+					cell(printed_table, 0, 2).should include(Msf::RankingName[module_detail.rank])
+				end
 
-        it 'should have rank name in third column' do
-          cell(printed_table, 0, 2).should include(Msf::RankingName[module_detail.rank])
-        end
-
-        it 'should have name in fourth column' do
-          cell(printed_table, 0, 3).should include(module_detail.name)
-        end
-      end
-    end
-  end
+				it 'should have name in fourth column' do
+					cell(printed_table, 0, 3).should include(module_detail.name)
+				end
+			end
+		end
+	end
 end