milvus/tests-deprecating/go/insert_test.go

package test

import (
	//"context"
	//"sync"
	"testing"

	"github.com/milvus-io/milvus/internal/util/typeutil"
)

type Timestamp = typeutil.Timestamp

//func GetInsertMsg(collectionName string, partitionTag string, entityId UniqueID) *msgpb.InsertOrDeleteMsg {
//	return &msgpb.InsertOrDeleteMsg{
//		CollectionName: collectionName,
//		PartitionTag:   partitionTag,
//		SegmentID:      UniqueID(entityId / 100),
//		Uid:            UniqueID(entityId),
//		Timestamp:      Timestamp(entityId),
//		ClientId:       0,
//	}
//}
//
//func GetDeleteMsg(collectionName string, entityId UniqueID) *msgpb.InsertOrDeleteMsg {
//	return &msgpb.InsertOrDeleteMsg{
//		CollectionName: collectionName,
//		Uid:            entityId,
//		Timestamp:      Timestamp(entityId + 100),
//	}
//}

func TestInsert(t *testing.T) {
	// TODO: fix test
	//return
	//ctx := context.Background()
	//var topics []string
	//topics = append(topics, "test")
	//topics = append(topics, "test1")
	//writerNode, _ := writer.NewWriteNode(ctx, "null", topics, 0)
	//var insertMsgs []*msgpb.InsertOrDeleteMsg
	//for i := 0; i < 120; i++ {
	//	insertMsgs = append(insertMsgs, GetInsertMsg("collection0", "tag01", UniqueID(i)))
	//}
	//wg := sync.WaitGroup{}
	//wg.Add(3)
	////var wg sync.WaitGroup
	//writerNode.InsertBatchData(ctx, insertMsgs, &wg)
	//var insertMsgs2 []*msgpb.InsertOrDeleteMsg
	//for i := 120; i < 200; i++ {
	//	insertMsgs2 = append(insertMsgs2, GetInsertMsg("collection0", "tag02", UniqueID(i)))
	//}
	//writerNode.InsertBatchData(ctx, insertMsgs2, &wg)
	//var deleteMsgs []*msgpb.InsertOrDeleteMsg
	//deleteMsgs = append(deleteMsgs, GetDeleteMsg("collection0", 2))
	//deleteMsgs = append(deleteMsgs, GetDeleteMsg("collection0", 120))
	//writerNode.DeleteBatchData(ctx, deleteMsgs)
}