Make test_net_info a bit more robust.

infra_libs/ts_mon/common/test/metric_store_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 functools
 import operator
 import time
 import unittest
 
 import mock
@@ skipping 109 matching lines @@
     self.assertEquals(24, self.store.get('foo', fields1, target_fields1))
     self.assertEquals(34, self.store.get('foo', fields2, target_fields2))
 
     self.assertIsNone(self.store.get('foo', fields3, None))
     self.assertIsNone(self.store.get('foo', (), None))
     self.assertIsNone(self.store.get('foo', fields1, target_fields2))
     self.assertEquals(44, self.store.get('foo', fields3, None, default=44))
 
     self.assertIsNone(self.store.get('bar', (), None))
 
+  def test_iter_field_values(self):
+    fields1 = (('field', 'value'),)
+    fields2 = (('field', 'value2'),)
+    target_fields1 = {'region': 'rrr'}
+
+    self.store.set('foo', fields1, None, 42)
+    self.store.set('foo', fields2, None, 43)
+    self.store.set('foo', fields2, target_fields1, 44)
+
+    field_values = list(self.store.iter_field_values('foo'))
+    self.assertEquals([
+        ((('field', 'value'),), 42),
+        ((('field', 'value2'),), 43),
+        ((('field', 'value2'),), 44),
+    ], sorted(field_values))
+
   def test_set_enforce_ge(self):
     self.store.set('foo', (('field', 'value'),), None, 42, enforce_ge=True)
     self.store.set('foo', (('field', 'value'),), None, 43, enforce_ge=True)
 
     with self.assertRaises(errors.MonitoringDecreasingValueError):
       self.store.set('foo', (('field', 'value'),), None, 42, enforce_ge=True)
 
   def test_incr(self):
     self.store.set('foo', (('field', 'value'),), None, 42)
     self.store.incr('foo', (('field', 'value'),), None, 4)
@@ skipping 32 matching lines @@
     fields = (('field', 'value'),)
     self.store.set('foo', fields, None, 42)  # Registered in setUp().
     self.store.set('bar', fields, None, 24)  # Unregistered.
     all_metrics = list(self.store.get_all())
     self.assertEqual(1, len(all_metrics))
     self.assertEqual('foo', all_metrics[0][1].name)
 
 
 class InProcessMetricStoreTest(MetricStoreTestBase, unittest.TestCase):
   METRIC_STORE_CLASS = metric_store.InProcessMetricStore