Commit bbf80b40 authored by Yorick Peterse's avatar Yorick Peterse

Added tests for removing soft removed objects

parent a2921246
...@@ -156,6 +156,15 @@ class RemoveSoftRemovedObjects < ActiveRecord::Migration ...@@ -156,6 +156,15 @@ class RemoveSoftRemovedObjects < ActiveRecord::Migration
end end
def remove_group_namespaces def remove_group_namespaces
admin_id = id_for_admin_user
unless admin_id
say 'Not scheduling soft removed groups for removal as no admin user ' \
'could be found. You will need to remove any such groups manually.'
return
end
# Left over groups can't be easily removed because we may also need to # Left over groups can't be easily removed because we may also need to
# remove memberships, repositories, and other associated data. As a result # remove memberships, repositories, and other associated data. As a result
# we'll just schedule a Sidekiq job to remove these. # we'll just schedule a Sidekiq job to remove these.
...@@ -163,10 +172,6 @@ class RemoveSoftRemovedObjects < ActiveRecord::Migration ...@@ -163,10 +172,6 @@ class RemoveSoftRemovedObjects < ActiveRecord::Migration
# As of January 5th, 2018 there are 36 groups that will be removed using # As of January 5th, 2018 there are 36 groups that will be removed using
# this code. # this code.
Namespace.select(:id).soft_removed_group.each_batch(of: 10) do |batch, index| Namespace.select(:id).soft_removed_group.each_batch(of: 10) do |batch, index|
# We need the ID of an admin user as the owners of the group may no longer
# exist (or might not even be set in `namespaces.owner_id`).
admin_id = id_for_admin_user
batch.each do |ns| batch.each do |ns|
schedule_group_removal(index * 5.minutes, ns.id, admin_id) schedule_group_removal(index * 5.minutes, ns.id, admin_id)
end end
...@@ -194,14 +199,7 @@ class RemoveSoftRemovedObjects < ActiveRecord::Migration ...@@ -194,14 +199,7 @@ class RemoveSoftRemovedObjects < ActiveRecord::Migration
end end
def id_for_admin_user def id_for_admin_user
return @id_for_admin_user if @id_for_admin_user User.where(admin: true).limit(1).pluck(:id).first
if (admin_id = User.where(admin: true).limit(1).pluck(:id).first)
@id_for_admin_user = admin_id
else
raise 'Can not remove soft removed groups as no admin user exists. ' \
'Please make sure at least one user with `admin` set to TRUE exists before proceeding.'
end
end end
def migrate_inline? def migrate_inline?
......
require 'spec_helper'
require Rails.root.join('db', 'post_migrate', '20171207150343_remove_soft_removed_objects.rb')
describe RemoveSoftRemovedObjects, :migration do
describe '#up' do
it 'removes various soft removed objects' do
5.times do
create_with_deleted_at(:issue)
end
regular_issue = create(:issue)
run_migration
expect(Issue.count).to eq(1)
expect(Issue.first).to eq(regular_issue)
end
it 'removes the temporary indexes once soft removed data has been removed' do
migration = described_class.new
run_migration
disable_migrations_output do
expect(migration.temporary_index_exists?(Issue)).to eq(false)
end
end
it 'removes routes of soft removed personal namespaces' do
namespace = create_with_deleted_at(:namespace)
group = create(:group)
expect(Route.where(source: namespace).exists?).to eq(true)
expect(Route.where(source: group).exists?).to eq(true)
run_migration
expect(Route.where(source: namespace).exists?).to eq(false)
expect(Route.where(source: group).exists?).to eq(true)
end
it 'schedules the removal of soft removed groups' do
group = create_with_deleted_at(:group)
admin = create(:user, admin: true)
expect_any_instance_of(GroupDestroyWorker)
.to receive(:perform)
.with(group.id, admin.id)
run_migration
end
it 'does not remove soft removed groups when no admin user could be found' do
create_with_deleted_at(:group)
expect_any_instance_of(GroupDestroyWorker)
.not_to receive(:perform)
run_migration
end
end
def run_migration
disable_migrations_output do
migrate!
end
end
def create_with_deleted_at(*args)
row = create(*args)
# We set "deleted_at" this way so we don't run into any column cache issues.
row.class.where(id: row.id).update_all(deleted_at: 1.year.ago)
row
end
end
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment