1 files changed, 55 insertions, 0 deletions
diff --git a/spec/migrations/20230105172120_sync_new_amount_used_with_amount_used_on_ci_namespace_monthly_usages_table_spec.rb b/spec/migrations/20230105172120_sync_new_amount_used_with_amount_used_on_ci_namespace_monthly_usages_table_spec.rb
new file mode 100644
index 00000000000..aa82ca2661b
--- /dev/null
+++ b/spec/migrations/20230105172120_sync_new_amount_used_with_amount_used_on_ci_namespace_monthly_usages_table_spec.rb
@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+require_migration!
+
+RSpec.describe SyncNewAmountUsedWithAmountUsedOnCiNamespaceMonthlyUsagesTable, migration: :gitlab_ci,
+feature_category: :continuous_integration do
+  let(:namespace_usages) { table(:ci_namespace_monthly_usages) }
+  let(:migration) { described_class.new }
+
+  before do
+    # Disabling the trigger temporarily to allow records being created with out-of-sync
+    # `new_amount_used` and `amount_used`. This will simulate existing records before
+    # we add the trigger.
+    ActiveRecord::Base.connection
+      .execute("ALTER TABLE ci_namespace_monthly_usages DISABLE TRIGGER sync_namespaces_amount_used_columns")
+
+    this_month = Time.now.utc.beginning_of_month
+    last_month = 1.month.ago.utc.beginning_of_month
+    last_year = 1.year.ago.utc.beginning_of_month
+
+    namespace_usages.create!(namespace_id: 1, date: last_year)
+    namespace_usages.create!(namespace_id: 1, date: this_month, amount_used: 10, new_amount_used: 0)
+    namespace_usages.create!(namespace_id: 1, date: last_month, amount_used: 20, new_amount_used: 0)
+
+    namespace_usages.create!(namespace_id: 2, date: last_year)
+    namespace_usages.create!(namespace_id: 2, date: this_month, amount_used: 30, new_amount_used: 0)
+    namespace_usages.create!(namespace_id: 2, date: last_month, amount_used: 40, new_amount_used: 0)
+
+    ActiveRecord::Base.connection
+      .execute("ALTER TABLE ci_namespace_monthly_usages ENABLE TRIGGER sync_namespaces_amount_used_columns")
+  end
+
+  describe '#up' do
+    it "doesnt change new_amount_used values" do
+      data = namespace_usages.all
+      data.each do |item|
+        expect { migration.up }.to not_change { item.new_amount_used }
+      end
+    end
+  end
+
+  describe '#down' do
+    it 'updates `new_amount_used` with values from `amount_used`' do
+      expect(namespace_usages.where(new_amount_used: 0).count).to eq(6)
+
+      migration.down
+
+      expect(namespace_usages.where(new_amount_used: 0).count).to eq(2)
+      expect(namespace_usages.order(:id).pluck(:new_amount_used))
+        .to contain_exactly(0, 0, 10, 20, 30, 40)
+    end
+  end
+end