Add webrtc unit test stub

This commit is contained in:
Brendan Coles
2015-12-14 17:01:26 +00:00
parent ebd2fa6088
commit 61a9b94ca1
2 changed files with 47 additions and 0 deletions

View File

@@ -0,0 +1,45 @@
#
# 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_Webrtc < Test::Unit::TestCase
class << self
def startup
$root_dir = '../../'
$:.unshift(File.expand_path($root_dir))
# load extension
require 'extensions/webrtc/extension'
# load config
BeEF::Core::Configuration.new(File.join($root_dir, 'config.yaml'))
config = BeEF::Core::Configuration.instance
config.load_extensions_config
@@webrtc_config = config.get('beef.extension.webrtc')
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(@@webrtc_config.has_key?('enable'))
assert(@@webrtc_config.has_key?('stunservers'))
assert(@@webrtc_config.has_key?('turnservers'))
end
end

View File

@@ -29,6 +29,7 @@ require './extensions/tc_event_logger'
require './extensions/tc_network' require './extensions/tc_network'
require './extensions/tc_qrcode' require './extensions/tc_qrcode'
require './extensions/tc_console' require './extensions/tc_console'
require './extensions/tc_webrtc'
#require './extensions/tc_dns' #require './extensions/tc_dns'
require './tc_grep' require './tc_grep'
require './tc_filesystem' require './tc_filesystem'
@@ -60,6 +61,7 @@ class TS_BeefTests
suite << TC_Redirector.suite suite << TC_Redirector.suite
suite << TC_DynamicReconstruction.suite suite << TC_DynamicReconstruction.suite
suite << TC_Console.suite suite << TC_Console.suite
suite << TC_Webrtc.suite
#suite << TC_Dns.suite #suite << TC_Dns.suite
return suite return suite