erpc_test.go

package erpc_test

import (
	"testing"
	"time"

	"github.com/henrylee2cn/erpc/v6"
)

func panic_call(erpc.CallCtx, *interface{}) (interface{}, *erpc.Status) {
	panic("panic_call")
}

func panic_push(erpc.PushCtx, *interface{}) *erpc.Status {
	panic("panic_push")
}

func TestPanic(t *testing.T) {
	srv := erpc.NewPeer(erpc.PeerConfig{
		CountTime:  true,
		ListenPort: 9090,
	})
	srv.RouteCallFunc(panic_call)
	srv.RoutePushFunc(panic_push)
	go srv.ListenAndServe()

	time.Sleep(2 * time.Second)

	cli := erpc.NewPeer(erpc.PeerConfig{})
	defer cli.Close()
	sess, stat := cli.Dial(":9090")
	if !stat.OK() {
		t.Fatal(stat)
	}
	stat = sess.Call("/panic/call", nil, nil).Status()
	if stat.OK() {
		t.Fatalf("/panic/call: expect error!")
	}
	t.Logf("/panic/call error: %v", stat)
	stat = sess.Push("/panic/push", nil)
	if !stat.OK() {
		t.Fatalf("/panic/push: expect ok!")
	}
	t.Logf("/panic/push: ok")
}