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
|
# 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
|
