From 4ceba73d5ad5530e906889800ebcae56193eec7f Mon Sep 17 00:00:00 2001
From: Thomas Walpole <twalpole@gmail.com>
Date: Sat, 23 Oct 2021 22:52:51 -0700
Subject: [PATCH] Fix deprecation warning in tests

---
 spec/selenium_spec_chrome.rb        | 10 ++++++++--
 spec/selenium_spec_chrome_remote.rb | 10 ++++++++--
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/spec/selenium_spec_chrome.rb b/spec/selenium_spec_chrome.rb
index 85fd4f029..be54e575f 100644
--- a/spec/selenium_spec_chrome.rb
+++ b/spec/selenium_spec_chrome.rb
@@ -181,15 +181,21 @@ module TestSessions
   end
 
   describe 'log access' do
+    let(:logs) do
+      session.driver.browser.then do |chrome_driver|
+        chrome_driver.respond_to?(:logs) ? chrome_driver : chrome_driver.manage
+      end.logs
+    end
+
     it 'does not error getting log types' do
       expect do
-        session.driver.browser.manage.logs.available_types
+        logs.available_types
       end.not_to raise_error
     end
 
     it 'does not error when getting logs' do
       expect do
-        session.driver.browser.manage.logs.get(:browser)
+        logs.get(:browser)
       end.not_to raise_error
     end
   end
diff --git a/spec/selenium_spec_chrome_remote.rb b/spec/selenium_spec_chrome_remote.rb
index 3f59abbe4..deb3d8d81 100644
--- a/spec/selenium_spec_chrome_remote.rb
+++ b/spec/selenium_spec_chrome_remote.rb
@@ -76,15 +76,21 @@ module TestSessions
   end
 
   describe 'log access' do
+    let(:logs) do
+      session.driver.browser.then do |chrome_driver|
+        chrome_driver.respond_to?(:logs) ? chrome_driver : chrome_driver.manage
+      end.logs
+    end
+
     it 'does not error when getting log types' do
       expect do
-        session.driver.browser.manage.logs.available_types
+        logs.available_types
       end.not_to raise_error
     end
 
     it 'does not error when getting logs' do
       expect do
-        session.driver.browser.manage.logs.get(:browser)
+        logs.get(:browser)
       end.not_to raise_error
     end
   end