From db2be8ba16f2cbcf306aef9bc7d03ecd72bd7afd Mon Sep 17 00:00:00 2001
From: Wade Alcorn <wade@bindshell.net>
Date: Sat, 31 Dec 2011 09:58:11 +1000
Subject: [PATCH] Added msfrpc-client gem check for unit tests

---
 test/unit/ts_beef.rb | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/test/unit/ts_beef.rb b/test/unit/ts_beef.rb
index b9574c808..c5d3c5ef6 100644
--- a/test/unit/ts_beef.rb
+++ b/test/unit/ts_beef.rb
@@ -38,6 +38,16 @@ rescue LoadError
   exit
 end
 
+if (ARGV[0] != 'no_msf')
+  begin
+    require 'msfrpc-client'
+  rescue LoadError
+    puts "The following instruction failed: require 'msfrpc-client'"
+    puts "Please run: sudo gem install msfrpc-client"
+    exit
+  end
+end
+
 require './core/main/network_stack/handlers/dynamicreconstruction.rb'
 require './core/filter/tc_base'
 require './core/filter/tc_command'