Update storage unittest (#7637)

Signed-off-by: Nyhmo <yihua.mo@zilliz.com>

internal/storage/binlog_writer_test.go

@@ -22,6 +22,9 @@ import (
 
 func TestBinlogWriterReader(t *testing.T) {
 	binlogWriter := NewInsertBinlogWriter(schemapb.DataType_Int32, 10, 20, 30, 40)
+	tp := binlogWriter.GetBinlogType()
+	assert.Equal(t, tp, InsertBinlog)
+
 	binlogWriter.SetEventTimeStamp(1000, 2000)
 	defer binlogWriter.Close()
 	eventWriter, err := binlogWriter.NextInsertEventWriter()

internal/storage/data_sorter_test.go

@@ -234,3 +234,92 @@ func TestDataSorter(t *testing.T) {
 	assert.Equal(t, []byte{255, 0}, dataSorter.InsertData.Data[108].(*BinaryVectorFieldData).Data)
 	assert.Equal(t, []float32{8, 9, 10, 11, 12, 13, 14, 15, 0, 1, 2, 3, 4, 5, 6, 7}, dataSorter.InsertData.Data[109].(*FloatVectorFieldData).Data)
 }
+
+func TestDataSorter_Len(t *testing.T) {
+	insertData := &InsertData{
+		Data: map[int64]FieldData{
+			1: &Int64FieldData{
+				NumRows: []int64{2},
+				Data:    []int64{6, 4},
+			},
+		},
+	}
+
+	dataSorter := &DataSorter{
+		InsertCodec: nil,
+		InsertData:  insertData,
+	}
+
+	n := dataSorter.Len()
+	assert.Equal(t, n, 0)
+
+	insertData = &InsertData{
+		Data: map[int64]FieldData{
+			0: &Int8FieldData{
+				NumRows: []int64{2},
+				Data:    []int8{3, 4},
+			},
+		},
+	}
+
+	dataSorter = &DataSorter{
+		InsertCodec: nil,
+		InsertData:  insertData,
+	}
+
+	n = dataSorter.Len()
+	assert.Equal(t, n, 0)
+}
+
+func TestDataSorter_Less(t *testing.T) {
+	insertData := &InsertData{
+		Data: map[int64]FieldData{
+			1: &Int64FieldData{
+				NumRows: []int64{2},
+				Data:    []int64{6, 4},
+			},
+		},
+	}
+
+	dataSorter := &DataSorter{
+		InsertCodec: nil,
+		InsertData:  insertData,
+	}
+
+	res := dataSorter.Less(1, 2)
+	assert.True(t, res)
+
+	insertData = &InsertData{
+		Data: map[int64]FieldData{
+			0: &Int8FieldData{
+				NumRows: []int64{2},
+				Data:    []int8{3, 4},
+			},
+		},
+	}
+
+	dataSorter = &DataSorter{
+		InsertCodec: nil,
+		InsertData:  insertData,
+	}
+
+	res = dataSorter.Less(1, 2)
+	assert.True(t, res)
+
+	insertData = &InsertData{
+		Data: map[int64]FieldData{
+			0: &Int64FieldData{
+				NumRows: []int64{2},
+				Data:    []int64{6, 4},
+			},
+		},
+	}
+
+	dataSorter = &DataSorter{
+		InsertCodec: nil,
+		InsertData:  insertData,
+	}
+
+	res = dataSorter.Less(-1, -2)
+	assert.True(t, res)
+}

internal/storage/event_writer_test.go

@@ -20,6 +20,16 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestEventTypeCode_String(t *testing.T) {
+	var code EventTypeCode = 127
+	res := code.String()
+	assert.Equal(t, res, "InvalidEventType")
+
+	code = DeleteEventType
+	res = code.String()
+	assert.Equal(t, res, "DeleteEventType")
+}
+
 func TestSizeofStruct(t *testing.T) {
 	var buf bytes.Buffer
 	err := binary.Write(&buf, binary.LittleEndian, baseEventHeader{})
@@ -74,5 +84,4 @@ func TestEventWriter(t *testing.T) {
 	assert.EqualValues(t, length, buffer.Len())
 	err = insertEvent.Close()
 	assert.Nil(t, err)
-
 }

internal/storage/stats_test.go

@@ -17,7 +17,7 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-func TestStatsInt64(t *testing.T) {
+func TestStatsWriter_StatsInt64(t *testing.T) {
 	data := []int64{1, 2, 3, 4, 5, 6, 7, 8, 9}
 	sw := &StatsWriter{}
 	err := sw.StatsInt64(data)
@@ -34,4 +34,8 @@ func TestStatsInt64(t *testing.T) {
 		Min: 1,
 	}
 	assert.Equal(t, stats, expectedStats)
+
+	msgs := []int64{}
+	err = sw.StatsInt64(msgs)
+	assert.Nil(t, err)
 }