1 files changed, 59 insertions, 0 deletions
diff --git a/spec/lib/gitlab/background_migration/disable_legacy_open_source_license_for_projects_less_than_one_mb_spec.rb b/spec/lib/gitlab/background_migration/disable_legacy_open_source_license_for_projects_less_than_one_mb_spec.rb
new file mode 100644
index 00000000000..205350f9df4
--- /dev/null
+++ b/spec/lib/gitlab/background_migration/disable_legacy_open_source_license_for_projects_less_than_one_mb_spec.rb
@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::BackgroundMigration::DisableLegacyOpenSourceLicenseForProjectsLessThanOneMb,
+               :migration,
+               schema: 20220906074449 do
+  let(:namespaces_table) { table(:namespaces) }
+  let(:projects_table) { table(:projects) }
+  let(:project_settings_table) { table(:project_settings) }
+  let(:project_statistics_table) { table(:project_statistics) }
+
+  subject(:perform_migration) do
+    described_class.new(start_id: project_settings_table.minimum(:project_id),
+                        end_id: project_settings_table.maximum(:project_id),
+                        batch_table: :project_settings,
+                        batch_column: :project_id,
+                        sub_batch_size: 2,
+                        pause_ms: 0,
+                        connection: ActiveRecord::Base.connection)
+                   .perform
+  end
+
+  it 'sets `legacy_open_source_license_available` to false only for projects less than 1 MB',
+     :aggregate_failures do
+    project_setting_1_mb = create_legacy_license_project_setting(repo_size: 1)
+    project_setting_2_mb = create_legacy_license_project_setting(repo_size: 2)
+    project_setting_quarter_mb = create_legacy_license_project_setting(repo_size: 0.25)
+    project_setting_half_mb = create_legacy_license_project_setting(repo_size: 0.5)
+
+    queries = ActiveRecord::QueryRecorder.new { perform_migration }
+
+    expect(queries.count).to eq(7)
+    expect(migrated_attribute(project_setting_1_mb)).to be_truthy
+    expect(migrated_attribute(project_setting_2_mb)).to be_truthy
+    expect(migrated_attribute(project_setting_quarter_mb)).to be_falsey
+    expect(migrated_attribute(project_setting_half_mb)).to be_falsey
+  end
+
+  private
+
+  # @param repo_size: Repo size in MB
+  def create_legacy_license_project_setting(repo_size:)
+    path = "path-for-repo-size-#{repo_size}"
+    namespace = namespaces_table.create!(name: "namespace-#{path}", path: "namespace-#{path}")
+    project_namespace =
+      namespaces_table.create!(name: "-project-namespace-#{path}", path: "project-namespace-#{path}", type: 'Project')
+    project = projects_table
+                .create!(name: path, path: path, namespace_id: namespace.id, project_namespace_id: project_namespace.id)
+
+    size_in_bytes = 1.megabyte * repo_size
+    project_statistics_table.create!(project_id: project.id, namespace_id: namespace.id, repository_size: size_in_bytes)
+    project_settings_table.create!(project_id: project.id, legacy_open_source_license_available: true)
+  end
+
+  def migrated_attribute(project_setting)
+    project_settings_table.find(project_setting.project_id).legacy_open_source_license_available
+  end
+end