hermez-node/coordinator/batch_test.go

package coordinator

import (
	"testing"

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

func TestBatchQueue(t *testing.T) {
	bq := BatchQueue{}

	bq.Push(&BatchInfo{
		batchNum: 0,
	})
	bq.Push(&BatchInfo{
		batchNum: 2,
	})
	bq.Push(&BatchInfo{
		batchNum: 1,
	})

	assert.Equal(t, uint64(0), bq.Pop().batchNum)
	assert.Equal(t, uint64(2), bq.Pop().batchNum)
	assert.Equal(t, uint64(1), bq.Pop().batchNum)
	assert.Nil(t, bq.Pop())
}