fix: fix TestAcquireJobWithCancel_Cancel flake (#18441)

This commit is contained in:
Jon Ayers
2025-06-18 22:51:13 -04:00
committed by GitHub
parent b0fa3275d2
commit 8b27983d14

View File

@ -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 {