Issue 623854: Support unit annotations in ts_mon metrics

infra_libs/ts_mon/common/test/metrics_test.py

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import sys
 import unittest
 
 import mock
 from infra_libs.ts_mon.protos import metrics_pb2
 
@@ skipping 44 matching lines @@
     return str(p).splitlines()
 
   def test_serialize_with_description(self):
     t = targets.DeviceTarget('reg', 'role', 'net', 'host')
     m = metrics.StringMetric('test', description='a custom description')
     m.set('val')
     p = metrics_pb2.MetricsCollection()
     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})
     with self.assertRaises(errors.MonitoringTooManyFieldsError):
       m.set('val', fields={'e': 5, 'f': 6, 'g': 7, 'h': 8})
 
   def test_populate_field_values(self):
     pb1 = metrics_pb2.MetricsData()
     m1 = metrics.Metric('foo', fields={'asdf': 1})
     m1._populate_fields(pb1, m1._normalized_fields)
@@ skipping 442 matching lines @@
     m.add(25)
     p = metrics_pb2.MetricsCollection()
     m.serialize_to(p, 1234, (), m.get(), t)
     self.assertEquals(1234000000, p.data[0].start_timestamp_us)
 
   def test_is_cumulative(self):
     cd = metrics.CumulativeDistributionMetric('test')
     ncd = metrics.NonCumulativeDistributionMetric('test2')
     self.assertTrue(cd.is_cumulative())
     self.assertFalse(ncd.is_cumulative())