fixed tests

Author: rekby

internal/grpcwrapper/rawtopic/rawtopicwriter/messages.go

@@ -184,6 +184,12 @@ func (r *WriteRequest) Size() int {
 	return 0
 }
 
+func (r *WriteRequest) FillCache() *WriteRequest {
+	r.Size()
+
+	return r
+}
+
 func (r *WriteRequest) Cut(count int) (head *WriteRequest, rest *WriteRequest) {
 	if count >= len(r.Messages) {
 		return r, nil

internal/topic/topicwriterinternal/writer_reconnector_test.go

@@ -127,7 +127,7 @@ func TestWriterImpl_Write(t *testing.T) {
 			const seqNo = 31
 
 			writeMessageReceived := make(empty.Chan)
-			e.stream.EXPECT().Send(&rawtopicwriter.WriteRequest{
+			e.stream.EXPECT().Send((&rawtopicwriter.WriteRequest{
 				Messages: []rawtopicwriter.MessageData{
 					{
 						SeqNo:            seqNo,
@@ -138,7 +138,7 @@ func TestWriterImpl_Write(t *testing.T) {
 					},
 				},
 				Codec: rawtopiccommon.CodecRaw,
-			}).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
+			}).FillCache()).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
 				close(writeMessageReceived)
 
 				return nil
@@ -523,24 +523,24 @@ func TestWriterImpl_Reconnect(t *testing.T) {
 		}
 
 		strm2 := newStream("strm2")
-		strm2.EXPECT().Send(&rawtopicwriter.WriteRequest{
+		strm2.EXPECT().Send((&rawtopicwriter.WriteRequest{
 			Messages: []rawtopicwriter.MessageData{
 				{SeqNo: 1},
 			},
 			Codec: rawtopiccommon.CodecRaw,
-		}).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
+		}).FillCache()).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
 			t.Logf("strm2 sent message and return retriable error")
 
 			return xerrors.Retryable(errors.New("retriable on strm2"))
 		})
 
 		strm3 := newStream("strm3")
-		strm3.EXPECT().Send(&rawtopicwriter.WriteRequest{
+		strm3.EXPECT().Send((&rawtopicwriter.WriteRequest{
 			Messages: []rawtopicwriter.MessageData{
 				{SeqNo: 1},
 			},
 			Codec: rawtopiccommon.CodecRaw,
-		}).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
+		}).FillCache()).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
 			t.Logf("strm3 sent message and return unretriable error")
 
 			return errors.New("strm3")
@@ -599,7 +599,7 @@ func TestWriterImpl_CloseWithFlush(t *testing.T) {
 		const seqNo = 36
 
 		writeCompleted := make(empty.Chan)
-		e.stream.EXPECT().Send(&rawtopicwriter.WriteRequest{
+		e.stream.EXPECT().Send((&rawtopicwriter.WriteRequest{
 			Messages: []rawtopicwriter.MessageData{
 				{
 					SeqNo:            seqNo,
@@ -610,7 +610,7 @@ func TestWriterImpl_CloseWithFlush(t *testing.T) {
 				},
 			},
 			Codec: rawtopiccommon.CodecRaw,
-		}).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
+		}).FillCache()).DoAndReturn(func(_ rawtopicwriter.ClientMessage) error {
 			close(writeCompleted)
 
 			return nil
@@ -720,7 +720,7 @@ func TestCreateRawMessageData(t *testing.T) {
 		req, err := createWriteRequest(newTestMessagesWithContent(), rawtopiccommon.CodecRaw)
 		require.NoError(t, err)
 		require.Equal(t,
-			rawtopicwriter.WriteRequest{
+			&rawtopicwriter.WriteRequest{
 				Messages: []rawtopicwriter.MessageData{},
 				Codec:    rawtopiccommon.CodecRaw,
 			},
@@ -740,7 +740,7 @@ func TestCreateRawMessageData(t *testing.T) {
 		})
 
 		require.NoError(t, err)
-		require.Equal(t, rawtopicwriter.WriteRequest{
+		require.Equal(t, &rawtopicwriter.WriteRequest{
 			Messages: []rawtopicwriter.MessageData{
 				{
 					SeqNo: 1,
@@ -763,7 +763,7 @@ func TestCreateRawMessageData(t *testing.T) {
 		req, err := createWriteRequest(newTestMessagesWithContent(1, 2, 3), rawtopiccommon.CodecRaw)
 		require.NoError(t, err)
 		require.Equal(t,
-			rawtopicwriter.WriteRequest{
+			&rawtopicwriter.WriteRequest{
 				Messages: []rawtopicwriter.MessageData{
 					{
 						SeqNo: 1,

tests/integration/topic_read_writer_test.go

@@ -18,9 +18,6 @@ import (
 	"testing"
 	"time"
 
-	"github.com/ydb-platform/ydb-go-sdk/v3/query"
-	"github.com/ydb-platform/ydb-go-sdk/v3/topic/topicreader"
-
 	"github.com/stretchr/testify/require"
 	"google.golang.org/grpc"
 
@@ -30,7 +27,9 @@ import (
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/version"
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/xtest"
 	"github.com/ydb-platform/ydb-go-sdk/v3/log"
+	"github.com/ydb-platform/ydb-go-sdk/v3/query"
 	"github.com/ydb-platform/ydb-go-sdk/v3/topic/topicoptions"
+	"github.com/ydb-platform/ydb-go-sdk/v3/topic/topicreader"
 	"github.com/ydb-platform/ydb-go-sdk/v3/topic/topicsugar"
 	"github.com/ydb-platform/ydb-go-sdk/v3/topic/topictypes"
 	"github.com/ydb-platform/ydb-go-sdk/v3/topic/topicwriter"