Add latest changes from gitlab-org/gitlab@master

2 files changed, 15 insertions, 1 deletions

diff --git a/app/services/award_emojis/add_service.rb b/app/services/award_emojis/add_service.rb
index eac15dabbf0..db465dcaf7e 100644
--- a/app/services/award_emojis/add_service.rb
+++ b/app/services/award_emojis/add_service.rb
@@ -16,7 +16,7 @@ module AwardEmojis
       award = awardable.award_emoji.create(name: name, user: current_user)
 
       if award.persisted?
-        TodoService.new.new_award_emoji(todoable, current_user) if todoable
+        after_create(award)
         success(award: award)
       else
         error(award.errors.full_messages, award: award)
@@ -25,6 +25,10 @@ module AwardEmojis
 
     private
 
+    def after_create(award)
+      TodoService.new.new_award_emoji(todoable, current_user) if todoable
+    end
+
     def todoable
       strong_memoize(:todoable) do
         case awardable
@@ -40,3 +44,5 @@ module AwardEmojis
     end
   end
 end
+
+AwardEmojis::AddService.prepend_if_ee('EE::AwardEmojis::AddService')
diff --git a/app/services/award_emojis/destroy_service.rb b/app/services/award_emojis/destroy_service.rb
index 3789a8403bc..a61a7911a9d 100644
--- a/app/services/award_emojis/destroy_service.rb
+++ b/app/services/award_emojis/destroy_service.rb
@@ -14,8 +14,16 @@ module AwardEmojis
       end
 
       award = awards.destroy_all.first # rubocop: disable DestroyAll
+      after_destroy(award)
 
       success(award: award)
     end
+
+    private
+
+    def after_destroy(award)
+    end
   end
 end
+
+AwardEmojis::DestroyService.prepend_if_ee('EE::AwardEmojis::DestroyService')