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);
GremlinDatabaseCreateUpdateData updateOptions = new GremlinDatabaseCreateUpdateData(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);
}
public async Task GremlinDatabaseCreateUpdateTests()
{
GremlinDatabaseCreateUpdateParameters gremlinDatabaseCreateUpdateParameters = new GremlinDatabaseCreateUpdateParameters(new GremlinDatabaseResource(databaseName), new CreateUpdateOptions(sampleThroughput, new AutoscaleSettings()));
var gremlinDatabaseResponse1 = await WaitForCompletionAsync(await CosmosDBManagementClient.GremlinResources.StartCreateUpdateGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName, gremlinDatabaseCreateUpdateParameters));
Assert.NotNull(gremlinDatabaseResponse1);
GremlinDatabaseResource gremlinDatabase1 = gremlinDatabaseResponse1.Value;
Assert.NotNull(gremlinDatabase1);
Assert.AreEqual(databaseName, gremlinDatabase1.Name);
var gremlinDatabaseResponse2 = await CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName);
Assert.NotNull(gremlinDatabaseResponse2);
GremlinDatabaseResource gremlinDatabase2 = gremlinDatabaseResponse2.Value;
Assert.NotNull(gremlinDatabase2);
Assert.AreEqual(databaseName, gremlinDatabase2.Name);
VerifyEqualGremlinDatabases(gremlinDatabase1, gremlinDatabase2);
var throughputResponse = CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseThroughputAsync(resourceGroupName, databaseAccountName, databaseName);
ThroughputSettingsData ThroughputSettingsData = throughputResponse.ConfigureAwait(false).GetAwaiter().GetResult();
Assert.NotNull(throughputSettings);
Assert.NotNull(throughputSettings.Name);
Assert.AreEqual(throughputSettings.Resource.Throughput, sampleThroughput);
Assert.AreEqual(gremlinDatabasesThroughputType, throughputSettings.Type);
GremlinDatabaseCreateUpdateParameters gremlinDatabaseCreateUpdateParameters2 = new GremlinDatabaseCreateUpdateParameters(new GremlinDatabaseResource(databaseName), new CreateUpdateOptions(sampleThroughput2, new AutoscaleSettings()));
var gremlinDatabaseResponse3 = await WaitForCompletionAsync(await CosmosDBManagementClient.GremlinResources.StartCreateUpdateGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName, gremlinDatabaseCreateUpdateParameters2));
Assert.NotNull(gremlinDatabaseResponse3);
GremlinDatabaseResource gremlinDatabase3 = gremlinDatabaseResponse3.Value;
Assert.NotNull(gremlinDatabase3);
Assert.AreEqual(databaseName, gremlinDatabase3.Name);
var gremlinDatabaseResponse4 = await CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName);
Assert.NotNull(gremlinDatabaseResponse4);
GremlinDatabaseResource gremlinDatabase4 = gremlinDatabaseResponse4.Value;
Assert.NotNull(gremlinDatabase4);
Assert.AreEqual(databaseName, gremlinDatabase4.Name);
VerifyEqualGremlinDatabases(gremlinDatabase3, gremlinDatabase4);
var throughputResponse2 = CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseThroughputAsync(resourceGroupName, databaseAccountName, databaseName);
ThroughputSettingsData throughputSettings2 = throughputResponse2.ConfigureAwait(false).GetAwaiter().GetResult();
Assert.NotNull(throughputSettings2);
Assert.NotNull(throughputSettings2.Name);
Assert.AreEqual(throughputSettings2.Resource.Throughput, sampleThroughput2);
Assert.AreEqual(gremlinDatabasesThroughputType, throughputSettings2.Type);
}
public async Task TearDown()
{
if (await GremlinDatabaseCollection.ExistsAsync(_databaseName))
{
var id = GremlinDatabaseCollection.Id;
id = GremlinDatabaseResource.CreateResourceIdentifier(id.SubscriptionId, id.ResourceGroupName, id.Name, _databaseName);
GremlinDatabaseResource database = ArmClient.GetGremlinDatabaseResource(id);
await database.DeleteAsync(WaitUntil.Completed);
}
}
public async Task GremlinDatabaseListTests()
{
List <GremlinDatabaseResource> gremlinDatabases = await CosmosDBManagementClient.GremlinResources.ListGremlinDatabasesAsync(resourceGroupName, databaseAccountName).ToEnumerableAsync();
Assert.NotNull(gremlinDatabases);
Assert.AreEqual(gremlinDatabases.Count, 1);
GremlinDatabaseResource gremlinDatabase = await(CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName));
VerifyEqualGremlinDatabases(gremlinDatabases[0], gremlinDatabase);
}
private void VerifyGremlinDatabases(GremlinDatabaseResource expectedValue, GremlinDatabaseResource actualValue)
{
Assert.AreEqual(expectedValue.Id, actualValue.Id);
Assert.AreEqual(expectedValue.Data.Name, actualValue.Data.Name);
Assert.AreEqual(expectedValue.Data.Location, actualValue.Data.Location);
Assert.AreEqual(expectedValue.Data.Tags, actualValue.Data.Tags);
Assert.AreEqual(expectedValue.Data.ResourceType, actualValue.Data.ResourceType);
Assert.AreEqual(expectedValue.Data.Options, actualValue.Data.Options);
Assert.AreEqual(expectedValue.Data.Resource.Id, actualValue.Data.Resource.Id);
Assert.AreEqual(expectedValue.Data.Resource.Rid, actualValue.Data.Resource.Rid);
Assert.AreEqual(expectedValue.Data.Resource.Ts, actualValue.Data.Resource.Ts);
Assert.AreEqual(expectedValue.Data.Resource.Etag, actualValue.Data.Resource.Etag);
}
