fix(skipWhile): fix and add tests for skipWhile

biggyspender · Jun 14, 2019 · 5925e45 · 5925e45
1 parent 17c9345
commit 5925e45
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/extensions/skipWhile.ts b/src/extensions/skipWhile.ts
@@ -12,11 +12,15 @@ function skipWhile<T>(this: Enumerable<T>, pred: IndexedPredicate<T>): Enumerabl
   const src = this
   return EnumerableGenerators.fromGenerator(function*() {
     let i = 0
+    let skip = true
     for (const item of src) {
-      const result = pred(item, i++)
-      if (result) {
-        continue
+      if (skip) {
+        const result = pred(item, i++)
+        if (result) {
+          continue
+        }
       }
+      skip = false
       yield item
     }
   })

diff --git a/test/blinq.test.ts b/test/blinq.test.ts
@@ -548,6 +548,9 @@ describe('blinq test', () => {
   it('skip', () => {
     expect([...blinq([1, 2, 3]).skip(1)]).toEqual([2, 3])
   })
+  it('skipWhile', () => {
+    expect([...blinq([1, 2, 3, 4, 1, 5]).skipWhile(x => x < 3)]).toEqual([3, 4, 1, 5])
+  })
   it('take', () => {
     expect([...blinq([1, 2, 3]).take(2)]).toEqual([1, 2])
   })