Updates the iperf_client unit tests to match aosp/1198235.
Bug: 147462211
Test: UT
Change-Id: Ibf0627458c5ab5dbb4dd909ad4c355509fce027b
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__':