Add new testing condition

umpc · Jul 6, 2017 · 9f7a133 · 9f7a133
1 parent 8d043c5
commit 9f7a133
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/iter_test.go b/iter_test.go
@@ -338,6 +338,13 @@ func TestCancelCustomIterCh(t *testing.T) {
 		defer ch.Close()
 
 		go func(ch IterChCloser) {
+			i := 0
+			for range ch.Records() {
+				if i == 50 {
+					ch.Close()
+				}
+				i++
+			}
 			if err := verifyRecords(ch.Records(), params.Reversed); err != nil {
 				if err.Error() != "Channel was nil." {
 					t.Fatal(err)