github.com/m3db/m3@v1.5.1-0.20231129193456-75a402aa583b/src/query/graphite/ts/sortable_series_test.go

// Copyright (c) 2019 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

package ts

import (
	"math"
	"math/rand"
	"sort"
	"testing"
	"time"

	"github.com/m3db/m3/src/query/graphite/context"
	xtest "github.com/m3db/m3/src/query/graphite/testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

type testSeries struct {
	name string
	data []float64
}

type testSortData struct {
	inputs []testSeries
	output []testSeries
}

func newTestSeriesValues(ctx context.Context, millisPerStep int, values []float64) Values {
	tsv := NewValues(ctx, millisPerStep, len(values))

	for i, n := range values {
		tsv.SetValueAt(i, n)
	}

	return tsv
}

func newTestSeriesList(ctx context.Context, start time.Time, inputs []testSeries, step int) []*Series {
	seriesList := make([]*Series, 0, len(inputs))

	for _, in := range inputs {
		series := NewSeries(ctx, in.name, start, newTestSeriesValues(ctx, step, in.data))
		seriesList = append(seriesList, series)
	}

	return seriesList
}

func validateOutputs(t *testing.T, step int, start time.Time, expected []testSeries, actual SeriesList) {
	require.Equal(t, len(expected), len(actual.Values))

	for i := range expected {
		a, e := actual.Values[i], expected[i].data

		require.Equal(t, len(e), a.Len())

		for step := 0; step < a.Len(); step++ {
			v := a.ValueAt(step)
			xtest.Equalish(t, e[step], v, "invalid value for %d", step)
		}

		assert.Equal(t, expected[i].name, a.Name())
		assert.Equal(t, step, a.MillisPerStep())
		assert.Equal(t, start, a.StartTime())
	}
}

func testSortImpl(ctx context.Context, t *testing.T, tests []testSortData, sr SeriesReducer, dir Direction) {
	var (
		startTime = time.Now()
		step      = 100
	)

	for _, test := range tests {
		series := newTestSeriesList(ctx, startTime, test.inputs, step)

		output, err := SortSeries(NewSeriesListWithSeries(series...), sr, dir)

		require.NoError(t, err)
		validateOutputs(t, step, startTime, test.output, output)
	}
}

func TestSortSeries(t *testing.T) {
	ctx := context.New()
	defer func() { _ = ctx.Close() }()

	testInput := []testSeries{
		{"foo", []float64{0, 601, 3, 4}},
		{"nan", []float64{math.NaN(), math.NaN(), math.NaN()}},
		{"bar", []float64{500, -8}},
		{"baz", []float64{600, -600, 3}},
		{"qux", []float64{100, 50000, 888, -1, -2}},
	}

	testSortImpl(ctx, t, []testSortData{
		{testInput, []testSeries{testInput[4], testInput[2], testInput[0], testInput[3], testInput[1]}},
	}, SeriesReducerAvg.Reducer(), Descending)

	testSortImpl(ctx, t, []testSortData{
		{testInput, []testSeries{testInput[0], testInput[3], testInput[4], testInput[2], testInput[1]}},
	}, SeriesReducerLast.Reducer(), Descending)

	testSortImpl(ctx, t, []testSortData{
		{testInput, []testSeries{testInput[4], testInput[0], testInput[3], testInput[2], testInput[1]}},
	}, SeriesReducerMax.Reducer(), Descending)

	testSortImpl(ctx, t, []testSortData{
		{testInput, []testSeries{testInput[4], testInput[3], testInput[2], testInput[0], testInput[1]}},
	}, SeriesReducerStdDev.Reducer(), Descending)

	testSortImpl(ctx, t, []testSortData{
		{testInput, []testSeries{testInput[1], testInput[3], testInput[0], testInput[2], testInput[4]}},
	}, SeriesReducerAvg.Reducer(), Ascending)

}

func TestSortSeriesStable(t *testing.T) {
	ctx := context.New()
	defer func() { _ = ctx.Close() }()

	constValues := newTestSeriesValues(ctx, 1000, []float64{1, 2, 3, 4})
	series := []*Series{
		NewSeries(ctx, "foo", time.Now(), constValues),
		NewSeries(ctx, "bar", time.Now(), constValues),
		NewSeries(ctx, "baz", time.Now(), constValues),
		NewSeries(ctx, "qux", time.Now(), constValues),
	}

	// Check that if input order is random that the same equal "lowest"
	// series is chosen deterministically each time.
	var lastOrder []string
	for i := 0; i < 100; i++ {
		rand.Shuffle(len(series), func(i, j int) {
			series[i], series[j] = series[j], series[i]
		})

		result, err := SortSeries(SeriesList{
			Values:      series,
			SortApplied: false,
		}, SeriesReducerMin.Reducer(), Descending)
		require.NoError(t, err)

		order := make([]string, 0, len(result.Values))
		for _, series := range result.Values {
			order = append(order, series.Name())
		}

		expectedOrder := lastOrder
		lastOrder = order
		if expectedOrder == nil {
			continue
		}

		require.Equal(t, expectedOrder, order)
	}
}

func TestSortSeriesByNameAndNaturalNumbers(t *testing.T) {
	ctx := context.New()
	defer func() { _ = ctx.Close() }()

	constValues := newTestSeriesValues(ctx, 1000, []float64{1, 2, 3, 4})
	series := []*Series{
		NewSeries(ctx, "server1", time.Now(), constValues),
		NewSeries(ctx, "server11", time.Now(), constValues),
		NewSeries(ctx, "server12", time.Now(), constValues),
		NewSeries(ctx, "server2", time.Now(), constValues),
	}

	sort.Sort(SeriesByNameAndNaturalNumbers(series))

	actual := make([]string, 0, len(series))
	for _, s := range series {
		actual = append(actual, s.Name())
	}

	require.Equal(t, []string{
		"server1",
		"server2",
		"server11",
		"server12",
	}, actual)
}