fix tests

centrifugal · Oct 20, 2024 · fe73421 · fe73421
1 parent 371b0fa
commit fe73421
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/internal/proxy/connect_handler_test.go b/internal/proxy/connect_handler_test.go
@@ -15,6 +15,8 @@ import (
 
 	"github.com/centrifugal/centrifuge"
 	"github.com/stretchr/testify/require"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
 )
 
 type grpcConnHandleTestCase struct {
@@ -227,7 +229,13 @@ func TestHandleConnectWithoutProxyServerStart(t *testing.T) {
 	cases := newConnHandleTestCases(httpTestCase, grpcTestCase)
 	for _, c := range cases {
 		reply, err := c.invokeHandle(context.Background())
-		require.ErrorIs(t, centrifuge.ErrorInternal, err, c.protocol)
+		if c.protocol == "grpc" {
+			st, ok := status.FromError(err)
+			require.True(t, ok, c.protocol)
+			require.Equal(t, codes.Unavailable, st.Code(), c.protocol)
+		} else {
+			require.Error(t, err, c.protocol)
+		}
 		require.Equal(t, centrifuge.ConnectReply{}, reply, c.protocol)
 	}
 }

diff --git a/internal/unihttpstream/config.go b/internal/unihttpstream/config.go
@@ -9,6 +9,6 @@ import (
 type Config struct {
 	// MaxRequestBodySize limits request body size.
 	MaxRequestBodySize    int
-	ConnectCodeTranslates tools.ConnectCodeToHTTPStatusTranslates
+	ConnectCodeTranslates tools.ConnectCodeToHTTPStatus
 	centrifuge.PingPongConfig
 }
diff --git a/internal/unisse/config.go b/internal/unisse/config.go
@@ -8,6 +8,6 @@ import (
 type Config struct {
 	// MaxRequestBodySize for POST requests when used.
 	MaxRequestBodySize    int
-	ConnectCodeTranslates tools.ConnectCodeToHTTPStatusTranslates
+	ConnectCodeTranslates tools.ConnectCodeToHTTPStatus
 	centrifuge.PingPongConfig
 }