Function TestQueuePushPopSingleGoroutine

tsdb/chunks/queue_test.go:64–116

(t *testing.T)

}

func TestQueuePushPopSingleGoroutine(t *testing.T) {
	t.Parallel()
	seed := time.Now().UnixNano()
	t.Log("seed:", seed)
	r := rand.New(rand.NewSource(seed))

	const maxSize = 500
	const maxIters = 50

	for maxCount := 1; maxCount < maxSize; maxCount++ {
		queue := newWriteJobQueue(maxCount, 1+(r.Int()%maxCount))

		elements := 0 // total elements in the queue
		lastWriteID := 0
		lastReadID := 0

		for range maxIters {
			if elements < maxCount {
				toWrite := r.Int() % (maxCount - elements)
				if toWrite == 0 {
					toWrite = 1
				}

				for i := 0; i < toWrite; i++ {
					lastWriteID++
					require.True(t, queue.push(chunkWriteJob{seriesRef: HeadSeriesRef(lastWriteID)}))

					elements++
				}
			}

			if elements > 0 {
				toRead := r.Int() % elements
				if toRead == 0 {
					toRead = 1
				}

				for i := 0; i < toRead; i++ {
					lastReadID++

					j, b := queue.pop()
					require.True(t, b)
					require.Equal(t, HeadSeriesRef(lastReadID), j.seriesRef)

					elements--
				}
			}

			require.Equal(t, elements, queue.length())
			queue.assertInvariants(t)
		}
	}
}

func TestQueuePushBlocksOnFullQueue(t *testing.T) {
	t.Parallel()