From 8a2270d3940e1cb5212ae8ceccb86425aa3135a2 Mon Sep 17 00:00:00 2001 From: Kyle Carberry Date: Tue, 1 Mar 2022 01:45:22 +0000 Subject: [PATCH] test: Fix flake with context.Cancelled in provisionerd This occurred because the context can cancel in the same time a response is sent. This isn't a bug, because the complete still occurs. --- provisionerd/provisionerd_test.go | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/provisionerd/provisionerd_test.go b/provisionerd/provisionerd_test.go index de27ec846df09..1eac1e77b49c4 100644 --- a/provisionerd/provisionerd_test.go +++ b/provisionerd/provisionerd_test.go @@ -466,15 +466,13 @@ func TestProvisionerd(t *testing.T) { }) require.NoError(t, err) <-shutdownCtx.Done() - err = stream.Send(&sdkproto.Provision_Response{ + return stream.Send(&sdkproto.Provision_Response{ Type: &sdkproto.Provision_Response_Complete{ Complete: &sdkproto.Provision_Complete{ Error: "some error", }, }, }) - require.NoError(t, err) - return nil }, }), })