From 58acf18973e2facb75787c8a0c9b5cfbea1638bc Mon Sep 17 00:00:00 2001
From: Alex Kotov <kotovalexarian@gmail.com>
Date: Thu, 13 Dec 2018 05:19:46 +0500
Subject: [PATCH] Render screenshots in Selenium WebDriver

---
 features/support/env.rb | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/features/support/env.rb b/features/support/env.rb
index 972dbb3..645924e 100644
--- a/features/support/env.rb
+++ b/features/support/env.rb
@@ -16,6 +16,10 @@ require 'capybara-screenshot/cucumber'
 Capybara.default_driver = :rack_test
 Capybara.javascript_driver = :selenium_headless
 
+Capybara::Screenshot.register_driver :selenium_headless do |driver, path|
+  driver.browser.save_screenshot path
+end
+
 # Capybara defaults to CSS3 selectors rather than XPath.
 # If you'd prefer to use XPath, just uncomment this line and adjust any
 # selectors in your step definitions to use the XPath syntax.