From 023245a7baffa58c353420b505e13b6d254ce420 Mon Sep 17 00:00:00 2001
From: Asim Aslam <asim@chuhnk.me>
Date: Fri, 25 May 2018 14:43:32 +0100
Subject: [PATCH] shutdown broker once done

---
 function_test.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/function_test.go b/function_test.go
index d2c828e5..47bcec7e 100644
--- a/function_test.go
+++ b/function_test.go
@@ -13,9 +13,13 @@ func TestFunction(t *testing.T) {
 	var wg sync.WaitGroup
 	wg.Add(1)
 
+	// cancellation context
+	ctx, cancel := context.WithCancel(context.Background())
+
 	// create service
 	fn := NewFunction(
 		Name("test.function"),
+		Context(ctx),
 		Registry(mock.NewRegistry()),
 		AfterStart(func() error {
 			wg.Done()
@@ -47,6 +51,8 @@ func TestFunction(t *testing.T) {
 		if rsp.Status != "ok" {
 			t.Fatalf("function response: %s", rsp.Status)
 		}
+
+		cancel()
 	}()
 
 	// run service