edit testcasebyId completed

Author: krishkumar84

cmd/portal-api/main.go

@@ -76,6 +76,7 @@ router.Handle("GET /api/admin/test",
 	router.HandleFunc("POST /api/question",question.CreateQuestion(storage))
 	router.HandleFunc("PUT /api/question/{id}",question.EditQuestionById(storage))
 	router.HandleFunc("POST /api/testcase",testcase.CreateTestCase(storage))
+	router.HandleFunc("PUT /api/testcase/{id}",testcase.EditTestCaseById(storage))
 	router.HandleFunc("GET /api/contest",contest.GetAllContests(storage))
 	router.HandleFunc("GET /api/contest/{id}",contest.GetContestById(storage))
 	router.HandleFunc("GET /api/question/{id}",question.GetQuestionById(storage))

pkg/http/handler/testcase/testcase.go

@@ -3,7 +3,8 @@ package testcase
 import (
 	"encoding/json"
 	"net/http"
-
+	"strings"
+	"fmt"
 	"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"
@@ -25,4 +26,29 @@ func CreateTestCase(storage storage.Storage) http.HandlerFunc {
 
 		response.WriteJson(w, http.StatusCreated, map[string]string{"test_case_id": testCaseId})
 	}
+}
+
+func EditTestCaseById(storage storage.Storage) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		path := r.URL.Path
+		id := path[strings.LastIndex(path, "/")+1:]
+		
+		if id == "" {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(fmt.Errorf("test case id is required")))
+			return
+		}
+
+		var testCaseReq types.TestCase
+		if err := json.NewDecoder(r.Body).Decode(&testCaseReq); err != nil {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(err))
+			return
+		}
+
+		if err := storage.EditTestCaseById(id, testCaseReq); err != nil {
+			response.WriteJson(w, http.StatusInternalServerError, response.GeneralError(err))
+			return
+		}
+
+		response.WriteJson(w, http.StatusOK, map[string]string{"status": "success", "message": "test case updated successfully"})
+	}
 }

pkg/storage/mongodb/mongodb.go

@@ -548,6 +548,45 @@ func (m *MongoDB) AddTestCaseToQuestion(questionId string, testCase types.TestCa
     return testCaseId, nil
 }
 
+func (m *MongoDB) EditTestCaseById(id string, updateData types.TestCase) error {
+    testCaseObjID, err := primitive.ObjectIDFromHex(id)
+    if err != nil {
+        return fmt.Errorf("invalid test case id format")
+    }
+
+    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+    defer cancel()
+
+    var testCase types.TestCase
+    err = m.db.Collection("test_cases").FindOne(ctx, bson.M{"_id": testCaseObjID}).Decode(&testCase)
+    if err != nil {
+        if err == mongo.ErrNoDocuments {
+            return fmt.Errorf("no test case found with the given id")
+        }
+        return fmt.Errorf("error checking test case existence: %v", err)
+    }
+
+    // Prepare the update
+    update := bson.M{}
+    if updateData.Input != nil {
+        update["input"] = updateData.Input
+    }
+    if updateData.ExpectedOutput != nil {
+        update["expected_output"] = updateData.ExpectedOutput
+    }
+    if updateData.Visibility != "" {
+        update["visibility"] = updateData.Visibility
+    }
+    if len(update) > 0 {
+        _, err = m.db.Collection("test_cases").UpdateOne(ctx, bson.M{"_id": testCaseObjID}, bson.M{"$set": update})
+        if err != nil {
+            return fmt.Errorf("failed to update test case: %v", err)
+        }
+    }
+
+    return nil
+}
+
 func (m *MongoDB) DeleteTestCaseFromQuestionById(questionId string, testCaseId string) error {
     questionObjID, err := primitive.ObjectIDFromHex(questionId)
     if err != nil {

pkg/storage/storage.go

@@ -25,4 +25,5 @@ type Storage interface {
 	AddQuestionToContest(contestId string, question types.Question) (string, error)
 	AddTestCaseToQuestion(questionId string, testCase types.TestCase) (string, error)
 	DeleteTestCaseFromQuestionById(questionId string, testCaseId string) error
+	EditTestCaseById(testCaseId string, testCase types.TestCase) error
 }