Wait for process to finish before flushing channel

main.go

@@ -62,15 +62,21 @@ func main() {
 			os.Exit(1)
 		}
 
+		ch := make(chan bool)
+
 		go func() {
 			io.Copy(io.MultiWriter(producer, os.Stdout), stdout)
+			ch <- true
 		}()
 
 		go func() {
 			io.Copy(io.MultiWriter(producer, os.Stderr), stderr)
+			ch <- true
 		}()
 
 		err = cmd.Run()
+		<-ch
+		<-ch
 		producer.Close()
 		if err != nil {
 			fmt.Printf("Err: %v\n", err)

main_test.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"fmt"
 	"io/ioutil"
 	"strings"
 	"testing"
@@ -10,7 +9,7 @@ import (
 func TestConsumerAndProducer(t *testing.T) {
 	p, err := NewProducer("redis://localhost:6379", "testKey")
 	if err != nil {
-		t.Errorf("Create Producer Error: %v", err)
+		t.Fatalf("Create Producer Error: %v", err)
 	}
 
 	_, err = p.Write([]byte("Here's a "))
@@ -25,8 +24,6 @@ func TestConsumerAndProducer(t *testing.T) {
 
 	p.Close()
 
-	fmt.Println("Writing done.")
-
 	c, err := NewConsumer("redis://localhost:6379", "testKey")
 	bytes, err := ioutil.ReadAll(c)
 	if err != nil {
@@ -42,7 +39,7 @@ func TestConsumerAndProducer(t *testing.T) {
 func TestLargeMessage(t *testing.T) {
 	p, err := NewProducer("redis://localhost:6379", "testKey2")
 	if err != nil {
-		t.Errorf("Create Producer Error: %v", err)
+		t.Fatalf("Create Producer Error: %v", err)
 	}
 
 	longString := strings.Repeat("a", 10000)
@@ -54,8 +51,6 @@ func TestLargeMessage(t *testing.T) {
 
 	p.Close()
 
-	fmt.Println("Writing done.")
-
 	c, err := NewConsumer("redis://localhost:6379", "testKey2")
 	bytes, err := ioutil.ReadAll(c)
 	if err != nil {