Add UseFreeLimit and FreeLimitExhaustionBehavior parameters to DB CRUD (#22884)

* Add UseFreeLimit and FreeLimitExhaustionBehavior parameters to database Get, Set and New

* Update ChangeLog.md

---------

Co-authored-by: Yabo Hu <[email protected]>

Author: mykolian

src/Sql/Sql.Test/ScenarioTests/DatabaseCrudTests.cs

@@ -234,5 +234,19 @@ public void TestRevalidateAndRevertAKVKeyForDatabaseWithPerDBCMK()
         {
             TestRunner.RunTestScript("Test-RevalidateAndRevertAKVKeyForDatabaseWithPerDBCMK");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestFreeLimitDatabaseCreate()
+        {
+            TestRunner.RunTestScript("Test-FreeLimitDatabaseCreate");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestFreeLimitDatabaseUpdateFreeLimitExhaustionBehavior()
+        {
+            TestRunner.RunTestScript("Test-FreeLimitDatabaseUpdateFreeLimitExhaustionBehavior");
+        }
     }
 }

src/Sql/Sql.Test/ScenarioTests/DatabaseCrudTests.ps1

@@ -443,6 +443,54 @@ function Test-CreateDatabaseWithPreferredEnclaveType
 	}
 }
 
+function Test-FreeLimitDatabaseCreate
+{
+	#Setup 
+	$location = "eastus2euap"
+	$rg = Create-ResourceGroupForTest
+	$server = Create-ServerForTest $rg $location
+
+	try
+	{
+		$databaseName = Get-DatabaseName
+		$db = New-AzSqlDatabase -ResourceGroupName $rg.ResourceGroupName -ServerName $server.ServerName -DatabaseName $databaseName -RequestedServiceObjectiveName GP_S_Gen5_2 -Edition GeneralPurpose -UseFreeLimit -FreeLimitExhaustionBehavior AutoPause
+		Assert-AreEqual true $db.UseFreeLimit
+		Assert-AreEqual AutoPause $db.FreeLimitExhaustionBehavior
+	}
+	finally
+	{
+		Remove-ResourceGroupForTest $rg
+	}
+}
+
+function Test-FreeLimitDatabaseUpdateFreeLimitExhaustionBehavior
+{
+	#Setup 
+	$location = "eastus2euap"
+	$rg = Create-ResourceGroupForTest
+	$server = Create-ServerForTest $rg $location
+
+	$databaseName = Get-DatabaseName 
+
+	$db = New-AzSqlDatabase -ResourceGroupName $rg.ResourceGroupName -ServerName $server.ServerName -DatabaseName $databaseName `
+		-RequestedServiceObjectiveName GP_S_Gen5_2 -Edition GeneralPurpose -UseFreeLimit -FreeLimitExhaustionBehavior AutoPause
+	Assert-AreEqual $db.DatabaseName $databaseName
+
+	try
+	{
+		$job = Set-AzSqlDatabase -ResourceGroupName $rg.ResourceGroupName -ServerName $server.ServerName -DatabaseName $databaseName -FreeLimitExhaustionBehavior BillOverUsage -AsJob
+		$job | Wait-Job
+		$db1 = $job.Output
+
+		Assert-AreEqual true $db1.UseFreeLimit
+		Assert-AreEqual BillOverUsage $db1.FreeLimitExhaustionBehavior
+	}
+	finally
+	{
+		Remove-ResourceGroupForTest $rg
+	}
+}
+
 <#
 	.SYNOPSIS
 	Tests updating a database