Function TestToMetadata

prompb/rwcommon/codec_test.go:87–158

(t *testing.T)

}

func TestToMetadata(t *testing.T) {
	sym := writev2.NewSymbolTable()

	for _, tc := range []struct {
		input    writev2.Metadata
		expected metadata.Metadata
	}{
		{
			input: writev2.Metadata{},
			expected: metadata.Metadata{
				Type: model.MetricTypeUnknown,
			},
		},
		{
			input: writev2.Metadata{
				Type: 12414, // Unknown.
			},
			expected: metadata.Metadata{
				Type: model.MetricTypeUnknown,
			},
		},
		{
			input: writev2.Metadata{
				Type:    writev2.Metadata_METRIC_TYPE_COUNTER,
				HelpRef: sym.Symbolize("help1"),
				UnitRef: sym.Symbolize("unit1"),
			},
			expected: metadata.Metadata{
				Type: model.MetricTypeCounter,
				Help: "help1",
				Unit: "unit1",
			},
		},
		{
			input: writev2.Metadata{
				Type:    writev2.Metadata_METRIC_TYPE_STATESET,
				HelpRef: sym.Symbolize("help2"),
			},
			expected: metadata.Metadata{
				Type: model.MetricTypeStateset,
				Help: "help2",
			},
		},
	} {
		t.Run("", func(t *testing.T) {
			ts := writev2.TimeSeries{Metadata: tc.input}
			meta, err := ts.ToMetadata(sym.Symbols())
			require.NoError(t, err)
			require.Equal(t, tc.expected, meta)
		})
	}

	t.Run("out of bounds unit ref", func(t *testing.T) {
		ts := writev2.TimeSeries{Metadata: writev2.Metadata{UnitRef: 999}}
		_, err := ts.ToMetadata(sym.Symbols())
		require.Error(t, err)
		require.Contains(t, err.Error(), "metadata unit_ref 999 outside of symbols table")
	})