From a7787b43cd17fcbb26ca05c52136f3c27b98671b Mon Sep 17 00:00:00 2001 From: Frostebite Date: Tue, 14 Feb 2023 01:03:24 +0000 Subject: [PATCH] locking test improvements --- .../cloud-runner/providers/aws/aws-task-runner.ts | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/src/model/cloud-runner/providers/aws/aws-task-runner.ts b/src/model/cloud-runner/providers/aws/aws-task-runner.ts index 1f284555..da501306 100644 --- a/src/model/cloud-runner/providers/aws/aws-task-runner.ts +++ b/src/model/cloud-runner/providers/aws/aws-task-runner.ts @@ -72,10 +72,15 @@ class AWSTaskRunner { CloudRunnerLogger.log(`Streaming...`); const { output, shouldCleanup } = await this.streamLogsUntilTaskStops(cluster, taskArn, streamName); - await new Promise((resolve) => resolve(10000)); - const taskData = await AWSTaskRunner.describeTasks(cluster, taskArn); - const containerState = taskData.containers?.[0]; - const exitCode = containerState?.exitCode; + let exitCode; + let containerState; + let taskData; + while (exitCode === undefined) { + await new Promise((resolve) => resolve(10000)); + taskData = await AWSTaskRunner.describeTasks(cluster, taskArn); + containerState = taskData.containers?.[0]; + exitCode = containerState?.exitCode; + } CloudRunnerLogger.log(`Container State: ${JSON.stringify(containerState, undefined, 4)}`); if (exitCode === undefined) { CloudRunnerLogger.logWarning(`Undefined exitcode for container`);