mirror of https://github.com/ceph/go-ceph
146 lines
4.1 KiB
Go
146 lines
4.1 KiB
Go
package rbd
|
|
|
|
import (
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetFeatures(t *testing.T) {
|
|
conn := radosConnect(t)
|
|
require.NotNil(t, conn)
|
|
defer conn.Shutdown()
|
|
|
|
poolname := GetUUID()
|
|
err := conn.MakePool(poolname)
|
|
require.NoError(t, err)
|
|
defer conn.DeletePool(poolname)
|
|
|
|
ioctx, err := conn.OpenIOContext(poolname)
|
|
require.NoError(t, err)
|
|
defer ioctx.Destroy()
|
|
|
|
name := GetUUID()
|
|
|
|
options := NewRbdImageOptions()
|
|
err = options.SetUint64(RbdImageOptionFeatures, FeatureLayering|FeatureStripingV2)
|
|
require.NoError(t, err)
|
|
// FeatureStripingV2 only works with additional arguments
|
|
err = options.SetUint64(RbdImageOptionStripeUnit, 1024*1024)
|
|
require.NoError(t, err)
|
|
err = options.SetUint64(RbdImageOptionStripeCount, 4)
|
|
require.NoError(t, err)
|
|
|
|
err = CreateImage(ioctx, name, 16*1024*1024, options)
|
|
require.NoError(t, err)
|
|
defer func() { assert.NoError(t, RemoveImage(ioctx, name)) }()
|
|
|
|
image, err := OpenImageReadOnly(ioctx, name, NoSnapshot)
|
|
assert.NoError(t, err)
|
|
defer func() { assert.NoError(t, image.Close()) }()
|
|
|
|
features, err := image.GetFeatures()
|
|
assert.NoError(t, err)
|
|
|
|
t.Run("compareBits", func(t *testing.T) {
|
|
hasLayering := (features & FeatureLayering) == FeatureLayering
|
|
hasStripingV2 := (features & FeatureStripingV2) == FeatureStripingV2
|
|
assert.True(t, hasLayering, "FeatureLayering is not set")
|
|
assert.True(t, hasStripingV2, "FeatureStripingV2 is not set")
|
|
})
|
|
|
|
t.Run("compareFeatureSet", func(t *testing.T) {
|
|
fs := FeatureSet(features)
|
|
assert.Contains(t, fs.Names(), FeatureNameLayering)
|
|
assert.Contains(t, fs.Names(), FeatureNameStripingV2)
|
|
})
|
|
}
|
|
|
|
func TestFeatureSet(t *testing.T) {
|
|
fsBits := FeatureSet(FeatureExclusiveLock | FeatureDeepFlatten)
|
|
fsNames := FeatureSetFromNames([]string{FeatureNameExclusiveLock, FeatureNameDeepFlatten})
|
|
assert.Equal(t, fsBits, fsNames)
|
|
|
|
fsBitsSorted := fsBits.Names()
|
|
sort.Strings(fsBitsSorted)
|
|
|
|
fsNamesSorted := fsNames.Names()
|
|
sort.Strings(fsNamesSorted)
|
|
|
|
assert.Equal(t, fsBitsSorted, fsNamesSorted)
|
|
}
|
|
|
|
func TestUpdateFeatures(t *testing.T) {
|
|
conn := radosConnect(t)
|
|
require.NotNil(t, conn)
|
|
defer conn.Shutdown()
|
|
|
|
poolname := GetUUID()
|
|
err := conn.MakePool(poolname)
|
|
require.NoError(t, err)
|
|
defer conn.DeletePool(poolname)
|
|
|
|
ioctx, err := conn.OpenIOContext(poolname)
|
|
require.NoError(t, err)
|
|
defer ioctx.Destroy()
|
|
|
|
name := GetUUID()
|
|
|
|
options := NewRbdImageOptions()
|
|
// test with FeatureExclusiveLock as that is mutable
|
|
err = options.SetUint64(RbdImageOptionFeatures, FeatureExclusiveLock)
|
|
require.NoError(t, err)
|
|
|
|
err = CreateImage(ioctx, name, 16*1024*1024, options)
|
|
require.NoError(t, err)
|
|
defer func() { assert.NoError(t, RemoveImage(ioctx, name)) }()
|
|
|
|
image, err := OpenImage(ioctx, name, NoSnapshot)
|
|
require.NoError(t, err)
|
|
defer func() { assert.NoError(t, image.Close()) }()
|
|
|
|
t.Run("imageNotOpen", func(t *testing.T) {
|
|
img, err := OpenImageReadOnly(ioctx, name, NoSnapshot)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, img)
|
|
|
|
err = img.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = img.UpdateFeatures(FeatureExclusiveLock, false)
|
|
assert.Equal(t, err, ErrImageNotOpen)
|
|
})
|
|
|
|
t.Run("verifyFeatureEnabled", func(t *testing.T) {
|
|
features, err := image.GetFeatures()
|
|
require.NoError(t, err)
|
|
|
|
hasExclusiveLock := (features & FeatureExclusiveLock) == FeatureExclusiveLock
|
|
require.True(t, hasExclusiveLock, "FeatureExclusiveLock is not set")
|
|
})
|
|
|
|
t.Run("disableFeature", func(t *testing.T) {
|
|
err = image.UpdateFeatures(FeatureExclusiveLock, false)
|
|
require.NoError(t, err)
|
|
|
|
features, err := image.GetFeatures()
|
|
require.NoError(t, err)
|
|
|
|
hasExclusiveLock := (features & FeatureExclusiveLock) == FeatureExclusiveLock
|
|
require.False(t, hasExclusiveLock, "FeatureExclusiveLock is set")
|
|
})
|
|
|
|
t.Run("enableFeature", func(t *testing.T) {
|
|
err = image.UpdateFeatures(FeatureExclusiveLock, true)
|
|
require.NoError(t, err)
|
|
|
|
features, err := image.GetFeatures()
|
|
require.NoError(t, err)
|
|
|
|
hasExclusiveLock := (features & FeatureExclusiveLock) == FeatureExclusiveLock
|
|
require.True(t, hasExclusiveLock, "FeatureExclusiveLock is not set")
|
|
})
|
|
}
|