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)); } } }