Add support for billy's selenium drivers

lib/capybara-screenshot.rb

@@ -172,6 +172,11 @@ def self.capybara_tmp_path=(path)
     driver.render(path, :full => true)
   end
 
+  register_driver :selenium_billy, &selenium_block
+  register_driver :selenium_headless_billy, &selenium_block
+  register_driver :selenium_chrome_billy, &selenium_block
+  register_driver :selenium_chrome_headless_billy, &selenium_block
+
   webkit_block = proc do |driver, path|
     if driver.respond_to?(:save_screenshot)
       driver.save_screenshot(path, webkit_options)

spec/unit/saver_spec.rb

@@ -298,6 +298,66 @@
     end
   end
 
+  describe "with selenium_billy driver" do
+    before do
+      allow(capybara_mock).to receive(:current_driver).and_return(:selenium_billy)
+    end
+
+    it 'saves via browser' do
+      browser_mock = double('browser')
+      expect(driver_mock).to receive(:browser).and_return(browser_mock)
+      expect(browser_mock).to receive(:save_screenshot).with(screenshot_path)
+
+      saver.save
+      expect(saver).to be_screenshot_saved
+    end
+  end
+
+  describe "with selenium_headless_billy driver" do
+    before do
+      allow(capybara_mock).to receive(:current_driver).and_return(:selenium_headless_billy)
+    end
+
+    it 'saves via browser' do
+      browser_mock = double('browser')
+      expect(driver_mock).to receive(:browser).and_return(browser_mock)
+      expect(browser_mock).to receive(:save_screenshot).with(screenshot_path)
+
+      saver.save
+      expect(saver).to be_screenshot_saved
+    end
+  end
+
+  describe "with selenium_chrome_billy driver" do
+    before do
+      allow(capybara_mock).to receive(:current_driver).and_return(:selenium_chrome_billy)
+    end
+
+    it 'saves via browser' do
+      browser_mock = double('browser')
+      expect(driver_mock).to receive(:browser).and_return(browser_mock)
+      expect(browser_mock).to receive(:save_screenshot).with(screenshot_path)
+
+      saver.save
+      expect(saver).to be_screenshot_saved
+    end
+  end
+
+  describe "with selenium_chrome_headless_billy driver" do
+    before do
+      allow(capybara_mock).to receive(:current_driver).and_return(:selenium_chrome_headless_billy)
+    end
+
+    it 'saves via browser' do
+      browser_mock = double('browser')
+      expect(driver_mock).to receive(:browser).and_return(browser_mock)
+      expect(browser_mock).to receive(:save_screenshot).with(screenshot_path)
+
+      saver.save
+      expect(saver).to be_screenshot_saved
+    end
+  end
+
   describe "with cuprite driver" do
     before do
       allow(capybara_mock).to receive(:current_driver).and_return(:cuprite)