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);
}
public async Task CassandraKeyspaceCreateUpdateTests()
{
var responseIsDatabaseNameExists = await CosmosDBManagementClient.DatabaseAccounts.CheckNameExistsAsync(databaseAccountName);
Assert.AreEqual(true, responseIsDatabaseNameExists.Value);
Assert.AreEqual(200, responseIsDatabaseNameExists.GetRawResponse().Status);
CassandraKeyspaceCreateUpdateParameters cassandraKeyspaceCreateUpdateParameters = new CassandraKeyspaceCreateUpdateParameters(new CassandraKeyspaceResource(keyspaceName), new CreateUpdateOptions(sampleThroughput, new AutoscaleSettings()));
var cassandraKeyspaceResponse1 = await WaitForCompletionAsync(await CosmosDBManagementClient.CassandraResources.StartCreateUpdateCassandraKeyspaceAsync(resourceGroupName, databaseAccountName, keyspaceName, cassandraKeyspaceCreateUpdateParameters));
CassandraKeyspaceResource cassandraKeyspace1 = cassandraKeyspaceResponse1.Value;
Assert.NotNull(cassandraKeyspace1);
Assert.AreEqual(keyspaceName, cassandraKeyspace1.Name);
var cassandraKeyspaceResponse2 = await CosmosDBManagementClient.CassandraResources.GetCassandraKeyspaceAsync(resourceGroupName, databaseAccountName, keyspaceName);
CassandraKeyspaceResource cassandraKeyspace2 = cassandraKeyspaceResponse2.Value;
Assert.NotNull(cassandraKeyspace2);
Assert.AreEqual(keyspaceName, cassandraKeyspace2.Name);
VerifyEqualCassandraDatabases(cassandraKeyspace1, cassandraKeyspace2);
var throughputResponse = CosmosDBManagementClient.CassandraResources.GetCassandraKeyspaceThroughputAsync(resourceGroupName, databaseAccountName, keyspaceName);
ThroughputSettingsData ThroughputSettingsData = throughputResponse.ConfigureAwait(false).GetAwaiter().GetResult();
Assert.NotNull(throughputSettings);
Assert.NotNull(throughputSettings.Name);
Assert.AreEqual(throughputSettings.Resource.Throughput, sampleThroughput);
Assert.AreEqual(cassandraKeyspacesThroughputType, throughputSettings.Type);
CassandraKeyspaceCreateUpdateParameters cassandraKeyspaceCreateUpdateParameters2 = new CassandraKeyspaceCreateUpdateParameters(new CassandraKeyspaceResource(keyspaceName), new CreateUpdateOptions(sampleThroughput2, new AutoscaleSettings()));
var cassandraKeyspaceResponse3 = await WaitForCompletionAsync(await CosmosDBManagementClient.CassandraResources.StartCreateUpdateCassandraKeyspaceAsync(resourceGroupName, databaseAccountName, keyspaceName, cassandraKeyspaceCreateUpdateParameters2));
CassandraKeyspaceResource cassandraKeyspace3 = cassandraKeyspaceResponse3.Value;
Assert.NotNull(cassandraKeyspace3);
Assert.AreEqual(keyspaceName, cassandraKeyspace3.Name);
var cassandraKeyspaceResponse4 = await CosmosDBManagementClient.CassandraResources.GetCassandraKeyspaceAsync(resourceGroupName, databaseAccountName, keyspaceName);
CassandraKeyspaceResource cassandraKeyspace4 = cassandraKeyspaceResponse4.Value;
Assert.NotNull(cassandraKeyspace4);
Assert.AreEqual(keyspaceName, cassandraKeyspace4.Name);
VerifyEqualCassandraDatabases(cassandraKeyspace3, cassandraKeyspace4);
var throughputResponse2 = CosmosDBManagementClient.CassandraResources.GetCassandraKeyspaceThroughputAsync(resourceGroupName, databaseAccountName, keyspaceName);
ThroughputSettingsData throughputSettings2 = throughputResponse2.ConfigureAwait(false).GetAwaiter().GetResult();
Assert.NotNull(throughputSettings2);
Assert.NotNull(throughputSettings2.Name);
Assert.AreEqual(throughputSettings2.Resource.Throughput, sampleThroughput2);
Assert.AreEqual(cassandraKeyspacesThroughputType, throughputSettings2.Type);
}
public async Task TearDown()
{
if (await CassandraKeyspaceCollection.ExistsAsync(_keyspaceName))
{
var id = CassandraKeyspaceCollection.Id;
id = CassandraKeyspaceResource.CreateResourceIdentifier(id.SubscriptionId, id.ResourceGroupName, id.Name, _keyspaceName);
CassandraKeyspaceResource keyspace = this.ArmClient.GetCassandraKeyspaceResource(id);
await keyspace.DeleteAsync(WaitUntil.Completed);
}
}
public async Task CassandraKeyspaceListTests()
{
List <CassandraKeyspaceResource> cassandraKeyspaces = await CosmosDBManagementClient.CassandraResources.ListCassandraKeyspacesAsync(resourceGroupName, databaseAccountName).ToEnumerableAsync();
Assert.NotNull(cassandraKeyspaces);
Assert.AreEqual(cassandraKeyspaces.Count, 1);
CassandraKeyspaceResource cassandraKeyspace = await(CosmosDBManagementClient.CassandraResources.GetCassandraKeyspaceAsync(resourceGroupName, databaseAccountName, keyspaceName));
VerifyEqualCassandraDatabases(cassandraKeyspaces[0], cassandraKeyspace);
}
private void VerifyCassandraKeyspaces(CassandraKeyspaceResource expectedValue, CassandraKeyspaceResource 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);
}
public async Task SetUp()
{
_cassandraKeyspace = await ArmClient.GetCassandraKeyspaceResource(_cassandraKeyspaceId).GetAsync();
}
