update ut mock anything (#19506)

Signed-off-by: yminer <yminer@vmware.com>
goharbor · Oct 31, 2023 · 6d81772 · 6d81772
1 parent 7c83e3a
commit 6d81772
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/server/middleware/quota/util_test.go b/src/server/middleware/quota/util_test.go
@@ -29,8 +29,8 @@ import (
 
 func Test_projectReferenceObject(t *testing.T) {
 	ctl := &project.Controller{}
-	ctl.On("GetByName", mock.Anything, "library").Return(&proModels.Project{ProjectID: 1}, nil)
-	ctl.On("GetByName", mock.Anything, "demo").Return(nil, fmt.Errorf("not found"))
+	ctl.On("GetByName", mock.AnythingOfType("context.todoCtx"), "library").Return(&proModels.Project{ProjectID: 1}, nil)
+	ctl.On("GetByName", mock.AnythingOfType("context.todoCtx"), "demo").Return(nil, fmt.Errorf("not found"))
 
 	originalProjectController := projectController
 	defer func() {