apps/workbench/test/integration_helper.rb

   1 require 'test_helper'
   2 require 'capybara/rails'
   3 require 'capybara/poltergeist'
   4 require 'uri'
   5 require 'yaml'
   6
   7 module WaitForAjax
   8   Capybara.default_wait_time = 5
   9   def wait_for_ajax
  10     Timeout.timeout(Capybara.default_wait_time) do
  11       loop until finished_all_ajax_requests?
  12     end
  13   end
  14
  15   def finished_all_ajax_requests?
  16     page.evaluate_script('jQuery.active').zero?
  17   end
  18 end
  19
  20 class ActionDispatch::IntegrationTest
  21   # Make the Capybara DSL available in all integration tests
  22   include Capybara::DSL
  23   include ApiFixtureLoader
  24   include WaitForAjax
  25
  26   @@API_AUTHS = self.api_fixture('api_client_authorizations')
  27
  28   def page_with_token(token, path='/')
  29     # Generate a page path with an embedded API token.
  30     # Typical usage: visit page_with_token('token_name', page)
  31     # The token can be specified by the name of an api_client_authorizations
  32     # fixture, or passed as a raw string.
  33     api_token = ((@@API_AUTHS.include? token) ?
  34                  @@API_AUTHS[token]['api_token'] : token)
  35     sep = (path.include? '?') ? '&' : '?'
  36     q_string = URI.encode_www_form('api_token' => api_token)
  37     "#{path}#{sep}#{q_string}"
  38   end
  39
  40   # Find a page element, but return false instead of raising an
  41   # exception if not found. Use this with assertions to explain that
  42   # the error signifies a failed test rather than an unexpected error
  43   # during a testing procedure.
  44   def find? *args
  45     begin
  46       find *args
  47     rescue Capybara::ElementNotFound
  48       false
  49     end
  50   end
  51
  52   @@screenshot_count = 1
  53   def screenshot
  54     image_file = "./tmp/workbench-fail-#{@@screenshot_count}.png"
  55     begin
  56       page.save_screenshot image_file
  57     rescue Capybara::NotSupportedByDriverError
  58       # C'est la vie.
  59     else
  60       puts "Saved #{image_file}"
  61       @@screenshot_count += 1
  62     end
  63   end
  64
  65   teardown do
  66     if not passed?
  67       screenshot
  68     end
  69     if Capybara.current_driver == :selenium
  70       page.execute_script("window.localStorage.clear()")
  71     end
  72     Capybara.reset_sessions!
  73   end
  74 end