lib/watir/container.rb in watir-1.6.6 vs lib/watir/container.rb in watir-1.6.7.rc1

@@ -818,11 +818,11 @@
     #                  - :index
     #                  - :value etc
     #   * what  - string that we are looking for, ex. the name, or id tag attribute or index of the object we are looking for.
     #   * types - what object types we will look at.
     #   * value - used for objects that have one name, but many values. ex. radio lists and checkboxes
-    def locate_input_element(how, what, types, value=nil)
+    def locate_input_element(how, what, types, value=nil, klass=nil)
       case how
       when :xpath
         return element_by_xpath(what)
       when :ole_object
         return what
@@ -833,9 +833,10 @@
       locator.specifier = [how, what, value]
       locator.document = document
       return locator.element if locator.fast_locate
       # todo: restrict search to elements.getElementsByTag('INPUT'); faster
       locator.elements = ole_inner_elements if locator.elements.nil?
+      locator.klass = klass if klass 
       locator.locate
     end
     
     # returns the ole object for the specified element
     def locate_tagged_element(tag, how, what)