Function TestRuler_TestShutdown

pkg/ruler/ruler_test.go:527–605

(t *testing.T)

}

func TestRuler_TestShutdown(t *testing.T) {
	tests := []struct {
		name       string
		shutdownFn func(*blockingQuerier, *Ruler)
	}{
		{
			name: "successful query after shutdown",
			shutdownFn: func(querier *blockingQuerier, ruler *Ruler) {
				// Wait query to start
				<-querier.queryStarted

				// The following cancel the context of the ruler service.
				ruler.StopAsync()

				// Simulate the completion of the query
				close(querier.queryBlocker)

				// Wait query to finish
				<-querier.queryFinished

				require.GreaterOrEqual(t, querier.successfulQueries.Load(), int64(1), "query failed to complete successfully failed to complete")
			},
		},
		{
			name: "query timeout while shutdown",
			shutdownFn: func(querier *blockingQuerier, ruler *Ruler) {
				// Wait query to start
				<-querier.queryStarted

				// The following cancel the context of the ruler service.
				ruler.StopAsync()

				// Wait query to finish
				<-querier.queryFinished

				require.Equal(t, querier.successfulQueries.Load(), int64(0), "query should not be succesfull")
			},
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			store := newMockRuleStore(mockRules, nil)
			cfg := defaultRulerConfig(t)
			mockQuerier := &blockingQuerier{
				queryBlocker:      make(chan struct{}),
				queryStarted:      make(chan struct{}),
				queryFinished:     make(chan struct{}),
				successfulQueries: atomic.NewInt64(0),
			}
			sleepQueriable := fixedQueryable(mockQuerier)

			d := &querier.MockDistributor{}

			d.On("QueryStream", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(
				&client.QueryStreamResponse{
					Chunkseries: []client.TimeSeriesChunk{},
				}, nil)