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