Add QRcode extension unit test stub
This commit is contained in:
46
test/unit/extensions/tc_qrcode.rb
Normal file
46
test/unit/extensions/tc_qrcode.rb
Normal file
@@ -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
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user