Make test_net_info a bit more robust.

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 212 matching lines @@
     m = metrics.CounterMetric('test', fields={'foo': 'bar'})
     m.increment()
     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):
     m = metrics.CounterMetric('test')
     self.assertTrue(m.is_cumulative())
 
+  def test_get_all(self):
+    m = metrics.CounterMetric('test')
+    m.increment()
+    m.increment({'foo': 'bar'})
+    m.increment({'foo': 'baz', 'moo': 'wibble'})
+    self.assertEqual([
+        ((), 1),
+        ((('foo', 'bar'),), 1),
+        ((('foo', 'baz'), ('moo', 'wibble')), 1),
+    ], sorted(m.get_all()))
+
 
 class GaugeMetricTest(TestBase):
 
   def test_populate_value(self):
     pb = metrics_pb2.MetricsData()
     m = metrics.GaugeMetric('test')
     m._populate_value(pb, 1, 1234)
     self.assertEquals(pb.gauge, 1)
 
   def test_set(self):
@@ skipping 263 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())