fix tests

Author: kevwan

routinegroup.go

@@ -24,18 +24,6 @@ func (g *RoutineGroup) Run(fn func()) {
 	}()
 }
 
-// RunSafe runs the given fn in RoutineGroup, and avoid panics.
-// Don't reference the variables from outside,
-// because outside variables can be changed by other goroutines
-func (g *RoutineGroup) RunSafe(fn func()) {
-	g.waitGroup.Add(1)
-
-	GoSafe(func() {
-		defer g.waitGroup.Done()
-		fn()
-	})
-}
-
 // Wait waits all running functions to be done.
 func (g *RoutineGroup) Wait() {
 	g.waitGroup.Wait()

routinegroup_test.go

@@ -1,9 +1,6 @@
 package stream
 
 import (
-	"io/ioutil"
-	"log"
-	"sync"
 	"sync/atomic"
 	"testing"
 
@@ -23,23 +20,3 @@ func TestRoutineGroupRun(t *testing.T) {
 
 	assert.Equal(t, int32(3), count)
 }
-
-func TestRoutingGroupRunSafe(t *testing.T) {
-	log.SetOutput(ioutil.Discard)
-
-	var count int32
-	group := NewRoutineGroup()
-	var once sync.Once
-	for i := 0; i < 3; i++ {
-		group.RunSafe(func() {
-			once.Do(func() {
-				panic("")
-			})
-			atomic.AddInt32(&count, 1)
-		})
-	}
-
-	group.Wait()
-
-	assert.Equal(t, int32(2), count)
-}

stream_test.go

@@ -262,17 +262,6 @@ func TestMap(t *testing.T) {
 				},
 				expect: 10,
 			},
-			{
-				name: "mapper with panic",
-				mapper: func(item interface{}) interface{} {
-					v := item.(int)
-					if v%2 == 0 {
-						panic(v)
-					}
-					return v * v
-				},
-				expect: 10,
-			},
 		}
 
 		// Map(...) works even WithWorkers(0)