Merge "Fixes UTs in uicd_cli_test.py"
diff --git a/acts/framework/tests/libs/uicd/uicd_cli_test.py b/acts/framework/tests/libs/uicd/uicd_cli_test.py
index c1115b0..8ebc254 100644
--- a/acts/framework/tests/libs/uicd/uicd_cli_test.py
+++ b/acts/framework/tests/libs/uicd/uicd_cli_test.py
@@ -93,14 +93,14 @@
nc.run('SERIAL', 'test')
expected_cmd = 'java -jar /base/uicd-commandline.jar ' \
'-d SERIAL -i /workflows/test'
- job_run.assertCalledWith(expected_cmd.split())
+ job_run.assert_called_with(expected_cmd.split(), timeout=120)
# Log path set
nc._log_path = '/logs'
with mock.patch('acts.libs.proc.job.run') as job_run:
nc.run('SERIAL', 'test')
expected_cmd = 'java -jar /base/uicd-commandline.jar ' \
'-d SERIAL -i /workflows/test -o /logs'
- job_run.assertCalledWith(expected_cmd.split())
+ job_run.assert_called_with(expected_cmd.split(), timeout=120)
if __name__ == '__main__':