Update supermentor role when changing mentored tracks (#6609)

* Update mentored tracks in command

* Update supermentor role when updating mentored tracks

app/commands/mentor/update_mentored_tracks.rb

@@ -0,0 +1,10 @@
+class Mentor::UpdateMentoredTracks
+  include Mandate
+
+  initialize_with :mentor, :tracks
+
+  def call
+    mentor.update!(mentored_tracks: tracks)
+    User::UpdateSupermentorRole.defer(mentor)
+  end
+end

app/controllers/api/mentoring/tracks_controller.rb

@@ -17,8 +17,7 @@ def mentored
 
   def update
     tracks = Track.where(slug: params[:track_slugs])
-
-    current_user.update!(mentored_tracks: tracks)
+    Mentor::UpdateMentoredTracks.(current_user, tracks)
 
     render json: {
       tracks: SerializeTracksForMentoring.(current_user.mentored_tracks, current_user)

test/commands/mentor/update_mentored_tracks_test.rb

@@ -0,0 +1,52 @@
+require "test_helper"
+
+class Mentor::UpdateMentoredTracksTest < ActiveSupport::TestCase
+  test "adds new mentored tracks" do
+    mentor = create :user
+    track = create :track
+
+    assert_empty mentor.mentored_tracks
+
+    Mentor::UpdateMentoredTracks.(mentor, [track])
+
+    assert_equal [track], mentor.mentored_tracks
+  end
+
+  test "keeps existing tracks" do
+    mentor = create :user
+    track = create :track
+    updated_at = Time.current - 2.weeks
+    track_mentorship = create(:user_track_mentorship, track:, user: mentor, updated_at:)
+
+    assert_equal [track], mentor.mentored_tracks
+
+    Mentor::UpdateMentoredTracks.(mentor, [track])
+
+    assert_equal [track], mentor.mentored_tracks
+    assert_equal updated_at, track_mentorship.updated_at
+  end
+
+  test "removes tracks no longer being mentored" do
+    mentor = create :user
+    track = create :track
+    updated_at = Time.current - 2.weeks
+    track_mentorship = create(:user_track_mentorship, track:, user: mentor, updated_at:)
+
+    assert_equal [track], mentor.mentored_tracks
+
+    Mentor::UpdateMentoredTracks.(mentor, [])
+
+    assert_empty mentor.mentored_tracks
+    assert_raises ActiveRecord::RecordNotFound do
+      track_mentorship.reload
+    end
+  end
+
+  test "updates supermentor role" do
+    mentor = create :user
+
+    User::UpdateSupermentorRole.expects(:defer).with(mentor)
+
+    Mentor::UpdateMentoredTracks.(mentor, [])
+  end
+end