go/grpcall
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[doc] 测试用例调整

Browse Source
This commit is contained in:
what 2023-05-30 17:47:53 +08:00
parent 3d596598c6
commit 050c77a5f7
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
grpcall_test.go
View File

@ -4,6 +4,7 @@ import (
"fmt" "fmt"
"os" "os"
"testing" "testing"
"time"
"google.golang.org/grpc" "google.golang.org/grpc"
) )
@ -18,12 +19,11 @@ func TestMain(m *testing.M) {
gc = _gc gc = _gc
} }
defer gc.Close()
// 执行测试 // 执行测试
code := m.Run() code := m.Run()
// 在这里执行一些清理工作，例如关闭数据库连接等 // 在这里执行一些清理工作，例如关闭数据库连接等
gc.Close()
// 退出测试 // 退出测试
os.Exit(code) os.Exit(code)
@ -41,6 +41,7 @@ func TestGetServiceMethods(t *testing.T) {
} }
} }
// 数据请求测试
func TestInvokeUnary(t *testing.T) { func TestInvokeUnary(t *testing.T) {
if resp, err := gc.Invoke("User.UserResourceStatus", "TestInt64Value", `123`, nil); err != nil { if resp, err := gc.Invoke("User.UserResourceStatus", "TestInt64Value", `123`, nil); err != nil {
t.Error(err) t.Error(err)
@ -49,6 +50,7 @@ func TestInvokeUnary(t *testing.T) {
} }
} }
// 服务流测试
func TestInvokeServStream(t *testing.T) { func TestInvokeServStream(t *testing.T) {
if resp, err := gc.Invoke("User.UserResourceStatus", "GetEvent", `{}`, nil); err != nil { if resp, err := gc.Invoke("User.UserResourceStatus", "GetEvent", `{}`, nil); err != nil {
t.Error(err) t.Error(err)
@ -75,6 +77,8 @@ func TestInvokeServStream(t *testing.T) {
<-flag <-flag
} }
} }
// 双向流测试
func TestInvokeBidiStream(t *testing.T) { func TestInvokeBidiStream(t *testing.T) {
if resp, err := gc.Invoke("User.UserResourceStatus", "TestBidiStream", `"hello"`, nil); err != nil { if resp, err := gc.Invoke("User.UserResourceStatus", "TestBidiStream", `"hello"`, nil); err != nil {
t.Error(err) t.Error(err)
@ -98,6 +102,8 @@ func TestInvokeBidiStream(t *testing.T) {
} }
flag <- true flag <- true
return return
case <-time.After(time.Second):
resp.Cancel()
} }
} }
}() }()