32 lines
693 B
Go
32 lines
693 B
Go
|
// Copyright (c) HashiCorp, Inc.
|
||
|
|
// SPDX-License-Identifier: BUSL-1.1
|
||
|
|
|
||
|
|
package state
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/nomad/ci"
|
||
|
|
"github.com/shoenig/test/must"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSliceIterator(t *testing.T) {
|
||
|
|
ci.Parallel(t)
|
||
|
|
|
||
|
|
sliceIterator := NewSliceIterator()
|
||
|
|
must.NotNil(t, sliceIterator)
|
||
|
|
|
||
|
|
// Add something and perform our tests to ensure the expected data is
|
||
|
|
// returned.
|
||
|
|
sliceIterator.Add("random-information")
|
||
|
|
must.Len(t, 1, sliceIterator.data)
|
||
|
|
must.Zero(t, sliceIterator.idx)
|
||
|
|
must.Nil(t, sliceIterator.WatchCh())
|
||
|
|
|
||
|
|
next1 := sliceIterator.Next()
|
||
|
|
next2 := sliceIterator.Next()
|
||
|
|
must.Eq(t, "random-information", next1)
|
||
|
|
must.Nil(t, next2)
|
||
|
|
must.Eq(t, 1, sliceIterator.idx)
|
||
|
|
}
|