Function testContainerIteratorAdvance

container_test.go:72–140

(t *testing.T, con container)

}

func testContainerIteratorAdvance(t *testing.T, con container) {
	values := []uint16{1, 2, 15, 16, 31, 32, 33, 9999}
	for _, v := range values {
		con.iadd(v)
	}

	cases := []struct {
		minval   uint16
		expected uint16
	}{
		{0, 1},
		{1, 1},
		{2, 2},
		{3, 15},
		{15, 15},
		{30, 31},
		{31, 31},
		{33, 33},
		{34, 9999},
		{9998, 9999},
		{9999, 9999},
	}

	t.Run("advance by using a new short iterator", func(t *testing.T) {
		for _, c := range cases {
			i := con.getShortIterator()
			i.advanceIfNeeded(c.minval)

			assert.True(t, i.hasNext())
			assert.Equal(t, c.expected, i.peekNext())
		}
	})

	t.Run("advance by using the same short iterator", func(t *testing.T) {
		i := con.getShortIterator()

		for _, c := range cases {
			i.advanceIfNeeded(c.minval)

			assert.True(t, i.hasNext())
			assert.Equal(t, c.expected, i.peekNext())
		}
	})

	t.Run("advance out of a container value", func(t *testing.T) {
		i := con.getShortIterator()

		i.advanceIfNeeded(33)
		assert.True(t, i.hasNext())
		assert.EqualValues(t, 33, i.peekNext())

		i.advanceIfNeeded(MaxUint16 - 1)
		assert.False(t, i.hasNext())

		i.advanceIfNeeded(MaxUint16)
		assert.False(t, i.hasNext())
	})