Add *new_only* parameter for process_iter() (#1667)

1 files changed, 17 insertions, 1 deletions

diff --git a/psutil/tests/test_system.py b/psutil/tests/test_system.py
index e04e120b..f2d07282 100755
--- a/psutil/tests/test_system.py
+++ b/psutil/tests/test_system.py
@@ -85,7 +85,7 @@ class TestSystemAPIs(unittest.TestCase):
             with self.assertRaises(psutil.AccessDenied):
                 list(psutil.process_iter())
 
-    def test_prcess_iter_w_params(self):
+    def test_prcess_iter_w_attrs(self):
         for p in psutil.process_iter(attrs=['pid']):
             self.assertEqual(list(p.info.keys()), ['pid'])
         with self.assertRaises(ValueError):
@@ -105,6 +105,22 @@ class TestSystemAPIs(unittest.TestCase):
                 self.assertGreaterEqual(p.info['pid'], 0)
             assert m.called
 
+    def test_process_iter_new_only(self):
+        ls1 = list(psutil.process_iter(attrs=['pid']))
+        ls2 = list(psutil.process_iter(attrs=['pid'], new_only=True))
+        self.assertGreater(len(ls1), len(ls2))
+        # assume no more than 3 new processes were created in the meantime
+        self.assertIn(len(ls2), [0, 1, 2, 3, 4, 5])
+
+        sproc = get_test_subprocess()
+        ls = list(psutil.process_iter(attrs=['pid'], new_only=True))
+        self.assertIn(len(ls2), [0, 1, 2, 3, 4, 5])
+        for p in ls:
+            if p.pid == sproc.pid:
+                break
+        else:
+            self.fail("subprocess not found")
+
     def test_wait_procs(self):
         def callback(p):
             pids.append(p.pid)