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
61
62
63
64
65
66
67
68
69
70
|
# frozen_string_literal: true
require 'webmock'
require 'webmock/rspec'
def webmock_allowed_hosts
%w[elasticsearch registry.gitlab.com-gitlab-org-test-elastic-image].tap do |hosts|
if ENV.key?('ELASTIC_URL')
hosts << URI.parse(ENV['ELASTIC_URL']).host
end
if ENV.key?('ZOEKT_INDEX_BASE_URL')
hosts.concat(allowed_host_and_ip(ENV['ZOEKT_INDEX_BASE_URL']))
end
if ENV.key?('ZOEKT_SEARCH_BASE_URL')
hosts.concat(allowed_host_and_ip(ENV['ZOEKT_SEARCH_BASE_URL']))
end
# The test Rails server usually runs on 127.0.0.1.
# On some development configurations Webpack or Vite may be running on a
# different address so explicitly allow connections to that host.
if Gitlab.config.webpack&.dev_server&.enabled
hosts << Gitlab.config.webpack.dev_server.host
end
if Object.const_defined?(:ViteRuby) && ViteRuby.env['VITE_ENABLED'] == "true"
hosts << ViteRuby.instance.config.host
hosts << ViteRuby.env['VITE_HMR_HOST']
end
end.compact.uniq
end
def allowed_host_and_ip(url)
host = URI.parse(url).host
ip_address = Addrinfo.ip(host).ip_address
allowed = [host, ip_address]
# Sometimes IPv6 address has square brackets around it
parsed_ip = IPAddr.new(ip_address)
allowed << "[#{ip_address}]" if parsed_ip.ipv6?
allowed
end
def with_net_connect_allowed
WebMock.allow_net_connect!
yield
ensure
webmock_enable!
end
# This prevents Selenium/WebMock from spawning thousands of connections
# while waiting for an element to appear via Capybara's find:
# https://github.com/teamcapybara/capybara/issues/2322#issuecomment-619321520
def webmock_enable_with_http_connect_on_start!
webmock_enable!(net_http_connect_on_start: true)
end
def webmock_enable!(options = {})
WebMock.disable_net_connect!(
{
allow_localhost: true,
allow: webmock_allowed_hosts
}.merge(options)
)
end
webmock_enable!
|
