98 lines
2.8 KiB
Ruby
98 lines
2.8 KiB
Ruby
require 'core/loader.rb'
|
|
|
|
# Notes
|
|
# We need to load vairables that 'beef' usually does for us
|
|
## config
|
|
config = BeEF::Core::Configuration.new('config.yaml')
|
|
## home_dir
|
|
$home_dir = Dir.pwd
|
|
## root_dir
|
|
$root_dir = Dir.pwd
|
|
|
|
require 'core/bootstrap.rb'
|
|
require 'rack/test'
|
|
require 'curb'
|
|
require 'rest-client'
|
|
require 'yaml'
|
|
require 'selenium-webdriver'
|
|
require 'browserstack/local'
|
|
|
|
# Require supports
|
|
Dir['spec/support/*.rb'].each do |f|
|
|
require f
|
|
end
|
|
|
|
ENV['RACK_ENV'] ||= 'test'
|
|
ARGV = []
|
|
|
|
## BrowserStack config
|
|
|
|
# Monkey patch to avoid reset sessions
|
|
class Capybara::Selenium::Driver < Capybara::Driver::Base
|
|
def reset!
|
|
if @browser
|
|
@browser.navigate.to('about:blank')
|
|
end
|
|
end
|
|
end
|
|
|
|
TASK_ID = (ENV['TASK_ID'] || 0).to_i
|
|
CONFIG_FILE = ENV['CONFIG_FILE'] || 'windows/windows_chrome_latest.config.yml'
|
|
CONFIG = YAML.safe_load(File.read("./spec/support/browserstack/#{CONFIG_FILE}"))
|
|
CONFIG['user'] = ENV['BROWSERSTACK_USERNAME'] || ''
|
|
CONFIG['key'] = ENV['BROWSERSTACK_ACCESS_KEY'] || ''
|
|
|
|
## DB config
|
|
ActiveRecord::Base.logger = nil
|
|
OTR::ActiveRecord.migrations_paths = [File.join('core', 'main', 'ar-migrations')]
|
|
OTR::ActiveRecord.configure_from_hash!(adapter:'sqlite3', database:':memory:')
|
|
ActiveRecord::Schema.verbose = false
|
|
context = ActiveRecord::Migration.new.migration_context
|
|
if context.needs_migration?
|
|
ActiveRecord::Migrator.new(:up, context.migrations, context.schema_migration).migrate
|
|
end
|
|
|
|
RSpec.configure do |config|
|
|
config.disable_monkey_patching!
|
|
config.bisect_runner = :shell
|
|
config.order = :random
|
|
Kernel.srand config.seed
|
|
config.include Rack::Test::Methods
|
|
config.expect_with :rspec do |c|
|
|
c.syntax = :expect
|
|
end
|
|
config.around do |example|
|
|
ActiveRecord::Base.transaction do
|
|
example.run
|
|
raise ActiveRecord::Rollback
|
|
end
|
|
end
|
|
# BrowserStack
|
|
config.around(:example, :run_on_browserstack => true) do |example|
|
|
@caps = CONFIG['common_caps'].merge(CONFIG['browser_caps'][TASK_ID])
|
|
@caps["name"] = ENV['name'] || example.metadata[:name] || example.metadata[:file_path].split('/').last.split('.').first
|
|
enable_local = @caps["browserstack.local"] && @caps["browserstack.local"].to_s == "true"
|
|
|
|
# Code to start browserstack local before start of test
|
|
if enable_local
|
|
@bs_local = BrowserStack::Local.new
|
|
bs_local_args = { "key" => CONFIG['key'], "forcelocal" => true }
|
|
@bs_local.start(bs_local_args)
|
|
@caps["browserstack.local"] = true
|
|
@caps['browserstack.localIdentifier'] = ENV['BROWSERSTACK_LOCAL_IDENTIFIER']
|
|
end
|
|
|
|
@driver = Selenium::WebDriver.for(:remote,
|
|
:url => "http://#{CONFIG['user']}:#{CONFIG['key']}@#{CONFIG['server']}/wd/hub",
|
|
:desired_capabilities => @caps)
|
|
|
|
begin
|
|
example.run
|
|
ensure
|
|
@driver.quit
|
|
# Code to stop browserstack local after end of test
|
|
@bs_local.stop if enable_local
|
|
end
|
|
end
|
|
end
|