mirror of https://github.com/milvus-io/milvus.git
52 lines
956 B
Go
52 lines
956 B
Go
package proxynode
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zilliztech/milvus-distributed/internal/allocator"
|
|
"github.com/zilliztech/milvus-distributed/internal/msgstream/pulsarms"
|
|
)
|
|
|
|
var trueCnt = 0
|
|
|
|
func checkFunc(timestamp Timestamp) bool {
|
|
ret := rand.Intn(2) == 1
|
|
if ret {
|
|
trueCnt++
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func TestTimeTick_Start(t *testing.T) {
|
|
fmt.Println("HHH")
|
|
}
|
|
|
|
func TestTimeTick_Start2(t *testing.T) {
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
masterAddr := Params.MasterAddress
|
|
tsoAllocator, err := allocator.NewTimestampAllocator(ctx, masterAddr)
|
|
assert.Nil(t, err)
|
|
err = tsoAllocator.Start()
|
|
assert.Nil(t, err)
|
|
|
|
msFactory := pulsarms.NewFactory()
|
|
tt := newTimeTick(ctx, tsoAllocator, Params.TimeTickInterval, checkFunc, msFactory)
|
|
|
|
defer func() {
|
|
cancel()
|
|
tsoAllocator.Close()
|
|
tt.Close()
|
|
}()
|
|
|
|
tt.Start()
|
|
|
|
<-ctx.Done()
|
|
|
|
}
|