Function TestDatabase_WaitFlushMetaCompleted

tsdb/database_test.go:435–473

(t *testing.T)

}

func TestDatabase_WaitFlushMetaCompleted(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	now := commontimeutil.Now()
	metaDB := memdb.NewMockMetadataDatabase(ctrl)
	db := &database{
		memMetaDB:      metaDB,
		isFlushing:     *atomic.NewBool(false),
		flushCondition: sync.NewCond(&sync.Mutex{}),
		statistics:     metrics.NewDatabaseStatistics("test"),
	}

	metaDB.EXPECT().Notify(gomock.Any()).DoAndReturn(func(event any) {
		time.Sleep(100 * time.Millisecond)
		mn := event.(*memdb.FlushEvent)
		mn.Callback(nil)
	})
	var wait sync.WaitGroup
	wait.Add(2)
	ch := make(chan struct{})
	go func() {
		ch <- struct{}{}
		err := db.FlushMeta()
		assert.NoError(t, err)
	}()
	<-ch
	time.Sleep(10 * time.Millisecond)
	go func() {
		db.WaitFlushMetaCompleted()
		wait.Done()
	}()
	go func() {
		db.WaitFlushMetaCompleted()
		wait.Done()
	}()
	wait.Wait()
	assert.True(t, commontimeutil.Now()-now >= 100*time.Millisecond.Milliseconds())
}

func TestDatabase_Drop(t *testing.T) {
	ctrl := gomock.NewController(t)