alloc_test.go

package alloc

import (
	"testing"

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

func TestAllocatorReservationUnique(t *testing.T) {
	assert := assert.New(t)
	alloc := New(5, 10, AddrSet{})
	a1, ok := alloc.Reserve()
	assert.GreaterOrEqual(a1, uint64(5), "allocated address %d should be in range", a1)
	assert.True(ok)
	a2, ok := alloc.Reserve()
	assert.True(ok)
	assert.NotEqual(a1, a2, "reserved same block twice")
}

func TestAllocatorAll(t *testing.T) {
	assert := assert.New(t)
	alloc := New(5, 10, AddrSet{})
	for i := 0; i < 10; i++ {
		_, ok := alloc.Reserve()
		assert.True(ok, "reservation failed early: %d", i)
	}
	_, ok := alloc.Reserve()
	assert.False(ok, "all addresses should be allocatd")
}

func TestAllocatorUsed(t *testing.T) {
	assert := assert.New(t)
	alloc := New(0, 3, AddrSet{0: unit{}, 2: unit{}})
	a, ok := alloc.Reserve()
	assert.True(ok, "should use last free slot")
	assert.Equal(uint64(1), a, "should not allocate used addresses")
}

func TestAllocatorFree(t *testing.T) {
	assert := assert.New(t)
	alloc := New(0, 10, AddrSet{})
	for i := 0; i < 10; i++ {
		alloc.Reserve()
	}
	alloc.Free(2)
	alloc.Free(3)
	a, ok := alloc.Reserve()
	assert.True(ok, "should use newly-freed space")
	assert.True(a == 2 || a == 3,
		"new address {} should be freed", a)
}