validate text query types

teamcapybara · Jun 20, 2020 · b81aed5 · b81aed5
1 parent 4db3463
commit b81aed5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/capybara/queries/text_query.rb b/lib/capybara/queries/text_query.rb
@@ -6,6 +6,8 @@ module Queries
     class TextQuery < BaseQuery
       def initialize(type = nil, expected_text, session_options:, **options) # rubocop:disable Style/OptionalArguments
         @type = type.nil? ? default_type : type
+        raise ArgumentError, '${@type} is not a valid type for a text query' unless valid_types.include?(@type)
+
         @options = options
         super(@options)
         self.session_options = session_options
@@ -89,6 +91,10 @@ def valid_keys
         COUNT_KEYS + %i[wait exact normalize_ws]
       end
 
+      def valid_types
+        %i[all visible]
+      end
+
       def check_visible_text?
         @type == :visible
       end