BigW Consortium Gitlab

wait_for_requests.rb 1.5 KB
Newer Older
1
require_relative './wait_for_requests'
2

3 4 5 6
module WaitForRequests
  extend self

  # This is inspired by http://www.salsify.com/blog/engineering/tearing-capybara-ajax-tests
7
  def block_and_wait_for_requests_complete
8
    Gitlab::Testing::RequestBlockerMiddleware.block_requests!
9
    wait_for('pending requests complete') do
10
      Gitlab::Testing::RequestBlockerMiddleware.num_active_requests.zero? && finished_all_requests?
11 12 13 14 15
    end
  ensure
    Gitlab::Testing::RequestBlockerMiddleware.allow_requests!
  end

16 17 18 19 20 21 22 23 24 25 26 27 28
  def wait_for_requests
    wait_for('JS requests') { finished_all_requests? }
  end

  private

  def finished_all_requests?
    return true unless javascript_test?

    finished_all_ajax_requests? &&
      finished_all_vue_resource_requests?
  end

29 30 31 32 33 34 35 36 37 38 39 40
  # Waits until the passed block returns true
  def wait_for(condition_name, max_wait_time: Capybara.default_max_wait_time, polling_interval: 0.01)
    wait_until = Time.now + max_wait_time.seconds
    loop do
      break if yield
      if Time.now > wait_until
        raise "Condition not met: #{condition_name}"
      else
        sleep(polling_interval)
      end
    end
  end
41 42

  def finished_all_vue_resource_requests?
43
    Capybara.page.evaluate_script('window.activeVueResources || 0').zero?
44 45 46
  end

  def finished_all_ajax_requests?
47
    return true if Capybara.page.evaluate_script('typeof jQuery === "undefined"')
48

49
    Capybara.page.evaluate_script('jQuery.active').zero?
50 51 52 53 54
  end

  def javascript_test?
    Capybara.current_driver == Capybara.javascript_driver
  end
55
end