New run shell implementation for DeviceUtils

build/android/pylib/device/adb_wrapper_test.py

 # Copyright 2013 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.
 
 """Tests for the AdbWrapper class."""
 
 import os
 import tempfile
 import time
 import unittest
@@ skipping 23 matching lines @@
     fi, path = tempfile.mkstemp()
     with os.fdopen(fi, 'wb') as f:
       f.write(contents)
     return path
 
   def testShell(self):
     output = self._adb.Shell('echo test', expect_rc=0)
     self.assertEqual(output.strip(), 'test')
     output = self._adb.Shell('echo test')
     self.assertEqual(output.strip(), 'test')
-    self.assertRaises(device_errors.AdbCommandFailedError, self._adb.Shell,
-        'echo test', expect_rc=1)
+    self.assertRaises(device_errors.AdbShellCommandFailedError,
+        self._adb.Shell, 'echo test', expect_rc=1)
 
   def testPushPull(self):
     path = self._MakeTempFile('foo')
     device_path = '/data/local/tmp/testfile.txt'
     local_tmpdir = os.path.dirname(path)
     self._adb.Push(path, device_path)
     self.assertEqual(self._adb.Shell('cat %s' % device_path), 'foo')
     self._adb.Pull(device_path, local_tmpdir)
     with open(os.path.join(local_tmpdir, 'testfile.txt'), 'r') as f:
       self.assertEqual(f.read(), 'foo')
@@ skipping 28 matching lines @@
       try:
         self._adb.Shell('start')
         break
       except device_errors.AdbCommandFailedError:
         time.sleep(1)
     self._adb.Remount()
 
 
 if __name__ == '__main__':
   unittest.main()