graceful shutdown in test

spikecurtis · spikecurtis · commit 13181b3e5404 · 2023-09-18T10:58:58.000Z
Signed-off-by: Spike Curtis &lt;spike@coder.com&gt;
diff --git a/provisionerd/provisionerd_test.go b/provisionerd/provisionerd_test.go
@@ -1030,6 +1030,9 @@ func TestProvisionerd(t *testing.T) {
 			}),
 		})
 		require.Condition(t, closedWithin(completeChan, testutil.WaitShort))
+		ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
+		defer cancel()
+		require.NoError(t, server.Shutdown(ctx))
 		require.NoError(t, server.Close())
 		assert.Equal(t, ops[len(ops)-1], "CompleteJob")
 		assert.Contains(t, ops[0:len(ops)-1], "Log: Cleaning Up | ")
