lib/polonium/dsl/selenium_dsl.rb in polonium-0.1.0 vs lib/polonium/dsl/selenium_dsl.rb in polonium-0.1.1

@@ -19,14 +19,17 @@
     def open_home_page
       selenium_driver.open(configuration.browser_url)
     end
 
     protected
-    Driver.public_instance_methods(false).each do |method_name|
-      delegate method_name, :to => :selenium_driver
+    def method_missing(method_name, *args, &block)
+      selenium_driver.__send__(method_name, *args, &block)
     end
+    delegate :open,
+             :type,
+             :to => :selenium_driver
 
-    def should_stop_driver?
+    def stop_driver?
       return false unless configuration.test_browser_mode?
       configuration.stop_driver?(passed?)
     end
   end
 end
\ No newline at end of file