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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::GithubImport::Importer::CollaboratorsImporter, feature_category: :importers do
subject(:importer) { described_class.new(project, client, parallel: parallel) }
let(:parallel) { true }
let(:project) { build(:project, id: 4, import_source: 'foo/bar', import_state: nil) }
let(:client) { instance_double(Gitlab::GithubImport::Client) }
let(:github_collaborator) do
{
id: 100500,
login: 'bob',
role_name: 'maintainer'
}
end
describe '#parallel?' do
context 'when parallel option is true' do
it { expect(importer).to be_parallel }
end
context 'when parallel option is false' do
let(:parallel) { false }
it { expect(importer).not_to be_parallel }
end
end
describe '#execute' do
context 'when running in parallel mode' do
it 'imports collaborators in parallel' do
expect(importer).to receive(:parallel_import)
importer.execute
end
end
context 'when running in sequential mode' do
let(:parallel) { false }
it 'imports collaborators in sequence' do
expect(importer).to receive(:sequential_import)
importer.execute
end
end
end
describe '#sequential_import' do
let(:parallel) { false }
it 'imports each collaborator in sequence' do
collaborator_importer = instance_double(Gitlab::GithubImport::Importer::CollaboratorImporter)
allow(importer)
.to receive(:each_object_to_import)
.and_yield(github_collaborator)
expect(Gitlab::GithubImport::Importer::CollaboratorImporter)
.to receive(:new)
.with(
an_instance_of(Gitlab::GithubImport::Representation::Collaborator),
project,
client
)
.and_return(collaborator_importer)
expect(collaborator_importer).to receive(:execute)
importer.sequential_import
end
end
describe '#parallel_import', :clean_gitlab_redis_cache do
before do
allow(Gitlab::Redis::SharedState).to receive(:with).and_return('OK')
allow(client).to receive(:collaborators).with(project.import_source, affiliation: 'direct')
.and_return([github_collaborator])
allow(client).to receive(:collaborators).with(project.import_source, affiliation: 'outside')
.and_return([])
end
it 'imports each collaborator in parallel' do
expect(Gitlab::GithubImport::ImportCollaboratorWorker).to receive(:perform_in)
.with(1, project.id, an_instance_of(Hash), an_instance_of(String))
waiter = importer.parallel_import
expect(waiter).to be_an_instance_of(Gitlab::JobWaiter)
expect(waiter.jobs_remaining).to eq(1)
end
context 'when collaborator is already imported' do
before do
Gitlab::Cache::Import::Caching.set_add(
"github-importer/already-imported/#{project.id}/collaborators",
github_collaborator[:id]
)
end
it "doesn't run importer on it" do
expect(Gitlab::GithubImport::ImportCollaboratorWorker).not_to receive(:perform_in)
waiter = importer.parallel_import
expect(waiter).to be_an_instance_of(Gitlab::JobWaiter)
expect(waiter.jobs_remaining).to eq(0)
end
end
end
describe '#each_object_to_import', :clean_gitlab_redis_cache do
let(:github_collaborator_2) { { id: 100501, login: 'alice', role_name: 'owner' } }
let(:github_collaborator_3) { { id: 100502, login: 'tom', role_name: 'guest' } }
before do
allow(client).to receive(:collaborators).with(project.import_source, affiliation: 'direct')
.and_return([github_collaborator, github_collaborator_2, github_collaborator_3])
allow(client).to receive(:collaborators).with(project.import_source, affiliation: 'outside')
.and_return([github_collaborator_3])
allow(Gitlab::GithubImport::ObjectCounter).to receive(:increment)
.with(project, :collaborator, :fetched)
end
it 'yields every direct collaborator who is not an outside collaborator to the supplied block' do
expect { |b| importer.each_object_to_import(&b) }
.to yield_successive_args(github_collaborator, github_collaborator_2)
expect(Gitlab::GithubImport::ObjectCounter).to have_received(:increment).twice
end
context 'when a collaborator has been already imported' do
before do
allow(importer).to receive(:already_imported?).and_return(true)
end
it 'does not yield anything' do
expect(Gitlab::GithubImport::ObjectCounter)
.not_to receive(:increment)
expect(importer)
.not_to receive(:mark_as_imported)
expect { |b| importer.each_object_to_import(&b) }
.not_to yield_control
end
end
end
describe '#id_for_already_imported_cache' do
it 'returns the ID of the given note' do
expect(importer.id_for_already_imported_cache(github_collaborator))
.to eq(100500)
end
end
end
|