public async Task GremlinDatabaseCreateAndUpdate()
{
var database = await CreateGremlinDatabase(null);
Assert.AreEqual(_databaseName, database.Data.Resource.Id);
// Seems bug in swagger definition
//Assert.AreEqual(TestThroughput1, database.Data.Options.Throughput);
bool ifExists = await GremlinDatabaseCollection.ExistsAsync(_databaseName);
Assert.True(ifExists);
// NOT WORKING API
//ThroughputSettingsData throughtput = await database.GetMongoDBCollectionThroughputAsync();
GremlinDatabaseResource database2 = await GremlinDatabaseCollection.GetAsync(_databaseName);
Assert.AreEqual(_databaseName, database2.Data.Resource.Id);
//Assert.AreEqual(TestThroughput1, database2.Data.Options.Throughput);
VerifyGremlinDatabases(database, database2);
var updateOptions = new GremlinDatabaseCreateOrUpdateContent(database.Id, _databaseName, database.Data.ResourceType, null,
new Dictionary <string, string>(),// TODO: use original tags see defect: https://github.com/Azure/autorest.csharp/issues/1590
AzureLocation.WestUS, database.Data.Resource, new CreateUpdateOptions {
Throughput = TestThroughput2
});
database = await(await GremlinDatabaseCollection.CreateOrUpdateAsync(WaitUntil.Started, _databaseName, updateOptions)).WaitForCompletionAsync();
Assert.AreEqual(_databaseName, database.Data.Resource.Id);
database2 = await GremlinDatabaseCollection.GetAsync(_databaseName);
VerifyGremlinDatabases(database, database2);
}
internal static async Task <GremlinDatabaseResource> CreateGremlinDatabase(string name, AutoscaleSettings autoscale, GremlinDatabaseCollection collection)
{
var cassandraKeyspaceCreateUpdateOptions = new GremlinDatabaseCreateOrUpdateContent(AzureLocation.WestUS,
new Models.GremlinDatabaseResource(name))
{
Options = BuildDatabaseCreateUpdateOptions(TestThroughput1, autoscale),
};
var databaseLro = await collection.CreateOrUpdateAsync(WaitUntil.Completed, name, cassandraKeyspaceCreateUpdateOptions);
return(databaseLro.Value);
}
public virtual ArmOperation <GremlinDatabaseResource> Update(WaitUntil waitUntil, GremlinDatabaseCreateOrUpdateContent content, CancellationToken cancellationToken = default)
{
Argument.AssertNotNull(content, nameof(content));
using var scope = _gremlinDatabaseGremlinResourcesClientDiagnostics.CreateScope("GremlinDatabaseResource.Update");
scope.Start();
try
{
var response = _gremlinDatabaseGremlinResourcesRestClient.CreateUpdateGremlinDatabase(Id.SubscriptionId, Id.ResourceGroupName, Id.Parent.Name, Id.Name, content, cancellationToken);
var operation = new CosmosDBArmOperation <GremlinDatabaseResource>(new GremlinDatabaseOperationSource(Client), _gremlinDatabaseGremlinResourcesClientDiagnostics, Pipeline, _gremlinDatabaseGremlinResourcesRestClient.CreateCreateUpdateGremlinDatabaseRequest(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;
}
}
public virtual async Task <ArmOperation <GremlinDatabaseResource> > CreateOrUpdateAsync(WaitUntil waitUntil, string databaseName, GremlinDatabaseCreateOrUpdateContent content, CancellationToken cancellationToken = default)
{
Argument.AssertNotNullOrEmpty(databaseName, nameof(databaseName));
Argument.AssertNotNull(content, nameof(content));
using var scope = _gremlinDatabaseGremlinResourcesClientDiagnostics.CreateScope("GremlinDatabaseCollection.CreateOrUpdate");
scope.Start();
try
{
var response = await _gremlinDatabaseGremlinResourcesRestClient.CreateUpdateGremlinDatabaseAsync(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, databaseName, content, cancellationToken).ConfigureAwait(false);
var operation = new CosmosDBArmOperation <GremlinDatabaseResource>(new GremlinDatabaseOperationSource(Client), _gremlinDatabaseGremlinResourcesClientDiagnostics, Pipeline, _gremlinDatabaseGremlinResourcesRestClient.CreateCreateUpdateGremlinDatabaseRequest(Id.SubscriptionId, Id.ResourceGroupName, Id.Name, databaseName, content).Request, response, OperationFinalStateVia.Location);
if (waitUntil == WaitUntil.Completed)
{
await operation.WaitForCompletionAsync(cancellationToken).ConfigureAwait(false);
}
return(operation);
}
catch (Exception e)
{
scope.Failed(e);
throw;
}
}