Replace selenium with selenium-webdriver gem

Replace gem require + update Session.new calls with headless firefox (:selenium_headless) module
This commit is contained in:
aussieklutz
2019-05-02 09:49:21 +10:00
committed by GitHub
parent 639d14c682
commit 5eb3b686f1

View File

@@ -6,9 +6,10 @@
require 'test/unit'
require 'capybara'
require 'capybara/rspec'
Capybara.run_server = false # we need to run our own BeEF server
require 'selenium/webdriver'
require 'selenium-webdriver'
class BeefTest
@@ -18,7 +19,7 @@ class BeefTest
end
def self.login(session = nil)
session = Capybara::Session.new(:selenium) if session.nil?
session = Capybara::Session.new(:selenium_headless) if session.nil?
session.visit(ATTACK_URL)
sleep 2.0
session.has_content?('BeEF Authentication')
@@ -41,7 +42,7 @@ class BeefTest
end
def self.new_victim
victim = Capybara::Session.new(:selenium)
victim = Capybara::Session.new(:selenium_headless)
victim.visit(VICTIM_URL)
victim
end