void EnableContainerReregistrationSetsReregisterFlag()
{
using (UndoContext undoContext = UndoContext.Current)
{
undoContext.Start();
BackupVaultServicesManagementClient client = GetServiceClient <BackupVaultServicesManagementClient>();
string containerId = ConfigurationManager.AppSettings["ContainerId"];
string friendlyName = ConfigurationManager.AppSettings["ContainerFriendlyName"];
EnableReregistrationRequest request = new EnableReregistrationRequest()
{
ContainerReregistrationState = new ContainerReregistrationState()
{
EnableReregistration = true,
},
};
OperationResponse response = client.Container.EnableMarsContainerReregistration(BackupServicesTestsBase.ResourceGroupName, BackupServicesTestsBase.ResourceName, containerId, request, GetCustomRequestHeaders());
// Response Validation
Assert.NotNull(response);
Assert.True(response.StatusCode == HttpStatusCode.NoContent, "Status code should be NoContent");
// Basic Validation
ListMarsContainerOperationResponse getResponse = client.Container.ListMarsContainersByTypeAndFriendlyName(BackupServicesTestsBase.ResourceGroupName, BackupServicesTestsBase.ResourceName, MarsContainerType.Machine, friendlyName, GetCustomRequestHeaders());
Assert.True(getResponse.ListMarsContainerResponse.Value.Any(marsContainer =>
{
return(marsContainer.ContainerType == MarsContainerType.Machine.ToString() &&
marsContainer.Properties != null &&
string.Equals(marsContainer.Properties.FriendlyName, friendlyName, StringComparison.OrdinalIgnoreCase) &&
marsContainer.Properties.CanReRegister == true);
}), "Reregistration doesn't appear to have been enabled for the input container");
}
}
/// <summary>
/// Enable container reregistration
/// </summary>
/// <param name="containerId"></param>
/// <returns></returns>
public void EnableMachineContainerReregistration(long containerId)
{
EnableReregistrationRequest request = new EnableReregistrationRequest()
{
ContainerReregistrationState = new ContainerReregistrationState()
{
EnableReregistration = true,
},
};
AzureBackupVaultClient.Container.EnableMarsContainerReregistration(containerId.ToString(), request, GetCustomRequestHeaders());
}
/// <summary>
/// Enable the container reregistration.
/// </summary>
/// <param name='operations'>
/// Reference to the
/// Microsoft.Azure.Management.BackupServices.IMarsContainerOperations.
/// </param>
/// <param name='resourceGroupName'>
/// Required.
/// </param>
/// <param name='resourceName'>
/// Required.
/// </param>
/// <param name='containerId'>
/// Required. MARS container ID.
/// </param>
/// <param name='enableReregistrationRequest'>
/// Required. Enable Reregistration Request.
/// </param>
/// <param name='customRequestHeaders'>
/// Optional. Request header parameters.
/// </param>
/// <returns>
/// The definition of a Operation Response.
/// </returns>
public static Task <OperationResponse> EnableMarsContainerReregistrationAsync(this IMarsContainerOperations operations, string resourceGroupName, string resourceName, string containerId, EnableReregistrationRequest enableReregistrationRequest, CustomRequestHeaders customRequestHeaders)
{
return(operations.EnableMarsContainerReregistrationAsync(resourceGroupName, resourceName, containerId, enableReregistrationRequest, customRequestHeaders, CancellationToken.None));
}
/// <summary>
/// Enable the container reregistration.
/// </summary>
/// <param name='operations'>
/// Reference to the
/// Microsoft.Azure.Management.BackupServices.IMarsContainerOperations.
/// </param>
/// <param name='resourceGroupName'>
/// Required.
/// </param>
/// <param name='resourceName'>
/// Required.
/// </param>
/// <param name='containerId'>
/// Required. MARS container ID.
/// </param>
/// <param name='enableReregistrationRequest'>
/// Required. Enable Reregistration Request.
/// </param>
/// <param name='customRequestHeaders'>
/// Optional. Request header parameters.
/// </param>
/// <returns>
/// The definition of a Operation Response.
/// </returns>
public static OperationResponse EnableMarsContainerReregistration(this IMarsContainerOperations operations, string resourceGroupName, string resourceName, string containerId, EnableReregistrationRequest enableReregistrationRequest, CustomRequestHeaders customRequestHeaders)
{
return(Task.Factory.StartNew((object s) =>
{
return ((IMarsContainerOperations)s).EnableMarsContainerReregistrationAsync(resourceGroupName, resourceName, containerId, enableReregistrationRequest, customRequestHeaders);
}
, operations, CancellationToken.None, TaskCreationOptions.None, TaskScheduler.Default).Unwrap().GetAwaiter().GetResult());
}
/// <summary>
/// Enable the container reregistration.
/// </summary>
/// <param name='containerId'>
/// Required. MARS container ID.
/// </param>
/// <param name='enableReregistrationRequest'>
/// Required. Enable Reregistration Request.
/// </param>
/// <param name='customRequestHeaders'>
/// Optional. Request header parameters.
/// </param>
/// <param name='cancellationToken'>
/// Cancellation token.
/// </param>
/// <returns>
/// The definition of a Operation Response.
/// </returns>
public async Task <OperationResponse> EnableMarsContainerReregistrationAsync(string containerId, EnableReregistrationRequest enableReregistrationRequest, CustomRequestHeaders customRequestHeaders, CancellationToken cancellationToken)
{
// Validate
if (containerId == null)
{
throw new ArgumentNullException("containerId");
}
if (enableReregistrationRequest == null)
{
throw new ArgumentNullException("enableReregistrationRequest");
}
// Tracing
bool shouldTrace = TracingAdapter.IsEnabled;
string invocationId = null;
if (shouldTrace)
{
invocationId = TracingAdapter.NextInvocationId.ToString();
Dictionary <string, object> tracingParameters = new Dictionary <string, object>();
tracingParameters.Add("containerId", containerId);
tracingParameters.Add("enableReregistrationRequest", enableReregistrationRequest);
tracingParameters.Add("customRequestHeaders", customRequestHeaders);
TracingAdapter.Enter(invocationId, this, "EnableMarsContainerReregistrationAsync", tracingParameters);
}
// Construct URL
string url = "";
url = url + "/Subscriptions/";
if (this.Client.Credentials.SubscriptionId != null)
{
url = url + Uri.EscapeDataString(this.Client.Credentials.SubscriptionId);
}
url = url + "/resourceGroups/";
url = url + Uri.EscapeDataString(this.Client.ResourceGroupName);
url = url + "/providers/";
url = url + "Microsoft.Backup";
url = url + "/";
url = url + "BackupVault";
url = url + "/";
url = url + Uri.EscapeDataString(this.Client.ResourceName);
url = url + "/backupContainers/";
url = url + Uri.EscapeDataString(containerId);
url = url + "/enableReRegister";
List <string> queryParameters = new List <string>();
queryParameters.Add("api-version=2015-03-15");
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 = new HttpMethod("PATCH");
httpRequest.RequestUri = new Uri(url);
// Set Headers
httpRequest.Headers.Add("Accept-Language", "en-us");
httpRequest.Headers.Add("x-ms-client-request-id", customRequestHeaders.ClientRequestId);
// Set Credentials
cancellationToken.ThrowIfCancellationRequested();
await this.Client.Credentials.ProcessHttpRequestAsync(httpRequest, cancellationToken).ConfigureAwait(false);
// Serialize Request
string requestContent = null;
JToken requestDoc = null;
JObject enableReregistrationRequestValue = new JObject();
requestDoc = enableReregistrationRequestValue;
if (enableReregistrationRequest.ContainerReregistrationState != null)
{
JObject propertiesValue = new JObject();
enableReregistrationRequestValue["properties"] = propertiesValue;
propertiesValue["enableReRegister"] = enableReregistrationRequest.ContainerReregistrationState.EnableReregistration;
}
requestContent = requestDoc.ToString(Newtonsoft.Json.Formatting.Indented);
httpRequest.Content = new StringContent(requestContent, Encoding.UTF8);
httpRequest.Content.Headers.ContentType = MediaTypeHeaderValue.Parse("application/json");
// 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.NoContent)
{
cancellationToken.ThrowIfCancellationRequested();
CloudException ex = CloudException.Create(httpRequest, requestContent, httpResponse, await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false));
if (shouldTrace)
{
TracingAdapter.Error(invocationId, ex);
}
throw ex;
}
// Create Result
OperationResponse result = null;
// Deserialize Response
if (statusCode == HttpStatusCode.NoContent)
{
cancellationToken.ThrowIfCancellationRequested();
string responseContent = await httpResponse.Content.ReadAsStringAsync().ConfigureAwait(false);
result = new OperationResponse();
JToken responseDoc = null;
if (string.IsNullOrEmpty(responseContent) == false)
{
responseDoc = JToken.Parse(responseContent);
}
if (responseDoc != null && responseDoc.Type != JTokenType.Null)
{
Guid operationIdInstance = Guid.Parse(((string)responseDoc));
result.OperationId = operationIdInstance;
}
}
result.StatusCode = statusCode;
if (shouldTrace)
{
TracingAdapter.Exit(invocationId, result);
}
return(result);
}
finally
{
if (httpResponse != null)
{
httpResponse.Dispose();
}
}
}
finally
{
if (httpRequest != null)
{
httpRequest.Dispose();
}
}
}
