rbd: added rbd info to validateRBDImageCount func

Signed-off-by: Oded Viner <oviner@redhat.com>
This commit is contained in:
Oded Viner 2024-10-31 16:14:02 +02:00
parent a7959d4721
commit e19572ae66
2 changed files with 59 additions and 9 deletions

View File

@ -189,12 +189,27 @@ func validateRBDImageCount(f *framework.Framework, count int, pool string) {
framework.Failf("failed to list rbd images: %v", err)
}
if len(imageList) != count {
var imageDetails []string // To collect details for all images
for _, image := range imageList {
_, imgInfoOutput, err := getImageInfo(f, image, pool)
if err != nil {
framework.Logf("Error getting image info: %v", err)
}
_, imgStatusOutput, err := getImageStatus(f, image, pool)
if err != nil {
framework.Logf("Error getting image status: %v", err)
}
// Collecting image details for printing
imageDetails = append(imageDetails, fmt.Sprintf(
"Pool: %s, Image: %s, Info: %s, Status: %s", pool, image, imgInfoOutput, imgStatusOutput))
}
framework.Failf(
"backend images not matching kubernetes resource count,image count %d kubernetes resource count %d"+
"\nbackend image Info:\n %v",
"\nbackend image Info:\n %v\n images information and status %v",
len(imageList),
count,
imageList)
imageList,
strings.Join(imageDetails, "\n"))
}
}
@ -475,7 +490,7 @@ var _ = Describe("RBD", func() {
framework.Failf("failed to create PVC: %v", err)
}
// validate created backend rbd images
validateRBDImageCount(f, 1, defaultRBDPool)
validateRBDImageCount(f, 11, defaultRBDPool)
validateOmapCount(f, 1, rbdType, defaultRBDPool, volumesType)
imageList, err := listRBDImages(f, defaultRBDPool)

View File

@ -1069,9 +1069,19 @@ type imageInfo struct {
ObjectSize int `json:"object_size"`
}
// imageStatus strongly typed JSON spec for image status.
type imageStatus struct {
Name string `json:"name"`
Pool string `json:"pool"`
Watchers int `json:"watchers"`
InUse bool `json:"in_use"` // Indicates if the image is currently in use.
Size int64 `json:"size"` // Total size of the image in bytes.
Format string `json:"format"` // Format of the image (e.g., raw, qcow2).
}
// getImageInfo queries rbd about the given image and returns its metadata, and returns
// error if provided image is not found.
func getImageInfo(f *framework.Framework, imageName, poolName string) (imageInfo, error) {
func getImageInfo(f *framework.Framework, imageName, poolName string) (imageInfo, string, error) {
// rbd --format=json info [image-spec | snap-spec]
var imgInfo imageInfo
@ -1080,18 +1090,43 @@ func getImageInfo(f *framework.Framework, imageName, poolName string) (imageInfo
fmt.Sprintf("rbd info %s %s --format json", rbdOptions(poolName), imageName),
rookNamespace)
if err != nil {
return imgInfo, fmt.Errorf("failed to get rbd info: %w", err)
return imgInfo, stdOut, fmt.Errorf("failed to get rbd info: %w", err)
}
if stdErr != "" {
return imgInfo, fmt.Errorf("failed to get rbd info: %v", stdErr)
return imgInfo, stdOut, fmt.Errorf("failed to get rbd info: %v", stdErr)
}
err = json.Unmarshal([]byte(stdOut), &imgInfo)
if err != nil {
return imgInfo, fmt.Errorf("unmarshal failed: %w. raw buffer response: %s",
return imgInfo, stdOut, fmt.Errorf("unmarshal failed: %w. raw buffer response: %s",
err, stdOut)
}
return imgInfo, nil
return imgInfo, stdOut, nil
}
// getImageStatus queries rbd about the given image and returns its metadata, and returns
// error if provided image is not found.
func getImageStatus(f *framework.Framework, imageName, poolName string) (imageStatus, string, error) {
// rbd --format=json status [image-spec | snap-spec]
var imgStatus imageStatus
stdOut, stdErr, err := execCommandInToolBoxPod(
f,
fmt.Sprintf("rbd status %s %s --format json", rbdOptions(poolName), imageName),
rookNamespace)
if err != nil {
return imgStatus, stdOut, fmt.Errorf("error retrieving rbd status: %w", err)
}
if stdErr != "" {
return imgStatus, stdOut, fmt.Errorf("failed to get rbd status: %v", stdErr)
}
err = json.Unmarshal([]byte(stdOut), &imgStatus)
if err != nil {
return imgStatus, stdOut, fmt.Errorf("unmarshal failed: %w. raw buffer response: %s",
err, stdOut)
}
return imgStatus, stdOut, nil
}
// validateStripe validate the stripe count, stripe unit and object size of the
@ -1107,7 +1142,7 @@ func validateStripe(f *framework.Framework,
return err
}
imgInfo, err := getImageInfo(f, imageData.imageName, defaultRBDPool)
imgInfo, _, err := getImageInfo(f, imageData.imageName, defaultRBDPool)
if err != nil {
return err
}