public static void WaitAllNodesToBeIdle(BatchAIManagementClient client, string rgName, string wsName, string clusterName)
{
while (true)
{
Cluster cluster = client.Clusters.Get(rgName, wsName, clusterName);
if (cluster.CurrentNodeCount == cluster.ScaleSettings.Manual.TargetNodeCount &&
cluster.CurrentNodeCount == cluster.NodeStateCounts.IdleNodeCount)
{
return;
}
TestUtilities.Wait(TimeSpan.FromSeconds(5));
}
}
public static void WaitJobSucceeded(BatchAIManagementClient client, string rgName, string wsName, string expName, String jobName)
{
while (true)
{
Job job = client.Jobs.Get(rgName, wsName, expName, jobName);
if (job.ExecutionState == ExecutionState.Succeeded ||
job.ExecutionState == ExecutionState.Failed)
{
Assert.Equal(job.ExecutionState, ExecutionState.Succeeded);
return;
}
else
{
TestUtilities.Wait(TimeSpan.FromSeconds(5));
}
}
}
