equalizer_test.go

package equalizer

import (
	"strings"
	"testing"

	"github.com/reugn/equalizer/internal/assert"
)

func TestEqualizer(t *testing.T) {
	offset := NewStepOffset(32, 5)
	eq, err := NewEqualizer(32, 8, offset)
	assert.IsNil(t, err)

	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 32))
	assert.Equal(t, eq.mask.Text(2), strings.Repeat("1", 8)+strings.Repeat("0", 24))
	assert.Equal(t, eq.mask.Bit(0), uint(0))

	eq.Failure(16)
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 16)+strings.Repeat("0", 16))

	assert.Equal(t, eq.TryAcquire(), false)
	assert.Equal(t, eq.TryAcquire(), false)
	assert.Equal(t, eq.TryAcquire(), false)
	assert.Equal(t, eq.TryAcquire(), true)

	eq.Success(10)
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 16)+strings.Repeat("0", 14)+
		strings.Repeat("1", 2))

	eq.Failure(1)
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 16)+strings.Repeat("0", 14)+
		strings.Repeat("1", 2))

	eq.Success(2)
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 16)+strings.Repeat("0", 11)+
		"110"+strings.Repeat("1", 2))

	eq.Failure(32)
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 8)+strings.Repeat("0", 24))

	eq.Reset()
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 32))

	eq.Failure(-1)
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 32))

	eq.Purge()
	assert.Equal(t, eq.tape.Text(2), strings.Repeat("1", 8)+strings.Repeat("0", 24))
}

func TestEqualizer_Errors(t *testing.T) {
	eq, err := NewEqualizer(32, 8, nil)
	assert.ErrorContains(t, err, "offset is nil")
	assert.IsNil(t, eq)

	offset := NewStepOffset(32, 5)
	eq, err = NewEqualizer(-1, 8, offset)
	assert.ErrorContains(t, err, "nonpositive size")
	assert.IsNil(t, eq)

	eq, err = NewEqualizer(32, -1, offset)
	assert.ErrorContains(t, err, "nonpositive reserved")
	assert.IsNil(t, eq)

	eq, err = NewEqualizer(8, 32, offset)
	assert.ErrorContains(t, err, "reserved must not exceed size")
	assert.IsNil(t, eq)
}