mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-02 20:09:57 +08:00
0b905078e7
See also #27675 This PR make previously merged refactory of datanode go online - Use write node to replace insert/delete node - Use write buffer manager to control all buffers - Use sync manager to control sync tasks instead of flush manager Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package datanode
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/cockroachdb/errors"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/milvus-io/milvus/pkg/mq/msgstream"
|
|
"github.com/milvus-io/milvus/pkg/util/tsoutil"
|
|
)
|
|
|
|
type MqStatsUpdaterSuite struct {
|
|
suite.Suite
|
|
|
|
producer *msgstream.MockMsgStream
|
|
updater *mqStatsUpdater
|
|
}
|
|
|
|
func (s *MqStatsUpdaterSuite) SetupTest() {
|
|
s.producer = msgstream.NewMockMsgStream(s.T())
|
|
s.updater = &mqStatsUpdater{
|
|
stats: make(map[int64]int64),
|
|
producer: s.producer,
|
|
config: &nodeConfig{
|
|
vChannelName: "by-dev-rootcoord-dml_0v0",
|
|
},
|
|
}
|
|
}
|
|
|
|
func (s *MqStatsUpdaterSuite) TestSend() {
|
|
s.Run("send_ok", func() {
|
|
s.producer.EXPECT().Produce(mock.Anything).Return(nil)
|
|
|
|
s.updater.mut.Lock()
|
|
s.updater.stats[100] = 1024
|
|
s.updater.mut.Unlock()
|
|
|
|
err := s.updater.send(tsoutil.GetCurrentTime(), []int64{100})
|
|
s.NoError(err)
|
|
|
|
s.updater.mut.Lock()
|
|
_, has := s.updater.stats[100]
|
|
s.updater.mut.Unlock()
|
|
s.False(has)
|
|
})
|
|
|
|
s.Run("send_error", func() {
|
|
s.SetupTest()
|
|
s.producer.EXPECT().Produce(mock.Anything).Return(errors.New("mocked"))
|
|
|
|
s.updater.mut.Lock()
|
|
s.updater.stats[100] = 1024
|
|
s.updater.mut.Unlock()
|
|
|
|
err := s.updater.send(tsoutil.GetCurrentTime(), []int64{100})
|
|
s.Error(err)
|
|
})
|
|
}
|
|
|
|
func TestMqStatsUpdater(t *testing.T) {
|
|
suite.Run(t, new(MqStatsUpdaterSuite))
|
|
}
|