mirror of https://github.com/milvus-io/milvus.git
31 lines
694 B
Go
31 lines
694 B
Go
package msgdispatcher
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/milvus-io/milvus-proto/go-api/v2/msgpb"
|
|
"github.com/milvus-io/milvus/pkg/log"
|
|
"github.com/milvus-io/milvus/pkg/mq/msgstream"
|
|
"github.com/milvus-io/milvus/pkg/util/paramtable"
|
|
)
|
|
|
|
func TestSendTimeout(t *testing.T) {
|
|
target := newTarget("test1", &msgpb.MsgPosition{}, nil)
|
|
|
|
time.Sleep(paramtable.Get().MQCfg.MaxTolerantLag.GetAsDuration(time.Second))
|
|
|
|
counter := 0
|
|
for i := 0; i < 10; i++ {
|
|
err := target.send(&msgstream.MsgPack{})
|
|
if err != nil {
|
|
log.Error("send failed", zap.Int("idx", i), zap.Error(err))
|
|
counter++
|
|
}
|
|
}
|
|
assert.Equal(t, counter, 0)
|
|
}
|