From 5eb3b686f1abc753832f0e7919efd03093b7df25 Mon Sep 17 00:00:00 2001
From: aussieklutz <aussieklutz@gmail.com>
Date: Thu, 2 May 2019 09:49:21 +1000
Subject: [PATCH] Replace selenium with selenium-webdriver gem

Replace gem require + update Session.new calls with headless firefox (:selenium_headless) module
---
 test/common/beef_test.rb | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/test/common/beef_test.rb b/test/common/beef_test.rb
index 94ecd47ab..be752c8da 100644
--- a/test/common/beef_test.rb
+++ b/test/common/beef_test.rb
@@ -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