From 6da7ce64affa73468c01712dc171925565472cc2 Mon Sep 17 00:00:00 2001 From: dudaodong Date: Mon, 26 Dec 2022 14:24:23 +0800 Subject: [PATCH] test: add unit test for RangeIterator --- iterator/iterator_test.go | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/iterator/iterator_test.go b/iterator/iterator_test.go index da62d3c..b762b5f 100644 --- a/iterator/iterator_test.go +++ b/iterator/iterator_test.go @@ -48,3 +48,28 @@ func TestSliceIterator(t *testing.T) { }) } + +func TestRangeIterator(t *testing.T) { + assert := internal.NewAssert(t, "TestRangeIterator") + + t.Run("range iterator: ", func(t *testing.T) { + iter := FromRange(1, 4, 1) + + item, ok := iter.Next() + assert.Equal(1, item) + assert.Equal(true, ok) + + item, ok = iter.Next() + assert.Equal(2, item) + assert.Equal(true, ok) + + item, ok = iter.Next() + assert.Equal(3, item) + assert.Equal(true, ok) + + _, ok = iter.Next() + assert.Equal(false, ok) + assert.Equal(false, iter.HasNext()) + }) + +}