feat: map flavor not found to ResourceExhausted grpc code

Author: Lukas Jarosch

pkg/driver/executor/errors.go

@@ -8,12 +8,17 @@ import (
 	"fmt"
 )
 
-// NoValidHost is a part of the error message returned when there is no valid host in the zone to deploy a VM.
-// Matches:
-//
-//	"No valid host was found."
-//	"No valid host was found. There are not enough hosts available."
-const NoValidHost = "No valid host was found"
+const (
+	// NoValidHost is a part of the error message returned when there is no valid host in the zone to deploy a VM.
+	// Matches:
+	//
+	//	"No valid host was found."
+	//	"No valid host was found. There are not enough hosts available."
+	NoValidHost = "No valid host was found"
+
+	// FlavorNotFound is part of the error message returned when a flavor cannot be resolved.
+	FlavorNotFound = "error resolving flavor"
+)
 
 var (
 	// ErrNotFound is returned when the requested resource could not be found.

pkg/driver/utils.go

@@ -68,7 +68,8 @@ func mapErrorToCode(err error) codes.Code {
 
 func mapErrorMessageToCode(err error) codes.Code {
 	errorMessage := err.Error()
-	if strings.Contains(errorMessage, executor.NoValidHost) {
+	if strings.Contains(errorMessage, executor.NoValidHost) ||
+		strings.Contains(errorMessage, executor.FlavorNotFound) {
 		return codes.ResourceExhausted
 	}
 	return codes.Internal