Add new test for Iter Close methods with and without a send timeout

umpc · Jul 6, 2017 · 8b578bf · 8b578bf
1 parent e7083b5
commit 8b578bf
Showing 1 changed file with 17 additions and 11 deletions.
diff --git a/iter_test.go b/iter_test.go
@@ -329,19 +329,25 @@ func TestCloseCustomIterCh(t *testing.T) {
 	earlierDate := time.Date(1, 1, 1, 0, 0, 0, 0, time.UTC)
 	laterDate := time.Now()
 
-	func() {
-		params := IterChParams{
-			LowerBound: earlierDate,
-			UpperBound: laterDate,
-		}
+	ch, err := sm.IterCh()
+	if err != nil {
+		t.Fatal(err)
+	}
 
-		ch, err := sm.CustomIterCh(params)
-		if err != nil {
-			t.Fatal(err)
-		}
+	ch.Close()
 
-		ch.Close()
-	}()
+	params := IterChParams{
+		SendTimeout: 5 * time.Minute,
+		LowerBound: earlierDate,
+		UpperBound: laterDate,
+	}
+
+	ch, err = sm.CustomIterCh(params)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	ch.Close()
 }
 
 func TestIterFunc(t *testing.T) {