Update wait time in OTA E2E tests (aws#2021)

tgsong · May 18, 2020 · 7227e77 · 7227e77
1 parent 5f6c746
commit 7227e77
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tools/ota_e2e_tests/aws_ota_test/aws_ota_test_case_disconnect_cancel_update.py b/tools/ota_e2e_tests/aws_ota_test/aws_ota_test_case_disconnect_cancel_update.py
@@ -77,10 +77,10 @@ def run(self):
         # Start an OTA Update.
         otaUpdateId = self._otaAwsAgent.quickCreateOtaUpdate(self._otaConfig, [self._protocol])
 
-        # Wait up to 3 minute until the job is in progress.
+        # Wait until the job is in progress.
         thing_name = self._otaAwsAgent._iotThing.thing_name
         exec_status = 'QUEUED'
-        timeout = time.time() + 180
+        timeout = time.time() + self._otaConfig['ota_timeout_sec']
         while exec_status == 'QUEUED' and time.time() < timeout:
             exec_status = self.get_job_exec_status(otaUpdateId, thing_name)
             time.sleep(1)

diff --git a/tools/ota_e2e_tests/aws_ota_test/aws_ota_test_case_disconnect_resume.py b/tools/ota_e2e_tests/aws_ota_test/aws_ota_test_case_disconnect_resume.py
@@ -77,10 +77,10 @@ def run(self):
         # Start an OTA Update.
         otaUpdateId = self._otaAwsAgent.quickCreateOtaUpdate(self._otaConfig, [self._protocol])
 
-        # Wait up to 3 minute until the job is in progress.
+        # Wait until the job is in progress.
         thing_name = self._otaAwsAgent._iotThing.thing_name
         exec_status = 'QUEUED'
-        timeout = time.time() + 180
+        timeout = time.time() + self._otaConfig['ota_timeout_sec']
         while exec_status == 'QUEUED' and time.time() < timeout:
             exec_status = self.get_job_exec_status(otaUpdateId, thing_name)
             time.sleep(1)