fix(routing/mock): wait for goroutine cleanup (#619)

routing/mock/centralized_test.go

@@ -2,6 +2,7 @@ package mockrouting
 
 import (
 	"context"
+	"sync"
 	"testing"
 	"time"
 
@@ -87,7 +88,12 @@ func TestCanceledContext(t *testing.T) {
 	// (we want the goroutine to keep trying to publish on a
 	// cancelled context until we've tested it doesnt do anything.)
 	done := make(chan struct{})
-	defer func() { done <- struct{}{} }()
+	doneWg := sync.WaitGroup{}
+	doneWg.Add(1)
+	defer func() {
+		done <- struct{}{}
+		doneWg.Wait()
+	}()
 
 	t.Log("async'ly announce infinite stream of providers for key")
 	i := 0
@@ -96,6 +102,7 @@ func TestCanceledContext(t *testing.T) {
 			select {
 			case <-done:
 				t.Log("exiting async worker")
+				doneWg.Done()
 				return
 			default:
 			}