grpcall/grpcall_test.go

package grpcall

import (
	"fmt"
	"os"
	"testing"
	"time"

	"google.golang.org/grpc"
)

var gc *Grpcall

func TestMain(m *testing.M) {
	if _gc, err := NewGrpcall("127.0.0.1:8888", nil, grpc.WithInsecure()); err != nil {
		// if _gc, err := NewGrpcall("192.168.0.6:6644", nil, grpc.WithInsecure()); err != nil {
		panic(err)
	} else {
		gc = _gc
	}

	// 执行测试
	code := m.Run()

	// 在这里执行一些清理工作，例如关闭数据库连接等
	gc.Close()

	// 退出测试
	os.Exit(code)
}

func TestGetServices(t *testing.T) {
	t.Log(gc.GetServices())
}

func TestGetServiceMethods(t *testing.T) {
	if s, err := gc.GetServices(); err != nil {
		t.Error(err)
	} else {
		t.Log(gc.GetServiceMethods(s[0]))
	}
}

// 数据请求测试
func TestInvokeUnary(t *testing.T) {
	if resp, err := gc.Invoke("User.UserResourceStatus", "TestInt64Value", `123`, nil); err != nil {
		t.Error(err)
	} else {
		t.Log(resp.Data())
	}
}

// 服务流测试
func TestInvokeServStream(t *testing.T) {
	if resp, err := gc.Invoke("User.UserResourceStatus", "GetEvent", `{}`, nil); err != nil {
		t.Error(err)
	} else if recv, err := resp.Recv(); err != nil {
		t.Error(err)
	} else if done, err := resp.Done(); err != nil {
		t.Error(err)
	} else {
		flag := make(chan bool)
		go func() {
			for {
				select {
				case msg := <-recv:
					fmt.Println("msg", msg, resp.Header())
				case err := <-done:
					if err != nil {
						t.Error(err)
					}
					flag <- true
					return
				}
			}
		}()
		<-flag
	}
}

// 双向流测试
func TestInvokeBidiStream(t *testing.T) {
	if resp, err := gc.Invoke("User.UserResourceStatus", "TestBidiStream", `"hello"`, nil); err != nil {
		t.Error(err)
	} else if send, err := resp.Send(); err != nil {
		t.Error(err)
	} else if recv, err := resp.Recv(); err != nil {
		t.Error(err)
	} else if done, err := resp.Done(); err != nil {
		t.Error(err)
	} else {
		flag := make(chan bool)

		go func() {
			timer := time.NewTimer(time.Second)
			defer timer.Stop()

			for {
				select {
				case msg := <-recv:
					fmt.Println("msg", msg, resp.Header())
				case err := <-done:
					if err != nil {
						t.Error(err)
					}
					flag <- true
					return
				case <-timer.C:
					resp.Cancel()
					timer.Reset(time.Second)
				}

			}
		}()

		for i := 0; i < 10; i++ {
			send <- fmt.Sprintf(`"abc %d"`, i)
		}

		<-flag
	}
}