[EE] Reduce diff with CE in Groups::UpdateService

Signed-off-by: Rémy Coutable <remy@rymai.me>

app/services/groups/update_service.rb

@@ -3,7 +3,6 @@
 module Groups
   class UpdateService < Groups::BaseService
     include UpdateVisibilityLevel
-    prepend ::EE::Groups::UpdateService # rubocop: disable Cop/InjectEnterpriseEditionModule
 
     def execute
       reject_parent_id!
@@ -12,9 +11,7 @@ module Groups
 
       return false unless valid_share_with_group_lock_change?
 
-      # Repository size limit comes as MB from the view
-      limit = @params.delete(:repository_size_limit)
-      group.repository_size_limit = Gitlab::Utils.try_megabytes_to_bytes(limit) if limit
+      before_assignment_hook(group, params)
 
       group.assign_attributes(params)
 
@@ -33,6 +30,10 @@ module Groups
 
     private
 
+    def before_assignment_hook(group, params)
+      # overriden in EE
+    end
+
     def after_update
       if group.previous_changes.include?(:visibility_level) && group.private?
         # don't enqueue immediately to prevent todos removal in case of a mistake
@@ -59,3 +60,5 @@ module Groups
     end
   end
 end
+
+Groups::UpdateService.prepend(EE::Groups::UpdateService)

ee/app/services/ee/groups/update_service.rb

@@ -17,6 +17,13 @@ module EE
 
       private
 
+      override :before_assignment_hook
+      def before_assignment_hook(group, params)
+        # Repository size limit comes as MB from the view
+        limit = params.delete(:repository_size_limit)
+        group.repository_size_limit = ::Gitlab::Utils.try_megabytes_to_bytes(limit) if limit
+      end
+
       def changes_file_template_project_id?
         return false unless params.key?(:file_template_project_id)