package bhsgo

import (
	bh "bhshmq/proto/source/bhome_msg"
	"fmt"
	"testing"
	"time"
)

func TestRegister(t *testing.T) {
	proc_id := "test_proc"
	proc := bh.ProcInfo{}
	proc.ProcId = []byte(proc_id)
	reply := bh.MsgCommonReply{}

	StartWorker(ClientCallback, ServerCallback, SubDataCallback)

	r := Register(&proc, &reply, 1000)
	if r {
		fmt.Println("register ok")
	} else {
		fmt.Println("register failed")
		return
	}

	r = HeartbeatEasy(1000)
	if r {
		fmt.Println("heartbeat ok")
	} else {
		fmt.Println("heartbeat failed")
	}

	topics := bh.MsgTopicList{}
	topics.TopicList = append(topics.TopicList, []byte("topic0"), []byte("topic1"))
	RegisterTopics(&topics, &reply, 0)
	if r {
		fmt.Println("reg topics ok")
	} else {
		fmt.Println("reg topics failed")
	}
	req := bh.MsgRequestTopic{}
	time.Sleep(time.Second * 1)
	req.Topic = []byte("topic0")
	req.Data = []byte("data0")
	// var msg_id []byte
	// AsyncRequest(&req, &msg_id)
	// fmt.Println(msg_id)
	// time.Sleep(time.Second * 5)

	pid := ""
	rr := bh.MsgRequestTopicReply{}
	for i := 0; i < 10000; i++ {
		if Request(&req, &pid, &rr, 3000) {
			fmt.Println("server:" + pid + ", reply:" + string(rr.Data))
		} else {
			e, s := GetLastError()
			fmt.Println("ec:", e, ", msg:"+s)
		}
	}
}