Beispiel #1
0
func WriteCheck(t *testing.T, expect []byte, box tl.Box) {
	var buff = new(bytes.Buffer)
	enc := bin.NewEncoder(buff)

	assert.NoError(t, box.Write(enc))
	assert.Equal(t, expect, buff.Bytes())
}
Beispiel #2
0
func Test_WriteDouble(t *testing.T) {
	data := []byte{0xbf, 0x7d, 0x1d, 0x38, 0x77, 0x37, 0xb6, 0x40}
	var val float64 = 5687.4657

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, proto.WriteDouble(val))
	assert.Equal(t, buff.Bytes(), data)
}
Beispiel #3
0
func Test_BytesBox_Write(t *testing.T) {
	data := []byte{0x9f, 0x37, 0xe1, 0xe, 0x4, 0x74, 0x65, 0x73, 0x74, 0x0, 0x0, 0x0}
	i := builtin.NewBytes([]byte("test"))

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, i.Box().Write(proto))
	assert.Equal(t, data, buff.Bytes())
}
Beispiel #4
0
func Test_StringBox_Write(t *testing.T) {
	data := []byte{0x24, 0x6e, 0x28, 0xb5, 0x4, 0x74, 0x65, 0x73, 0x74, 0x0, 0x0, 0x0}
	i := builtin.NewString("test")

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, i.Box().Write(proto))
	assert.Equal(t, data, buff.Bytes())
}
Beispiel #5
0
func Test_WriteStringLong(t *testing.T) {
	var val string = strings.Repeat("tes", 101)
	var data []byte = []byte{0xfe, 0x2f, 0x1, 0x0, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x74, 0x65, 0x73, 0x0}

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, proto.WriteString(val))
	assert.Equal(t, buff.Bytes(), data)
}
Beispiel #6
0
func Test_IntBox_Write(t *testing.T) {
	data := []byte{0xda, 0x9b, 0x50, 0xa8, 0xda, 0xbc, 0x34, 0x13}
	i := builtin.NewInt(322223322)

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, i.Box().Write(proto))
	assert.Equal(t, data, buff.Bytes())
}
Beispiel #7
0
func Test_WriteInt(t *testing.T) {
	data := []byte{0x59, 0x1, 0x0, 0x0}
	var val int32 = 345

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, proto.WriteInt(val))
	assert.Equal(t, buff.Bytes(), data)
}
Beispiel #8
0
func Test_WriteBytesShort(t *testing.T) {
	var val string = strings.Repeat("test", 31)
	var data []byte = []byte{0x7c, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x74, 0x65, 0x73, 0x74, 0x0, 0x0, 0x0}

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, proto.WriteBytes([]byte(val)))
	assert.Equal(t, buff.Bytes(), data)
}
Beispiel #9
0
func Test_WriteLong(t *testing.T) {
	data := []byte{0x24, 0x46, 0x49, 0x3c, 0x47, 0x5, 0x2, 0x0}
	var val int64 = 568753465673252

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, proto.WriteLong(val))
	assert.Equal(t, buff.Bytes(), data)
}
Beispiel #10
0
func Test_WriteTypeBegin(t *testing.T) {
	data := []byte{0x7f, 0x2, 0xe2, 0x5c}
	var val uint32 = 0x5ce2027f

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, proto.WriteTypeBegin(val))
	assert.Equal(t, buff.Bytes(), data)
}
Beispiel #11
0
func Test_LongBox_Write(t *testing.T) {
	data := []byte{0xba, 0x6c, 0x7, 0x22, 0xda, 0xbc, 0x34, 0x13, 0x0, 0x0, 0x0, 0x0}
	i := builtin.NewLong(322223322)

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, i.Box().Write(proto))
	assert.Equal(t, data, buff.Bytes())
}
Beispiel #12
0
func Test_DoubleBox_Write(t *testing.T) {
	data := []byte{0x54, 0xc1, 0x10, 0x22, 0xef, 0xc9, 0xc3, 0x42, 0x95, 0x77, 0xdf, 0x40}
	i := builtin.NewDouble(32222.3322)

	var buff = new(bytes.Buffer)
	proto := bin.NewEncoder(buff)

	assert.NoError(t, i.Box().Write(proto))
	assert.Equal(t, data, buff.Bytes())
}
Beispiel #13
0
// Test request handling with RPCResult.
func Test_Processor_WithRPCResult(t *testing.T) {
	proc := mock.NewProcessorFactory(
		&TestHandler{},
		bin.NewEncoderFactory(),
		bin.NewDecoderFactory(),
		dummy.NewInternalWaitress(time.Minute*5),
		tl.NewBaseIDBrokerFactory(0),
	).GetProcessor(context.TODO(), 10)

	// Prepare request
	reqEncoder := bin.NewEncoder(new(bytes.Buffer))
	conn, _ := uuid.NewV4()
	session, _ := uuid.NewV4()
	reqMsgId, _ := proc.NewMessageID(*session, time.Now(), 0)
	assert.NoError(t, (&mock.FuncEmpty{}).Box().Write(reqEncoder))
	res := &testProcessorResult{}

	// Fire request
	err := proc.Process(tl.RequestContext{*conn, *session, reqMsgId, 3},
		reqEncoder.Bytes(), res)
	assert.NoError(t, err)

	assert.Equal(t, len(res.requests), 0, "Requests should be zero")
	assert.Equal(t, len(res.responses), 2,
		"Responses should be one ack and one RpcResult")

	// Check ack response
	checkAck(t, proc, res.responses[0].ResponseContext, res.responses[0].data)

	// check result with temporary waitress
	resWaitress := dummy.NewInternalWaitress(time.Minute * 5)
	assert.NoError(t, resWaitress.WaitResult(
		*session, reqMsgId, mock.TL_ID_FUNC_EMPTY_RES))

	resResCtx, resMsgData := res.responses[1].ResponseContext, res.responses[1].data
	assert.EqualValues(t, 5, resResCtx.SeqNo)
	assert.EqualValues(t, 1, resResCtx.MessageID&1)
	resBox := &builtin.ObjectBox{}
	assert.NoError(t,
		resBox.Read(bin.NewDecoder(bytes.NewBuffer(resMsgData)),
			resWaitress, *session))
	resBare, err := resBox.Bare()
	assert.NoError(t, err)
	assert.EqualValues(t, reqMsgId, resBare.(*builtin.RpcResult).ReqMsgId)

	// Waitress should wait for one ack
	waitressRes, err := proc.GetAck(*session, resResCtx.MessageID)
	assert.NoError(t, err)
	assert.False(t, waitressRes)
}
func Test_RPCResult_Write1(t *testing.T) {
	m := &builtin.RpcResult{
		ReqMsgId: 345,
		Result:   &MockRPCResult{[]int32{34, 35}}}
	buff := bytes.NewBuffer([]byte{})
	encoder := bin.NewEncoder(buff)

	assert.NoError(t, m.Write(encoder))
	assert.Equal(t, []byte{
		0x59, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
		0x15, 0xc4, 0xb5, 0x1c,
		0x2, 0x0, 0x0, 0x0,
		0x22, 0x0, 0x0, 0x0,
		0x23, 0x0, 0x0, 0x0}, buff.Bytes())
}
func Test_RPCResultBox_WriteError1(t *testing.T) {
	m := &builtin.RpcResult{
		ReqMsgId: 345,
		RpcError: &builtin.RpcError{34, "test"},
	}
	buff := bytes.NewBuffer([]byte{})
	encoder := bin.NewEncoder(buff)

	assert.NoError(t, m.Box().Write(encoder))
	assert.Equal(t, []byte{
		0x1, 0x6d, 0x5c, 0xf3,
		0x59, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0,
		0x19, 0xca, 0x44, 0x21,
		0x22, 0x0, 0x0, 0x0,
		0x4, 0x74, 0x65, 0x73,
		0x74, 0x0, 0x0, 0x0}, buff.Bytes())
}
Beispiel #16
0
func Test_ProcessorAck1(t *testing.T) {
	proc := mock.NewProcessorFactory(
		&TestHandler{},
		bin.NewEncoderFactory(),
		bin.NewDecoderFactory(),
		dummy.NewInternalWaitress(time.Minute*5),
		tl.NewBaseIDBrokerFactory(0),
	).GetProcessor(context.TODO(), 10)

	// Prepare request
	reqEncoder := bin.NewEncoder(new(bytes.Buffer))
	conn, _ := uuid.NewV4()
	session, _ := uuid.NewV4()

	ackMsgId1, _ := proc.NewMessageID(*session, time.Now(), 1)
	ackMsgId2, _ := proc.NewMessageID(*session, time.Now(), 1)
	reqMsgId, err := proc.NewMessageID(*session, time.Now(), 0)
	assert.NoError(t, (&mock.MsgsAck{[]int64{
		ackMsgId1,
		ackMsgId2,
	}}).Box().Write(reqEncoder))
	res := &testProcessorResult{}

	// Register two fake acks in waitress
	assert.NoError(t, proc.WaitAck(*session, ackMsgId1))
	assert.NoError(t, proc.WaitAck(*session, ackMsgId2))

	// Fire request
	err = proc.Process(tl.RequestContext{*conn, *session, reqMsgId, 2},
		reqEncoder.Bytes(), res)
	assert.NoError(t, err)

	assert.Empty(t, res.requests, "requests should be empty")
	assert.Empty(t, res.responses, "responses should be empty")

	// check acks in waitress
	ackRes1, err := proc.GetAck(*session, ackMsgId1)
	assert.NoError(t, err)
	assert.True(t, ackRes1)
	ackRes2, err := proc.GetAck(*session, ackMsgId2)
	assert.NoError(t, err)
	assert.True(t, ackRes2)
}
Beispiel #17
0
func Test_Processor_MessageContaitner(t *testing.T) {
	proc := mock.NewProcessorFactory(
		&TestHandler{},
		bin.NewEncoderFactory(),
		bin.NewDecoderFactory(),
		dummy.NewInternalWaitress(time.Minute*5),
		tl.NewBaseIDBrokerFactory(0),
	).GetProcessor(context.TODO(), 10)

	// Prepare request
	reqEncoder := bin.NewEncoder(new(bytes.Buffer))
	conn, _ := uuid.NewV4()
	session, _ := uuid.NewV4()
	reqMsgId, _ := proc.NewMessageID(*session, time.Now(), 0)
	assert.NoError(t, (&mock.MsgContainer{
		[]mock.Message{
			mock.Message{34, 3, 67,
				*(builtin.NewObject(&mock.FuncEmpty{}))},
		}}).Box().Write(reqEncoder))
	res := &testProcessorResult{}

	// fire
	err := proc.Process(
		tl.RequestContext{*conn, *session, reqMsgId, 2},
		reqEncoder.Bytes(),
		res)
	assert.NoError(t, err)
	assert.Empty(t, res.responses, "responses should be empty")

	// Check requests
	// check result
	resResCtx, resMsgData := res.requests[0].RequestContext, res.requests[0].data
	assert.EqualValues(t, 3, resResCtx.SeqNo)
	assert.EqualValues(t, 34, 34)
	resBox := &builtin.ObjectBox{}
	assert.NoError(t,
		resBox.Read(bin.NewDecoder(bytes.NewBuffer(resMsgData)),
			proc, *session))
	resBare, err := resBox.Bare()
	assert.NoError(t, err)
	assert.EqualValues(t, &mock.FuncEmpty{}, resBare)
}
Beispiel #18
0
func Test_Write_SingleMultiInt32R4_BadN(t *testing.T) {
	what := &mock.SingleMultiInt32R4{[]int32{1, 2}}
	proto := bin.NewEncoder(new(bytes.Buffer))
	err := what.Write(proto)
	assert.Error(t, err)
}