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
|
# frozen_string_literal: true
require 'spec_helper'
require_migration!
RSpec.describe RemoveOrphanGroupTokenUsers, :migration, :sidekiq_inline do
subject(:migration) { described_class.new }
let(:users) { table(:users) }
let!(:orphan_bot) do
create_bot(username: 'orphan_bot', email: 'orphan_bot@bot.com').tap do |bot|
namespaces.create!(type: 'User', path: 'n1', name: 'n1', owner_id: bot.id)
end
end
let!(:valid_used_bot) do
create_bot(username: 'used_bot', email: 'used_bot@bot.com').tap do |bot|
group = namespaces.create!(type: 'Group', path: 'used_bot_group', name: 'used_bot_group')
members.create!(user_id: bot.id,
source_id: group.id,
member_namespace_id: group.id,
source_type: 'Group',
access_level: 10,
notification_level: 0)
end
end
let!(:different_bot) do
create_bot(username: 'other_bot', email: 'other_bot@bot.com', user_type: 5)
end
let(:personal_access_tokens) { table(:personal_access_tokens) }
let(:members) { table(:members) }
let(:namespaces) { table(:namespaces) }
before do
stub_feature_flags(user_destroy_with_limited_execution_time_worker: false)
end
it 'removes orphan project bot and its tokens', :aggregate_failures do
expect(DeleteUserWorker)
.to receive(:perform_async)
.with(orphan_bot.id, orphan_bot.id, skip_authorization: true)
.and_call_original
migrate!
expect(users.count).to eq 2
expect(personal_access_tokens.count).to eq 2
expect(personal_access_tokens.find_by(user_id: orphan_bot.id)).to eq nil
end
context "when DeleteUserWorker doesn't fit anymore" do
it 'removes project bot tokens only', :aggregate_failures do
allow(DeleteUserWorker).to receive(:respond_to?).and_call_original
allow(DeleteUserWorker).to receive(:respond_to?).with(:perform_async).and_return(false)
migrate!
expect(users.count).to eq 3
expect(personal_access_tokens.count).to eq 2
expect(personal_access_tokens.find_by(user_id: orphan_bot.id)).to eq nil
end
end
private
def create_bot(**params)
users.create!({ projects_limit: 0, state: 'active', user_type: 6 }.merge(params)).tap do |bot|
personal_access_tokens.create!(user_id: bot.id, name: "BOT##{bot.id}")
end
end
end
|
