diff --git a/test/common/beef_test.rb b/test/common/beef_test.rb new file mode 100644 index 000000000..5151093a2 --- /dev/null +++ b/test/common/beef_test.rb @@ -0,0 +1,10 @@ +require 'test/unit' + +class BeefTest + + def self.save_screenshot(session) + Dir.mkdir(BEEF_TEST_DIR) if not File.directory?(BEEF_TEST_DIR) + session.driver.browser.save_screenshot(BEEF_TEST_DIR + Time.now.strftime("%Y-%m-%d--%H-%M-%S-%N") + ".png") + end + +end \ No newline at end of file diff --git a/test/common/test_constants.rb b/test/common/test_constants.rb new file mode 100644 index 000000000..e909afec3 --- /dev/null +++ b/test/common/test_constants.rb @@ -0,0 +1 @@ +BEEF_TEST_DIR = "/tmp/beef-test/" \ No newline at end of file diff --git a/test/integration/tc_login.rb b/test/integration/tc_login.rb index 696d94133..8dcba90d7 100644 --- a/test/integration/tc_login.rb +++ b/test/integration/tc_login.rb @@ -1,25 +1,20 @@ require 'test/unit' - -BEEF_TEST_DIR = "/tmp/beef-test/" +require '../common/test_constants' +require '../common/beef_test' class TC_login < Test::Unit::TestCase - def save_screenshot(session) - Dir.mkdir(BEEF_TEST_DIR) if not File.directory?(BEEF_TEST_DIR) - session.driver.browser.save_screenshot(BEEF_TEST_DIR + Time.now.strftime("%Y-%m-%d--%H-%M-%S-%N") + ".png") - end - def test_log_in session = Capybara::Session.new(:selenium) session.visit('http://localhost:3000/ui/panel') - save_screenshot(session) + BeefTest.save_screenshot(session) session.has_content?('BeEF Authentication') session.fill_in 'user', :with => 'beef' session.fill_in 'pass', :with => 'beef' - save_screenshot(session) + BeefTest.save_screenshot(session) session.click_button('Login') session.has_content?('logout') - save_screenshot(session) + BeefTest.save_screenshot(session) session end @@ -28,9 +23,9 @@ class TC_login < Test::Unit::TestCase session = test_log_in session.has_content?('logout') session.click_link('Logout') - save_screenshot(session) + BeefTest.save_screenshot(session) session.has_content?('BeEF Authentication') - save_screenshot(session) + BeefTest.save_screenshot(session) session end