diff --git a/test/unit/extensions/tc_qrcode.rb b/test/unit/extensions/tc_qrcode.rb new file mode 100644 index 000000000..59defe0e6 --- /dev/null +++ b/test/unit/extensions/tc_qrcode.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_Qrcode < Test::Unit::TestCase + + class << self + + def startup + $root_dir = '../../' + $:.unshift(File.expand_path($root_dir)) + + # load extension + require 'extensions/qrcode/extension' + + # load config + BeEF::Core::Configuration.new(File.join($root_dir, 'config.yaml')) + config = BeEF::Core::Configuration.instance + config.load_extensions_config + @@qrcode_config = config.get('beef.extension.qrcode') + 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(@@qrcode_config.has_key?('enable')) + assert(@@qrcode_config.has_key?('targets')) + assert(@@qrcode_config.has_key?('qrsize')) + assert(@@qrcode_config.has_key?('qrborder')) + end + +end diff --git a/test/unit/ts_unit.rb b/test/unit/ts_unit.rb index 984b4aa78..0eae06f7a 100644 --- a/test/unit/ts_unit.rb +++ b/test/unit/ts_unit.rb @@ -27,6 +27,7 @@ require './extensions/tc_proxy' require './extensions/tc_requester' require './extensions/tc_event_logger' require './extensions/tc_network' +require './extensions/tc_qrcode' #require './extensions/tc_dns' require './tc_grep' require './tc_filesystem' @@ -53,6 +54,7 @@ class TS_BeefTests suite << TC_Proxy.suite suite << TC_EventLogger.suite suite << TC_Network.suite + suite << TC_Qrcode.suite suite << TC_Hooks.suite suite << TC_Redirector.suite suite << TC_DynamicReconstruction.suite