diff --git a/src/VirtualClient/VirtualClient.Actions/NASParallelBench/NASParallelBenchExecutor.cs b/src/VirtualClient/VirtualClient.Actions/NASParallelBench/NASParallelBenchExecutor.cs index 492fe8572e..e5d1f7a27d 100644 --- a/src/VirtualClient/VirtualClient.Actions/NASParallelBench/NASParallelBenchExecutor.cs +++ b/src/VirtualClient/VirtualClient.Actions/NASParallelBench/NASParallelBenchExecutor.cs @@ -7,15 +7,12 @@ namespace VirtualClient.Actions using System.Collections.Generic; using System.Net; using System.Net.Http; - using System.Runtime.InteropServices; using System.Threading; using System.Threading.Tasks; using Microsoft.Extensions.DependencyInjection; - using Microsoft.Extensions.Logging; using Newtonsoft.Json.Linq; using VirtualClient.Common; using VirtualClient.Common.Extensions; - using VirtualClient.Common.Platform; using VirtualClient.Common.Telemetry; using VirtualClient.Contracts; @@ -146,6 +143,8 @@ private async Task BuildBinariesAsync(EventContext telemetryContext, Cancellatio IApiClientManager clientManager = this.Dependencies.GetService(); var apiClient = clientManager.GetOrCreateApiClient(IPAddress.Loopback.ToString(), IPAddress.Loopback); + await apiClient.PollForHeartbeatAsync(TimeSpan.FromMinutes(40), cancellationToken); + HttpResponseMessage response = await apiClient.GetStateAsync(nameof(this.NpbBuildState), cancellationToken) .ConfigureAwait(false);