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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::BackgroundMigration::BackfillProjectFeaturePackageRegistryAccessLevel do
let(:non_null_project_features) { { pages_access_level: 20 } }
let(:namespaces) { table(:namespaces) }
let(:projects) { table(:projects) }
let(:project_features) { table(:project_features) }
let(:namespace1) { namespaces.create!(name: 'namespace 1', path: 'namespace1') }
let(:namespace2) { namespaces.create!(name: 'namespace 2', path: 'namespace2') }
let(:namespace3) { namespaces.create!(name: 'namespace 3', path: 'namespace3') }
let(:namespace4) { namespaces.create!(name: 'namespace 4', path: 'namespace4') }
let(:namespace5) { namespaces.create!(name: 'namespace 5', path: 'namespace5') }
let(:namespace6) { namespaces.create!(name: 'namespace 6', path: 'namespace6') }
let(:project1) do
projects.create!(namespace_id: namespace1.id, project_namespace_id: namespace1.id, packages_enabled: false)
end
let(:project2) do
projects.create!(namespace_id: namespace2.id, project_namespace_id: namespace2.id, packages_enabled: nil)
end
let(:project3) do
projects.create!(
namespace_id: namespace3.id,
project_namespace_id: namespace3.id,
packages_enabled: true,
visibility_level: Gitlab::VisibilityLevel::PRIVATE
)
end
let(:project4) do
projects.create!(
namespace_id: namespace4.id,
project_namespace_id: namespace4.id,
packages_enabled: true, visibility_level: Gitlab::VisibilityLevel::INTERNAL)
end
let(:project5) do
projects.create!(
namespace_id: namespace5.id,
project_namespace_id: namespace5.id,
packages_enabled: true,
visibility_level: Gitlab::VisibilityLevel::PUBLIC
)
end
let(:project6) do
projects.create!(namespace_id: namespace6.id, project_namespace_id: namespace6.id, packages_enabled: false)
end
let!(:project_feature1) do
project_features.create!(
project_id: project1.id,
package_registry_access_level: ProjectFeature::ENABLED,
**non_null_project_features
)
end
let!(:project_feature2) do
project_features.create!(
project_id: project2.id,
package_registry_access_level: ProjectFeature::ENABLED,
**non_null_project_features
)
end
let!(:project_feature3) do
project_features.create!(
project_id: project3.id,
package_registry_access_level: ProjectFeature::DISABLED,
**non_null_project_features
)
end
let!(:project_feature4) do
project_features.create!(
project_id: project4.id,
package_registry_access_level: ProjectFeature::DISABLED,
**non_null_project_features
)
end
let!(:project_feature5) do
project_features.create!(
project_id: project5.id,
package_registry_access_level: ProjectFeature::DISABLED,
**non_null_project_features
)
end
let!(:project_feature6) do
project_features.create!(
project_id: project6.id,
package_registry_access_level: ProjectFeature::ENABLED,
**non_null_project_features
)
end
subject(:perform_migration) do
described_class.new(start_id: project1.id,
end_id: project5.id,
batch_table: :projects,
batch_column: :id,
sub_batch_size: 2,
pause_ms: 0,
connection: ActiveRecord::Base.connection)
.perform
end
it 'backfills project_features.package_registry_access_level', :aggregate_failures do
perform_migration
expect(project_feature1.reload.package_registry_access_level).to eq(ProjectFeature::DISABLED)
expect(project_feature2.reload.package_registry_access_level).to eq(ProjectFeature::DISABLED)
expect(project_feature3.reload.package_registry_access_level).to eq(ProjectFeature::PRIVATE)
expect(project_feature4.reload.package_registry_access_level).to eq(ProjectFeature::ENABLED)
expect(project_feature5.reload.package_registry_access_level).to eq(ProjectFeature::PUBLIC)
expect(project_feature6.reload.package_registry_access_level).to eq(ProjectFeature::ENABLED)
end
end
|