prometheus/prompb/custom_test.go

package prompb

import (
	"testing"

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

func TestOptimizedMarshal(t *testing.T) {
	var got []byte

	tests := []struct {
		name string
		m    *MinimizedWriteRequest
	}{
		// {
		// 	name: "empty",
		// 	m:    &MinimizedWriteRequest{},
		// },
		{
			name: "simple",
			m: &MinimizedWriteRequest{
				Timeseries: []MinimizedTimeSeries{
					{
						LabelSymbols: []uint32{
							0, 10,
							10, 3,
							13, 3,
							16, 6,
							22, 3,
							25, 5,
							30, 3,
							33, 7,
						},
						Samples:    []Sample{{Value: 1, Timestamp: 0}},
						Exemplars:  []Exemplar{{Labels: []Label{{Name: "f", Value: "g"}}, Value: 1, Timestamp: 0}},
						Histograms: nil,
					},
					{
						LabelSymbols: []uint32{
							0, 10,
							10, 3,
							13, 3,
							16, 6,
							22, 3,
							25, 5,
							30, 3,
							33, 7,
						},
						Samples:    []Sample{{Value: 2, Timestamp: 1}},
						Exemplars:  []Exemplar{{Labels: []Label{{Name: "h", Value: "i"}}, Value: 2, Timestamp: 1}},
						Histograms: nil,
					},
				},
				// 40 chars
				Symbols: "abcdefghijabcdefghijabcdefghijabcdefghij",
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got = got[:0]
			// should be the same as the standard marshal
			expected, err := tt.m.Marshal()
			assert.NoError(t, err)
			got, err = tt.m.OptimizedMarshal(got)
			assert.NoError(t, err)
			assert.Equal(t, expected, got)

			// round trip
			m := &MinimizedWriteRequest{}
			assert.NoError(t, m.Unmarshal(got))
			assert.Equal(t, tt.m, m)
		})
	}
}