Skip to content

Commit 640e657

Browse files
author
Meha Kaushik
committed
updating tests
1 parent 21ed968 commit 640e657

File tree

15 files changed

+8129
-6
lines changed

15 files changed

+8129
-6
lines changed

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/CassandraOperationsTests.cs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,5 +47,12 @@ public void TestCassandraCreateUpdateGetCmdlets()
4747
{
4848
TestController.NewInstance.RunPowerShellTest(_logger, "Test-CassandraCreateUpdateGetCmdlets");
4949
}
50+
51+
[Fact]
52+
[Trait(Category.AcceptanceType, Category.CheckIn)]
53+
public void TestCassandraMigrateThroughputCmdlets()
54+
{
55+
TestController.NewInstance.RunPowerShellTest(_logger, "Test-CassandraMigrateThroughputCmdlets");
56+
}
5057
}
5158
}

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/CassandraOperationsTests.ps1

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,61 @@ function Test-CassandraThroughputCmdlets
292292
}
293293
}
294294

295+
<#
296+
.SYNOPSIS
297+
Test Cassandra migrate throughput cmdlets
298+
#>
299+
function Test-CassandraMigrateThroughputCmdlets
300+
{
301+
302+
$AccountName = "db2725"
303+
$rgName = "CosmosDBResourceGroup2510"
304+
$KeyspaceName = "KeyspaceName3"
305+
$TableName = "tableName"
306+
307+
$ThroughputValue = 1200
308+
$TableThroughputValue = 800
309+
310+
$Autoscale = "Autoscale"
311+
$Manual = "Manual"
312+
313+
Try{
314+
$Column1 = New-AzCosmosDBCassandraColumn -Name "ColumnA" -Type "int"
315+
$Column2 = New-AzCosmosDBCassandraColumn -Name "ColumnB" -Type "ascii"
316+
$clusterkey1 = New-AzCosmosDBCassandraClusterKey -Name "ColumnB" -OrderBy "Asc"
317+
$schema = New-AzCosmosDBCassandraSchema -Column $Column1,$Column2 -ClusterKey $clusterkey1 -PartitionKey "ColumnA"
318+
319+
$NewKeyspace = New-AzCosmosDBCassandraKeyspace -AccountName $AccountName -ResourceGroupName $rgName -Name $KeyspaceName -Throughput $ThroughputValue
320+
$Throughput = Get-AzCosmosDBCassandraKeyspaceThroughput -AccountName $AccountName -ResourceGroupName $rgName -Name $KeyspaceName
321+
Assert-AreEqual $Throughput.Throughput $ThroughputValue
322+
Assert-AreEqual $Throughput.AutoscaleSettings.MaxThroughput 0
323+
324+
$AutoscaleThroughput = Migrate-AzCosmosDBCassandraKeyspaceThroughput -InputObject $NewKeyspace -ThroughputType $Autoscale
325+
Assert-AreNotEqual $AutoscaleThroughput.AutoscaleSettings.MaxThroughput 0
326+
327+
$CosmosDBAccount = Get-AzCosmosDBAccount -ResourceGroupName $rgName -Name $AccountName #get parent object
328+
$ManualThroughput = Migrate-AzCosmosDBCassandraKeyspaceThroughput -ParentObject $CosmosDBAccount -Name $KeyspaceName -ThroughputType $Manual
329+
Assert-AreEqual $ManualThroughput.AutoscaleSettings.MaxThroughput 0
330+
331+
$NewTable = New-AzCosmosDBCassandraTable -AccountName $AccountName -ResourceGroupName $rgName -KeyspaceName $KeyspaceName -Name $TableName -Schema $schema -Throughput $TableThroughputValue
332+
$TableThroughput = Get-AzCosmosDBCassandraTableThroughput -AccountName $AccountName -ResourceGroupName $rgName -KeyspaceName $KeyspaceName -Name $TableName
333+
Assert-AreEqual $TableThroughput.Throughput $TableThroughputValue
334+
335+
$AutoscaledTableThroughput = Migrate-AzCosmosDBCassandraTableThroughput -AccountName $AccountName -ResourceGroupName $rgName -KeyspaceName $KeyspaceName -Name $TableName -ThroughputType $Autoscale
336+
Assert-AreNotEqual $AutoscaledTableThroughput.AutoscaleSettings.MaxThroughput 0
337+
338+
$ManuaTableThroughput = Migrate-AzCosmosDBCassandraTableThroughput -InputObject $NewTable -ThroughputType $Manual
339+
Assert-AreEqual $ManuaTableThroughput.AutoscaleSettings.MaxThroughput 0
340+
341+
Remove-AzCosmosDBCassandraTable -InputObject $NewTable
342+
Remove-AzCosmosDBCassandraKeyspace -InputObject $NewKeyspace
343+
}
344+
Finally{
345+
Remove-AzCosmosDBCassandraTable -AccountName $AccountName -ResourceGroupName $rgName -KeyspaceName $KeyspaceName -Name $TableName
346+
Remove-AzCosmosDBCassandraKeyspace -AccountName $AccountName -ResourceGroupName $rgName -Name $KeyspaceName
347+
}
348+
}
349+
295350
function Validate-EqualColumns($Column1, $Column2)
296351
{
297352
Assert-AreEqual $Column1.Name $Column2.Name

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/GremlinOperationsTests.cs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,5 +47,12 @@ public void TestGremlinThroughputCmdlets()
4747
{
4848
TestController.NewInstance.RunPowerShellTest(_logger, "Test-GremlinThroughputCmdlets");
4949
}
50+
51+
[Fact]
52+
[Trait(Category.AcceptanceType, Category.CheckIn)]
53+
public void TestGremlinMigrateThroughputCmdlets()
54+
{
55+
TestController.NewInstance.RunPowerShellTest(_logger, "Test-GremlinMigrateThroughputCmdlets");
56+
}
5057
}
5158
}

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/GremlinOperationsTests.ps1

Lines changed: 53 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -305,4 +305,57 @@ function Test-GremlinThroughputCmdlets
305305
Remove-AzCosmosDBGremlinGraph -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $GraphName
306306
Remove-AzCosmosDBGremlinDatabase -AccountName $AccountName -ResourceGroupName $rgName -Name $DatabaseName
307307
}
308+
}
309+
310+
<#
311+
.SYNOPSIS
312+
Test Gremlin migrate throughput cmdlets
313+
#>
314+
function Test-GremlinMigrateThroughputCmdlets
315+
{
316+
$AccountName = "db1002"
317+
$rgName = "CosmosDBResourceGroup2510"
318+
$DatabaseName = "dbName4"
319+
$GraphName = "graphName"
320+
321+
$PartitionKeyPathValue = "/foo"
322+
$PartitionKeyKindValue = "Hash"
323+
324+
$ThroughputValue = 1200
325+
326+
$GraphThroughputValue = 800
327+
328+
$Autoscale = "Autoscale"
329+
$Manual = "Manual"
330+
331+
Try{
332+
$NewDatabase = New-AzCosmosDBGremlinDatabase -AccountName $AccountName -ResourceGroupName $rgName -Name $DatabaseName -Throughput $ThroughputValue
333+
$Throughput = Get-AzCosmosDBGremlinDatabaseThroughput -AccountName $AccountName -ResourceGroupName $rgName -Name $DatabaseName
334+
Assert-AreEqual $Throughput.Throughput $ThroughputValue
335+
Assert-AreEqual $Throughput.AutoscaleSettings.MaxThroughput 0
336+
337+
$AutoscaleThroughput = Migrate-AzCosmosDBGremlinDatabaseThroughput -InputObject $NewDatabase -ThroughputType $Autoscale
338+
Assert-AreNotEqual $AutoscaleThroughput.AutoscaleSettings.MaxThroughput 0
339+
340+
$CosmosDBAccount = Get-AzCosmosDBAccount -ResourceGroupName $rgName -Name $AccountName #get parent object
341+
$ManualThroughput = Migrate-AzCosmosDBGremlinDatabaseThroughput -ParentObject $CosmosDBAccount -Name $DatabaseName -ThroughputType $Manual
342+
Assert-AreEqual $ManualThroughput.AutoscaleSettings.MaxThroughput 0
343+
344+
$NewGraph = New-AzCosmosDBGremlinGraph -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Throughput $GraphThroughputValue -Name $GraphName -PartitionKeyPath $PartitionKeyPathValue -PartitionKeyKind $PartitionKeyKindValue
345+
$GraphThroughput = Get-AzCosmosDBGremlinGraphThroughput -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $GraphName
346+
Assert-AreEqual $GraphThroughput.Throughput $GraphThroughputValue
347+
348+
$AutoscaledGraphThroughput = Migrate-AzCosmosDBGremlinGraphThroughput -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $GraphName -ThroughputType $Autoscale
349+
Assert-AreNotEqual $AutoscaledGraphThroughput.AutoscaleSettings.MaxThroughput 0
350+
351+
$ManualGraphThroughput = Migrate-AzCosmosDBGremlinGraphThroughput -InputObject $NewGraph -ThroughputType $Manual
352+
Assert-AreEqual $ManualGraphThroughput.AutoscaleSettings.MaxThroughput 0
353+
354+
Remove-AzCosmosDBGremlinGraph -InputObject $NewGraph
355+
Remove-AzCosmosDBGremlinDatabase -InputObject $NewDatabase
356+
}
357+
Finally{
358+
Remove-AzCosmosDBGremlinGraph -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $GraphName
359+
Remove-AzCosmosDBGremlinDatabase -AccountName $AccountName -ResourceGroupName $rgName -Name $DatabaseName
360+
}
308361
}

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/MongoOperationsTests.cs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,5 +47,12 @@ public void TestMongoThroughputCmdlets()
4747
{
4848
TestController.NewInstance.RunPowerShellTest(_logger, "Test-MongoThroughputCmdlets");
4949
}
50+
51+
[Fact]
52+
[Trait(Category.AcceptanceType, Category.CheckIn)]
53+
public void TestMongoMigrateThroughputCmdlets()
54+
{
55+
TestController.NewInstance.RunPowerShellTest(_logger, "Test-MongoMigrateThroughputCmdlets");
56+
}
5057
}
5158
}

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/MongoOperationsTests.ps1

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -358,4 +358,55 @@ function Validate-EqualLists($list1, $list2)
358358
{
359359
Assert-true($list1 -contains $ele)
360360
}
361+
}
362+
363+
<#
364+
.SYNOPSIS
365+
Test Mongo migrate throughput cmdlets
366+
#>
367+
function Test-MongoMigrateThroughputCmdlets
368+
{
369+
370+
$AccountName = "db001"
371+
$rgName = "CosmosDBResourceGroup3668"
372+
$DatabaseName = "dbName4"
373+
$CollectionName = "collectionName"
374+
375+
$ShardKey = "shardKeyPath"
376+
$ThroughputValue = 1200
377+
$CollectionThroughputValue = 800
378+
379+
$Autoscale = "Autoscale"
380+
$Manual = "Manual"
381+
382+
Try{
383+
$NewDatabase = New-AzCosmosDBMongoDBDatabase -AccountName $AccountName -ResourceGroupName $rgName -Name $DatabaseName -Throughput $ThroughputValue
384+
$Throughput = Get-AzCosmosDBMongoDBDatabaseThroughput -AccountName $AccountName -ResourceGroupName $rgName -Name $DatabaseName
385+
Assert-AreEqual $Throughput.Throughput $ThroughputValue
386+
Assert-AreEqual $Throughput.AutoscaleSettings.MaxThroughput 0
387+
388+
$AutoscaleThroughput = Migrate-AzCosmosDBMongoDBDatabaseThroughput -InputObject $NewDatabase -ThroughputType $Autoscale
389+
Assert-AreNotEqual $AutoscaleThroughput.AutoscaleSettings.MaxThroughput 0
390+
391+
$CosmosDBAccount = Get-AzCosmosDBAccount -ResourceGroupName $rgName -Name $AccountName #get parent object
392+
$ManualThroughput = Migrate-AzCosmosDBMongoDBDatabaseThroughput -ParentObject $CosmosDBAccount -Name $DatabaseName -ThroughputType $Manual
393+
Assert-AreEqual $ManualThroughput.AutoscaleSettings.MaxThroughput 0
394+
395+
$NewCollection = New-AzCosmosDBMongoDBCollection -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Throughput $CollectionThroughputValue -Name $CollectionName -Shard $ShardKey
396+
$CollectionThroughput = Get-AzCosmosDBMongoDBCollectionThroughput -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $CollectionName
397+
Assert-AreEqual $CollectionThroughput.Throughput $CollectionThroughputValue
398+
399+
$AutoscaledCollectionThroughput = Migrate-AzCosmosDBMongoDBCollectionThroughput -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $CollectionName -ThroughputType $Autoscale
400+
Assert-AreNotEqual $AutoscaledCollectionThroughput.AutoscaleSettings.MaxThroughput 0
401+
402+
$ManualCollectionThroughput = Migrate-AzCosmosDBMongoDBCollectionThroughput -InputObject $NewCollection -ThroughputType $Manual
403+
Assert-AreEqual $ManualCollectionThroughput.AutoscaleSettings.MaxThroughput 0
404+
405+
Remove-AzCosmosDBMongoDBCollection -InputObject $NewCollection
406+
Remove-AzCosmosDBMongoDBDatabase -InputObject $NewDatabase
407+
}
408+
Finally{
409+
Remove-AzCosmosDBMongoDBCollection -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $CollectionName
410+
Remove-AzCosmosDBMongoDBDatabase -AccountName $AccountName -ResourceGroupName $rgName -Name $DatabaseName
411+
}
361412
}

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/SqlOperationsTests.ps1

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -573,7 +573,7 @@ function Test-SqlThroughputCmdlets
573573

574574
<#
575575
.SYNOPSIS
576-
Test SQL throughput cmdlets using all parameter sets
576+
Test SQL migrate throughput cmdlets
577577
#>
578578
function Test-SqlMigrateThroughputCmdlets
579579
{
@@ -599,7 +599,7 @@ function Test-SqlMigrateThroughputCmdlets
599599
Assert-AreEqual $Throughput.AutoscaleSettings.MaxThroughput 0
600600

601601
$AutoscaleThroughput = Migrate-AzCosmosDBSqlDatabaseThroughput -InputObject $NewDatabase -ThroughputType $Autoscale
602-
Assert-NotEqual $UpdatedThroughput.AutoscaleSettings.MaxThroughput 0
602+
Assert-AreNotEqual $UpdatedThroughput.AutoscaleSettings.MaxThroughput 0
603603

604604
$CosmosDBAccount = Get-AzCosmosDBAccount -ResourceGroupName $rgName -Name $AccountName #get parent object
605605
$ManualThroughput = Migrate-AzCosmosDBSqlDatabaseThroughput -ParentObject $CosmosDBAccount -Name $DatabaseName -ThroughputType $Manual
@@ -610,7 +610,7 @@ function Test-SqlMigrateThroughputCmdlets
610610
Assert-AreEqual $ContainerThroughput.Throughput $ContainerThroughputValue
611611

612612
$AutoscaledContainerThroughput = Migrate-AzCosmosDBSqlContainerThroughput -AccountName $AccountName -ResourceGroupName $rgName -DatabaseName $DatabaseName -Name $ContainerName -ThroughputType $Autoscale
613-
Assert-NotEqual $UpdatedContainerThroughput.AutoscaleSettings.MaxThroughput 0
613+
Assert-AreNotEqual $UpdatedContainerThroughput.AutoscaleSettings.MaxThroughput 0
614614

615615
$ManualContainerThroughput = Migrate-AzCosmosDBSqlContainerThroughput -InputObject $NewContainer -ThroughputType $Manual
616616
Assert-AreEqual $UpdatedContainerThroughput.AutoscaleSettings.MaxThroughput 0

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/TableOperationsTests.cs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,5 +47,12 @@ public void TestTableThroughputCmdlets()
4747
{
4848
TestController.NewInstance.RunPowerShellTest(_logger, "Test-TableThroughputCmdlets");
4949
}
50+
51+
[Fact]
52+
[Trait(Category.AcceptanceType, Category.CheckIn)]
53+
public void TestTableMigrateThroughputCmdlets()
54+
{
55+
TestController.NewInstance.RunPowerShellTest(_logger, "Test-TableMigrateThroughputCmdlets");
56+
}
5057
}
5158
}

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/TableOperationsTests.ps1

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -161,4 +161,41 @@ function Test-TableThroughputCmdlets
161161
Finally{
162162
Remove-AzCosmosDBTable -AccountName $AccountName -ResourceGroupName $rgName -Name $TableName
163163
}
164+
}
165+
166+
<#
167+
.SYNOPSIS
168+
Test Cassandra migrate throughput cmdlets
169+
#>
170+
function Test-TableMigrateThroughputCmdlets
171+
{
172+
173+
$AccountName = "db2527"
174+
$rgName = "CosmosDBResourceGroup2510"
175+
$TableName = "tableName4"
176+
177+
$ThroughputValue = 1200
178+
$TableThroughputValue = 800
179+
180+
$Autoscale = "Autoscale"
181+
$Manual = "Manual"
182+
183+
Try{
184+
$NewTable = New-AzCosmosDBTable -AccountName $AccountName -ResourceGroupName $rgName -Name $TableName -Throughput $ThroughputValue
185+
$Throughput = Get-AzCosmosDBTableThroughput -AccountName $AccountName -ResourceGroupName $rgName -Name $TableName
186+
Assert-AreEqual $Throughput.Throughput $ThroughputValue
187+
Assert-AreEqual $Throughput.AutoscaleSettings.MaxThroughput 0
188+
189+
$AutoscaleThroughput = Migrate-AzCosmosDBTableThroughput -InputObject $NewTable -ThroughputType $Autoscale
190+
Assert-AreNotEqual $AutoscaleThroughput.AutoscaleSettings.MaxThroughput 0
191+
192+
$CosmosDBAccount = Get-AzCosmosDBAccount -ResourceGroupName $rgName -Name $AccountName #get parent object
193+
$ManualThroughput = Migrate-AzCosmosDBTableThroughput -ParentObject $CosmosDBAccount -Name $TableName -ThroughputType $Manual
194+
Assert-AreEqual $ManualThroughput.AutoscaleSettings.MaxThroughput 0
195+
196+
Remove-AzCosmosDBTable -InputObject $NewTable
197+
}
198+
Finally{
199+
Remove-AzCosmosDBTable -AccountName $AccountName -ResourceGroupName $rgName -Name $TableName
200+
}
164201
}

0 commit comments

Comments
 (0)