dispatcher_test.go

package gorpc

import (
	"bytes"
	"fmt"
	"io"
	"reflect"
	"sync/atomic"
	"testing"
	"time"
	"unsafe"
)

func TestDispatcherNewHandlerNoFuncs(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.NewHandlerFunc()
	})
}

func TestDispatcherNewFuncClientNoFuncs(t *testing.T) {
	c := NewTCPClient(getRandomAddr())

	d := NewDispatcher()
	testPanic(t, func() {
		d.NewFuncClient(c)
	})
}

func TestDispatcherNewServiceClientNoService(t *testing.T) {
	c := NewTCPClient(getRandomAddr())

	d := NewDispatcher()
	testPanic(t, func() {
		d.NewServiceClient("foobar", c)
	})
}

func TestDispatcherAddNonFunc(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foobar", 123)
	})
}

func TestDispatcherEmptyFuncName(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("", func() {})
	})
}

func TestDispatcherInvalidFirstArgType(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(clientAddr bool, request string) {})
	})
}

func TestDispatcherInvalidSecondResType(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (response int, err float64) { return })
	})
}

func TestDispatcherTooManyArgs(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(clientAddr string, foo, bar int) {})
	})
}

func TestDispatcherTooManyRes(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (response int, err error, foobar string) { return })
	})
}

func TestDispatcherFuncArg(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(req func()) {})
	})
}

func TestDispatcherChanArg(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(req chan int) {})
	})
}

func TestDispatcherInterfaceArg(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(req io.Reader) {})
	})
	testPanic(t, func() {
		d.AddFunc("foo", func(req interface{}) {})
	})
}

func TestDispatcherUnsafePointerArg(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(req unsafe.Pointer) {})
	})
}

func TestDispatcherFuncRes(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (res func()) { return })
	})
}

func TestDispatcherChanRes(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (res chan int) { return })
	})
}

func TestDispatcherInterfaceRes(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (res io.Reader) { return })
	})
	testPanic(t, func() {
		d.AddFunc("foo", func() (res interface{}) { return })
	})
}

func TestDispatcherUnsafePointerRes(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (res unsafe.Pointer) { return })
	})
}

func TestDispatcherStructWithInvalidFields(t *testing.T) {
	type InvalidMsg struct {
		B int
		A io.Reader
	}

	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(req *InvalidMsg) {})
	})
}

func TestDispatcherInvalidMap(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(req map[string]interface{}) {})
	})
}

func TestDispatcherPassStructArgByValue(t *testing.T) {
	type RequestType struct {
		a int
		B string
	}

	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(request RequestType) {})
	})
}

func TestDispatcherReturnStructResByValue(t *testing.T) {
	type ResponseType struct {
		A int
		b string
	}

	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (response ResponseType) { return })
	})
}

func TestDispatcherPassStructArgNoExportedFields(t *testing.T) {
	type RequestTypeNoExport struct {
		a int
		b string
	}

	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func(req *RequestTypeNoExport) {})
	})
}

func TestDispatcherReturnStructResNoExportedFields(t *testing.T) {
	type ResponseTypeNoExport struct {
		a int
		b string
	}

	d := NewDispatcher()
	testPanic(t, func() {
		d.AddFunc("foo", func() (resp *ResponseTypeNoExport) { return })
	})
}

func TestDispatcherMultipleFuncs(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("foo", func(a int) int { return a })
	d.AddFunc("bar", func(b string) string { return b })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqa := 4327
		res, err := dc.Call("foo", reqa)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		resa, ok := res.(int)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected int", res)
		}
		if resa != reqa {
			t.Fatalf("Unexpected response: %d. Expected %d", resa, reqa)
		}

		reqb := "aaa"
		if res, err = dc.Call("bar", reqb); err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		resb, ok := res.(string)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected string", res)
		}
		if resb != reqb {
			t.Fatalf("Unexpected response: [%s]. Expected [%s]", resb, reqb)
		}
	})
}

func TestDispatcherStructsWithIdenticalFields(t *testing.T) {
	type Struct1 struct {
		A int
	}
	type Struct2 struct {
		A int
	}

	d := NewDispatcher()

	d.AddFunc("foo", func(request *Struct1) *Struct2 {
		return &Struct2{
			A: request.A,
		}
	})

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqs := &Struct1{
			A: 12356,
		}
		res, err := dc.Call("foo", reqs)
		if err != nil {
			t.Fatalf("Unepxected error: [%s]", err)
		}
		ress, ok := res.(*Struct2)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected *Struct2", res)
		}
		if ress.A != reqs.A {
			t.Fatalf("Unexpected response: [%+v]. Expected [%+v]", ress, reqs)
		}
	})
}

func TestDispatcherInvalidArgType(t *testing.T) {
	d := NewDispatcher()
	d.AddFunc("foo", func(request string) {})
	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		res, err := dc.Call("foo", 1234)
		if err == nil {
			t.Fatalf("Expected non-nil error")
		}
		if res != nil {
			t.Fatalf("Expected nil response. Got %+v", res)
		}
	})
}

func TestDispatcherUnknownFuncCall(t *testing.T) {
	d := NewDispatcher()
	d.AddFunc("foo", func(request string) {})
	testDispatcherFunc(t, d, func(dc *DispatcherClient) { testUnknownFuncs(t, dc) })
}

func testUnknownFuncs(t *testing.T, dc *DispatcherClient) {
	res, err := dc.Call("UnknownFunc", 1234)
	if err == nil {
		t.Fatalf("Expected non-nil error")
	}
	if res != nil {
		t.Fatalf("Expected nil response. Got %+v", res)
	}

	res, err = dc.CallTimeout("foobar", 1234, time.Second)
	if err == nil {
		t.Fatalf("Expected non-nil error")
	}
	if res != nil {
		t.Fatalf("Expected nil response. Got %+v", res)
	}

	ar, err := dc.CallAsync("UnknownFunc", "foobar")
	if err != nil {
		t.Fatalf("Unexpected error: [%s]", err)
	}
	<-ar.Done
	if ar.Error == nil {
		t.Fatalf("Unexpected non-nil error")
	}
	if ar.Response != nil {
		t.Fatalf("Expected nil response. Got %+v", ar.Response)
	}

	if err = dc.Send("FoobarFunc", "aaa"); err != nil {
		t.Fatalf("Unepxected error: [%s]", err)
	}

	b := dc.NewBatch()
	br := b.Add("UnknownFunc", 123)
	b.AddSkipResponse("Aaaa", nil)
	if err = b.Call(); err != nil {
		t.Fatalf("Unexpected error: [%s]", err)
	}
	if br.Error == nil {
		t.Fatalf("Unexpected non-nil error")
	}
	if br.Response != nil {
		t.Fatalf("Expected nil response. Got %+v", br.Response)
	}
}

func TestDispatcherEchoFuncCall(t *testing.T) {
	d := NewDispatcher()
	d.AddFunc("Echo", func(request string) string { return request })
	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		res, err := dc.Call("Echo", "foobar")
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(string)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected string", ress)
		}
		if ress != "foobar" {
			t.Fatalf("Unexpected response: [%s]. Expected [foobar]", ress)
		}
	})
}

func TestDispatcherEchoGobEncoding(t *testing.T) {
	d := NewDispatcher()
	d.AddFunc("Echo", func(request *time.Time) *time.Time { return request })
	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		// time.Time implements gob encoding
		tt := time.Now()
		res, err := dc.Call("Echo", &tt)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		rest, ok := res.(*time.Time)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected %T", rest, &tt)
		}
		if *rest != tt {
			t.Fatalf("Unexpected response: [%#v]. Expected [%#v]", *rest, tt)
		}
	})
}

func TestDispatcherEchoEmptyStruct(t *testing.T) {
	d := NewDispatcher()
	d.AddFunc("Echo", func(request map[string]struct{}) map[string]struct{} { return request })
	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		m := map[string]struct{}{
			"foo": struct{}{},
			"bar": struct{}{},
		}
		res, err := dc.Call("Echo", m)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		resm, ok := res.(map[string]struct{})
		if !ok {
			t.Fatalf("Unexpected response type %T. Expected %T", resm, m)
		}
		if !reflect.DeepEqual(m, resm) {
			t.Fatalf("Unexpected response: [%#v]. Expected [%#v]", resm, m)
		}
	})
}

type sBinaryEnc struct {
	u uint32
	s string
}

func (s *sBinaryEnc) MarshalBinary() (b []byte, err error) {
	b = append(b, byte(s.u>>24), byte(s.u>>16), byte(s.u>>8), byte(s.u))
	b = append(b, s.s...)
	return b, nil
}

func (s *sBinaryEnc) UnmarshalBinary(b []byte) error {
	s.u = (uint32(b[0]) << 24) | (uint32(b[1]) << 16) | (uint32(b[2]) << 8) | uint32(b[3])
	s.s = string(b[4:])
	return nil
}

func TestDispatcherEchoBinaryEncoding(t *testing.T) {
	d := NewDispatcher()
	d.AddFunc("Echo", func(request *sBinaryEnc) *sBinaryEnc { return request })
	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		s := &sBinaryEnc{
			u: 1893243243,
			s: "foobarbaz",
		}
		res, err := dc.Call("Echo", s)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(*sBinaryEnc)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected %T", ress, s)
		}
		if *ress != *s {
			t.Fatalf("Unexpected response: [%#v]. Expected [%#v]", *ress, *s)
		}
	})
}

func TestDispatcherStructArgCall(t *testing.T) {
	type RequestArg struct {
		A int
		B string
	}

	type ResponseArg struct {
		C string
		D int
		T time.Time
	}

	tt := time.Now()
	d := NewDispatcher()
	d.AddFunc("fooBar", func(request *RequestArg) *ResponseArg {
		return &ResponseArg{
			C: request.B,
			D: request.A,
			T: tt,
		}
	})

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		// verify call by reference
		reqs := &RequestArg{
			A: 123,
			B: "7822",
		}
		res, err := dc.Call("fooBar", reqs)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(*ResponseArg)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected *ResponseArg", ress)
		}
		if ress.C != reqs.B || ress.D != reqs.A {
			t.Fatalf("Unexpected response: [%+v]. Expected &ResponseArg{C:%s, D:%d}", ress, reqs.B, reqs.A)
		}

		// verify call by value
		reqs.A = 7889
		reqs.B = "alkjjal"
		if res, err = dc.Call("fooBar", *reqs); err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		if ress, ok = res.(*ResponseArg); !ok {
			t.Fatalf("Unexpected response type: %T. Expected *ResponseArg", ress)
		}
		if ress.C != reqs.B || ress.D != reqs.A || ress.T != tt {
			t.Fatalf("Unexpected response: [%+v]. Expected &ResponseArg{C:%s, D:%d, T:%s}", ress, reqs.B, reqs.A, tt)
		}
	})
}

func TestDispatcherRecursiveStructArg(t *testing.T) {
	type RecMsg struct {
		A   int
		Rec *RecMsg
	}

	d := NewDispatcher()
	d.AddFunc("foo", func(req *RecMsg) *RecMsg { return req })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqs := &RecMsg{
			A: 1,
			Rec: &RecMsg{
				A: 2,
			},
		}
		res, err := dc.Call("foo", reqs)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(*RecMsg)
		if !ok {
			t.Fatalf("Unepxected response type: %T. Expected *RecMsg", ress)
		}
		if ress.A != reqs.A || ress.Rec.A != reqs.Rec.A {
			t.Fatalf("Unepxected respons: [%+v]. Expected [%+v]", ress, reqs)
		}
	})
}

func TestDispatcherMapArgCall(t *testing.T) {
	d := NewDispatcher()

	type MapT map[string]int
	d.AddFunc("foo", func(m MapT) MapT { return m })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqm := MapT{
			"foo": 1,
			"bar": 42,
		}
		res, err := dc.Call("foo", reqm)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		resm, ok := res.(MapT)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected %T", res, reqm)
		}
		if resm["foo"] != reqm["foo"] || resm["bar"] != reqm["bar"] {
			t.Fatalf("Unexpected response: [%+v]. Expected [%+v]", resm, reqm)
		}
	})
}

func TestDispatcherArrayArgCall(t *testing.T) {
	d := NewDispatcher()

	type ArrT [3]byte
	d.AddFunc("foo", func(m ArrT) ArrT { return m })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqm := ArrT{'a', 'b', 'c'}
		res, err := dc.Call("foo", reqm)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		resm, ok := res.(ArrT)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected %T", res, reqm)
		}
		if !bytes.Equal(resm[:], reqm[:]) {
			t.Fatalf("Unexpected response: [%+v]. Expected [%+v]", resm, reqm)
		}
	})
}

func TestDispatcherSliceArgCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("foo", func(m []byte) []byte { return m })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqm := []byte("foobar")
		res, err := dc.Call("foo", reqm)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		resm, ok := res.([]byte)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected %T", res, reqm)
		}
		if !bytes.Equal(resm, reqm) {
			t.Fatalf("Unexpected response: [%+v]. Expected [%+v]", resm, reqm)
		}
	})
}

func TestDispatcherNoArgNoResCall(t *testing.T) {
	d := NewDispatcher()

	noArgNoResCalls := 0
	d.AddFunc("NoArgNoRes", func() { noArgNoResCalls++ })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		N := 10
		for i := 0; i < N; i++ {
			res, err := dc.Call("NoArgNoRes", "ignoreThis")
			if err != nil {
				t.Fatalf("Unexpected error: [%s]", err)
			}
			if res != nil {
				t.Fatalf("Unexpected response: [%+v]", res)
			}
		}

		if noArgNoResCalls != N {
			t.Fatalf("Unepxected number of NoArgNoRes calls: %d. Expected %d", noArgNoResCalls, N)
		}
	})
}

func TestDispatcherOneArgNoResCall(t *testing.T) {
	d := NewDispatcher()

	serverS := 0
	clientS := 0
	d.AddFunc("OneArgNoRes", func(n int) { serverS += n })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		for i := 0; i < 10; i++ {
			res, err := dc.Call("OneArgNoRes", i)
			if err != nil {
				t.Fatalf("Unexpected error: [%s]", err)
			}
			if res != nil {
				t.Fatalf("Unexpected response: [%+v]", res)
			}
			clientS += i
		}

		if clientS != serverS {
			t.Fatalf("Unepxected serverS=%d. Expected %d", serverS, clientS)
		}
	})
}

func TestDispatcherTwoArgNoResCall(t *testing.T) {
	d := NewDispatcher()

	serverS := 0
	clientS := 0
	d.AddFunc("TwoArgNoRes", func(clientAddr string, n int) { serverS += n })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		for i := 0; i < 10; i++ {
			res, err := dc.Call("TwoArgNoRes", i)
			if err != nil {
				t.Fatalf("Unexpected error: [%s]", err)
			}
			if res != nil {
				t.Fatalf("Unexpected response: [%+v]", res)
			}
			clientS += i
		}

		if clientS != serverS {
			t.Fatalf("Unepxected serverS=%d. Expected %d", serverS, clientS)
		}
	})
}

func TestDispatcherNoArgErrorResCall(t *testing.T) {
	d := NewDispatcher()

	var returnErr error
	d.AddFunc("NoArgErrorRes", func() error { return returnErr })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		returnErr = nil
		res, err := dc.Call("NoArgErrorRes", nil)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		if res != nil {
			t.Fatalf("Unexpected response: [%+v]", res)
		}

		returnErr = fmt.Errorf("foobar")
		if res, err = dc.Call("NoArgErrorRes", nil); err == nil {
			t.Fatalf("Unexpected nil error")
		}
		if err.Error() != returnErr.Error() {
			t.Fatalf("Unexpected error: [%s]. Expected [%s]", err, returnErr)
		}
		if res != nil {
			t.Fatalf("Unexpected response: [%+v]", res)
		}
	})
}

func TestDispatcherOneArgErrorResCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("OneArgErrorRes", func(r string) error { return fmt.Errorf("%s", r) })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqs := "foobar"
		res, err := dc.Call("OneArgErrorRes", reqs)
		if err == nil {
			t.Fatalf("Unexpected nil error")
		}
		if err.Error() != reqs {
			t.Fatalf("Unexpected error: [%s]. Expected [%s]", err, reqs)
		}
		if res != nil {
			t.Fatalf("Unexpected response: [%+v]", res)
		}
	})
}

func TestDispatcherTwoArgErrorResCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("TwoArgErrorRes", func(clientAddr string, r int) error { return fmt.Errorf("%d", r) })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		res, err := dc.Call("TwoArgErrorRes", 123)
		if err == nil {
			t.Fatalf("Unexpected nil error")
		}
		if err.Error() != fmt.Sprintf("%d", 123) {
			t.Fatalf("Unexpected error: [%s]. Expected [123]", err)
		}
		if res != nil {
			t.Fatalf("Unexpected response: [%+v]", res)
		}
	})
}

func TestDispatcherNoArgOneResCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("NoArgOneResCall", func() string { return "foobar" })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		res, err := dc.Call("NoArgOneResCall", nil)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(string)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected string", res)
		}
		if ress != "foobar" {
			t.Fatalf("Unexpected response [%s]. Expected [foobar]", ress)
		}
	})
}

func TestDispatcherOneArgOneResCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("OneArgOneResCall", func(req int) int { return req })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqs := 42
		res, err := dc.Call("OneArgOneResCall", reqs)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(int)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected int", res)
		}
		if ress != reqs {
			t.Fatalf("Unexpected response [%d]. Expected [%d]", ress, reqs)
		}
	})
}

func TestDispatcherOneArgTwoResCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("OneArgTwoResCall", func(req int) (int, error) { return req, nil })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqs := 442
		res, err := dc.Call("OneArgTwoResCall", reqs)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(int)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected int", res)
		}
		if ress != reqs {
			t.Fatalf("Unexpected response [%d]. Expected [%d]", ress, reqs)
		}
	})
}

func TestDispatcherTwoArgOneResCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("TwoArgOneResCall", func(clientAddr string, req int) int { return req })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqs := 142
		res, err := dc.Call("TwoArgOneResCall", reqs)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(int)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected int", res)
		}
		if ress != reqs {
			t.Fatalf("Unexpected response [%d]. Expected [%d]", ress, reqs)
		}
	})
}

func TestDispatcherTwoArgTwoResCall(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("TwoArgTwoResCall", func(clientAddr string, req int) (int, error) { return req, nil })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		reqs := 1423
		res, err := dc.Call("TwoArgTwoResCall", reqs)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(int)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected int", res)
		}
		if ress != reqs {
			t.Fatalf("Unexpected response [%d]. Expected [%d]", ress, reqs)
		}
	})
}

func TestDispatcherSend(t *testing.T) {
	d := NewDispatcher()

	N := 10
	ch := make(chan struct{}, N)
	serverS := uint32(0)
	clientS := uint32(0)
	d.AddFunc("Sum", func(n int) {
		atomic.AddUint32(&serverS, uint32(n))
		ch <- struct{}{}
	})

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		for i := 0; i < N; i++ {
			if err := dc.Send("Sum", i); err != nil {
				t.Fatalf("Unexpected error in Send(): [%s]", err)
			}
			clientS += uint32(i)
		}
		for i := 0; i < N; i++ {
			<-ch
		}
		if atomic.LoadUint32(&serverS) != clientS {
			t.Fatalf("Unepxected serverS=%d. Should be %d", serverS, clientS)
		}
	})
}

func TestDispatcherSendWrongFunc(t *testing.T) {
	d := NewDispatcher()

	ch := make(chan struct{})
	d.AddFunc("Foo", func(n int) { close(ch) })

	N := 10
	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		for i := 0; i < N; i++ {
			if err := dc.Send("BarUnknown", i); err != nil {
				t.Fatalf("Unexpected error: [%s]", err)
			}
		}
		if err := dc.Send("Foo", 10); err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		<-ch
	})
}

func TestDispatcherCallAsync(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("aaa", func(x int) int { return x })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		N := 10
		ar := make([]*AsyncResult, N)
		var err error
		for i := 0; i < N; i++ {
			ar[i], err = dc.CallAsync("aaa", i)
			if err != nil {
				t.Fatalf("Unexpected error in CallAsync: [%s]", err)
			}
		}
		for i := 0; i < N; i++ {
			r := ar[i]
			<-r.Done
			if r.Error != nil {
				t.Fatalf("Unexpected error: [%s]", r.Error)
			}
			ress, ok := r.Response.(int)
			if !ok {
				t.Fatalf("Unexpected response type: %T. Expected int", r.Response)
			}
			if ress != i {
				t.Fatalf("Unexpected response: [%d]. Expected [%d]", ress, i)
			}
		}
	})
}

type testService struct{ state int }

func (s *testService) Inc()         { s.state++ }
func (s *testService) Add(n int)    { s.state += n }
func (s *testService) Get() int     { return s.state }
func (s *testService) privateFunc() {}

func TestDispatcherServicePassByValue(t *testing.T) {
	d := NewDispatcher()
	testPanic(t, func() {
		d.AddService("aaa", testService{})
	})
}

func TestDispatcherServiceWithoutName(t *testing.T) {
	service := &testService{}

	d := NewDispatcher()
	testPanic(t, func() {
		d.AddService("", service)
	})
}

type testServiceWithoutMethods struct{}

func (s *testServiceWithoutMethods) privateMethod1() {}
func (s *testServiceWithoutMethods) privateMethod2() {}

func TestDispatcherServiceWithoutPublicMethods(t *testing.T) {
	service := &testServiceWithoutMethods{}

	d := NewDispatcher()
	testPanic(t, func() {
		d.AddService("foobar", service)
	})
}

func TestDispatcherServiceUnknownService(t *testing.T) {
	service := &testService{}

	sd := NewDispatcher()
	sd.AddService("foobar", service)

	cd := NewDispatcher()
	cd.AddService("barbaz", service)

	c, s := getClientServer(t, sd)
	defer s.Stop()
	defer c.Stop()

	dc := cd.NewServiceClient("barbaz", c)

	res, err := dc.Call("Inc", nil)
	if err == nil {
		t.Fatalf("Error expected")
	}
	if res != nil {
		t.Fatalf("Unexpected response: [%+v]. Expected nil", res)
	}
}

func TestDispatcherServiceUnknownMethodCall(t *testing.T) {
	d := NewDispatcher()
	d.AddService("qwerty", &testService{})
	testDispatcherService(t, d, "qwerty", func(dc *DispatcherClient) { testUnknownFuncs(t, dc) })
}

func TestDispatcherServicePrivateMethodCall(t *testing.T) {
	service := &testService{}

	d := NewDispatcher()
	d.AddService("qwerty", service)

	testDispatcherService(t, d, "qwerty", func(dc *DispatcherClient) {
		res, err := dc.Call("privateFunc", nil)
		if err == nil {
			t.Fatalf("Error expected")
		}
		if res != nil {
			t.Fatalf("Expected nil response. Got %+v", res)
		}
	})
}

func TestDispatcherService(t *testing.T) {
	service := &testService{}

	d := NewDispatcher()
	d.AddService("qwerty", service)

	testDispatcherService(t, d, "qwerty", func(dc *DispatcherClient) {
		res, err := dc.Call("Add", 123)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		if res != nil {
			t.Fatalf("Unexpected response: [%+v]", res)
		}
		if service.state != 123 {
			t.Fatalf("Unexpected service state: %d. Expected 123", service.state)
		}

		if res, err = dc.Call("Inc", nil); err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		if res != nil {
			t.Fatalf("Unexpected response: [%+v]", res)
		}
		if service.state != 124 {
			t.Fatalf("Unexpected service state: %d. Expected 124", service.state)
		}

		if res, err = dc.Call("Get", nil); err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(int)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected int", res)
		}
		if ress != service.state {
			t.Fatalf("Unexpected response: [%d]. Expected [%d]", ress, service.state)
		}
	})
}

func TestDispatcherServiceMultiple(t *testing.T) {
	d := NewDispatcher()

	service1 := &testService{}
	service2 := &testService{}

	d.AddService("service1", service1)
	d.AddService("service2", service2)

	c, s := getClientServer(t, d)
	defer s.Stop()
	defer c.Stop()

	dc1 := d.NewServiceClient("service1", c)
	dc2 := d.NewServiceClient("service2", c)

	if _, err := dc1.Call("Inc", nil); err != nil {
		t.Fatalf("Unexpected error: [%s]", err)
	}
	if service1.state != 1 {
		t.Fatalf("Unexpected service1 state: %d. Expected 1", service1.state)
	}
	if service2.state != 0 {
		t.Fatalf("Unexpected service2 state: %d. Expected 0", service1.state)
	}

	if _, err := dc2.Call("Add", 42); err != nil {
		t.Fatalf("Unexpected error: [%s]", err)
	}
	if service1.state != 1 {
		t.Fatalf("Unexpected service1 state: %d. Expected 1", service1.state)
	}
	if service2.state != 42 {
		t.Fatalf("Unexpected service2 state: %d. Expected 42", service2.state)
	}
}

func TestDispatcherBatchSkipResponse(t *testing.T) {
	d := NewDispatcher()

	N := 10
	ch := make(chan struct{}, N+1)
	d.AddFunc("Echo", func(x int) int {
		ch <- struct{}{}
		return x
	})

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		b := dc.NewBatch()

		for i := 0; i < N; i++ {
			b.AddSkipResponse("Echo", i)
		}
		r := b.Add("Echo", 1234)

		select {
		case <-r.Done:
			t.Fatalf("<-Done must be blocked until DispatcherBatch.Call() is called")
		default:
		}

		if err := b.Call(); err != nil {
			t.Fatalf("Error in DispatcherBatch.Call(): [%s]", err)
		}

		select {
		case <-r.Done:
		default:
			t.Fatalf("<-Done must be unblocked after DispatcherBatch.Call()")
		}

		if r.Error != nil {
			t.Fatalf("Unexpected error after DispatcherBatch.Call(): [%s]", r.Error)
		}
		if r.Response.(int) != 1234 {
			t.Fatalf("Unexpected response returned: %+v. Expected 1234", r.Response)
		}

		for i := 0; i < N+1; i++ {
			<-ch
		}
	})
}

func TestDispatcherBatchMultiFunc(t *testing.T) {
	d := NewDispatcher()

	d.AddFunc("Echo", func(x int) int { return x })
	d.AddFunc("F42", func() int { return 42 })

	testDispatcherFunc(t, d, func(dc *DispatcherClient) {
		b := dc.NewBatch()

		N := 10
		echoResult := make([]*BatchResult, N)
		f42Result := make([]*BatchResult, N)
		for i := 0; i < N; i++ {
			echoResult[i] = b.Add("Echo", i)
			f42Result[i] = b.Add("F42", nil)

			select {
			case <-echoResult[i].Done:
				t.Fatalf("<-Done must be blocked until DispatcherBatch.Call() is called")
			default:
			}

			select {
			case <-f42Result[i].Done:
				t.Fatalf("<-Done must be blocked until DispatcherBatch.Call() is called")
			default:
			}
		}

		if err := b.Call(); err != nil {
			t.Fatalf("error in DispatcherBatch.Call(): [%s]", err)
		}

		for i := 0; i < N; i++ {
			r := echoResult[i]
			select {
			case <-r.Done:
			default:
				t.Fatalf("%d. <-Done must be unblocked after DispatcherBatch.Call()", i)
			}
			if r.Error != nil {
				t.Fatalf("%d. Unexpected error in DispatcherBatch result: [%s]", i, r.Error)
			}
			if r.Response.(int) != i {
				t.Fatalf("%d. Unexpected response in DispatcherBatch result: %+v. Expected %d", i, r.Response, i)
			}
		}

		for i := 0; i < N; i++ {
			r := f42Result[i]
			select {
			case <-r.Done:
			default:
				t.Fatalf("%d. <-Done must be unblocked after DispatcherBatch.Call()", i)
			}
			if r.Error != nil {
				t.Fatalf("%d. Unexpected error in DispatcherBatch result: [%s]", i, r.Error)
			}
			if r.Response.(int) != 42 {
				t.Fatalf("%d. Unexpected response in DispatcherBatch result: %+v. Expected 42", i, r.Response)
			}
		}
	})
}

type testNilService struct{}

func (s *testNilService) Foo(x int) int { return x }

func TestDispatcherNilService(t *testing.T) {
	d := NewDispatcher()

	var service *testNilService
	d.AddService("nil", service)

	testDispatcherService(t, d, "nil", func(dc *DispatcherClient) {
		res, err := dc.Call("Foo", 123)
		if err != nil {
			t.Fatalf("Unexpected error: [%s]", err)
		}
		ress, ok := res.(int)
		if !ok {
			t.Fatalf("Unexpected response type: %T. Expected int", res)
		}
		if ress != 123 {
			t.Fatalf("Unexpected response: %d. Expected %d", ress, 123)
		}
	})
}

func TestDispatcherFuncAndService(t *testing.T) {
	d := NewDispatcher()

	service := &testService{}
	d.AddService("foo", service)

	barCalls := 0
	d.AddFunc("bar", func() { barCalls++ })

	c, s := getClientServer(t, d)
	defer s.Stop()
	defer c.Stop()

	dcc := d.NewFuncClient(c)
	dcs := d.NewServiceClient("foo", c)

	res, err := dcc.Call("bar", nil)
	if err != nil {
		t.Fatalf("Unexpected error: [%s]", err)
	}
	if res != nil {
		t.Fatalf("Unexpected response: [%+v]", res)
	}
	if barCalls != 1 {
		t.Fatalf("Unexpected barCalls=%d. Expected 1", barCalls)
	}

	if res, err = dcs.Call("Add", 123); err != nil {
		t.Fatalf("Unexpected error: [%s]", err)
	}
	if res != nil {
		t.Fatalf("Unexpected response: [%+v]", res)
	}
	if service.state != 123 {
		t.Fatalf("Unexpected service state: %d. Expected 123", service.state)
	}
}

func testDispatcherService(t *testing.T, d *Dispatcher, serviceName string, f func(dc *DispatcherClient)) {
	c, s := getClientServer(t, d)
	defer s.Stop()
	defer c.Stop()

	dc := d.NewServiceClient(serviceName, c)
	f(dc)
}

func testDispatcherFunc(t *testing.T, d *Dispatcher, f func(dc *DispatcherClient)) {
	c, s := getClientServer(t, d)
	defer s.Stop()
	defer c.Stop()

	dc := d.NewFuncClient(c)
	f(dc)
}

func getClientServer(t *testing.T, d *Dispatcher) (c *Client, s *Server) {
	addr := "./dispatcher-test.sock"
	s = NewUnixServer(addr, d.NewHandlerFunc())
	if err := s.Start(); err != nil {
		t.Fatalf("Error when starting server: [%s]", err)
	}

	c = NewUnixClient(addr)
	c.Start()
	return
}

func testPanic(t *testing.T, f func()) {
	defer func() {
		if r := recover(); r != nil {
			return
		}
		t.Fatalf("Panic expected")
	}()
	f()
}