1
0
mirror of https://github.com/duke-git/lancet.git synced 2026-02-17 11:12:28 +08:00

feat: add RepeateFn function

This commit is contained in:
dudaodong
2022-04-15 16:32:55 +08:00
parent f66c0938e5
commit dfc6b868fb
2 changed files with 40 additions and 0 deletions

View File

@@ -42,6 +42,28 @@ func TestRepeat(t *testing.T) {
assert.Equal(1, <-intStream)
}
func TestRepeatFn(t *testing.T) {
assert := internal.NewAssert(t, "TestRepeatFn")
done := make(chan any)
defer close(done)
fn := func() any {
s := "a"
return s
}
c := NewChannel()
dataStream := c.Take(done, c.RepeatFn(done, fn), 3)
// for v := range dataStream {
// t.Log(v) //a, a, a
// }
assert.Equal("a", <-dataStream)
assert.Equal("a", <-dataStream)
assert.Equal("a", <-dataStream)
}
func TestTake(t *testing.T) {
assert := internal.NewAssert(t, "TestRepeat")