verifying EnvironmentID (#4484)

Signed-off-by: Dhanush0369 <[email protected]> Co-authored-by: Sarthak Jain <[email protected]> Co-authored-by: Saranya Jena <[email protected]>
litmuschaos · Apr 5, 2024 · c1467b7 · c1467b7
1 parent cc74f22
commit c1467b7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/chaoscenter/graphql/server/pkg/chaos_infrastructure/service.go b/chaoscenter/graphql/server/pkg/chaos_infrastructure/service.go
@@ -86,6 +86,11 @@ func (in *infraService) RegisterInfra(c context.Context, projectID string, input
 		return nil, errors.New("infra already exists in this project with the same name")
 	}
 
+	// Check if EnvironmentID is valid
+	if _, err := in.envOperator.GetEnvironmentDetails(c, input.EnvironmentID, projectID); err != nil {
+		return nil, errors.New("Invalid EnvironmentID")
+	}
+
 	if (*input.InfraNsExists && input.InfraNamespace == nil) || (*input.InfraNsExists && *input.InfraNamespace == "") {
 		return nil, errors.New("InfraNamespace parameter is required if InfraNsExists is true")
 	}