Close unused Capybara sessions

test/integration/tc_jools.rb

@@ -19,7 +19,7 @@ class TC_Jools < Test::Unit::TestCase
         return new Jools([]);"
         jools_obj = victim.execute_script(script)
         assert_not_nil jools_obj
-        victim.reset_session!
+        victim.driver.browser.close
     end
 
     #test simple jools rule example
@@ -44,6 +44,7 @@ class TC_Jools < Test::Unit::TestCase
             return result.state;"
        result = victim.execute_script(script)
        assert_equal result,'on'
+       victim.driver.browser.close
     end
 
     #test jools chaining example
@@ -110,5 +111,6 @@ class TC_Jools < Test::Unit::TestCase
        assert_not_equal results[1]['color'], 'green'
        assert_equal results[2]['color'],'yellow'
        assert_not_equal results[3]['color'], 'yellow'
+       victim.driver.browser.close
     end
 end

test/integration/tc_login.rb

@@ -22,12 +22,14 @@ class TC_Login < Test::Unit::TestCase
     sleep 20.0
     session.has_content?('logout')
     BeefTest.save_screenshot(session)
+    session.driver.browser.close
   end
 
   def test_beef_test_login_function
     session = BeefTest.login
     session.has_content?('logout')
     BeefTest.save_screenshot(session)
+    session.driver.browser.close
   end
 
   def test_log_out
@@ -35,6 +37,7 @@ class TC_Login < Test::Unit::TestCase
     session.click_link('Logout')
     session.has_content?('BeEF Authentication')
     BeefTest.save_screenshot(session)
+    session.driver.browser.close
   end
 
   def test_beef_test_logout_function
@@ -42,6 +45,7 @@ class TC_Login < Test::Unit::TestCase
     session = BeefTest.logout(session)
     session.has_content?('BeEF Authentication')
     BeefTest.save_screenshot(session)
+    session.driver.browser.close
   end
 
   def test_logs_tab
@@ -57,6 +61,7 @@ class TC_Login < Test::Unit::TestCase
 
     BeefTest.save_screenshot(session)
     BeefTest.logout(session)
+    session.driver.browser.close
   end
 
   def test_hooking_browser
@@ -79,6 +84,8 @@ class TC_Login < Test::Unit::TestCase
     BeefTest.save_screenshot(victim)
 
     BeefTest.logout(attacker)
+    attacker.driver.browser.close
+    victim.driver.browser.close
   end
 
 end