watir_pump 0.3.6 → 0.3.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f6e3ece528bd0040e107b8e711a7701e1ef995e2
-  data.tar.gz: 427f32cf5b32977392de849dc500801f0feaa8c2
+  metadata.gz: 6ec31a031d8aedd7d62a03f29e2e3b7926efdc68
+  data.tar.gz: 7089d63c641acb31767e957abb5854247d92aba6
 SHA512:
-  metadata.gz: e05ee7ce157807acafc2c30ee45abad858d0a95fe966c8944a0b687ff72978a567a5520bd84f68770123d48f2cfca0ee7fe36b595773a4273024746c51a4fee0
-  data.tar.gz: 935f8cfc562db1be85227e4a638596e5d53340dbcb7130ca8406fc832675fbc170510bb32b95ef237c90d1332f0cc0762d87b992a08e87da4fe5e830507015d9
+  metadata.gz: c9c9372a6656ff88907f40f71346d232f378cce8f7324f072d17ba1085fd4efc603d2d1132815d629cc7fd327102d7bd4b58892f842c3397cae8df5ff817641b
+  data.tar.gz: 1ef6400e082d0c038cbfa5ad4606a5078ff2cc376840764d86ec03f32a8048f3824a84eb047cbec9226ee320645164d5b71495117055a70dfe278494e2672277

data/lib/watir_pump/component.rb

@@ -126,8 +126,26 @@ module WatirPump
           instance_exec(*args, &p)
         end
       end
-      alias element query
-      alias elements query
+
+      def element(name, p)
+        define_method(name) do |*args|
+          ret = instance_exec(*args, &p)
+          unless ret.is_a?(Watir::Element)
+            raise 'element method did not return a Watir::Element'
+          end
+          ret
+        end
+      end
+
+      def elements(name, p)
+        define_method(name) do |*args|
+          ret = instance_exec(*args, &p)
+          unless ret.is_a?(Watir::ElementCollection)
+            raise 'elements method did not return a Watir::ElementCollection'
+          end
+          ret
+        end
+      end
 
       def region(name, loc_method = nil, *loc_args, &blk)
         klass = Class.new(Component) { instance_exec(&blk) }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: watir_pump
 version: !ruby/object:Gem::Version
-  version: 0.3.6
+  version: 0.3.7
 platform: ruby
 authors:
 - Bartek Wilczek