Issue 623854: Support unit annotations in ts_mon metrics

infra_libs/ts_mon/common/test/metrics_test.py

Index: infra_libs/ts_mon/common/test/metrics_test.py
diff --git a/infra_libs/ts_mon/common/test/metrics_test.py b/infra_libs/ts_mon/common/test/metrics_test.py
index a09795d63afa2a5b42f006352600dd4c4b242625..e89ebc4de979b284e48179ef6ac33a6fccd34636 100644
--- a/infra_libs/ts_mon/common/test/metrics_test.py
+++ b/infra_libs/ts_mon/common/test/metrics_test.py
@@ -62,6 +62,15 @@ class MetricTest(TestBase):
     m.serialize_to(p, 1234, (('bar', 1), ('baz', False)), m.get(), t)
     return str(p).splitlines()
 
+  def test_serialize_with_units(self):
+    t = targets.DeviceTarget('reg', 'role', 'net', 'host')
+    m = metrics.GaugeMetric('test', units=metrics.MetricsDataUnits.SECONDS)
+    m.set(1)
+    p = metrics_pb2.MetricsCollection()
+    m.serialize_to(p, 1234, (('bar', 1), ('baz', False)), m.get(), t)
+    self.assertEquals(p.data[0].units, metrics.MetricsDataUnits.SECONDS)
+    return str(p).splitlines()
+
   def test_serialize_too_many_fields(self):
     m = metrics.StringMetric('test', fields={'a': 1, 'b': 2, 'c': 3, 'd': 4})
     m.set('val', fields={'e': 5, 'f': 6, 'g': 7})