1 files changed, 44 insertions, 26 deletions
diff --git a/tests/mobly/controllers/android_device_lib/sl4a_client_test.py b/tests/mobly/controllers/android_device_lib/sl4a_client_test.py
index 08560d4..bf11b07 100755
--- a/tests/mobly/controllers/android_device_lib/sl4a_client_test.py
+++ b/tests/mobly/controllers/android_device_lib/sl4a_client_test.py
@@ -22,43 +22,60 @@ from tests.lib import mock_android_device
 
 
 class Sl4aClientTest(jsonrpc_client_test_base.JsonRpcClientTestBase):
-  """Unit tests for mobly.controllers.android_device_lib.sl4a_client.
-  """
+  """Unit tests for mobly.controllers.android_device_lib.sl4a_client."""
 
   @mock.patch('socket.create_connection')
-  @mock.patch('mobly.controllers.android_device_lib.snippet_client.'
-              'utils.start_standing_subprocess')
-  @mock.patch('mobly.controllers.android_device_lib.snippet_client.'
-              'utils.get_available_host_port')
-  def test_start_app_and_connect(self, mock_get_port,
-                                 mock_start_standing_subprocess,
-                                 mock_create_connection):
+  @mock.patch(
+      'mobly.controllers.android_device_lib.snippet_client.'
+      'utils.start_standing_subprocess'
+  )
+  @mock.patch(
+      'mobly.controllers.android_device_lib.snippet_client.'
+      'utils.get_available_host_port'
+  )
+  def test_start_app_and_connect(
+      self,
+      mock_get_port,
+      mock_start_standing_subprocess,
+      mock_create_connection,
+  ):
     self.setup_mock_socket_file(mock_create_connection)
-    self._setup_mock_instrumentation_cmd(mock_start_standing_subprocess,
-                                         resp_lines=[b'\n'])
+    self._setup_mock_instrumentation_cmd(
+        mock_start_standing_subprocess, resp_lines=[b'\n']
+    )
     client = self._make_client()
     client.start_app_and_connect()
     self.assertEqual(8080, client.device_port)
 
   @mock.patch('socket.create_connection')
-  @mock.patch('mobly.controllers.android_device_lib.snippet_client.'
-              'utils.start_standing_subprocess')
-  @mock.patch('mobly.controllers.android_device_lib.snippet_client.'
-              'utils.get_available_host_port')
-  def test_app_not_installed(self, mock_get_port,
-                             mock_start_standing_subprocess,
-                             mock_create_connection):
+  @mock.patch(
+      'mobly.controllers.android_device_lib.snippet_client.'
+      'utils.start_standing_subprocess'
+  )
+  @mock.patch(
+      'mobly.controllers.android_device_lib.snippet_client.'
+      'utils.get_available_host_port'
+  )
+  def test_app_not_installed(
+      self,
+      mock_get_port,
+      mock_start_standing_subprocess,
+      mock_create_connection,
+  ):
     self.setup_mock_socket_file(mock_create_connection)
-    self._setup_mock_instrumentation_cmd(mock_start_standing_subprocess,
-                                         resp_lines=[b'\n'])
+    self._setup_mock_instrumentation_cmd(
+        mock_start_standing_subprocess, resp_lines=[b'\n']
+    )
     client = self._make_client(adb_proxy=mock_android_device.MockAdbProxy())
-    with self.assertRaisesRegex(jsonrpc_client_base.AppStartError,
-                                '.* SL4A is not installed on .*'):
+    with self.assertRaisesRegex(
+        jsonrpc_client_base.AppStartError, '.* SL4A is not installed on .*'
+    ):
       client.start_app_and_connect()
 
   def _make_client(self, adb_proxy=None):
     adb_proxy = adb_proxy or mock_android_device.MockAdbProxy(
-        installed_packages=['com.googlecode.android_scripting'])
+        installed_packages=['com.googlecode.android_scripting']
+    )
     ad = mock.Mock()
     ad.adb = adb_proxy
     ad.build_info = {
@@ -67,11 +84,12 @@ class Sl4aClientTest(jsonrpc_client_test_base.JsonRpcClientTestBase):
     }
     return sl4a_client.Sl4aClient(ad=ad)
 
-  def _setup_mock_instrumentation_cmd(self, mock_start_standing_subprocess,
-                                      resp_lines):
+  def _setup_mock_instrumentation_cmd(
+      self, mock_start_standing_subprocess, resp_lines
+  ):
     mock_proc = mock_start_standing_subprocess()
     mock_proc.stdout.readline.side_effect = resp_lines
 
 
-if __name__ == "__main__":
+if __name__ == '__main__':
   unittest.main()