From ebd2fa60886d1ecba1428920ae756d5be5fe469d Mon Sep 17 00:00:00 2001
From: Brendan Coles <bcoles@gmail.com>
Date: Mon, 14 Dec 2015 16:57:31 +0000
Subject: [PATCH] Add console unit test stub

---
 test/unit/extensions/tc_console.rb | 46 ++++++++++++++++++++++++++++++
 test/unit/ts_unit.rb               |  2 ++
 2 files changed, 48 insertions(+)
 create mode 100644 test/unit/extensions/tc_console.rb

diff --git a/test/unit/extensions/tc_console.rb b/test/unit/extensions/tc_console.rb
new file mode 100644
index 000000000..fe6d7f100
--- /dev/null
+++ b/test/unit/extensions/tc_console.rb
@@ -0,0 +1,46 @@
+#
+# Copyright (c) 2006-2015 Wade Alcorn - wade@bindshell.net
+# Browser Exploitation Framework (BeEF) - http://beefproject.com
+# See the file 'doc/COPYING' for copying permission
+#
+require 'test/unit'
+
+class TC_Console < Test::Unit::TestCase
+
+  class << self
+
+    def startup
+      $root_dir = '../../'
+      $:.unshift(File.expand_path($root_dir))
+
+      # load extension
+      require 'extensions/console/extension'
+
+      # load config
+      BeEF::Core::Configuration.new(File.join($root_dir, 'config.yaml'))
+      config = BeEF::Core::Configuration.instance
+      config.load_extensions_config
+      @@console_config = config.get('beef.extension.console')
+      @@console_config_shell = config.get('beef.extension.console.shell')
+    end
+
+    def shutdown
+      $root_dir = nil
+    end
+
+  end
+
+  # Connects to in-memory database (does not test anything)
+  def test_01_database
+    DataMapper.setup(:default, 'sqlite3::memory:')
+    DataMapper.auto_migrate!
+  end
+
+  # Checks for required settings in config file
+  def test_02_config
+    assert(@@console_config.has_key?('enable'))
+    assert(@@console_config_shell.has_key?('historyfolder'))
+    assert(@@console_config_shell.has_key?('historyfile'))
+  end
+
+end
diff --git a/test/unit/ts_unit.rb b/test/unit/ts_unit.rb
index 0eae06f7a..16a81f588 100644
--- a/test/unit/ts_unit.rb
+++ b/test/unit/ts_unit.rb
@@ -28,6 +28,7 @@ require './extensions/tc_requester'
 require './extensions/tc_event_logger'
 require './extensions/tc_network'
 require './extensions/tc_qrcode'
+require './extensions/tc_console'
 #require './extensions/tc_dns'
 require './tc_grep'
 require './tc_filesystem'
@@ -58,6 +59,7 @@ class TS_BeefTests
     suite << TC_Hooks.suite
     suite << TC_Redirector.suite
     suite << TC_DynamicReconstruction.suite
+    suite << TC_Console.suite
     #suite << TC_Dns.suite
 
     return suite