test: handle io timeout failures

natesales · Apr 5, 2024 · d8a1dd6 · d8a1dd6
1 parent 499d6ca
commit d8a1dd6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/main_test.go b/main_test.go
@@ -176,7 +176,9 @@ func TestMainInvalidUpstream(t *testing.T) {
 		"-s", "127.127.127.127:1",
 		"example.com",
 	)
-	if !(err != nil && strings.Contains(err.Error(), "connection refused")) {
+	if !(err != nil &&
+		(strings.Contains(err.Error(), "connection refused") ||
+			strings.Contains(err.Error(), "i/o timeout"))) {
 		t.Errorf("expected connection error, got %+v", err)
 	}
 }