do not error if context is canceled

hemilabs · Jul 19, 2024 · 33ed38e · 33ed38e
1 parent 5c4c994
commit 33ed38e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/service/tbc/tbcfork_test.go b/service/tbc/tbcfork_test.go
@@ -734,6 +734,9 @@ func waitForPort(ctx context.Context, port string) error {
 	for {
 		select {
 		case <-ctx.Done():
+			if errors.Is(ctx.Err(), context.Canceled) {
+				return nil
+			}
 			return ctx.Err()
 		case <-time.After(100 * time.Millisecond):
 			conn, err := net.DialTimeout("tcp", net.JoinHostPort("localhost", port), 100*time.Millisecond)