Merge branch '299054-fj-update-groups-repository-storage-class' into 'master'

Include logic to move storage in Groups::RepositoryStorageMove class

See merge request gitlab-org/gitlab!52513

ee/app/models/groups/repository_storage_move.rb

@@ -5,11 +5,26 @@
 # balance storage capacity.
 module Groups
   class RepositoryStorageMove < ApplicationRecord
+    extend ::Gitlab::Utils::Override
+    include RepositoryStorageMovable
+
     self.table_name = 'group_repository_storage_moves'
 
     belongs_to :container, class_name: 'Group', inverse_of: :repository_storage_moves, foreign_key: :group_id
     alias_attribute :group, :container
 
     scope :with_groups, -> { includes(container: :route) }
+
+    override :schedule_repository_storage_update_worker
+    def schedule_repository_storage_update_worker
+      # No-op. It will be implemented in https://gitlab.com/gitlab-org/gitlab/-/issues/299059
+    end
+
+    private
+
+    override :error_key
+    def error_key
+      :group
+    end
   end
 end

ee/spec/factories/group_repository_storage_moves.rb

+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :group_repository_storage_move, class: 'Groups::RepositoryStorageMove' do
+    container { association(:group) }
+
+    source_storage_name { 'default' }
+
+    trait :scheduled do
+      state { Groups::RepositoryStorageMove.state_machines[:state].states[:scheduled].value }
+    end
+
+    trait :started do
+      state { Groups::RepositoryStorageMove.state_machines[:state].states[:started].value }
+    end
+
+    trait :replicated do
+      state { Groups::RepositoryStorageMove.state_machines[:state].states[:replicated].value }
+    end
+
+    trait :finished do
+      state { Groups::RepositoryStorageMove.state_machines[:state].states[:finished].value }
+    end
+
+    trait :failed do
+      state { Groups::RepositoryStorageMove.state_machines[:state].states[:failed].value }
+    end
+  end
+end

ee/spec/models/groups/repository_storage_move_spec.rb

@@ -3,7 +3,10 @@
 require 'spec_helper'
 
 RSpec.describe Groups::RepositoryStorageMove, type: :model do
-  describe 'associations' do
-    it { is_expected.to belong_to(:container).class_name('Group') }
+  it_behaves_like 'handles repository moves', check_worker: false do
+    let_it_be_with_refind(:container) { create(:group) }
+
+    let(:repository_storage_factory_key) { :group_repository_storage_move }
+    let(:error_key) { :group }
   end
 end

spec/support/shared_examples/models/concerns/repository_storage_movable_shared_examples.rb

 # frozen_string_literal: true
 
-RSpec.shared_examples 'handles repository moves' do
+RSpec.shared_examples 'handles repository moves' do |check_worker: true|
   describe 'associations' do
     it { is_expected.to belong_to(:container) }
   end
@@ -33,7 +33,7 @@ RSpec.shared_examples 'handles repository moves' do
       subject { build(repository_storage_factory_key, container: container) }
 
       it "does not allow the container to be read-only on create" do
-        container.update!(repository_read_only: true)
+        container.set_repository_read_only!
 
         expect(subject).not_to be_valid
         expect(subject.errors[error_key].first).to match(/is read only/)
@@ -61,23 +61,25 @@ RSpec.shared_examples 'handles repository moves' do
     context 'when in the default state' do
       subject(:storage_move) { create(repository_storage_factory_key, container: container, destination_storage_name: 'test_second_storage') }
 
-      context 'and transits to scheduled' do
-        it 'triggers the corresponding repository storage worker' do
-          expect(repository_storage_worker).to receive(:perform_async).with(container.id, 'test_second_storage', storage_move.id)
+      if check_worker
+        context 'and transits to scheduled' do
+          it 'triggers the corresponding repository storage worker' do
+            expect(repository_storage_worker).to receive(:perform_async).with(container.id, 'test_second_storage', storage_move.id)
 
-          storage_move.schedule!
+            storage_move.schedule!
 
-          expect(container).to be_repository_read_only
-        end
+            expect(container).to be_repository_read_only
+          end
 
-        context 'when the transition fails' do
-          it 'does not trigger the corresponding repository storage worker and adds an error' do
-            allow(storage_move.container).to receive(:set_repository_read_only!).and_raise(StandardError, 'foobar')
-            expect(repository_storage_worker).not_to receive(:perform_async)
+          context 'when the transition fails' do
+            it 'does not trigger the corresponding repository storage worker and adds an error' do
+              allow(storage_move.container).to receive(:set_repository_read_only!).and_raise(StandardError, 'foobar')
+              expect(repository_storage_worker).not_to receive(:perform_async)
 
-            storage_move.schedule!
+              storage_move.schedule!
 
-            expect(storage_move.errors[error_key]).to include('foobar')
+              expect(storage_move.errors[error_key]).to include('foobar')
+            end
           end
         end
       end