fix: fix TestAcquireJobWithCancel_Cancel flake (#18441)

sreya · web-flow · commit 8b27983d149a · 2025-06-18T22:51:13.000-04:00
diff --git a/coderd/provisionerdserver/provisionerdserver.go b/coderd/provisionerdserver/provisionerdserver.go
@@ -368,7 +368,7 @@ func (s *server) AcquireJobWithCancel(stream proto.DRPCProvisionerDaemon_Acquire
 		je = <-jec
 	case je = <-jec:
 	}
-	if xerrors.Is(je.err, context.Canceled) {
+	if database.IsQueryCanceledError(je.err) {
 		s.Logger.Debug(streamCtx, "successful cancel")
 		err := stream.Send(&proto.AcquiredJob{})
 		if err != nil {

Original file line number	Diff line number	Diff line change
`@@ -368,7 +368,7 @@ func (s *server) AcquireJobWithCancel(stream proto.DRPCProvisionerDaemon_Acquire`
`368`	`368`	`je = <-jec`
`369`	`369`	`case je = <-jec:`
`370`	`370`	`}`
`371`		`- if xerrors.Is(je.err, context.Canceled) {`
	`371`	`+ if database.IsQueryCanceledError(je.err) {`
`372`	`372`	`s.Logger.Debug(streamCtx, "successful cancel")`
`373`	`373`	`err := stream.Send(&proto.AcquiredJob{})`
`374`	`374`	`if err != nil {`