package grpcall

import (
	"fmt"
	"io"
	"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(ioutil.ReadAll(resp.Recv()))
		str := ""
		decoder := resp.RecvDecoder()
		t.Log(decoder.Decode(&str), str)
	}
}

// 服务流测试
func TestInvokeServStream(t *testing.T) {
	if resp, err := gc.Invoke("User.UserResourceStatus", "GetEvent", `{}`, nil); err != nil {
		t.Error(err)
	} else {
		defer resp.Close()
		decoder := resp.RecvDecoder()

		for {
			data := struct {
				Uuid string `json:"uuid"`
				Type string `json:"type"`
			}{}

			if err := decoder.Decode(&data); err != nil {
				if err != io.EOF && err != io.ErrClosedPipe {
					t.Error(err)
				}
				break
			}
			fmt.Println(data)
		}
	}
}

// 双向流测试
func TestInvokeBidiStream(t *testing.T) {
	if resp, err := gc.Invoke("User.UserResourceStatus", "TestBidiStream", `"hello"`, nil); err != nil {
		t.Error(err)
	} else {
		defer resp.Close()

		decoder := resp.RecvDecoder()

		go func() {
			for {
				data := struct {
					Uuid string `json:"uuid"`
					Type string `json:"type"`
				}{}

				if err := decoder.Decode(&data); err != nil {
					if err != io.EOF && err != io.ErrClosedPipe {
						t.Error(err)
					}
					return
				}
				fmt.Println(333, data)
			}
		}()

		for i := 0; i < 10; i++ {
			fmt.Println("消息发送, ", i, resp.Send(fmt.Sprintf(`"abc %d"`, i)))
		}

		time.Sleep(time.Second)
	}
}