blob: 5e2e8ad53e091f24321d8a1209ad7fc7bcdd1d59 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
# frozen_string_literal: true
require_relative 'wait_helpers'
module WaitForRequests
include WaitHelpers
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_requests? &&
finished_all_js_requests?
end
end
private
def finished_all_rack_requests?
Gitlab::Testing::RequestBlockerMiddleware.num_active_requests == 0
end
def finished_all_js_requests?
return true unless javascript_test?
finished_all_ajax_requests?
end
def finished_all_ajax_requests?
Capybara.page.evaluate_script('window.pendingRequests || window.pendingApolloRequests || window.pendingRailsUJSRequests || 0').zero? # rubocop:disable Style/NumericPredicate
end
end
|