Change delete ensembler api response from int to struct (#364)

caraml-dev · Oct 5, 2023 · 22d15b1 · 22d15b1
1 parent 7fa5d36
commit 22d15b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/turing/api/ensemblers_api.go b/api/turing/api/ensemblers_api.go
@@ -213,7 +213,7 @@ func (c EnsemblersController) DeleteEnsembler(
 		return Error(httpStatus, "failed to delete the ensembler", err.Error())
 	}
 
-	return Ok(ensembler.GetID())
+	return Ok(map[string]int{"id": int(ensembler.GetID())})
 }
 
 func (c EnsemblersController) checkActiveRouterVersion(options EnsemblersPathOptions) (int, error) {

diff --git a/api/turing/api/ensemblers_api_test.go b/api/turing/api/ensemblers_api_test.go
@@ -917,7 +917,7 @@ func TestEnsemblerController_DeleteEnsembler(t *testing.T) {
 				mlflowSvc.On("DeleteExperiment", mock.Anything, "1", true).Return(nil)
 				return mlflowSvc
 			},
-			expected: Ok(models.ID(2)),
+			expected: Ok(map[string]int{"id": 2}),
 		},
 		"success": {
 			vars: RequestVars{
@@ -962,7 +962,7 @@ func TestEnsemblerController_DeleteEnsembler(t *testing.T) {
 				mlflowSvc.On("DeleteExperiment", mock.Anything, "1", true).Return(nil)
 				return mlflowSvc
 			},
-			expected: Ok(models.ID(2)),
+			expected: Ok(map[string]int{"id": 2}),
 		},
 	}
 	for name, tt := range tests {