Merge "Updates the iperf_client unit tests to match aosp/1198235."
diff --git a/acts/framework/tests/controllers/iperf_client_test.py b/acts/framework/tests/controllers/iperf_client_test.py
index e004672..b35cf38 100644
--- a/acts/framework/tests/controllers/iperf_client_test.py
+++ b/acts/framework/tests/controllers/iperf_client_test.py
@@ -128,13 +128,11 @@
 
         with mock.patch('acts.controllers.iperf_client.'
                         'IPerfClientOverAdb._android_device') as adb_device:
-            adb_device.run_iperf_client.return_value = ('', [expected_output])
+            adb_device.adb.shell.return_value = 'output'
             client.start('127.0.0.1', 'IPERF_ARGS', 'TAG')
 
         mock_open.assert_called_with(file_path, 'w')
-        mock_open().__enter__().write.assert_called_with(
-            expected_output
-        )
+        mock_open().__enter__().write.assert_called_with('output')
 
 
 if __name__ == '__main__':