// Returns a Volume object with info about the volume at the specified
// mount point. The mount point should be the path to a disk or partition.
// For example, "/", "/mnt/data", etc.
func (service *VolumeService) getVolume(path string) *models.Volume {
mountpoint, err := platform.GetMountPointFromPath(path)
if err != nil {
mountpoint = "/"
service.logger.Error("Cannot determine mountpoint of file '%s': %v",
path, err)
}
if _, keyExists := service.volumes[mountpoint]; !keyExists {
service.volumes[mountpoint] = models.NewVolume(mountpoint)
}
return service.volumes[mountpoint]
}
func TestClaimedReserveReleasePath(t *testing.T) {
_, filename, _, _ := runtime.Caller(0)
volume := models.NewVolume(filename)
assert.EqualValues(t, 0, volume.ClaimedSpace())
assert.Equal(t, filename, volume.MountPoint())
err := volume.Reserve("/path/to/file_0", 1000)
require.Nil(t, err)
assert.EqualValues(t, 1000, volume.ClaimedSpace())
volume.Release("/this/file/was/never/reserved")
assert.EqualValues(t, 1000, volume.ClaimedSpace())
volume.Release("/path/to/file_0")
assert.EqualValues(t, 0, volume.ClaimedSpace())
}
func TestReservations(t *testing.T) {
_, filename, _, _ := runtime.Caller(0)
volume := models.NewVolume(filename)
paths := []string{"p1", "p2", "p3", "p4", "p5"}
for i, path := range paths {
err := volume.Reserve(path, uint64(1000+i))
assert.Nil(t, err)
}
reservations := volume.Reservations()
require.Equal(t, len(paths), len(reservations))
for i, path := range paths {
bytes, keyExists := reservations[path]
assert.True(t, keyExists)
assert.EqualValues(t, uint64(1000+i), bytes)
// Releasing path should remove it from reservations
volume.Release(path)
}
assert.Empty(t, volume.Reservations())
}
// This functional/behavioral test goes through some more realistic
// usage scenarios.
func TestVolume(t *testing.T) {
_, filename, _, _ := runtime.Caller(0)
volume := models.NewVolume(filename)
// Make sure we can reserve space that's actually there.
initialSpace, err := volume.AvailableSpace()
require.Nil(t, err)
numBytes := initialSpace / 3
err = volume.Reserve("/path/to/file_1", numBytes)
require.Nil(t, err)
err = volume.Reserve("/path/to/file_2", numBytes)
require.Nil(t, err)
// Make sure we're tracking the available space correctly.
bytesAvailable, err := volume.AvailableSpace()
require.Nil(t, err)
expectedBytesAvailable := (initialSpace - (2 * numBytes))
assert.Equal(t, expectedBytesAvailable, bytesAvailable)
// Make sure a request for too much space is rejected
err = volume.Reserve("/path/to/file_3", numBytes*2)
require.NotNil(t, err)
// Free the two chunks of space we just requested.
volume.Release("/path/to/file_1")
volume.Release("/path/to/file_2")
// Make sure it was freed.
bytesAvailable, err = volume.AvailableSpace()
require.Nil(t, err)
assert.Equal(t, initialSpace, bytesAvailable)
// Now we should have enough space for this.
err = volume.Reserve("/path/to/file_4", numBytes*2)
require.Nil(t, err)
}