DeleteTestCaseFromQuestionById completed

Author: krishkumar84

cmd/portal-api/main.go

@@ -80,6 +80,7 @@ router.Handle("GET /api/admin/test",
 	router.HandleFunc("POST /api/contest/{id}/question", contest.AddQuestionToContest(storage))
 	router.HandleFunc("DELETE /api/contest/{contestId}/question/{questionId}", contest.DeleteQuestionFromContestById(storage))
 	router.HandleFunc("POST /api/question/{id}/testcase", question.AddTestCaseToQuestion(storage))
+	router.HandleFunc("DELETE /api/question/{questionId}/testcase/{testCaseId}", question.DeleteTestCaseFromQuestionById(storage))
 
 	//start server
 

pkg/http/handler/question/question.go

@@ -79,4 +79,33 @@ func AddTestCaseToQuestion(storage storage.Storage) http.HandlerFunc {
 
 		response.WriteJson(w, http.StatusCreated, map[string]string{"test_case_id": testCaseId})
 	}
+}
+
+func DeleteTestCaseFromQuestionById(storage storage.Storage) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		path := r.URL.Path
+		parts := strings.Split(path, "/")
+		if len(parts) < 6 {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(fmt.Errorf("invalid URL format")))
+			return
+		}
+		questionId := parts[3]
+		testCaseId := parts[5]
+		
+		fmt.Printf("Extracted question ID: %s\n", questionId)
+		fmt.Printf("Extracted test case ID: %s\n", testCaseId)
+		
+		if questionId == "" || testCaseId == "" {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(fmt.Errorf("question id and test case id are required")))
+			return
+		}
+
+		err := storage.DeleteTestCaseFromQuestionById(questionId, testCaseId)
+		if err != nil {
+			response.WriteJson(w, http.StatusInternalServerError, response.GeneralError(err))
+			return
+		}
+
+		response.WriteJson(w, http.StatusOK, map[string]string{"message": "test case deleted successfully"})
+	}
 }

pkg/storage/mongodb/mongodb.go

@@ -449,4 +449,49 @@ func (m *MongoDB) AddTestCaseToQuestion(questionId string, testCase types.TestCa
     fmt.Printf("Updated question. Modified count: %d\n", result.ModifiedCount)
 
     return testCaseId, nil
+}
+
+func (m *MongoDB) DeleteTestCaseFromQuestionById(questionId string, testCaseId string) error {
+    questionObjID, err := primitive.ObjectIDFromHex(questionId)
+    if err != nil {
+        return fmt.Errorf("invalid question id format")
+    }
+
+    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+    defer cancel()
+
+    // Check if question exists
+    var question types.Question
+    err = m.db.Collection("questions").FindOne(ctx, bson.M{"_id": questionObjID}).Decode(&question)
+    if err != nil {
+        if err == mongo.ErrNoDocuments {
+            return fmt.Errorf("no question found with the given id")
+        }
+        return fmt.Errorf("error checking question existence: %v", err)
+    }
+
+    found := false
+    for _, tcID := range question.TestCaseIDs {
+        if tcID == testCaseId {
+            found = true
+            break
+        }
+    }
+    if !found {
+        return fmt.Errorf("no test case found with the given id in the question")
+    }
+
+    filter := bson.M{"_id": questionObjID}
+    update := bson.M{"$pull": bson.M{"test_case_ids": testCaseId}}
+    
+    result, err := m.db.Collection("questions").UpdateOne(ctx, filter, update)
+    if err != nil {
+        return fmt.Errorf("failed to update question: %v", err)
+    }
+
+    if result.ModifiedCount == 0 {
+        return fmt.Errorf("no test case found with the given id in the question")
+    }
+
+    return nil
 }

pkg/storage/storage.go

@@ -22,4 +22,5 @@ type Storage interface {
 	GetQuestionById(id string) ([]bson.M, error)
 	AddQuestionToContest(contestId string, question types.Question) (string, error)
 	AddTestCaseToQuestion(questionId string, testCase types.TestCase) (string, error)
+	DeleteTestCaseFromQuestionById(questionId string, testCaseId string) error
 }