updated(task_manager_api): fixing the task_contorller test

solo21-12 · Aug 15, 2024 · dedb712 · dedb712
1 parent db90856
commit dedb712
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/bootstrap/env.go b/bootstrap/env.go
@@ -19,7 +19,6 @@ type Env struct {
 }
 
 func NewEnv() *Env {
-	// Initialize viper to read from environment variables
 	viper.AutomaticEnv()
 
 	env := Env{}

diff --git a/tests/task_tests/controller_tests/task_controller_test.go b/tests/task_tests/controller_tests/task_controller_test.go
@@ -67,7 +67,7 @@ func (suite *TaskControllerSuite) TestCreateTask_Success() {
 
 	suite.usecase.
 		EXPECT().
-		CreateTask(taskCreate, gomock.Any()).
+		CreateTask(gomock.Eq(taskCreate), gomock.Any()).
 		Return(taskDTO, nil).
 		Times(1)
 
@@ -81,6 +81,12 @@ func (suite *TaskControllerSuite) TestCreateTask_Success() {
 
 	// Check the response status
 	suite.Equal(http.StatusCreated, response.StatusCode)
+
+	// Optional: Check the response body if needed
+	var responseDTO domain.TaskDTO
+	err = json.NewDecoder(response.Body).Decode(&responseDTO)
+	suite.NoError(err)
+	suite.Equal(taskDTO, responseDTO)
 }
 
 func (suite *TaskControllerSuite) TestCreateTask_Failure() {