Another queue unit test to test batching

queue/queue_test.go

@@ -136,6 +136,42 @@ func Test_NewQueueWriteBatchSizeSingle(t *testing.T) {
 	}
 }
 
+func Test_NewQueueWriteBatchSizeDouble(t *testing.T) {
+	q := New(1024, 1, 60*time.Second)
+	defer q.Close()
+
+	if _, err := q.Write(testStmtsFoo, nil); err != nil {
+		t.Fatalf("failed to write: %s", err.Error())
+	}
+	if _, err := q.Write(testStmtsBar, nil); err != nil {
+		t.Fatalf("failed to write: %s", err.Error())
+	}
+
+	// Just test that I get a batch size, each time.
+	select {
+	case req := <-q.C:
+		if exp, got := 1, len(req.Statements); exp != got {
+			t.Fatalf("received wrong length slice, exp %d, got %d", exp, got)
+		}
+		if req.Statements[0].Sql != "SELECT * FROM foo" {
+			t.Fatalf("received wrong SQL")
+		}
+	case <-time.NewTimer(5 * time.Second).C:
+		t.Fatalf("timed out waiting for statement")
+	}
+	select {
+	case req := <-q.C:
+		if exp, got := 1, len(req.Statements); exp != got {
+			t.Fatalf("received wrong length slice, exp %d, got %d", exp, got)
+		}
+		if req.Statements[0].Sql != "SELECT * FROM bar" {
+			t.Fatalf("received wrong SQL")
+		}
+	case <-time.NewTimer(5 * time.Second).C:
+		t.Fatalf("timed out waiting for statement")
+	}
+}
+
 func Test_NewQueueWriteBatchSizeSingleChan(t *testing.T) {
 	q := New(1024, 1, 60*time.Second)
 	defer q.Close()