diff --git a/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/EnrollmentAppServiceImplTest.java b/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/EnrollmentAppServiceImplTest.java
index 85abc0e42..3fe814556 100644
--- a/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/EnrollmentAppServiceImplTest.java
+++ b/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/EnrollmentAppServiceImplTest.java
@@ -1046,6 +1046,17 @@ public class EnrollmentAppServiceImplTest
             return 1;
         }
 
+        @Override
+        public int updateTrainStudent(TrainStudent student)
+        {
+            if (student != null && student.getsId() != null)
+            {
+                studentStore.put(student.getsId(), student);
+                return 1;
+            }
+            return 0;
+        }
+
         @Override
         public TrainStudent selectTrainStudentByUserId(Long userId)
         {
diff --git a/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/TrainStudentServiceImplTest.java b/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/TrainStudentServiceImplTest.java
index c1949c65a..7daa899cd 100644
--- a/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/TrainStudentServiceImplTest.java
+++ b/RuoYi-Vue/ruoyi-train/src/test/java/com/ruoyi/train/service/impl/TrainStudentServiceImplTest.java
@@ -307,6 +307,17 @@ public class TrainStudentServiceImplTest
             return 1;
         }
 
+        @Override
+        public int updateTrainStudent(TrainStudent student)
+        {
+            if (student != null && student.getsId() != null)
+            {
+                store.put(student.getsId(), student);
+                return 1;
+            }
+            return 0;
+        }
+
         @Override
         public TrainStudent selectTrainStudentByUserId(Long userId)
         {