private static Mock <IAzureProxy> GetMockAzureProxy(AzureProxyReturnValues azureProxyReturnValues)
{
var azureProxy = new Mock <IAzureProxy>();
azureProxy.Setup(a => a.GetBatchJobAndTaskStateAsync(It.IsAny <string>())).Returns(Task.FromResult(azureProxyReturnValues.BatchJobAndTaskState));
azureProxy.Setup(a => a.GetNextBatchJobIdAsync(It.IsAny <string>())).Returns(Task.FromResult(azureProxyReturnValues.NextBatchJobId));
azureProxy.Setup(a => a.GetAccessibleStorageAccountsAsync()).Returns(Task.FromResult(azureProxyReturnValues.AccessibleStorageAccounts));
azureProxy.Setup(a => a.GetAccessibleContainerRegistriesAsync()).Returns(Task.FromResult(azureProxyReturnValues.AccessibleContainerRegistries));
azureProxy.Setup(a => a.GetStorageAccountKeyAsync(It.IsAny <StorageAccountInfo>())).Returns(Task.FromResult(azureProxyReturnValues.StorageAccountKey));
azureProxy.Setup(a => a.GetVmSizesAndPricesAsync()).Returns(Task.FromResult(azureProxyReturnValues.VmSizesAndPrices));
azureProxy.Setup(a => a.GetBatchAccountQuotasAsync()).Returns(Task.FromResult(azureProxyReturnValues.BatchQuotas));
azureProxy.Setup(a => a.GetBatchActiveNodeCountByVmSize()).Returns(azureProxyReturnValues.ActiveNodeCountByVmSize);
azureProxy.Setup(a => a.GetBatchActiveJobCount()).Returns(azureProxyReturnValues.ActiveJobCount);
azureProxy.Setup(a => a.GetBatchActivePoolCount()).Returns(azureProxyReturnValues.ActivePoolCount);
azureProxy.Setup(a => a.DownloadBlobAsync(It.IsAny <Uri>())).Returns(Task.FromResult(azureProxyReturnValues.DownloadedBlobContent));
return(azureProxy);
}
private static Mock <IAzureProxy> GetMockAzureProxy(AzureProxyReturnValues azureProxyReturnValues)
{
var azureProxy = new Mock <IAzureProxy>();
azureProxy.Setup(a => a.GetBatchJobAndTaskStateAsync(It.IsAny <string>())).Returns(Task.FromResult(azureProxyReturnValues.BatchJobAndTaskState));
azureProxy.Setup(a => a.GetNextBatchJobIdAsync(It.IsAny <string>())).Returns(Task.FromResult(azureProxyReturnValues.NextBatchJobId));
azureProxy.Setup(a => a.GetStorageAccountInfoAsync("defaultstorageaccount")).Returns(Task.FromResult(azureProxyReturnValues.StorageAccountInfos["defaultstorageaccount"]));
azureProxy.Setup(a => a.GetStorageAccountInfoAsync("storageaccount1")).Returns(Task.FromResult(azureProxyReturnValues.StorageAccountInfos["storageaccount1"]));
azureProxy.Setup(a => a.GetContainerRegistryInfoAsync("registryServer1/imageName1:tag1")).Returns(Task.FromResult(azureProxyReturnValues.ContainerRegistryInfo));
azureProxy.Setup(a => a.GetStorageAccountKeyAsync(It.IsAny <StorageAccountInfo>())).Returns(Task.FromResult(azureProxyReturnValues.StorageAccountKey));
azureProxy.Setup(a => a.GetVmSizesAndPricesAsync()).Returns(Task.FromResult(azureProxyReturnValues.VmSizesAndPrices));
azureProxy.Setup(a => a.GetBatchAccountQuotasAsync()).Returns(Task.FromResult(azureProxyReturnValues.BatchQuotas));
azureProxy.Setup(a => a.GetBatchActiveNodeCountByVmSize()).Returns(azureProxyReturnValues.ActiveNodeCountByVmSize);
azureProxy.Setup(a => a.GetBatchActiveJobCount()).Returns(azureProxyReturnValues.ActiveJobCount);
azureProxy.Setup(a => a.GetBatchActivePoolCount()).Returns(azureProxyReturnValues.ActivePoolCount);
azureProxy.Setup(a => a.DownloadBlobAsync(It.IsAny <Uri>())).Returns(Task.FromResult(azureProxyReturnValues.DownloadedBlobContent));
azureProxy.Setup(a => a.LocalFileExists(It.IsAny <string>())).Returns(azureProxyReturnValues.LocalFileExists);
return(azureProxy);
}
private static async Task <TesState> GetNewTesTaskStateAsync(TesResources resources, AzureProxyReturnValues azureProxyReturnValues)
{
var tesTask = GetTesTask();
tesTask.Resources = resources;
await ProcessTesTaskAndGetBatchJobArgumentsAsync(tesTask, GetMockConfig(), GetMockAzureProxy(azureProxyReturnValues));
return(tesTask.State);
}
