private static void AssertLongrunningOperation(RemoteAppManagementClient client, RemoteAppOperationStatus status, String operationTrackingId)
{
int retries = 10;
RemoteAppOperationStatusResult operationResult = null;
//check for the operation status to success
while (retries > 0)
{
operationResult = null;
Assert.DoesNotThrow(() =>
{
operationResult = client.OperationResults.Get(operationTrackingId);
});
Assert.Equal(operationResult.StatusCode, HttpStatusCode.OK);
if (operationResult.RemoteAppOperationResult.Status == RemoteAppOperationStatus.Success ||
operationResult.RemoteAppOperationResult.Status == RemoteAppOperationStatus.Failed)
{
break;
}
else
{
TestUtilities.Wait(1000);
--retries;
}
}
Assert.True(retries > 0, "Failed getting the operation status to success or fail after 10 retries.");
Assert.NotNull(operationResult);
Assert.Equal(status, operationResult.RemoteAppOperationResult.Status);
}
private void EnsureStorageInRegion(string region)
{
OperationResultWithTrackingId responseWithTrackingId = null;
RemoteAppOperationStatusResult operationalResponse = null;
const int waitPeriodMilliseconds = 5 * 1000;
const int maxIterations = 60;
int counter = 0;
responseWithTrackingId = CallClient_ThrowOnError(() => Client.TemplateImages.EnsureStorageInRegion(region));
if (responseWithTrackingId.TrackingId != null)
{
task.SetStatus(Commands_RemoteApp.WaitingForStorageVerificationToCompleteMessage);
do
{
Thread.Sleep(waitPeriodMilliseconds);
counter++;
operationalResponse = CallClient_ThrowOnError(() => Client.OperationResults.Get(responseWithTrackingId.TrackingId));
}while ((operationalResponse.RemoteAppOperationResult.Status == RemoteAppOperationStatus.Pending ||
operationalResponse.RemoteAppOperationResult.Status == RemoteAppOperationStatus.InProgress) &&
counter < maxIterations);
if (counter >= maxIterations || operationalResponse.RemoteAppOperationResult.Status != RemoteAppOperationStatus.Success)
{
throw new RemoteAppServiceException(Commands_RemoteApp.StorageCreationFailedError, ErrorCategory.OperationTimeout);
}
}
}
public static int SetUpDefaultRemoteAppOperationResult(Mock <IRemoteAppManagementClient> clientMock, string trackingId)
{
ISetup <IRemoteAppManagementClient, Task <RemoteAppOperationStatusResult> > setup = null;
RemoteAppOperationStatusResult response = new RemoteAppOperationStatusResult()
{
RequestId = "77394",
StatusCode = HttpStatusCode.OK,
RemoteAppOperationResult = new OperationResult()
{
Description = "The Operation has completed successfully",
ErrorDetails = null,
Status = RemoteAppOperationStatus.Success
}
};
mockOperationResult = new List <OperationResult>()
{
new OperationResult()
{
Description = response.RemoteAppOperationResult.Description,
ErrorDetails = response.RemoteAppOperationResult.ErrorDetails,
Status = response.RemoteAppOperationResult.Status
}
};
setup = clientMock.Setup(c => c.OperationResults.GetAsync(trackingId, It.IsAny <CancellationToken>()));
setup.Returns(Task.Factory.StartNew(() => response));
return(mockOperationResult.Count);
}
private void GetUsageDetails(CollectionUsageDetailsResult detailsUsage)
{
RemoteAppOperationStatusResult operationResult = null;
// The request is async and we have to wait for the usage details to be produced here
do
{
System.Threading.Thread.Sleep(5000);
operationResult = CallClient(() => Client.OperationResults.Get(detailsUsage.UsageDetails.OperationTrackingId), Client.OperationResults);
}while (operationResult.RemoteAppOperationResult.Status != RemoteAppOperationStatus.Success &&
operationResult.RemoteAppOperationResult.Status != RemoteAppOperationStatus.Failed);
if (operationResult.RemoteAppOperationResult.Status == RemoteAppOperationStatus.Success)
{
WriteUsageDetails(detailsUsage);
}
else
{
ErrorRecord error = RemoteAppCollectionErrorState.CreateErrorRecordFromString(
Commands_RemoteApp.DetailedUsageFailureMessage,
String.Empty,
Client.Collections,
ErrorCategory.ResourceUnavailable);
WriteError(error);
}
}
public override void ExecuteCmdlet()
{
RemoteAppOperationStatusResult response = null;
response = CallClient(() => Client.OperationResults.Get(TrackingId), Client.OperationResults);
if (response != null)
{
WriteObject(response.RemoteAppOperationResult);
}
}
/// <summary>
/// Gets the status of a long-running operation.
/// </summary>
/// <param name='trackingId'>
/// Required. The operation tracking id.
/// </param>
/// <param name='cancellationToken'>
/// Cancellation token.
/// </param>
/// <returns>
/// Operation response for OperationResultOperations.Get.
/// </returns>
public async Task <RemoteAppOperationStatusResult> GetAsync(string trackingId, CancellationToken cancellationToken)
{
// Validate
if (trackingId == null)
{
throw new ArgumentNullException("trackingId");
}
// Tracing
bool shouldTrace = TracingAdapter.IsEnabled;
string invocationId = null;
if (shouldTrace)
{
invocationId = TracingAdapter.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("trackingId", trackingId);
TracingAdapter.Enter(invocationId, this, "GetAsync", tracingParameters);
}
// Construct URL
string url = "";
url = url + "/";
if (this.Client.Credentials.SubscriptionId != null)
{
url = url + Uri.EscapeDataString(this.Client.Credentials.SubscriptionId);
}
url = url + "/services/";
if (this.Client.RdfeNamespace != null)
{
url = url + Uri.EscapeDataString(this.Client.RdfeNamespace);
}
url = url + "/operationResults/";
url = url + Uri.EscapeDataString(trackingId);
List <string> queryParameters = new List <string>();
queryParameters.Add("api-version=2014-09-01");
if (queryParameters.Count > 0)
{
url = url + "?" + string.Join("&", queryParameters);
}
string baseUrl = this.Client.BaseUri.AbsoluteUri;
// Trim '/' character from the end of baseUrl and beginning of url.
if (baseUrl[baseUrl.Length - 1] == '/')
{
baseUrl = baseUrl.Substring(0, baseUrl.Length - 1);
}
if (url[0] == '/')
{
url = url.Substring(1);
}
url = baseUrl + "/" + url;
url = url.Replace(" ", "%20");
// Create HTTP transport objects
HttpRequestMessage httpRequest = null;
try
{
httpRequest = new HttpRequestMessage();
httpRequest.Method = HttpMethod.Get;
httpRequest.RequestUri = new Uri(url);
// Set Headers
httpRequest.Headers.Add("Accept", "application/json; charset=utf-8");
httpRequest.Headers.Add("x-ms-version", "2014-08-01");
// Set Credentials
cancellationToken.ThrowIfCancellationRequested();
await this.Client.Credentials.ProcessHttpRequestAsync(httpRequest, cancellationToken).ConfigureAwait(false);
// Send Request
HttpResponseMessage httpResponse = null;
try
{
if (shouldTrace)
{
TracingAdapter.SendRequest(invocationId, httpRequest);
}
cancellationToken.ThrowIfCancellationRequested();
httpResponse = await this.Client.HttpClient.SendAsync(httpRequest, cancellationToken).ConfigureAwait(false);
if (shouldTrace)
{
TracingAdapter.ReceiveResponse(invocationId, httpResponse);
}
HttpStatusCode statusCode = httpResponse.StatusCode;
if (statusCode != HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
CloudException ex = CloudException.Create(httpRequest, null, httpResponse, await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false));
if (shouldTrace)
{
TracingAdapter.Error(invocationId, ex);
}
throw ex;
}
// Create Result
RemoteAppOperationStatusResult result = null;
// Deserialize Response
if (statusCode == HttpStatusCode.OK)
{
cancellationToken.ThrowIfCancellationRequested();
string responseContent = await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
result = new RemoteAppOperationStatusResult();
JToken responseDoc = null;
if (string.IsNullOrEmpty(responseContent) == false)
{
responseDoc = JToken.Parse(responseContent);
}
if (responseDoc != null && responseDoc.Type != JTokenType.Null)
{
OperationResult remoteAppOperationResultInstance = new OperationResult();
result.RemoteAppOperationResult = remoteAppOperationResultInstance;
JToken statusValue = responseDoc["Status"];
if (statusValue != null && statusValue.Type != JTokenType.Null)
{
RemoteAppOperationStatus statusInstance = ((RemoteAppOperationStatus)Enum.Parse(typeof(RemoteAppOperationStatus), ((string)statusValue), true));
remoteAppOperationResultInstance.Status = statusInstance;
}
JToken descriptionValue = responseDoc["Description"];
if (descriptionValue != null && descriptionValue.Type != JTokenType.Null)
{
string descriptionInstance = ((string)descriptionValue);
remoteAppOperationResultInstance.Description = descriptionInstance;
}
JToken errorDetailsValue = responseDoc["ErrorDetails"];
if (errorDetailsValue != null && errorDetailsValue.Type != JTokenType.Null)
{
string errorDetailsInstance = ((string)errorDetailsValue);
remoteAppOperationResultInstance.ErrorDetails = errorDetailsInstance;
}
}
}
result.StatusCode = statusCode;
if (httpResponse.Headers.Contains("x-ms-request-id"))
{
result.RequestId = httpResponse.Headers.GetValues("x-ms-request-id").FirstOrDefault();
}
if (shouldTrace)
{
TracingAdapter.Exit(invocationId, result);
}
return(result);
}
finally
{
if (httpResponse != null)
{
httpResponse.Dispose();
}
}
}
finally
{
if (httpRequest != null)
{
httpRequest.Dispose();
}
}
}
public override void ExecuteCmdlet()
{
DateTime today = DateTime.Now;
CollectionUsageDetailsResult detailsUsage = null;
string locale = String.Empty;
RemoteAppOperationStatusResult operationResult = null;
int maxRetryCount = 60;
if (String.IsNullOrWhiteSpace(UsageMonth))
{
UsageMonth = today.Month.ToString();
}
if (String.IsNullOrWhiteSpace(UsageYear))
{
UsageYear = today.Year.ToString();
}
locale = System.Globalization.CultureInfo.CurrentCulture.ToString();
detailsUsage = CallClient(() => Client.Collections.GetUsageDetails(CollectionName, UsageYear, UsageMonth, locale), Client.Collections);
if (detailsUsage == null)
{
return;
}
// The request is async and we have to wait for the usage details to be produced here
do
{
System.Threading.Thread.Sleep(5000);
operationResult = CallClient(() => Client.OperationResults.Get(detailsUsage.UsageDetails.OperationTrackingId), Client.OperationResults);
}while(operationResult.RemoteAppOperationResult.Status != RemoteAppOperationStatus.Success &&
operationResult.RemoteAppOperationResult.Status != RemoteAppOperationStatus.Failed &&
--maxRetryCount > 0);
if (operationResult.RemoteAppOperationResult.Status == RemoteAppOperationStatus.Success)
{
WriteUsageDetails(detailsUsage);
}
else
{
if (operationResult.RemoteAppOperationResult.Status == RemoteAppOperationStatus.Failed)
{
ErrorRecord error = RemoteAppCollectionErrorState.CreateErrorRecordFromString(
Commands_RemoteApp.DetailedUsageFailureMessage,
String.Empty,
Client.Collections,
ErrorCategory.ResourceUnavailable);
WriteError(error);
}
else if (maxRetryCount <= 0)
{
ErrorRecord error = RemoteAppCollectionErrorState.CreateErrorRecordFromString(
String.Format(System.Globalization.CultureInfo.InvariantCulture,
Commands_RemoteApp.RequestTimedOutFormat,
detailsUsage.UsageDetails.OperationTrackingId),
String.Empty,
Client.Collections,
ErrorCategory.OperationTimeout);
WriteError(error);
}
}
}
