Update mock

descope · Mar 18, 2024 · 1fb2f72 · 1fb2f72
1 parent 0b5df83
commit 1fb2f72
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/descope/tests/mocks/mgmt/managementmock.go b/descope/tests/mocks/mgmt/managementmock.go
@@ -1162,6 +1162,10 @@ type MockProject struct {
 	ImportAssert func(req *descope.ImportProjectRequest)
 	ImportError  error
 
+	ValidateImportAssert   func(req *descope.ImportProjectRequest)
+	ValidateImportError    error
+	ValidateImportResponse *descope.ValidateImportProjectResponse
+
 	UpdateNameAssert func(name string)
 	UpdateNameError  error
 
@@ -1185,7 +1189,10 @@ func (m *MockProject) Import(_ context.Context, req *descope.ImportProjectReques
 }
 
 func (m *MockProject) ValidateImport(_ context.Context, req *descope.ImportProjectRequest) (*descope.ValidateImportProjectResponse, error) {
-	return nil, m.ImportError
+	if m.ValidateImportAssert != nil {
+		m.ValidateImportAssert(req)
+	}
+	return m.ValidateImportResponse, m.ValidateImportError
 }
 
 func (m *MockProject) UpdateName(_ context.Context, name string) error {