Merge pull request #1 from SamuelTissot/feature/waitgroup

added waitgroup to the Queue.

Author: SamuelTissot

queue.go

@@ -4,6 +4,10 @@
 // interface `Job`
 package bqueue
 
+import "sync"
+
+var WG sync.WaitGroup
+
 // Queue that process jobs reveived
 type Queue struct {
 	maxWorker   int
@@ -35,6 +39,7 @@ func (q *Queue) Start() {
 
 // CollectJob Adds a job to the Queue
 func (q *Queue) CollectJob(job Job) {
+	WG.Add(1)
 	q.JobReceived <- job
 }
 

queue_test.go

@@ -3,8 +3,11 @@ package bqueue
 import (
 	"bytes"
 	"errors"
+	"fmt"
 	"log"
 	"os"
+	"strconv"
+	"strings"
 	"testing"
 	"time"
 
@@ -37,6 +40,30 @@ func Test_Collect_and_process_job(t *testing.T) {
 
 }
 
+func Test_WG(t *testing.T) {
+	q := New(1)
+	q.Start()
+
+	//var output string
+	var output string
+	output += captureStout(func() {
+		for i := 0; i <= 200; i++ {
+			j := aJob{strconv.Itoa(i)}
+			q.CollectJob(j)
+
+		}
+		WG.Wait()
+	})
+
+	// look for all instances of "test job: N"
+	// not very efficient but minimizes false positive
+	for i := 0; i <= 200; i++ {
+		if !strings.Contains(output, fmt.Sprintf("test job: %d", i)) {
+			t.FailNow()
+		}
+	}
+}
+
 func captureStout(f func()) string {
 	var buf bytes.Buffer
 	output := ""

worker.go

@@ -52,4 +52,6 @@ func (w *worker) do(j Job) {
 	if err != nil {
 		fmt.Println(err)
 	}
+
+	WG.Done()
 }