lib/gogo_driver.rb in gogo_driver-0.1.1 vs lib/gogo_driver.rb in gogo_driver-0.1.2

@@ -1,23 +1,25 @@
 require 'selenium-webdriver'
 
 class GogoDriver
   attr_accessor :driver
 
-  def initialize
-    @driver = Selenium::WebDriver.for(:chrome)
+  def initialize(browser=:chrome)
+    @driver = Selenium::WebDriver.for(browser)
   end
 
   def go(url)
+    logging "[VISITE] #{url}..."
     @driver.navigate.to(url)
   end
 
   def reload
     @driver.navigate.refresh
   end
 
   def find(selector)
+    logging "[FIND] #{selector}..."
     @driver.find_element(css: selector)
   end
 
   def finds(selector)
     @driver.find_elements(css: selector)
@@ -34,31 +36,44 @@
   rescue Selenium::WebDriver::Error::NoSuchElementError
     false
   end
 
   def click(selector)
+    logging "[CLICK] #{selector}..."
     has?(selector) ? find(selector).click : false
   end
 
   def submit
+    logging "[SUBMIT] ..."
     $focus.submit if $focus
   end
 
   def method_missing(method, *args, &block)
     @driver.respond_to?(method) ? @driver.send(method, *args, &block) : super
   end
+
+  private
+    def logging(text)
+      puts text
+    end
 end
 
 class Selenium::WebDriver::Element
   def fill(text)
     $focus = self
+    "[FILL] #{selector}..."
     send_key(text)
   end
 
   def find(selector)
     find_element(css: selector)
   end
 
   def finds(selector)
     find_elements(css: selector)
   end
+
+  private
+    def logging(text)
+      puts text
+    end
 end