Skip to content

Commit

Permalink
rbd: added rbd info to validateRBDImageCount func
Browse files Browse the repository at this point in the history
Signed-off-by: Oded Viner <[email protected]>
  • Loading branch information
OdedViner committed Nov 20, 2024
1 parent a7959d4 commit a1e1a5c
Show file tree
Hide file tree
Showing 2 changed files with 45 additions and 14 deletions.
19 changes: 17 additions & 2 deletions e2e/rbd.go
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
imgInfoStr, 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, imgInfoStr, 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"))
}
}

Expand Down
40 changes: 28 additions & 12 deletions e2e/rbd_helper.go
Original file line number Diff line number Diff line change
Expand Up @@ -1071,27 +1071,36 @@ type imageInfo struct {

// 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) (string, error) {
// rbd --format=json info [image-spec | snap-spec]
var imgInfo imageInfo

stdOut, stdErr, err := execCommandInToolBoxPod(
f,
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 stdOut, fmt.Errorf("failed to get rbd info: %w", err)
}
if stdErr != "" {
return imgInfo, fmt.Errorf("failed to get rbd info: %v", stdErr)
return stdOut, fmt.Errorf("failed to get rbd info: %v", stdErr)
}
err = json.Unmarshal([]byte(stdOut), &imgInfo)
return 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) (string, error) {
// rbd --format=json status [image-spec | snap-spec]
stdOut, stdErr, err := execCommandInToolBoxPod(
f,
fmt.Sprintf("rbd status %s %s --format json", rbdOptions(poolName), imageName),
rookNamespace)
if err != nil {
return imgInfo, fmt.Errorf("unmarshal failed: %w. raw buffer response: %s",
err, stdOut)
return stdOut, fmt.Errorf("error retrieving rbd status: %w", err)
}

return imgInfo, nil
if stdErr != "" {
return stdOut, fmt.Errorf("failed to get rbd info: %v", stdErr)
}
return stdOut, nil
}

// validateStripe validate the stripe count, stripe unit and object size of the
Expand All @@ -1102,15 +1111,22 @@ func validateStripe(f *framework.Framework,
stripeCount,
objectSize int,
) error {
var imgInfo imageInfo

imageData, err := getImageInfoFromPVC(pvc.Namespace, pvc.Name, f)
if err != nil {
return err
}

imgInfo, err := getImageInfo(f, imageData.imageName, defaultRBDPool)
imgInfoStr, err := getImageInfo(f, imageData.imageName, defaultRBDPool)
if err != nil {
return err
}

err = json.Unmarshal([]byte(imgInfoStr), &imgInfo)
if err != nil {
return fmt.Errorf("unmarshal failed: %w. raw buffer response: %s", err, imgInfoStr)
}

if imgInfo.ObjectSize != objectSize {
return fmt.Errorf("objectSize %d does not match expected %d", imgInfo.ObjectSize, objectSize)
Expand Down

0 comments on commit a1e1a5c

Please sign in to comment.