| Index: build/android/pylib/device/device_utils_test.py
|
| diff --git a/build/android/pylib/device/device_utils_test.py b/build/android/pylib/device/device_utils_test.py
|
| index bc50e6dc3fef3c09af83e3139172f24b6934d670..fb91daa41e24445ddc9991164ba20245c0740308 100755
|
| --- a/build/android/pylib/device/device_utils_test.py
|
| +++ b/build/android/pylib/device/device_utils_test.py
|
| @@ -16,13 +16,13 @@ import datetime
|
| import logging
|
| import os
|
| import re
|
| -import signal
|
| import sys
|
| import unittest
|
|
|
| from pylib import android_commands
|
| from pylib import cmd_helper
|
| from pylib import constants
|
| +from pylib import device_signal
|
| from pylib.device import adb_wrapper
|
| from pylib.device import device_errors
|
| from pylib.device import device_utils
|
| @@ -641,40 +641,34 @@ class DeviceUtilsGetDevicePieWrapper(DeviceUtilsTest):
|
| @mock.patch('time.sleep', mock.Mock())
|
| class DeviceUtilsKillAllTest(DeviceUtilsTest):
|
|
|
| - def testKillAll_noMatchingProcesses(self):
|
| - with self.assertCall(self.call.adb.Shell('ps'),
|
| - 'USER PID PPID VSIZE RSS WCHAN PC NAME\n'):
|
| + def testKillAll_noMatchingProcessesFailure(self):
|
| + with self.assertCall(self.call.device.GetPids('test_process'), {}):
|
| with self.assertRaises(device_errors.CommandFailedError):
|
| self.device.KillAll('test_process')
|
|
|
| + def testKillAll_noMatchingProcessesQuiet(self):
|
| + with self.assertCall(self.call.device.GetPids('test_process'), {}):
|
| + self.assertEqual(0, self.device.KillAll('test_process', quiet=True))
|
| +
|
| def testKillAll_nonblocking(self):
|
| with self.assertCalls(
|
| - (self.call.adb.Shell('ps'),
|
| - 'USER PID PPID VSIZE RSS WCHAN PC NAME\n'
|
| - 'u0_a1 1234 174 123456 54321 ffffffff 456789ab some.process\n'),
|
| + (self.call.device.GetPids('some.process'), {'some.process': '1234'}),
|
| (self.call.adb.Shell('kill -9 1234'), '')):
|
| self.assertEquals(1,
|
| self.device.KillAll('some.process', blocking=False))
|
|
|
| def testKillAll_blocking(self):
|
| with self.assertCalls(
|
| - (self.call.adb.Shell('ps'),
|
| - 'USER PID PPID VSIZE RSS WCHAN PC NAME\n'
|
| - 'u0_a1 1234 174 123456 54321 ffffffff 456789ab some.process\n'),
|
| + (self.call.device.GetPids('some.process'), {'some.process': '1234'}),
|
| (self.call.adb.Shell('kill -9 1234'), ''),
|
| - (self.call.adb.Shell('ps'),
|
| - 'USER PID PPID VSIZE RSS WCHAN PC NAME\n'
|
| - 'u0_a1 1234 174 123456 54321 ffffffff 456789ab some.process\n'),
|
| - (self.call.adb.Shell('ps'),
|
| - 'USER PID PPID VSIZE RSS WCHAN PC NAME\n')):
|
| + (self.call.device.GetPids('some.process'), {'some.process': '1234'}),
|
| + (self.call.device.GetPids('some.process'), {})):
|
| self.assertEquals(1,
|
| self.device.KillAll('some.process', blocking=True))
|
|
|
| def testKillAll_root(self):
|
| with self.assertCalls(
|
| - (self.call.adb.Shell('ps'),
|
| - 'USER PID PPID VSIZE RSS WCHAN PC NAME\n'
|
| - 'u0_a1 1234 174 123456 54321 ffffffff 456789ab some.process\n'),
|
| + (self.call.device.GetPids('some.process'), {'some.process': '1234'}),
|
| (self.call.device.NeedsSU(), True),
|
| (self.call.adb.Shell("su -c sh -c 'kill -9 1234'"), '')):
|
| self.assertEquals(1,
|
| @@ -682,12 +676,10 @@ class DeviceUtilsKillAllTest(DeviceUtilsTest):
|
|
|
| def testKillAll_sigterm(self):
|
| with self.assertCalls(
|
| - (self.call.adb.Shell('ps'),
|
| - 'USER PID PPID VSIZE RSS WCHAN PC NAME\n'
|
| - 'u0_a1 1234 174 123456 54321 ffffffff 456789ab some.process\n'),
|
| + (self.call.device.GetPids('some.process'), {'some.process': '1234'}),
|
| (self.call.adb.Shell('kill -15 1234'), '')):
|
| self.assertEquals(1,
|
| - self.device.KillAll('some.process', signum=signal.SIGTERM))
|
| + self.device.KillAll('some.process', signum=device_signal.SIGTERM))
|
|
|
|
|
| class DeviceUtilsStartActivityTest(DeviceUtilsTest):
|
|
|