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();
GremlinDatabase 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 TearDown()
{
GremlinDatabase database = await GremlinDatabaseCollection.GetIfExistsAsync(_databaseName);
if (database != null)
{
await database.DeleteAsync(WaitUntil.Completed);
}
}
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);
GremlinDatabase gremlinDatabase1 = gremlinDatabaseResponse1.Value;
Assert.NotNull(gremlinDatabase1);
Assert.AreEqual(databaseName, gremlinDatabase1.Name);
var gremlinDatabaseResponse2 = await CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName);
Assert.NotNull(gremlinDatabaseResponse2);
GremlinDatabase 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);
GremlinDatabase gremlinDatabase3 = gremlinDatabaseResponse3.Value;
Assert.NotNull(gremlinDatabase3);
Assert.AreEqual(databaseName, gremlinDatabase3.Name);
var gremlinDatabaseResponse4 = await CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName);
Assert.NotNull(gremlinDatabaseResponse4);
GremlinDatabase 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 GremlinDatabaseListTests()
{
List <GremlinDatabase> gremlinDatabases = await CosmosDBManagementClient.GremlinResources.ListGremlinDatabasesAsync(resourceGroupName, databaseAccountName).ToEnumerableAsync();
Assert.NotNull(gremlinDatabases);
Assert.AreEqual(gremlinDatabases.Count, 1);
GremlinDatabase gremlinDatabase = await(CosmosDBManagementClient.GremlinResources.GetGremlinDatabaseAsync(resourceGroupName, databaseAccountName, databaseName));
VerifyEqualGremlinDatabases(gremlinDatabases[0], gremlinDatabase);
}
private void VerifyEqualGremlinDatabases(GremlinDatabase expectedValue, GremlinDatabase actualValue)
{
Assert.AreEqual(expectedValue.Id, actualValue.Id);
Assert.AreEqual(expectedValue.Name, actualValue.Name);
Assert.AreEqual(expectedValue.Location, actualValue.Location);
Assert.AreEqual(expectedValue.Tags, actualValue.Tags);
Assert.AreEqual(expectedValue.Type, actualValue.Type);
Assert.AreEqual(expectedValue.Options, actualValue.Options);
Assert.AreEqual(expectedValue.Resource.Id, actualValue.Resource.Id);
Assert.AreEqual(expectedValue.Resource.Rid, actualValue.Resource.Rid);
Assert.AreEqual(expectedValue.Resource.Ts, actualValue.Resource.Ts);
Assert.AreEqual(expectedValue.Resource.Etag, actualValue.Resource.Etag);
}
