add testcase to question added

Author: krishkumar84

cmd/portal-api/main.go

@@ -77,6 +77,7 @@ router.Handle("GET /api/admin/test",
 	router.HandleFunc("GET /api/contest/{id}",contest.GetContestById(storage))
 	router.HandleFunc("GET /api/question/{id}",question.GetQuestionById(storage))
 	router.HandleFunc("POST /api/contest/{id}/question", contest.AddQuestionToContest(storage))
+	router.HandleFunc("POST /api/question/{id}/testcase", question.AddTestCaseToQuestion(storage))
 
 	//start server
 

pkg/http/handler/question/question.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net/http"
 	"strings"
+
 	"github.com/krishkumar84/bdcoe-golang-portal/pkg/storage"
 	"github.com/krishkumar84/bdcoe-golang-portal/pkg/types"
 	"github.com/krishkumar84/bdcoe-golang-portal/pkg/utils/response"
@@ -45,4 +46,37 @@ func GetQuestionById(storage storage.Storage) http.HandlerFunc {
 		}
 		response.WriteJson(w, http.StatusOK, question)
 	}
+}
+
+func AddTestCaseToQuestion(storage storage.Storage) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		path := r.URL.Path
+		parts := strings.Split(path, "/")
+		if len(parts) < 5 {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(fmt.Errorf("invalid URL format")))
+			return
+		}
+		questionId := parts[3]
+		
+		fmt.Printf("Extracted question ID: %s\n", questionId)
+		
+		if questionId == "" {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(fmt.Errorf("question id is required")))
+			return
+		}
+
+		var testCase types.TestCase
+		if err := json.NewDecoder(r.Body).Decode(&testCase); err != nil {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(err))
+			return
+		}
+
+		testCaseId, err := storage.AddTestCaseToQuestion(questionId, testCase)
+		if err != nil {
+			response.WriteJson(w, http.StatusInternalServerError, response.GeneralError(err))
+			return
+		}
+
+		response.WriteJson(w, http.StatusCreated, map[string]string{"test_case_id": testCaseId})
+	}
 }

pkg/storage/mongodb/mongodb.go

@@ -348,4 +348,39 @@ func (m *MongoDB) AddQuestionToContest(contestId string, question types.Question
     fmt.Printf("Updated contest. Modified count: %d\n", result.ModifiedCount)
 
     return questionId, nil
+}
+
+func (m *MongoDB) AddTestCaseToQuestion(questionId string, testCase types.TestCase) (string, error) {
+    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+    defer cancel()
+
+    fmt.Printf("Received question ID: %s\n", questionId)
+
+    // Create test case
+    testCaseResult, err := m.db.Collection("test_cases").InsertOne(ctx, testCase)
+    if err != nil {
+        return "", fmt.Errorf("failed to create test case: %v", err)
+    }
+    testCaseId := testCaseResult.InsertedID.(primitive.ObjectID).Hex()
+
+    fmt.Printf("Created test case with ID: %s\n", testCaseId)
+
+    // Convert question ID to ObjectID
+    questionObjID, err := primitive.ObjectIDFromHex(questionId)
+    if err != nil {
+        return "", fmt.Errorf("invalid question id format: %v", err)
+    }
+
+    // Update question with test case ID
+    filter := bson.M{"_id": questionObjID}
+    update := bson.M{"$push": 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)
+    }
+
+    fmt.Printf("Updated question. Modified count: %d\n", result.ModifiedCount)
+
+    return testCaseId, nil
 }

pkg/storage/storage.go

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