open-nomad/nomad/structs/csi_test.go

package structs

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestCSIVolumeClaim(t *testing.T) {
	vol := NewCSIVolume("", 0)
	vol.AccessMode = CSIVolumeAccessModeMultiNodeSingleWriter
	vol.Schedulable = true

	alloc := &Allocation{ID: "a1"}
	alloc2 := &Allocation{ID: "a2"}

	vol.ClaimRead(alloc)
	require.True(t, vol.CanReadOnly())
	require.True(t, vol.CanWrite())
	require.True(t, vol.ClaimRead(alloc))

	vol.ClaimWrite(alloc)
	require.True(t, vol.CanReadOnly())
	require.False(t, vol.CanWrite())
	require.False(t, vol.ClaimWrite(alloc2))

	vol.ClaimRelease(alloc)
	require.True(t, vol.CanReadOnly())
	require.True(t, vol.CanWrite())
}