From d0f18b1168205cef6d8affd533a7a07503b107b3 Mon Sep 17 00:00:00 2001
From: Xi Chen <xichen@uber.com>
Date: Wed, 11 Apr 2018 12:37:18 -0400
Subject: [PATCH] Reset message type as well

---
 encoding/protobuf/reset.go      | 1 +
 encoding/protobuf/reset_test.go | 8 ++++++++
 2 files changed, 9 insertions(+)

diff --git a/encoding/protobuf/reset.go b/encoding/protobuf/reset.go
index f4d6045..433bb2e 100644
--- a/encoding/protobuf/reset.go
+++ b/encoding/protobuf/reset.go
@@ -32,6 +32,7 @@ func resetMetricWithMetadatasProto(pb *metricpb.MetricWithMetadatas) {
 	if pb == nil {
 		return
 	}
+	pb.Type = metricpb.MetricWithMetadatas_UNKNOWN
 	resetCounterWithMetadatasProto(pb.CounterWithMetadatas)
 	resetBatchTimerWithMetadatasProto(pb.BatchTimerWithMetadatas)
 	resetGaugeWithMetadatasProto(pb.GaugeWithMetadatas)
diff --git a/encoding/protobuf/reset_test.go b/encoding/protobuf/reset_test.go
index 6a9ed75..2119e61 100644
--- a/encoding/protobuf/reset_test.go
+++ b/encoding/protobuf/reset_test.go
@@ -74,12 +74,14 @@ func TestResetMetricWithMetadatasProtoNilProto(t *testing.T) {
 
 func TestResetMetricWithMetadatasProtoOnlyCounter(t *testing.T) {
 	input := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_COUNTER_WITH_METADATAS,
 		CounterWithMetadatas: &metricpb.CounterWithMetadatas{
 			Counter:   testCounterBeforeResetProto,
 			Metadatas: testMetadatasBeforeResetProto,
 		},
 	}
 	expected := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_UNKNOWN,
 		CounterWithMetadatas: &metricpb.CounterWithMetadatas{
 			Counter:   testCounterAfterResetProto,
 			Metadatas: testMetadatasAfterResetProto,
@@ -93,12 +95,14 @@ func TestResetMetricWithMetadatasProtoOnlyCounter(t *testing.T) {
 
 func TestResetMetricWithMetadatasProtoOnlyBatchTimer(t *testing.T) {
 	input := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_BATCH_TIMER_WITH_METADATAS,
 		BatchTimerWithMetadatas: &metricpb.BatchTimerWithMetadatas{
 			BatchTimer: testBatchTimerBeforeResetProto,
 			Metadatas:  testMetadatasBeforeResetProto,
 		},
 	}
 	expected := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_UNKNOWN,
 		BatchTimerWithMetadatas: &metricpb.BatchTimerWithMetadatas{
 			BatchTimer: testBatchTimerAfterResetProto,
 			Metadatas:  testMetadatasAfterResetProto,
@@ -112,12 +116,14 @@ func TestResetMetricWithMetadatasProtoOnlyBatchTimer(t *testing.T) {
 
 func TestResetMetricWithMetadatasProtoOnlyGauge(t *testing.T) {
 	input := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_GAUGE_WITH_METADATAS,
 		GaugeWithMetadatas: &metricpb.GaugeWithMetadatas{
 			Gauge:     testGaugeBeforeResetProto,
 			Metadatas: testMetadatasBeforeResetProto,
 		},
 	}
 	expected := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_UNKNOWN,
 		GaugeWithMetadatas: &metricpb.GaugeWithMetadatas{
 			Gauge:     testGaugeAfterResetProto,
 			Metadatas: testMetadatasAfterResetProto,
@@ -131,6 +137,7 @@ func TestResetMetricWithMetadatasProtoOnlyGauge(t *testing.T) {
 
 func TestResetMetricWithMetadatasProtoAll(t *testing.T) {
 	input := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_GAUGE_WITH_METADATAS,
 		CounterWithMetadatas: &metricpb.CounterWithMetadatas{
 			Counter:   testCounterBeforeResetProto,
 			Metadatas: testMetadatasBeforeResetProto,
@@ -145,6 +152,7 @@ func TestResetMetricWithMetadatasProtoAll(t *testing.T) {
 		},
 	}
 	expected := &metricpb.MetricWithMetadatas{
+		Type: metricpb.MetricWithMetadatas_UNKNOWN,
 		CounterWithMetadatas: &metricpb.CounterWithMetadatas{
 			Counter:   testCounterAfterResetProto,
 			Metadatas: testMetadatasAfterResetProto,