Adjust error checking in tests

janos · May 21, 2018 · 2c1d00e · 2c1d00e
1 parent 8d5ff75
commit 2c1d00e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/client/http/http_client_test.go b/client/http/http_client_test.go
@@ -94,15 +94,15 @@ func TestClientRetryFailure(t *testing.T) {
 	}()
 
 	r, err := New(&Options{RetryTimeMax: 1 * time.Second}).Get(fmt.Sprintf("http://localhost:%d", port))
-	if err == nil || !strings.Contains(err.Error(), "getsockopt: connection refused") {
+	if err == nil || !strings.Contains(err.Error(), "connection refused") {
 		t.Errorf("expected connection refused error, got %#v", err)
 	}
 	if r != nil {
 		t.Error("unexpected not-nil response")
 	}
 
 	r, err = (&http.Client{Transport: Transport(&Options{RetryTimeMax: 1 * time.Second})}).Get(fmt.Sprintf("http://localhost:%d", port))
-	if err == nil || !strings.Contains(err.Error(), "getsockopt: connection refused") {
+	if err == nil || !strings.Contains(err.Error(), "connection refused") {
 		t.Errorf("expected connection refused error, got %#v", err)
 	}
 	if r != nil {

diff --git a/servers/grpc/grpc_test.go b/servers/grpc/grpc_test.go
@@ -104,8 +104,8 @@ func TestServerShutdown(t *testing.T) {
 	s.Shutdown(context.Background())
 
 	r, err = c.Greet(context.Background(), &hello.GreetRequest{Name: name})
-	if err.Error() != "rpc error: code = Unavailable desc = all SubConns are in TransientFailure" {
-		t.Fatal(err)
+	if !strings.Contains(err.Error(), "connection refused") {
+		t.Fatalf("unexpected error: %q", err)
 	}
 }