#802: add test for disappearing keys which wrap

giampaolo · May 8, 2017 · 726e179 · 726e179
1 parent ef42840
commit 726e179
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/psutil/tests/test_misc.py b/psutil/tests/test_misc.py
@@ -439,6 +439,25 @@ def test_changing_keys(self):
         input = {'disk1': nt(8, 8, 8)}
         self.assertEqual(wrap_numbers(input, 'disk_io'), input)
 
+    def test_changing_keys_w_wrap(self):
+        input = {'disk1': nt(50, 50, 50),
+                 'disk2': nt(100, 100, 100)}
+        self.assertEqual(wrap_numbers(input, 'disk_io'), input)
+        # disk 2 wraps
+        input = {'disk1': nt(50, 50, 50),
+                 'disk2': nt(100, 100, 10)}
+        self.assertEqual(wrap_numbers(input, 'disk_io'),
+                         {'disk1': nt(50, 50, 50),
+                          'disk2': nt(100, 100, 110)})
+        # disk 2 disappears
+        input = {'disk1': nt(50, 50, 50)}
+        self.assertEqual(wrap_numbers(input, 'disk_io'), input)
+        # then it appears again; the old wrap is supposed to be
+        # gone.
+        input = {'disk1': nt(50, 50, 50),
+                 'disk2': nt(100, 100, 100)}
+        self.assertEqual(wrap_numbers(input, 'disk_io'), input)
+
 
 # ===================================================================
 # --- Example script tests