diff --git a/test/unit/extensions/tc_webrtc.rb b/test/unit/extensions/tc_webrtc.rb new file mode 100644 index 000000000..2bcb74ada --- /dev/null +++ b/test/unit/extensions/tc_webrtc.rb @@ -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 diff --git a/test/unit/ts_unit.rb b/test/unit/ts_unit.rb index 16a81f588..856f65b74 100644 --- a/test/unit/ts_unit.rb +++ b/test/unit/ts_unit.rb @@ -29,6 +29,7 @@ require './extensions/tc_event_logger' require './extensions/tc_network' require './extensions/tc_qrcode' require './extensions/tc_console' +require './extensions/tc_webrtc' #require './extensions/tc_dns' require './tc_grep' require './tc_filesystem' @@ -60,6 +61,7 @@ class TS_BeefTests suite << TC_Redirector.suite suite << TC_DynamicReconstruction.suite suite << TC_Console.suite + suite << TC_Webrtc.suite #suite << TC_Dns.suite return suite