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); }