From 08d50512e9e1b0f150eed48a8fae0521f44ea34c Mon Sep 17 00:00:00 2001
From: antisnatchor <antisnatchor@gmail.com>
Date: Sun, 4 Mar 2012 16:22:37 +0100
Subject: [PATCH] Added bootstrap unit tests.

---
 test/unit/core/tc_bootstrap.rb | 40 ++++++++++++++++++++++++++++++++++
 test/unit/ts_unit.rb           |  5 +++--
 2 files changed, 43 insertions(+), 2 deletions(-)
 create mode 100644 test/unit/core/tc_bootstrap.rb

diff --git a/test/unit/core/tc_bootstrap.rb b/test/unit/core/tc_bootstrap.rb
new file mode 100644
index 000000000..1f4a1f525
--- /dev/null
+++ b/test/unit/core/tc_bootstrap.rb
@@ -0,0 +1,40 @@
+#
+#   Copyright 2012 Wade Alcorn wade@bindshell.net
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+require 'test/unit'
+
+class TC_Bootstrap < Test::Unit::TestCase
+
+  def setup
+    $root_dir = "../../"
+    $:.unshift File.join( %w{ ../../ } )
+    BeEF::Core::Configuration.new("#{$root_dir}/config.yaml")
+  end
+
+  def teardown
+    $root_dir = nil
+  end
+
+  #
+  # Test the core is functional
+  #
+  def test_bootstrap
+    assert_nothing_raised do
+      require 'core/bootstrap'
+    end
+  end
+
+end
+
diff --git a/test/unit/ts_unit.rb b/test/unit/ts_unit.rb
index b9f603c67..61f27f9b8 100644
--- a/test/unit/ts_unit.rb
+++ b/test/unit/ts_unit.rb
@@ -22,6 +22,7 @@ require './core/filter/tc_command'
 require './core/tc_loader'
 require './core/tc_core'
 require './core/tc_api'
+require './core/tc_bootstrap'
 require './core/tc_modules'
 require './core/tc_social_engineering'
 require './core/tc_autorun'
@@ -44,6 +45,7 @@ class TS_BeefTests
     suite << TC_Filter.suite
     suite << TC_Loader.suite
     suite << TC_Core.suite
+    suite << TC_Bootstrap.suite
     suite << TC_Api.suite
     suite << TC_Modules.suite
     suite << TC_Filesystem.suite
@@ -64,5 +66,4 @@ class TS_BeefTests
   end
 end
 
-Test::Unit::UI::Console::TestRunner.run(TS_BeefTests)
-
+Test::Unit::UI::Console::TestRunner.run(TS_BeefTests)
\ No newline at end of file