From 2393d6137bbe91088ee965ab0a4acd6928856bb0 Mon Sep 17 00:00:00 2001 From: Guangwen Feng Date: Wed, 27 May 2020 14:38:33 +0800 Subject: [PATCH] Add unit test case for func Type in record.go (#7082) Signed-off-by: Guangwen Feng --- tsdb/record/record_test.go | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/tsdb/record/record_test.go b/tsdb/record/record_test.go index 76a5b412d..de2ce30cf 100644 --- a/tsdb/record/record_test.go +++ b/tsdb/record/record_test.go @@ -117,3 +117,26 @@ func TestRecord_Corrupted(t *testing.T) { testutil.Equals(t, err, encoding.ErrInvalidSize) }) } + +func TestRecord_Type(t *testing.T) { + var enc Encoder + var dec Decoder + + series := []RefSeries{{Ref: 100, Labels: labels.FromStrings("abc", "123")}} + recordType := dec.Type(enc.Series(series, nil)) + testutil.Equals(t, Series, recordType) + + samples := []RefSample{{Ref: 123, T: 12345, V: 1.2345}} + recordType = dec.Type(enc.Samples(samples, nil)) + testutil.Equals(t, Samples, recordType) + + tstones := []tombstones.Stone{{Ref: 1, Intervals: tombstones.Intervals{{Mint: 1, Maxt: 2}}}} + recordType = dec.Type(enc.Tombstones(tstones, nil)) + testutil.Equals(t, Tombstones, recordType) + + recordType = dec.Type(nil) + testutil.Equals(t, Invalid, recordType) + + recordType = dec.Type([]byte{0}) + testutil.Equals(t, Invalid, recordType) +}