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()) }
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) }
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()) }
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()) }
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) }
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()) }
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) }
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) }
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) }
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) }
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()) }
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()) }
// 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()) }
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) }
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) }
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) }