Added tests for closeAll

dustin · Mar 11, 2014 · 6bf6082 · 6bf6082
1 parent 987fd94
commit 6bf6082
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/util_test.go b/util_test.go
@@ -4,6 +4,24 @@ import (
 	"testing"
 )
 
+func TestCloseAllEmpty(t *testing.T) {
+	closeAll([]chan int{})
+}
+
+func TestCloseAllTwo(t *testing.T) {
+	ichan := make(chan int)
+	i2chan := make(chan int)
+	closeAll([]chan int{ichan, i2chan})
+	_, ok := <-ichan
+	if ok {
+		t.Errorf("ichan wasn't closed")
+	}
+	_, ok = <-i2chan
+	if ok {
+		t.Errorf("bchan wasn't closed")
+	}
+}
+
 func buildTestChans(n int) []chan int {
 	s := make([]chan int, 0, n)
 	for i := 0; i < n; i++ {