def testIsChildProcess(self): """Test IsChildProcess with no name.""" mock_pstree_output = 'a(1)-+-b(2)\n\t|-c(3)\n\t|-foo(4)-bar(5)' self.rc.AddCmdResult(partial_mock.Ignore(), output=mock_pstree_output) self.assertTrue(osutils.IsChildProcess(4)) self.assertTrue(osutils.IsChildProcess(4, name='foo')) self.assertFalse(osutils.IsChildProcess(5, name='foo'))
def _GetDDPid(self): """Get the Pid of dd.""" try: pids = cros_build_lib.RunCommand(['pgrep', 'dd'], capture_output=True, print_cmd=False).output for pid in pids.splitlines(): if osutils.IsChildProcess(int(pid), name='dd'): return int(pid) return -1 except cros_build_lib.RunCommandError: # If dd isn't still running, then we assume that it is finished. return -1