| 83 | } |
| 84 | |
| 85 | func TestRangeIterator(t *testing.T) { |
| 86 | t.Parallel() |
| 87 | |
| 88 | assert := internal.NewAssert(t, "TestRangeIterator") |
| 89 | |
| 90 | t.Run("range iterator: ", func(t *testing.T) { |
| 91 | iter := FromRange(1, 4, 1) |
| 92 | |
| 93 | item, ok := iter.Next() |
| 94 | assert.Equal(1, item) |
| 95 | assert.Equal(true, ok) |
| 96 | |
| 97 | item, ok = iter.Next() |
| 98 | assert.Equal(2, item) |
| 99 | assert.Equal(true, ok) |
| 100 | |
| 101 | item, ok = iter.Next() |
| 102 | assert.Equal(3, item) |
| 103 | assert.Equal(true, ok) |
| 104 | |
| 105 | _, ok = iter.Next() |
| 106 | assert.Equal(false, ok) |
| 107 | assert.Equal(false, iter.HasNext()) |
| 108 | |
| 109 | iter.Reset() |
| 110 | |
| 111 | item, ok = iter.Next() |
| 112 | assert.Equal(1, item) |
| 113 | assert.Equal(true, ok) |
| 114 | |
| 115 | item, ok = iter.Next() |
| 116 | assert.Equal(2, item) |
| 117 | assert.Equal(true, ok) |
| 118 | |
| 119 | item, ok = iter.Next() |
| 120 | assert.Equal(3, item) |
| 121 | assert.Equal(true, ok) |
| 122 | |
| 123 | _, ok = iter.Next() |
| 124 | assert.Equal(false, ok) |
| 125 | assert.Equal(false, iter.HasNext()) |
| 126 | }) |
| 127 | |
| 128 | t.Run("range iterator reset: ", func(t *testing.T) { |
| 129 | iter := FromRange(1, 4, 1) |
| 130 | |
| 131 | item, ok := iter.Next() |
| 132 | assert.Equal(1, item) |
| 133 | assert.Equal(true, ok) |
| 134 | |
| 135 | item, ok = iter.Next() |
| 136 | assert.Equal(2, item) |
| 137 | assert.Equal(true, ok) |
| 138 | |
| 139 | iter.Reset() |
| 140 | |
| 141 | item, ok = iter.Next() |
| 142 | assert.Equal(1, item) |