From 9f7a13337f091be0d4ec53f252c12579bc7949e1 Mon Sep 17 00:00:00 2001
From: Justin Lowery <j@justin.sx>
Date: Thu, 6 Jul 2017 06:11:19 -0400
Subject: [PATCH] Add new testing condition

---
 iter_test.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/iter_test.go b/iter_test.go
index a869446..74991db 100644
--- 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)