From bb2a449faf62ff1ce8d6c3bdc7082a01c85c780b Mon Sep 17 00:00:00 2001
From: John Mulligan <jmulligan@redhat.com>
Date: Tue, 19 Mar 2024 07:40:57 -0400
Subject: [PATCH] rbd: add a test to verify that encrypted volumes can be
 resized

Add a test to verify that an encrypted volume can be resized to
the desired unencrypted size similar to how the rbd command line
tool is documented as able to do.

Fixes: #972

Original-Version-By: Will Gorman <will.gorman@gmail.com>
Signed-off-by: John Mulligan <jmulligan@redhat.com>
---
 rbd/encryption_test.go | 66 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/rbd/encryption_test.go b/rbd/encryption_test.go
index c7a6faf45..3bc066158 100644
--- a/rbd/encryption_test.go
+++ b/rbd/encryption_test.go
@@ -128,3 +128,69 @@ func TestEncryptionLoad(t *testing.T) {
 	conn.DeletePool(poolname)
 	conn.Shutdown()
 }
+
+func TestEncryptedResize(t *testing.T) {
+	conn := radosConnect(t)
+	defer conn.Shutdown()
+
+	poolname := GetUUID()
+	err := conn.MakePool(poolname)
+	assert.NoError(t, err)
+	defer conn.DeletePool(poolname)
+
+	imageName := "resizeme"
+	imageSize := uint64(50) * 1024 * 1024
+	encOpts := EncryptionOptionsLUKS2{
+		Alg:        EncryptionAlgorithmAES256,
+		Passphrase: []byte("test-password"),
+	}
+
+	t.Run("create", func(t *testing.T) {
+		ioctx, err := conn.OpenIOContext(poolname)
+		require.NoError(t, err)
+		defer ioctx.Destroy()
+
+		err = CreateImage(ioctx, imageName, imageSize, NewRbdImageOptions())
+		require.NoError(t, err)
+
+		image, err := OpenImage(ioctx, imageName, NoSnapshot)
+		require.NoError(t, err)
+		defer image.Close()
+
+		s, err := image.GetSize()
+		require.NoError(t, err)
+		t.Logf("image size before encryption: %d", s)
+
+		err = image.EncryptionFormat(encOpts)
+		require.NoError(t, err)
+
+		s, err = image.GetSize()
+		require.NoError(t, err)
+		t.Logf("image size after encryption: %d", s)
+	})
+
+	t.Run("resize", func(t *testing.T) {
+		ioctx, err := conn.OpenIOContext(poolname)
+		require.NoError(t, err)
+		defer ioctx.Destroy()
+
+		image, err := OpenImage(ioctx, imageName, NoSnapshot)
+		require.NoError(t, err)
+		defer image.Close()
+
+		err = image.EncryptionLoad(encOpts)
+		assert.NoError(t, err)
+
+		s, err := image.GetSize()
+		require.NoError(t, err)
+		t.Logf("image size before resize: %d", s)
+		assert.NotEqual(t, s, imageSize)
+
+		err = image.Resize(imageSize)
+		assert.NoError(t, err)
+
+		s, err = image.GetSize()
+		require.NoError(t, err)
+		t.Logf("image size after resize of encrypted image: %d", s)
+	})
+}