From b32db76da625f13ed7451bb94d6af6d7b11d7b55 Mon Sep 17 00:00:00 2001 From: Nuno Cruces Date: Fri, 25 Apr 2025 00:40:52 +0100 Subject: [PATCH] Fix flake. --- driver/driver_test.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/driver/driver_test.go b/driver/driver_test.go index 081b8c6..8b3fad2 100644 --- a/driver/driver_test.go +++ b/driver/driver_test.go @@ -249,8 +249,9 @@ func Test_nested_context(t *testing.T) { cancel() var terr interface{ Temporary() bool } - if inner.Next() || !errors.As(inner.Err(), &terr) || !terr.Temporary() { - t.Fatalf("got %v, want temporary", inner.Err()) + if inner.Next() || !errors.Is(inner.Err(), context.Canceled) && + (!errors.As(inner.Err(), &terr) || !terr.Temporary()) { + t.Fatalf("got %v, want cancellation", inner.Err()) } want(outer, 1)