cornucopia 0.1.55 → 0.2.0

data/lib/cornucopia/capybara/install_extensions.rb

@@ -4,10 +4,10 @@ if Object.const_defined?("Capybara") &&
     Capybara.const_defined?("Node") &&
     Capybara::Node.const_defined?("Document")
   unless Capybara::Node::Document.included_modules.include?(Cornucopia::Capybara::FinderExtensions)
-    Capybara::Node::Document.include Cornucopia::Capybara::FinderExtensions
+    Capybara::Node::Document.prepend Cornucopia::Capybara::FinderExtensions
   end
   unless Capybara::Node::Document.included_modules.include?(Cornucopia::Capybara::MatcherExtensions)
-    Capybara::Node::Document.include Cornucopia::Capybara::MatcherExtensions
+    Capybara::Node::Document.prepend Cornucopia::Capybara::MatcherExtensions
   end
 end
 
@@ -15,10 +15,10 @@ if Object.const_defined?("Capybara") &&
     Capybara.const_defined?("Node") &&
     Capybara::Node.const_defined?("Simple")
   unless Capybara::Node::Simple.included_modules.include?(Cornucopia::Capybara::FinderExtensions)
-    Capybara::Node::Simple.include Cornucopia::Capybara::FinderExtensions
+    Capybara::Node::Simple.prepend Cornucopia::Capybara::FinderExtensions
   end
   unless Capybara::Node::Simple.included_modules.include?(Cornucopia::Capybara::MatcherExtensions)
-    Capybara::Node::Simple.include Cornucopia::Capybara::MatcherExtensions
+    Capybara::Node::Simple.prepend Cornucopia::Capybara::MatcherExtensions
   end
 end
 
@@ -26,10 +26,10 @@ if Object.const_defined?("Capybara") &&
     Capybara.const_defined?("Node") &&
     Capybara::Node.const_defined?("Element")
   unless Capybara::Node::Element.included_modules.include?(Cornucopia::Capybara::FinderExtensions)
-    Capybara::Node::Element.include Cornucopia::Capybara::FinderExtensions
+    Capybara::Node::Element.prepend Cornucopia::Capybara::FinderExtensions
   end
   unless Capybara::Node::Element.included_modules.include?(Cornucopia::Capybara::MatcherExtensions)
-    Capybara::Node::Element.include Cornucopia::Capybara::MatcherExtensions
+    Capybara::Node::Element.prepend Cornucopia::Capybara::MatcherExtensions
   end
   unless Capybara::Node::Element.included_modules.include?(Cornucopia::Capybara::SelectableValues)
     Capybara::Node::Element.include Cornucopia::Capybara::SelectableValues

data/lib/cornucopia/capybara/matcher_extensions.rb

@@ -10,62 +10,109 @@ module Cornucopia
     module MatcherExtensions
       extend ActiveSupport::Concern
 
-      included do
-        alias_method :__cornucopia_capybara_orig_assert_selector, :assert_selector
-        alias_method :__cornucopia_capybara_orig_assert_no_selector, :assert_no_selector
-        alias_method :__cornucopia_capybara_orig_has_selector?, :has_selector?
-        alias_method :__cornucopia_capybara_orig_has_no_selector?, :has_no_selector?
-
-        define_method :assert_selector do |*args|
-          __cornucopia_assert_selector_function(:assert_selector, *args)
-        end
+      def assert_selector(*args, &block)
+        retry_count    = 0
+        result         = nil
+        no_option_args = args.clone
+        options        = no_option_args.pop if no_option_args.last.is_a?(Hash)
+        options        ||= {}
+
+        support_options, normal_options = __cornucopia__extract_selector_support_options(**options)
+
+        begin
+          retry_count += 1
+          if __cornucopia__call_super?(:assert_selector, *no_option_args, **normal_options, &block)
+            result = super(*no_option_args, normal_options, &block)
+          end
+        rescue Selenium::WebDriver::Error::StaleElementReferenceError
+          retry if __cornucopia__retry_selector?(retry_count, support_options)
 
-        define_method :assert_no_selector do |*args|
-          __cornucopia_assert_selector_function(:assert_no_selector, *args)
+          result = __cornucopia__analyze_selector(:assert_selector, support_options, *no_option_args, **normal_options)
+        rescue
+          result = __cornucopia__analyze_selector(:assert_selector, support_options, *no_option_args, **normal_options)
         end
 
-        define_method :has_selector? do |*args|
-          new_args = args.dup
-          options  = (new_args.pop if new_args.length > 1 && new_args[-1].is_a?(Hash)) || {}
+        result
+      end
+
+      def assert_no_selector(*args, &block)
+        retry_count    = 0
+        result         = nil
+        no_option_args = args.clone
+        options        = no_option_args.pop if no_option_args.last.is_a?(Hash)
+        options        ||= {}
 
-          if Cornucopia::Util::Configuration.ignore_has_selector_errors
-            options = { __cornucopia_no_analysis: true }.merge options
+        support_options, normal_options = __cornucopia__extract_selector_support_options(**options)
+
+        begin
+          retry_count += 1
+          if __cornucopia__call_super?(:assert_no_selector, *no_option_args, **normal_options, &block)
+            result = super(*no_option_args, **normal_options, &block)
           end
+        rescue Selenium::WebDriver::Error::StaleElementReferenceError
+          retry if __cornucopia__retry_selector?(retry_count, support_options)
 
-          __cornucopia_assert_selector_function(:has_selector?, *new_args, options)
+          result = __cornucopia__analyze_selector(:assert_no_selector, support_options, *no_option_args, **normal_options)
+        rescue
+          result = __cornucopia__analyze_selector(:assert_no_selector, support_options, *no_option_args, **normal_options)
         end
 
-        define_method :has_no_selector? do |*args|
-          new_args = args.dup
-          options  = (new_args.pop if new_args.length > 1 && new_args[-1].is_a?(Hash)) || {}
+        result
+      end
+
+      def has_selector?(*args, **options, &block)
+        new_args    = args.dup
+        retry_count = 0
+        result      = nil
 
-          if Cornucopia::Util::Configuration.ignore_has_selector_errors
-            options = { __cornucopia_no_analysis: true }.merge options
+        support_options, normal_options = __cornucopia__extract_selector_support_options(**options)
+
+        begin
+          retry_count += 1
+          if __cornucopia__call_super?(:has_selector?, *new_args, **options, &block)
+            __cornucopia_with_no_analysis do
+              result = super(*new_args, **options, &block)
+
+              unless Cornucopia::Util::Configuration.ignore_has_selector_errors
+                result ||= __cornucopia__analyze_selector(:has_no_selector?, support_options, *new_args, **normal_options)
+              end
+            end
           end
-          __cornucopia_assert_selector_function(:has_no_selector?, *new_args, options)
+        rescue Selenium::WebDriver::Error::StaleElementReferenceError
+          retry if __cornucopia__retry_selector?(retry_count, support_options)
+
+          result = __cornucopia__analyze_selector(:has_selector?, support_options, *new_args, **normal_options)
+        rescue
+          result = __cornucopia__analyze_selector(:has_selector?, support_options, *new_args, **normal_options)
         end
-      end
 
+        result
+      end
 
-      def __cornucopia_assert_selector_function(assert_selector_function, *args)
+      def has_no_selector?(*args, **options, &block)
+        new_args    = args.dup
         retry_count = 0
         result      = nil
 
-        support_options = if [:has_selector?, :has_no_selector?].include?(assert_selector_function)
-                            {}
-                          else
-                            __cornucopia__extract_selector_support_options(*args)
-                          end
+        support_options, normal_options = __cornucopia__extract_selector_support_options(**options)
 
         begin
           retry_count += 1
-          result      = send("__cornucopia_capybara_orig_#{assert_selector_function}", *args)
+          if __cornucopia__call_super?(:has_no_selector?, *new_args, **options, &block)
+            __cornucopia_with_no_analysis do
+              result = super(*new_args, **options, &block)
+
+              unless Cornucopia::Util::Configuration.ignore_has_selector_errors
+                result ||= __cornucopia__analyze_selector(:has_no_selector?, support_options, *new_args, **normal_options)
+              end
+            end
+          end
         rescue Selenium::WebDriver::Error::StaleElementReferenceError
-          retry if __cornucopia__retry_selector(retry_count, support_options)
+          retry if __cornucopia__retry_selector?(retry_count, support_options)
 
-          result = __cornucopia__analyze_selector(assert_selector_function, support_options, *args)
-        rescue Exception
-          result = __cornucopia__analyze_selector(assert_selector_function, support_options, *args)
+          result = __cornucopia__analyze_selector(:has_no_selector?, support_options, *new_args, **normal_options)
+        rescue
+          result = __cornucopia__analyze_selector(:has_no_selector?, support_options, *new_args, **normal_options)
         end
 
         result
@@ -73,18 +120,22 @@ module Cornucopia
 
       private
 
-      def __cornucopia__extract_selector_support_options(*args)
-        support_options = {}
+      attr_reader :__cornucopia_block_analysis
 
-        if args[-1].is_a?(Hash)
-          support_options[:__cornucopia_no_analysis]      = args[-1].delete(:__cornucopia_no_analysis)
-          support_options[:__cornucopia_retry_with_found] = args[-1].delete(:__cornucopia_retry_with_found)
-        end
+      def __cornucopia__call_super?(_funciton_name, *_args, **_options, &_block)
+        true
+      end
+
+      def __cornucopia__extract_selector_support_options(**options)
+        support_options = options.slice(:__cornucopia_no_analysis, :__cornucopia_retry_with_found)
+        normal_options  = options.slice(*(options.keys - %i[__cornucopia_no_analysis __cornucopia_retry_with_found]))
 
-        support_options
+        support_options[:__cornucopia_no_analysis] = true if __cornucopia_block_analysis
+
+        [support_options, normal_options]
       end
 
-      def __cornucopia__analyze_selector(function_name, support_options, *args)
+      def __cornucopia__analyze_selector(function_name, support_options, *args, **options)
         return_value = nil
         error        = $!
 
@@ -97,26 +148,38 @@ module Cornucopia
                                                                                 {},
                                                                                 support_options,
                                                                                 function_name,
-                                                                                *args)
+                                                                                *args,
+                                                                                **options)
 
           if find_action.perform_analysis(Cornucopia::Util::Configuration.retry_match_with_found ||
                                               support_options[:__cornucopia_retry_with_found])
             return_value = find_action.simple_run({ __cornucopia_no_analysis: true }) rescue nil
             return_value ||= find_action.return_value
           else
-            raise error
+            raise error if error
           end
         else
-          raise error
+          raise error if error
         end
 
         return_value
       end
 
-      def __cornucopia__retry_selector(retry_count, support_options)
+      def __cornucopia__retry_selector?(retry_count, support_options)
         retry_count <= Cornucopia::Util::Configuration.selenium_cache_retry_count &&
             !support_options[:__cornucopia_no_analysis]
       end
+
+      def __cornucopia_with_no_analysis
+        orig_block = __cornucopia_block_analysis
+
+        begin
+          @__cornucopia_block_analysis = true
+          yield
+        ensure
+          @__cornucopia_block_analysis = orig_block
+        end
+      end
     end
   end
 end

data/lib/cornucopia/capybara/page_diagnostics/window_iterator.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "digest"
+# require ::File.expand_path("../../util/configuration", File.dirname(__FILE__))
+require ::File.expand_path("../../util/report_builder", File.dirname(__FILE__))
+require ::File.expand_path("../../util/report_table", File.dirname(__FILE__))
+
+module Cornucopia
+  module Capybara
+    class PageDiagnostics
+      class WindowIterator
+        def initialize(window_handles, current_window, diagnostics)
+          @window_handles = window_handles
+          @current_window = current_window
+          @diagnostics    = diagnostics
+        end
+
+        def each(&block)
+          begin
+            @diagnostics.allow_other_windows = false
+
+            if @window_handles.length > 1
+              @window_handles.each do |window_handle|
+                unless @current_window && @current_window == window_handle
+                  switched = @diagnostics.execute_driver_function(:switch_to_window,
+                                                                  "could not switch windows",
+                                                                  window_handle)
+                  if switched != "could not switch windows"
+                    block.yield @diagnostics
+                  end
+                end
+              end
+            end
+          ensure
+            @diagnostics.allow_other_windows = true
+            @diagnostics.execute_driver_function(:switch_to_window, "could not switch windows", @current_window)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/cornucopia/capybara/page_diagnostics.rb

@@ -4,6 +4,7 @@ require "digest"
 # require ::File.expand_path("../util/configuration", File.dirname(__FILE__))
 require ::File.expand_path("../util/report_builder", File.dirname(__FILE__))
 require ::File.expand_path("../util/report_table", File.dirname(__FILE__))
+require ::File.expand_path("page_diagnostics/window_iterator", File.dirname(__FILE__))
 
 module Cornucopia
   module Capybara
@@ -104,36 +105,6 @@ module Cornucopia
         end
       end
 
-      class WindowIterator
-        def initialize(window_handles, current_window, diagnostics)
-          @window_handles = window_handles
-          @current_window = current_window
-          @diagnostics    = diagnostics
-        end
-
-        def each(&block)
-          begin
-            @diagnostics.allow_other_windows = false
-
-            if @window_handles.length > 1
-              @window_handles.each do |window_handle|
-                unless @current_window && @current_window == window_handle
-                  switched = @diagnostics.execute_driver_function(:switch_to_window,
-                                                                  "could not switch windows",
-                                                                  window_handle)
-                  if switched != "could not switch windows"
-                    block.yield @diagnostics
-                  end
-                end
-              end
-            end
-          ensure
-            @diagnostics.allow_other_windows = true
-            @diagnostics.execute_driver_function(:switch_to_window, "could not switch windows", @current_window)
-          end
-        end
-      end
-
       def other_windows
         if @allow_other_windows
           Cornucopia::Capybara::PageDiagnostics::WindowIterator.new(@window_handles, @current_window, self)

data/lib/cornucopia/capybara/synchronizable.rb

@@ -15,7 +15,7 @@ module Cornucopia
       def synchronize_test(seconds = nil, options = {}, &block)
         seconds ||= ::Capybara.respond_to?(:default_max_wait_time) ? ::Capybara.default_max_wait_time : ::Capybara.default_wait_time
 
-        document.synchronize(seconds, options) do
+        document.synchronize(seconds, **options) do
           raise ::Capybara::ElementNotFound unless block.yield
         end
       end

data/lib/cornucopia/site_prism/class_extensions.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require 'active_support/concern'
+
+module Cornucopia
+  module SitePrism
+    module ClassExtensions
+      def to_capybara_node
+        @__corunucopia_base_node || super
+      end
+    end
+  end
+end

data/lib/cornucopia/site_prism/element_extensions.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'active_support/concern'
+require ::File.expand_path("class_extensions", File.dirname(__FILE__))
 
 module Cornucopia
   module SitePrism
@@ -9,16 +10,18 @@ module Cornucopia
 
       included do
         ::Capybara::Session::DSL_METHODS.each do |method|
-          alias_method "__cornucopia_site_prism_orig_#{method}".to_sym, method
+          next if Cornucopia::SitePrism::ClassExtensions.methods.include?(method)
 
-          define_method method do |*args, &block|
+          Cornucopia::SitePrism::ClassExtensions.define_singleton_method(method) do |*args, **options, &block|
             if @__corunucopia_base_node
-              @__corunucopia_base_node.send method, *args, &block
+              @__corunucopia_base_node.send(method, *args, **options, &block)
             else
-              send "__cornucopia_site_prism_orig_#{method}", *args, &block
+              super(*args, **options, &block)
             end
           end
         end
+
+        self.prepend Cornucopia::SitePrism::ClassExtensions
       end
 
       module ClassMethods

data/lib/cornucopia/site_prism/install_extensions.rb

@@ -13,6 +13,6 @@ if Object.const_defined?("SitePrism") &&
     SitePrism::Section.include Cornucopia::SitePrism::ElementExtensions
   end
   unless SitePrism::Section.included_modules.include?(Cornucopia::SitePrism::SectionExtensions)
-    SitePrism::Section.include Cornucopia::SitePrism::SectionExtensions
+    SitePrism::Section.prepend Cornucopia::SitePrism::SectionExtensions
   end
 end

data/lib/cornucopia/site_prism/page_application.rb

@@ -66,8 +66,8 @@ module Cornucopia
           end
         end
 
-        def respond_to?(method_sym, include_private = false)
-          if self.current_instance.respond_to?(method_sym, include_private)
+        def respond_to_missing?(method_name, include_private = false)
+          if self.current_instance.respond_to_missing?(method_sym, include_private)
             true
           else
             super
@@ -114,7 +114,7 @@ module Cornucopia
         end
       end
 
-      def respond_to?(method_sym, include_private = false)
+      def respond_to_missing?(method_sym, include_private = false)
         method_name = method_sym.to_s
 
         if is_page_name?(method_name)

data/lib/cornucopia/site_prism/section_extensions.rb

@@ -5,18 +5,10 @@ require 'active_support/concern'
 module Cornucopia
   module SitePrism
     module SectionExtensions
-      extend ActiveSupport::Concern
+      def initialize(*args)
+        super(*args)
 
-      included do |base|
-        base.class_exec do
-          alias :__cornucopia_site_prism_orig_initialize :initialize
-
-          def initialize(*args)
-            __cornucopia_site_prism_orig_initialize(*args)
-
-            self.owner_node = args[0].owner_node
-          end
-        end
+        self.owner_node = args[0].owner_node
       end
     end
   end

data/lib/cornucopia/util/report_builder.rb

@@ -313,7 +313,7 @@ module Cornucopia
 
       def delete_old_folders
         old_directories = Dir[File.join(index_folder_name, "#{@base_folder_name}_*")].
-            map { |dir| File.directory?(dir) ? dir : nil }.compact
+            map { |dir| File.directory?(dir) ? dir : nil }.compact.sort
 
         if Array.wrap(old_directories).length > MAX_OLD_FOLDERS
           old_directories.each_with_index do |dir, index|

data/lib/cornucopia/version.rb

@@ -1,3 +1,3 @@
 module Cornucopia
-  VERSION = "0.1.55"
+  VERSION = "0.2.0"
 end

data/spec/dummy/app/assets/config/manifest.js

@@ -0,0 +1,2 @@
+// app/assets/config/manifest.js
+//

data/spec/dummy/config/application.rb

@@ -6,7 +6,7 @@ require File.expand_path('../boot', __FILE__)
 # require "active_record/railtie"
 require "action_controller/railtie"
 require "action_mailer/railtie"
-require "sprockets/railtie"
+# require "sprockets/railtie"
 # require "rails/test_unit/railtie"
 
 Bundler.require(*Rails.groups)

data/spec/lib/capybara/finder_diagnostics_spec.rb

@@ -81,7 +81,7 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
         expect(contents_frame.errors[0].tables[0].rows[2].sub_tables[0].rows[0].values[0].text).to be == ":css"
         expect(contents_frame.errors[0].tables[0].rows[2].sub_tables[0].rows[1].labels[0].text).to be == "1"
         expect(contents_frame.errors[0].tables[0].rows[2].sub_tables[0].rows[1].values[0].text).to be == "#base-contentss"
-        expect(contents_frame.errors[0].tables[0].rows[3].labels[0].text).to be == "exception"
+        expect(contents_frame.errors[0].tables[0].rows[3].labels[0].text).to be == "message"
         expect(contents_frame.errors[0].tables[0].rows[3].values[0].text).to be == "Unable to find css \"#base-contentss\""
         expect(contents_frame.errors[0].tables[0].rows[4].labels[0].text).to be == "backtrace"
         expect(contents_frame.errors[0].tables[0].rows[4].expands[0]).to be
@@ -93,7 +93,7 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
         expect(contents_frame.errors[0].more_details.details.rows[0].labels[0].text).to be == "name"
         expect(contents_frame.errors[0].more_details.details.rows[0].values[0].text).to be == "Capybara::ElementNotFound"
         expect(contents_frame.errors[0].more_details.details.rows[1].labels[0].text).to be == "support_options"
-        expect(contents_frame.errors[0].more_details.details.rows[1].values[0].text).to be == "{:__cornucopia_no_analysis=>true, :__cornucopia_retry_with_found=>nil}"
+        expect(contents_frame.errors[0].more_details.details.rows[1].values[0].text).to be == "{:__cornucopia_no_analysis=>true}"
         expect(contents_frame.errors[0].more_details.details.rows[2].labels[0].text).to be == "page_url"
         expect(contents_frame.errors[0].more_details.details.rows[2].values[0].text).to match %r{sample_report/index\.html}
         expect(contents_frame.errors[0].more_details.details.rows[3].labels[0].text).to be == "title"
@@ -156,8 +156,8 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
         expect(contents_frame.errors[0].tables[0].rows[2].sub_tables[0].rows[0].values[0].text).to be == ":css"
         expect(contents_frame.errors[0].tables[0].rows[2].sub_tables[0].rows[1].labels[0].text).to be == "1"
         expect(contents_frame.errors[0].tables[0].rows[2].sub_tables[0].rows[1].values[0].text).to be == "a"
-        expect(contents_frame.errors[0].tables[0].rows[3].labels[0].text).to be == "exception"
-        expect(contents_frame.errors[0].tables[0].rows[3].values[0].text).to match /Ambiguous match, found .+ elements matching css "a"/
+        expect(contents_frame.errors[0].tables[0].rows[3].labels[0].text).to be == "message"
+        expect(contents_frame.errors[0].tables[0].rows[3].values[0].text).to match /Ambiguous match, found .+ elements matching visible css "a"/
         expect(contents_frame.errors[0].tables[0].rows[4].labels[0].text).to be == "backtrace"
         expect(contents_frame.errors[0].tables[0].rows[4].expands[0]).to be
         expect(contents_frame.errors[0].tables[0].rows[4].mores[0]).to be
@@ -173,7 +173,7 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
         expect(contents_frame.errors[0].more_details.details.rows[2].labels[0].text).to be == "guessed_types"
         expect(contents_frame.errors[0].more_details.details.rows[2].values[0].text).to be
         expect(contents_frame.errors[0].more_details.details.rows[3].labels[0].text).to be == "support_options"
-        expect(contents_frame.errors[0].more_details.details.rows[3].values[0].text).to be == "{:__cornucopia_no_analysis=>true, :__cornucopia_retry_with_found=>nil}"
+        expect(contents_frame.errors[0].more_details.details.rows[3].values[0].text).to be == "{:__cornucopia_no_analysis=>true}"
         expect(contents_frame.errors[0].more_details.details.rows[4].labels[0].text).to be == "page_url"
         expect(contents_frame.errors[0].more_details.details.rows[4].values[0].text).to match %r{sample_report/index\.html}
         expect(contents_frame.errors[0].more_details.details.rows[5].labels[0].text).to be == "title"
@@ -279,8 +279,6 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
         expect(contents_frame.errors[0].more_details.details.rows[1].labels[0].text).to be == "args"
         expect(contents_frame.errors[0].more_details.details.rows[1].sub_tables[0].rows[0].labels[0].text).to be == "1"
         expect(contents_frame.errors[0].more_details.details.rows[1].sub_tables[0].rows[0].values[0].text).to be == "a"
-        expect(contents_frame.errors[0].more_details.details.rows[1].sub_tables[0].rows[1].labels[0].text).to be == "2"
-        expect(contents_frame.errors[0].more_details.details.rows[1].sub_tables[0].rows[1].values[0].text).to be == "{:visible=>true}"
         expect(contents_frame.errors[0].more_details.details.rows[2].labels[0].text).to be == "guessed_types"
         expect(contents_frame.errors[0].more_details.details.rows[2].values[0].text).to be
         expect(contents_frame.errors[0].more_details.details.rows[3].labels[0].text).to be == "page_url"
@@ -539,7 +537,7 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
               to receive(:execute_script).at_least(1).times.and_raise(::Capybara::NotSupportedByDriverError)
 
           base_object = ::Capybara.current_session.find(:css, "#hidden-div", visible: false)
-          button      = base_object.first("Still cool!", count: 1)
+          button      = base_object.first("input[type='button']", count: 1)
 
           expect(button.class).to be == ::Capybara::Node::Element
 
@@ -574,7 +572,7 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
               to receive(:execute_script).at_least(1).times.and_raise(::Capybara::NotSupportedByDriverError)
 
           base_object = ::Capybara.current_session.find(:css, "#hidden-div", visible: false)
-          button      = base_object.assert_selector("Still cool!", count: 1)
+          button      = base_object.assert_selector("input[type='button']", count: 1)
 
           expect(button).to be true
 
@@ -617,18 +615,18 @@ describe Cornucopia::Capybara::FinderDiagnostics, type: :feature do
           allow_any_instance_of(::Cornucopia::Capybara::FinderDiagnostics::FindAction).
               to receive(:all_other_elements).and_return []
           allow(empty_find_button).to receive(:size).and_return 0
-          allow_any_instance_of(::Capybara::Node::Element).
+          allow_any_instance_of(Cornucopia::Capybara::FinderDiagnostics::FindAction).
               to receive(:all).
                      and_call_original
-          allow_any_instance_of(::Capybara::Node::Element).
+          allow_any_instance_of(Cornucopia::Capybara::FinderDiagnostics::FindAction).
               to receive(:all).
                      with("Still cool!", { count: 1 }).
                      and_return(find_button)
-          allow_any_instance_of(::Capybara::Node::Element).
+          allow_any_instance_of(Cornucopia::Capybara::FinderDiagnostics::FindAction).
               to receive(:all).
                      with(instance_of(Symbol), "Still cool!", { visible: false, __cornucopia_no_analysis: true }).
                      and_return(empty_find_button)
-          allow_any_instance_of(::Capybara::Node::Element).
+          allow_any_instance_of(Cornucopia::Capybara::FinderDiagnostics::FindAction).
               to receive(:all).
                      with("Still cool!", { visible: false, __cornucopia_no_analysis: true }).
                      and_return(empty_find_button)