kevin 4 lat temu
rodzic
commit
02b509e0a4
1 zmienionych plików z 34 dodań i 0 usunięć
  1. 34 0
      rpcx/internal/codes/accept_test.go

+ 34 - 0
rpcx/internal/codes/accept_test.go

@@ -0,0 +1,34 @@
+package codes
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func TestAccept(t *testing.T) {
+	tests := []struct {
+		name   string
+		err    error
+		accept bool
+	}{
+		{
+			name:   "nil error",
+			err:    nil,
+			accept: true,
+		},
+		{
+			name:   "deadline error",
+			err:    status.Error(codes.DeadlineExceeded, "deadline"),
+			accept: false,
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			assert.Equal(t, test.accept, Acceptable(test.err))
+		})
+	}
+}