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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
# frozen_string_literal: true
module QA
RSpec.shared_context(
'with gitlab group migration',
:import,
:orchestrated,
requires_admin: 'creates a user via API'
) do
let!(:import_wait_duration) { { max_duration: 120, sleep_interval: 2 } }
# source instance objects
#
let!(:source_gitlab_address) { ENV["QA_IMPORT_SOURCE_URL"] || raise("QA_IMPORT_SOURCE_URL is required!") }
let!(:source_admin_api_client) do
Runtime::API::Client.new(
source_gitlab_address,
personal_access_token: Runtime::Env.admin_personal_access_token || raise("Admin access token missing!"),
is_new_session: false
)
end
let!(:source_bulk_import_enabled) do
Runtime::ApplicationSettings.get_application_settings(api_client: source_admin_api_client)[:bulk_import_enabled]
end
let!(:source_admin_user) do
Resource::User.fabricate_via_api! do |usr|
usr.api_client = source_admin_api_client
usr.username = Runtime::Env.admin_username || "root"
end.tap(&:set_public_email)
end
let!(:source_group) do
Resource::Sandbox.fabricate_via_api! do |group|
group.api_client = source_admin_api_client
group.path = "source-group-for-import-#{SecureRandom.hex(4)}"
group.avatar = File.new(File.join(Runtime::Path.fixtures_path, 'designs', 'tanuki.jpg'), "r")
end
end
# target instance objects
#
let!(:admin_api_client) { Runtime::API::Client.as_admin }
let!(:target_bulk_import_enabled) do
Runtime::ApplicationSettings.get_application_settings(api_client: admin_api_client)[:bulk_import_enabled]
end
let!(:admin_user) do
Resource::User.fabricate_via_api! do |usr|
usr.api_client = admin_api_client
usr.username = Runtime::Env.admin_username || "root"
end.tap(&:set_public_email)
end
let!(:user) do
Resource::User.fabricate_via_api! do |usr|
usr.api_client = admin_api_client
usr.username = "target-user-#{SecureRandom.hex(6)}"
end
end
let!(:api_client) { Runtime::API::Client.new(user: user) }
let!(:target_sandbox) do
Resource::Sandbox.fabricate_via_api! do |group|
group.api_client = admin_api_client
end
end
let(:destination_group_path) { source_group.path }
let(:imported_group) do
Resource::BulkImportGroup.fabricate_via_api! do |group|
group.api_client = api_client
group.sandbox = target_sandbox
group.source_group = source_group
group.source_gitlab_address = source_gitlab_address
group.destination_group_path = destination_group_path
group.import_access_token = source_admin_api_client.personal_access_token
end
end
let(:import_failures) do
imported_group.import_details.sum([]) { |details| details[:failures] }
end
let(:cleanup!) {}
def expect_group_import_finished_successfully
imported_group # trigger import
status = nil
Support::Retrier.retry_until(**import_wait_duration, raise_on_failure: false) do
status = imported_group.import_status
%w[finished failed].include?(status)
end
# finished status means success, all other statuses are considered to fail the test
expect(status).to eq('finished'), "Expected import to finish successfully, but status was: #{status}"
end
before do
enable_bulk_import(source_admin_api_client) if source_admin_user && !source_bulk_import_enabled
enable_bulk_import(admin_api_client) unless target_bulk_import_enabled
target_sandbox.add_member(user, Resource::Members::AccessLevel::OWNER)
end
after do |example|
# Checking for failures in the test currently makes test very flaky due to catching unrelated failures
# Log failures for easier debugging
Runtime::Logger.error("Import failures: #{import_failures}") if example.exception && !import_failures.empty?
rescue StandardError
# rescue when import did not happen at all and checking import failues will raise an error
ensure
# make sure cleanup runs last
cleanup!
end
def enable_bulk_import(api_client)
Runtime::ApplicationSettings.set_application_settings(api_client: api_client, bulk_import_enabled: true)
end
end
end
|
