test: Fix flake with context.Cancelled in provisionerd (#386)

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.
This commit is contained in:
Kyle Carberry
2022-03-01 01:56:54 +00:00
committed by GitHub
parent fd5eceb0b8
commit ea5efbd37f
+1 -3
View File
@@ -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
},
}),
})