Fix race condition in pushing / pulling volume to local registry

Signed-off-by: Guillaume Tardif <guillaume.tardif@gmail.com>

vm/internal/handler/pull_test.go

@@ -144,15 +144,13 @@ func TestPullVolume(t *testing.T) {
 			return err
 		}
 
+		t.Log("image pushed successfully")
 		return nil
 	})
 	if err != nil {
 		t.Fatal(err)
 	}
 
-	// The sleep is to ensure the image is present in the registry after the `ImagePush` operation.
-	time.Sleep(3 * time.Second)
-
 	_, err = cli.ImageRemove(context.Background(), imageID, types.ImageRemoveOptions{
 		Force: true,
 	})
@@ -169,8 +167,11 @@ func TestPullVolume(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	// Pull volume from registry
-	err = h.PullVolume(c)
+	// Ths is to ensure the image is present in the registry after the `ImagePush` operation, might take a bit of time.
+	err = retry(10, 1*time.Second, func() error {
+		// Pull volume from registry
+		return h.PullVolume(c)
+	})
 
 	require.NoError(t, err)
 	require.Equal(t, http.StatusCreated, rec.Code)