public async Task CassandraKeyspaceCreateAndUpdate()
{
var keyspace = await CreateCassandraKeyspace(null);
Assert.AreEqual(_keyspaceName, keyspace.Data.Resource.Id);
// Seems bug in swagger definition
//Assert.AreEqual(TestThroughput1, keyspace.Data.Options.Throughput);
bool ifExists = await CassandraKeyspaceCollection.ExistsAsync(_keyspaceName);
Assert.True(ifExists);
// NOT WORKING API
//ThroughputSettingsData throughtput = await keyspace.GetMongoDBCollectionThroughputAsync();
CassandraKeyspaceResource keyspace2 = await CassandraKeyspaceCollection.GetAsync(_keyspaceName);
Assert.AreEqual(_keyspaceName, keyspace2.Data.Resource.Id);
//Assert.AreEqual(TestThroughput1, keyspace2.Data.Options.Throughput);
VerifyCassandraKeyspaces(keyspace, keyspace2);
// TODO: use original tags see defect: https://github.com/Azure/autorest.csharp/issues/1590
var updateOptions = new CassandraKeyspaceCreateOrUpdateContent(AzureLocation.WestUS, keyspace.Data.Resource)
{
Options = new CreateUpdateOptions {
Throughput = TestThroughput2
}
};
keyspace = (await CassandraKeyspaceCollection.CreateOrUpdateAsync(WaitUntil.Completed, _keyspaceName, updateOptions)).Value;
Assert.AreEqual(_keyspaceName, keyspace.Data.Resource.Id);
keyspace2 = await CassandraKeyspaceCollection.GetAsync(_keyspaceName);
VerifyCassandraKeyspaces(keyspace, keyspace2);
}
internal static async Task <CassandraKeyspaceResource> CreateCassandraKeyspace(string name, AutoscaleSettings autoscale, CassandraKeyspaceCollection collection)
{
var cassandraKeyspaceCreateUpdateOptions = new CassandraKeyspaceCreateOrUpdateContent(AzureLocation.WestUS,
new Models.CassandraKeyspaceResource(name))
{
Options = BuildDatabaseCreateUpdateOptions(TestThroughput1, autoscale),
};
var keyspaceLro = await collection.CreateOrUpdateAsync(WaitUntil.Completed, name, cassandraKeyspaceCreateUpdateOptions);
return(keyspaceLro.Value);
}
public virtual async Task <ArmOperation <CassandraKeyspaceResource> > CreateOrUpdateAsync(WaitUntil waitUntil, string keyspaceName, CassandraKeyspaceCreateOrUpdateContent content, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(keyspaceName, nameof(keyspaceName));
Argument.AssertNotNull(content, nameof(content));
using var scope = _cassandraKeyspaceCassandraResourcesClientDiagnostics.CreateScope("CassandraKeyspaceCollection.CreateOrUpdate");
scope.Start();
try
{
var response = await _cassandraKeyspaceCassandraResourcesRestClient.CreateUpdateCassandraKeyspaceAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, keyspaceName, content, cancellationToken).ConfigureAwait(false);
var operation = new CosmosDBArmOperation <CassandraKeyspaceResource>(new CassandraKeyspaceOperationSource(Client), _cassandraKeyspaceCassandraResourcesClientDiagnostics, Pipeline, _cassandraKeyspaceCassandraResourcesRestClient.CreateCreateUpdateCassandraKeyspaceRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, keyspaceName, content).Request, response, OperationFinalStateVia.Location);
if (waitUntil == WaitUntil.Completed)
{
await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
public virtual ArmOperation <CassandraKeyspaceResource> Update(WaitUntil waitUntil, CassandraKeyspaceCreateOrUpdateContent content, CancellationToken cancellationToken = default)
{
Argument.AssertNotNull(content, nameof(content));
using var scope = _cassandraKeyspaceCassandraResourcesClientDiagnostics.CreateScope("CassandraKeyspaceResource.Update");
scope.Start();
try
{
var response = _cassandraKeyspaceCassandraResourcesRestClient.CreateUpdateCassandraKeyspace(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, content, cancellationToken);
var operation = new CosmosDBArmOperation <CassandraKeyspaceResource>(new CassandraKeyspaceOperationSource(Client), _cassandraKeyspaceCassandraResourcesClientDiagnostics, Pipeline, _cassandraKeyspaceCassandraResourcesRestClient.CreateCreateUpdateCassandraKeyspaceRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, content).Request, response, OperationFinalStateVia.Location);
if (waitUntil == WaitUntil.Completed)
{
operation.WaitForCompletion(cancellationToken);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}
