go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-02-13 13:48:51 +02:00
Code Issues Releases Activity

test(transport): add transport Listener test (#1735)

Browse Source 
* test(tansport): add transport Listener test
This commit is contained in:
haiyux 2021-12-31 11:15:39 +08:00 committed by GitHub
parent 3625634d3c
commit 7f003a8742
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 32 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
transport/grpc/server_test.go
View File

@ -4,6 +4,7 @@ import (
"context"
"crypto/tls"
"fmt"
"net"
"net/url"
"strings"
"testing"
@ -214,3 +215,10 @@ func TestServer_unaryServerInterceptor(t *testing.T) {
assert.NoError(t, err)
assert.Equal(t, "hi", rv.(*testResp).Data)
}
func TestListener(t *testing.T) {
lis := &net.TCPListener{}
s := &Server{}
Listener(lis)(s)
assert.Equal(t, s.lis, lis)
}

22
transport/http/client_test.go
View File

@ -5,12 +5,14 @@ import (
"context"
"crypto/tls"
"encoding/json"
"fmt"
"io"
nethttp "net/http"
"strconv"
"testing"
"time"
"github.com/go-kratos/kratos/v2/errors"
kratosErrors "github.com/go-kratos/kratos/v2/errors"
"github.com/go-kratos/kratos/v2/middleware"
"github.com/go-kratos/kratos/v2/registry"
"github.com/stretchr/testify/assert"
@ -116,8 +118,16 @@ func (*mockDiscovery) Watch(ctx context.Context, serviceName string) (registry.W
type mockWatcher struct{}
func (*mockWatcher) Next() ([]*registry.ServiceInstance, error) {
return nil, nil
func (m *mockWatcher) Next() ([]*registry.ServiceInstance, error) {
instance := &registry.ServiceInstance{
ID: "1",
Name: "kratos",
Version: "v1",
Metadata: map[string]string{},
Endpoints: []string{fmt.Sprintf("http://127.0.0.1:9001?isSecure=%s", strconv.FormatBool(false))},
}
time.Sleep(time.Millisecond * 500)
return []*registry.ServiceInstance{instance}, nil
}
func (*mockWatcher) Stop() error {
@ -202,9 +212,9 @@ func TestDefaultErrorDecoder(t *testing.T) {
}
err2 := DefaultErrorDecoder(context.TODO(), resp2)
assert.Error(t, err2)
assert.Equal(t, int32(500), err2.(*errors.Error).GetCode())
assert.Equal(t, "hi", err2.(*errors.Error).GetMessage())
assert.Equal(t, "FOO", err2.(*errors.Error).GetReason())
assert.Equal(t, int32(500), err2.(*kratosErrors.Error).GetCode())
assert.Equal(t, "hi", err2.(*kratosErrors.Error).GetMessage())
assert.Equal(t, "FOO", err2.(*kratosErrors.Error).GetReason())
}
func TestCodecForResponse(t *testing.T) {

8
transport/http/server_test.go
View File

@ -6,6 +6,7 @@ import (
"encoding/json"
"fmt"
"io"
"net"
"net/http"
"strings"
"testing"
@ -236,3 +237,10 @@ func TestTLSConfig(t *testing.T) {
TLSConfig(v)(o)
assert.Equal(t, v, o.tlsConf)
}
func TestListener(t *testing.T) {
lis := &net.TCPListener{}
s := &Server{}
Listener(lis)(s)
assert.Equal(t, s.lis, lis)
}