public async Task MachineExtensions_UpdateAsync()
{
_context = MockContext.Start(GetType().FullName);
Initialize();
PopulateExtensions();
const string newCommand = "echo 'goodbye'";
MachineExtensionUpdate extensionUpdate = _isLinux
? new MachineExtensionUpdate
{
Settings = new Hashtable {
{ "commandToExecute", newCommand },
},
}
: new MachineExtensionUpdate
{
Settings = new Hashtable {
{ "commandToExecute", newCommand },
},
};
MachineExtension extension = await _client.MachineExtensions.UpdateAsync(RESOURCE_GROUP_NAME, MACHINE_NAME, CUSTOM_SCRIPT_EXTENSION_NAME, extensionUpdate).ConfigureAwait(false);
Assert.Equal(newCommand, ((JObject)extension.Settings).Value <string>("commandToExecute"));
}
/// <summary>
/// The operation to create or update the extension.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group.
/// </param>
/// <param name='name'>
/// The name of the machine where the extension should be created or updated.
/// </param>
/// <param name='extensionName'>
/// The name of the machine extension.
/// </param>
/// <param name='extensionParameters'>
/// Parameters supplied to the Create Machine Extension operation.
/// </param>
public static MachineExtension Update(this IMachineExtensionsOperations operations, string resourceGroupName, string name, string extensionName, MachineExtensionUpdate extensionParameters)
{
return(operations.UpdateAsync(resourceGroupName, name, extensionName, extensionParameters).GetAwaiter().GetResult());
}
/// <summary>
/// The operation to create or update the extension.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// The name of the resource group.
/// </param>
/// <param name='name'>
/// The name of the machine where the extension should be created or updated.
/// </param>
/// <param name='extensionName'>
/// The name of the machine extension.
/// </param>
/// <param name='extensionParameters'>
/// Parameters supplied to the Create Machine Extension operation.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <MachineExtension> BeginUpdateAsync(this IMachineExtensionsOperations operations, string resourceGroupName, string name, string extensionName, MachineExtensionUpdate extensionParameters, CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.BeginUpdateWithHttpMessagesAsync(resourceGroupName, name, extensionName, extensionParameters, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
public virtual MachineExtensionUpdateOperation Update(bool waitForCompletion, MachineExtensionUpdate extensionParameters, CancellationToken cancellationToken = default)
{
if (extensionParameters == null)
{
throw new ArgumentNullException(nameof(extensionParameters));
}
using var scope = _machineExtensionClientDiagnostics.CreateScope("MachineExtension.Update");
scope.Start();
try
{
var response = _machineExtensionRestClient.Update(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, extensionParameters, cancellationToken);
var operation = new MachineExtensionUpdateOperation(ArmClient, _machineExtensionClientDiagnostics, Pipeline, _machineExtensionRestClient.CreateUpdateRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, extensionParameters).Request, response);
if (waitForCompletion)
{
operation.WaitForCompletion(cancellationToken);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public Response Update(string subscriptionId, string resourceGroupName, string name, string extensionName, MachineExtensionUpdate extensionParameters, CancellationToken cancellationToken = default)
{
if (subscriptionId == null)
{
throw new ArgumentNullException(nameof(subscriptionId));
}
if (resourceGroupName == null)
{
throw new ArgumentNullException(nameof(resourceGroupName));
}
if (name == null)
{
throw new ArgumentNullException(nameof(name));
}
if (extensionName == null)
{
throw new ArgumentNullException(nameof(extensionName));
}
if (extensionParameters == null)
{
throw new ArgumentNullException(nameof(extensionParameters));
}
using var message = CreateUpdateRequest(subscriptionId, resourceGroupName, name, extensionName, extensionParameters);
_pipeline.Send(message, cancellationToken);
switch (message.Response.Status)
{
case 200:
case 201:
return(message.Response);
default:
throw _clientDiagnostics.CreateRequestFailedException(message.Response);
}
}
internal HttpMessage CreateUpdateRequest(string subscriptionId, string resourceGroupName, string name, string extensionName, MachineExtensionUpdate extensionParameters)
{
var message = _pipeline.CreateMessage();
var request = message.Request;
request.Method = RequestMethod.Patch;
var uri = new RawRequestUriBuilder();
uri.Reset(endpoint);
uri.AppendPath("/subscriptions/", false);
uri.AppendPath(subscriptionId, true);
uri.AppendPath("/resourceGroups/", false);
uri.AppendPath(resourceGroupName, true);
uri.AppendPath("/providers/Microsoft.ConnectedVMwarevSphere/virtualMachines/", false);
uri.AppendPath(name, true);
uri.AppendPath("/extensions/", false);
uri.AppendPath(extensionName, true);
uri.AppendQuery("api-version", apiVersion, true);
request.Uri = uri;
request.Headers.Add("Accept", "application/json");
request.Headers.Add("Content-Type", "application/json");
var content = new Utf8JsonRequestContent();
content.JsonWriter.WriteObjectValue(extensionParameters);
request.Content = content;
message.SetProperty("SDKUserAgent", _userAgent);
return(message);
}
