VictoriaMetrics/lib/prompb/prompb_test.go

package prompb_test

import (
	"bytes"
	"testing"

	"github.com/VictoriaMetrics/VictoriaMetrics/lib/prompb"
	"github.com/VictoriaMetrics/VictoriaMetrics/lib/prompbmarshal"
)

func TestWriteRequestUnmarshalProtobuf(t *testing.T) {
	var wr prompb.WriteRequest

	f := func(data []byte) {
		t.Helper()

		// Verify that the marshaled protobuf is unmarshaled properly
		if err := wr.UnmarshalProtobuf(data); err != nil {
			t.Fatalf("cannot unmarshal protobuf: %s", err)
		}

		// Compare the unmarshaled wr with the original wrm.
		var wrm prompbmarshal.WriteRequest
		for _, ts := range wr.Timeseries {
			var labels []prompbmarshal.Label
			for _, label := range ts.Labels {
				labels = append(labels, prompbmarshal.Label{
					Name:  label.Name,
					Value: label.Value,
				})
			}
			var samples []prompbmarshal.Sample
			for _, sample := range ts.Samples {
				samples = append(samples, prompbmarshal.Sample{
					Value:     sample.Value,
					Timestamp: sample.Timestamp,
				})
			}
			var exemplars []prompbmarshal.Exemplar
			for _, exemplar := range ts.Exemplars {
				exemplarLabels := make([]prompbmarshal.Label, len(exemplar.Labels))
				for i, label := range exemplar.Labels {
					exemplarLabels[i] = prompbmarshal.Label{
						Name:  label.Name,
						Value: label.Value,
					}
				}
				exemplars = append(exemplars, prompbmarshal.Exemplar{
					Labels:    exemplarLabels,
					Value:     exemplar.Value,
					Timestamp: exemplar.Timestamp,
				})
			}
			wrm.Timeseries = append(wrm.Timeseries, prompbmarshal.TimeSeries{
				Labels:    labels,
				Samples:   samples,
				Exemplars: exemplars,
			})
		}
		dataResult := wrm.MarshalProtobuf(nil)
		if !bytes.Equal(dataResult, data) {
			t.Fatalf("unexpected data obtained after marshaling\ngot\n%X\nwant\n%X", dataResult, data)
		}
	}

	var data []byte
	wrm := &prompbmarshal.WriteRequest{}

	wrm.Reset()
	data = wrm.MarshalProtobuf(data[:0])
	f(data)

	wrm.Reset()
	wrm.Timeseries = []prompbmarshal.TimeSeries{
		{
			Labels: []prompbmarshal.Label{
				{
					Name:  "__name__",
					Value: "process_cpu_seconds_total",
				},
				{
					Name:  "instance",
					Value: "host-123:4567",
				},
				{
					Name:  "job",
					Value: "node-exporter",
				},
			},
		},
	}
	data = wrm.MarshalProtobuf(data[:0])
	f(data)

	wrm.Reset()
	wrm.Timeseries = []prompbmarshal.TimeSeries{
		{
			Samples: []prompbmarshal.Sample{
				{
					Value:     123.3434,
					Timestamp: 8939432423,
				},
				{
					Value:     -123.3434,
					Timestamp: 18939432423,
				},
			},
		},
	}
	data = wrm.MarshalProtobuf(data[:0])
	f(data)

	wrm.Reset()
	wrm.Timeseries = []prompbmarshal.TimeSeries{
		{
			Labels: []prompbmarshal.Label{
				{
					Name:  "__name__",
					Value: "process_cpu_seconds_total",
				},
				{
					Name:  "instance",
					Value: "host-123:4567",
				},
				{
					Name:  "job",
					Value: "node-exporter",
				},
			},
			Samples: []prompbmarshal.Sample{
				{
					Value:     123.3434,
					Timestamp: 8939432423,
				},
				{
					Value:     -123.3434,
					Timestamp: 18939432423,
				},
			},
			Exemplars: []prompbmarshal.Exemplar{
				{
					Labels: []prompbmarshal.Label{
						{Name: "trace-id",
							Value: "123456",
						},
						{Name: "log_id",
							Value: "987664"},
					},
					Value:     12345.6,
					Timestamp: 456,
				},
			},
		},
	}
	data = wrm.MarshalProtobuf(data[:0])
	f(data)

	wrm.Reset()
	wrm.Timeseries = []prompbmarshal.TimeSeries{
		{
			Labels: []prompbmarshal.Label{
				{
					Name:  "__name__",
					Value: "process_cpu_seconds_total",
				},
				{
					Name:  "instance",
					Value: "host-123:4567",
				},
				{
					Name:  "job",
					Value: "node-exporter",
				},
			},
			Samples: []prompbmarshal.Sample{
				{
					Value:     123.3434,
					Timestamp: 8939432423,
				},
				{
					Value:     -123.3434,
					Timestamp: 18939432423,
				},
			},
			Exemplars: []prompbmarshal.Exemplar{
				{
					Labels: []prompbmarshal.Label{
						{
							Name:  "trace-id",
							Value: "123456",
						},
					},
					Value:     12345.6,
					Timestamp: 456,
				},
			},
		},
		{
			Labels: []prompbmarshal.Label{
				{
					Name:  "foo",
					Value: "bar",
				},
			},
			Samples: []prompbmarshal.Sample{
				{
					Value: 9873,
				},
			},
			Exemplars: []prompbmarshal.Exemplar{
				{
					Labels: []prompbmarshal.Label{
						{
							Name:  "trace-id",
							Value: "123456",
						},
						{
							Name:  "log_id",
							Value: "987654",
						},
					},
					Value:     12345.6,
					Timestamp: 456,
				},
			},
		},
	}
	data = wrm.MarshalProtobuf(data[:0])
	f(data)
}