feat: add test field to UserStageModel

Author: wangeguo

migrations/0010_add_test_to_user_stages.sql

@@ -0,0 +1,7 @@
+-- Migration to add test column to user_stages table
+-- Stores test result status (passed/failed) for user stage progress
+
+ALTER TABLE user_stages ADD COLUMN test TEXT NOT NULL DEFAULT 'failed';
+
+-- Update existing records to maintain consistency
+UPDATE user_stages SET test = 'failed' WHERE test IS NULL;

openapi.json

@@ -1161,6 +1161,7 @@
           "course_slug",
           "stage_slug",
           "status",
+          "test",
           "started_at"
         ],
         "properties": {
@@ -1188,6 +1189,10 @@
           "status": {
             "type": "string",
             "description": "Current progress status (in_progress, completed)"
+          },
+          "test": {
+            "type": "string",
+            "description": "Test result status (passed, failed)"
           }
         }
       },

src/model/stage.rs

@@ -125,6 +125,9 @@ pub struct UserStageModel {
     /// Current progress status (in_progress, completed)
     pub status: String,
 
+    /// Test result status (passed, failed)
+    pub test: String,
+
     /// Timestamp when the stage was started
     pub started_at: DateTime<Utc>,
 
@@ -142,11 +145,18 @@ impl UserStageModel {
             stage_id,
             stage_slug: String::new(),
             status: "in_progress".to_string(),
+            test: "failed".to_string(),
             started_at: Utc::now(),
             completed_at: None,
         }
     }
 
+    /// Marks the stage as test passed
+    pub fn passed(mut self) -> Self {
+        self.test = "passed".to_string();
+        self
+    }
+
     /// Marks the stage as completed with current timestamp
     pub fn complete(mut self) -> Self {
         self.status = "completed".to_string();

src/repository/stage.rs

@@ -343,8 +343,8 @@ impl StageRepository {
             r#"
             WITH inserted AS (
                 INSERT INTO user_stages (
-                    id, user_course_id, stage_id, status, started_at
-                ) VALUES ($1, $2, $3, $4, $5)
+                    id, user_course_id, stage_id, status, test, started_at
+                ) VALUES ($1, $2, $3, $4, $5, $6)
                 RETURNING *
             )
             SELECT
@@ -361,6 +361,7 @@ impl StageRepository {
         .bind(user_stage.user_course_id)
         .bind(user_stage.stage_id)
         .bind(&user_stage.status)
+        .bind(&user_stage.test)
         .bind(user_stage.started_at)
         .fetch_one(&mut **tx)
         .await?;
@@ -381,7 +382,8 @@ impl StageRepository {
                 UPDATE user_stages
                 SET
                     status = $2,
-                    completed_at = $3
+                    test = $3,
+                    completed_at = $4
                 WHERE id = $1
                 RETURNING *
             )
@@ -397,6 +399,7 @@ impl StageRepository {
         )
         .bind(user_stage.id)
         .bind(&user_stage.status)
+        .bind(&user_stage.test)
         .bind(user_stage.completed_at)
         .fetch_one(&mut **tx)
         .await?;

src/response/stage.rs

@@ -116,6 +116,9 @@ pub struct UserStageResponse {
     /// Current progress status (in_progress, completed)
     pub status: String,
 
+    /// Test result status (passed, failed)
+    pub test: String,
+
     /// Timestamp when the stage was started
     pub started_at: DateTime<Utc>,
 
@@ -129,6 +132,7 @@ impl From<UserStageModel> for UserStageResponse {
             course_slug: model.course_slug,
             stage_slug: model.stage_slug,
             status: model.status,
+            test: model.test,
             started_at: model.started_at,
             completed_at: model.completed_at,
         }

src/service/stage.rs

@@ -107,7 +107,7 @@ impl StageService {
         let mut tx = ctx.database.pool().begin().await?;
 
         // Mark the stage as completed.
-        user_stage = user_stage.complete();
+        user_stage = user_stage.passed().complete();
         let completed_stage = StageRepository::update_user_stage(&mut tx, &user_stage).await?;
 
         // Update user course and create next stage if needed.
@@ -166,6 +166,6 @@ impl StageService {
             StageRepository::get_user_stage(&ctx.database, user_id, course_slug, stage_slug)
                 .await?;
 
-        Ok(UserStageStatusResponse { status: user_stage.status, test: "failed".into() })
+        Ok(UserStageStatusResponse { status: user_stage.status, test: user_stage.test })
     }
 }