lib/symbiont/pages.rb in symbiont-0.3.0 vs lib/symbiont/pages.rb in symbiont-0.4.0

@@ -1,32 +1,127 @@
 module Symbiont
   module Page
     include Helpers
 
     def view
-      no_url_is_provided if url.nil?
-      driver.goto(url)
+      no_url_is_provided if asserted_url.nil?
+      driver.goto(asserted_url)
     end
 
     def has_correct_url?
       no_url_matches_is_provided if url_match.nil?
       !(driver.url =~ url_match).nil?
     end
 
     def has_correct_title?
-      no_title_is_provided if page_title.nil?
-      !(driver.title.match(page_title)).nil?
+      no_title_is_provided if asserted_title.nil?
+      !(driver.title.match(asserted_title)).nil?
     end
 
-    def url
-      self.class.url
+    def asserted_url
+      self.class.asserted_url
     end
 
     def url_match
       self.class.url_match
     end
 
-    def page_title
-      self.class.page_title
+    def asserted_title
+      self.class.asserted_title
     end
+
+    def url
+      driver.url
+    end
+
+    def markup
+      driver.html
+    end
+
+    def text
+      driver.text
+    end
+
+    def title
+      driver.title
+    end
+
+    def visit(url)
+      driver.goto(url)
+    end
+
+    def screenshot(file)
+      driver.wd.save_screenshot(file)
+    end
+
+    def run_script(script, *args)
+      driver.execute_script(script, *args)
+    end
+
+    def get_cookie(name)
+      for cookie in driver.cookies.to_a
+        if cookie[:name] == name
+          return cookie[:value]
+        end
+      end
+      nil
+    end
+
+    def clear_cookies
+      driver.cookies.clear
+    end
+
+    def refresh
+      driver.refresh
+    end
+
+    # @param block [Proc] the code that generates the alert
+    # @return [String] the message contained in the alert message box
+    def will_alert(&block)
+      yield
+      value = nil
+      if driver.alert.exists?
+        value = driver.alert.text
+        driver.alert.ok
+      end
+      value
+    end
+
+    # @param response [Boolean] true to accept the confirmation, false to cancel it
+    # @param block [Proc] the code that generates the confirmation
+    # @return [String] the message contained in the confirmation message box
+    def will_confirm(response, &block)
+      yield
+      value = nil
+      if driver.alert.exists?
+        value = driver.alert.text
+        response ? driver.alert.ok : driver.alert.close
+      end
+      value
+    end
+
+    # @param response [String] the value to be used in the prompt
+    # @param block [Proc] the code that generates the prompt
+    # @return [Hash] :message for the prompt message, :default_value for
+    # the value that the prompt had before the response was applied
+    def will_prompt(response, &block)
+      cmd = "window.prompt = function(text, value) {window.__lastWatirPrompt = {message: text, default_value: value}; return '#{response}';}"
+      driver.wd.execute_script(cmd)
+      yield
+      result = driver.wd.execute_script('return window.__lastWatirPrompt')
+      result && result.dup.each_key { |k| result[k.to_sym] = result.delete(k) }
+      result
+    end
+
+    alias_method :current_url, :url
+    alias_method :page_url, :url
+    alias_method :html, :markup
+    alias_method :page_text, :text
+    alias_method :page_title, :title
+    alias_method :navigate_to, :visit
+    alias_method :goto, :visit
+    alias_method :save_screenshot, :screenshot
+    alias_method :execute_script, :run_script
+    alias_method :remove_cookies, :clear_cookies
+    alias_method :refresh_page, :refresh
   end
 end