80 lines
2.1 KiB
Ruby
80 lines
2.1 KiB
Ruby
module WaitForRequests
|
|
extend self
|
|
|
|
# This is inspired by http://www.salsify.com/blog/engineering/tearing-capybara-ajax-tests
|
|
def block_and_wait_for_requests_complete
|
|
block_requests { wait_for_all_requests }
|
|
end
|
|
|
|
# Block all requests inside block with 503 response
|
|
def block_requests
|
|
Gitlab::Testing::RequestBlockerMiddleware.block_requests!
|
|
yield
|
|
ensure
|
|
Gitlab::Testing::RequestBlockerMiddleware.allow_requests!
|
|
end
|
|
|
|
# Slow down requests inside block by injecting `sleep 0.2` before each response
|
|
def slow_requests
|
|
Gitlab::Testing::RequestBlockerMiddleware.slow_requests!
|
|
yield
|
|
ensure
|
|
Gitlab::Testing::RequestBlockerMiddleware.allow_requests!
|
|
end
|
|
|
|
# Wait for client-side AJAX requests
|
|
def wait_for_requests
|
|
wait_for('JS requests complete', max_wait_time: 2 * Capybara.default_max_wait_time) do
|
|
finished_all_js_requests?
|
|
end
|
|
end
|
|
|
|
# Wait for active Rack requests and client-side AJAX requests
|
|
def wait_for_all_requests
|
|
wait_for('pending requests complete') do
|
|
finished_all_rack_reqiests? &&
|
|
finished_all_js_requests?
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def finished_all_rack_reqiests?
|
|
Gitlab::Testing::RequestBlockerMiddleware.num_active_requests.zero?
|
|
end
|
|
|
|
def finished_all_js_requests?
|
|
return true unless javascript_test?
|
|
|
|
finished_all_ajax_requests? &&
|
|
finished_all_vue_resource_requests?
|
|
end
|
|
|
|
# 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
|
|
|
|
def finished_all_vue_resource_requests?
|
|
Capybara.page.evaluate_script('window.activeVueResources || 0').zero?
|
|
end
|
|
|
|
def finished_all_ajax_requests?
|
|
return true if Capybara.page.evaluate_script('typeof jQuery === "undefined"')
|
|
|
|
Capybara.page.evaluate_script('jQuery.active').zero?
|
|
end
|
|
|
|
def javascript_test?
|
|
Capybara.current_driver == Capybara.javascript_driver
|
|
end
|
|
end
|