test: add more tests (#1119)

master
Kevin Wan 3 years ago committed by GitHub
parent 4c6234f108
commit 9b114e3251
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,16 +1,45 @@
package internal
import (
"sync"
"testing"
"github.com/stretchr/testify/assert"
"github.com/tal-tech/go-zero/core/stat"
"github.com/tal-tech/go-zero/zrpc/internal/mock"
"google.golang.org/grpc"
)
func TestWithMetrics(t *testing.T) {
func TestRpcServer(t *testing.T) {
metrics := stat.NewMetrics("foo")
opt := WithMetrics(metrics)
var options rpcServerOptions
opt(&options)
assert.Equal(t, metrics, options.metrics)
server := NewRpcServer("localhost:54321", WithMetrics(metrics))
server.SetName("mock")
var wg sync.WaitGroup
var grpcServer *grpc.Server
var lock sync.Mutex
wg.Add(1)
go func() {
err := server.Start(func(server *grpc.Server) {
lock.Lock()
mock.RegisterDepositServiceServer(server, new(mock.DepositServer))
grpcServer = server
lock.Unlock()
wg.Done()
})
assert.Nil(t, err)
}()
wg.Wait()
lock.Lock()
grpcServer.GracefulStop()
lock.Unlock()
}
func TestRpcServer_WithBadAddress(t *testing.T) {
server := NewRpcServer("localhost:111111")
server.SetName("mock")
err := server.Start(func(server *grpc.Server) {
mock.RegisterDepositServiceServer(server, new(mock.DepositServer))
})
assert.NotNil(t, err)
}

Loading…
Cancel
Save