From 3ec19fec99cf339704ec908c0fcca5faca1eec4a Mon Sep 17 00:00:00 2001
From: Ben Waugh <ben@delonge.com.au>
Date: Tue, 27 Dec 2011 18:31:43 +1000
Subject: [PATCH] MSF starts/stops with unit tests

Added BeEF.rc with MSF starting parameters
changed ts_beef.rb to startup and shutdown MSF before/after unit tests
---
 test/unit/BeEF.rc    |  2 ++
 test/unit/ts_beef.rb | 20 ++++++++++++++++++++
 2 files changed, 22 insertions(+)
 create mode 100644 test/unit/BeEF.rc

diff --git a/test/unit/BeEF.rc b/test/unit/BeEF.rc
new file mode 100644
index 000000000..326dc1681
--- /dev/null
+++ b/test/unit/BeEF.rc
@@ -0,0 +1,2 @@
+
+load msgrpc Pass=abc123
diff --git a/test/unit/ts_beef.rb b/test/unit/ts_beef.rb
index 62916f90f..c14d94c13 100644
--- a/test/unit/ts_beef.rb
+++ b/test/unit/ts_beef.rb
@@ -48,6 +48,19 @@ require './tc_grep'
 require './tc_filesystem'
 require './extensions/tc_metasploit'
 
+begin
+  child = fork do
+	Signal.trap("TERM") do 
+	  puts "Shutting Down"
+	  exit 99
+	end
+      puts "Starting MSF..."
+  	%x{cd ../msf;./msfconsole -r ../unit/BeEF.rc}
+  	exit 99
+  end
+  sleep 25
+end
+
 class TS_BeefTests
   def self.suite
     suite = Test::Unit::TestSuite.new(name="BeEF TestSuite")
@@ -64,3 +77,10 @@ class TS_BeefTests
 end
 
 Test::Unit::UI::Console::TestRunner.run(TS_BeefTests)
+
+begin
+  puts "\nShutting down MSF ()...\n"
+  Process.kill("TERM", child)
+  Process.wait
+  sleep 240
+end