Fix streaming ingestion policy error

microsoft · Oct 16, 2023 · 5c83163 · 5c83163
1 parent e97138e
commit 5c83163
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 14 deletions.
diff --git a/code/DeltaKustoLib/CommandModel/Policies/AlterStreamingIngestionPolicyCommand.cs b/code/DeltaKustoLib/CommandModel/Policies/AlterStreamingIngestionPolicyCommand.cs
@@ -72,18 +72,10 @@ public override string ToScript(ScriptingContext? context = null)
             var entityType = ExtractEntityType(commandBlock);
             var nameReferences = commandBlock.GetDescendants<NameReference>();
             var entityNameReference = nameReferences.Last();
-            //  Weird parser behaviour:  if the db (or db + cluster) is used
-            //  in the name, the policy body is encoded differently
-            var policyText = nameReferences.Count == 1
-                ? QuotedText.FromLiteral(
-                    commandBlock.GetUniqueDescendant<LiteralExpression>(
-                        "StreamingIngestion",
-                        e => e.NameInParent == "StreamingIngestionPolicy"))
-                : QuotedText.FromText(
-                    commandBlock
-                    .GetUniqueDescendant<SkippedTokens>(
-                        "StreamingIngestion")
-                    .ToString());
+            var policyText = QuotedText.FromLiteral(
+                commandBlock.GetUniqueDescendant<LiteralExpression>(
+                    "StreamingIngestion",
+                    e => e.NameInParent == "StreamingIngestionPolicy"));
             var policy = Deserialize<JsonDocument>(policyText.Text);
 
             if (policy == null)

diff --git a/code/DeltaKustoUnitTest/CommandParsing/Policies/AlterStreamingIngestionPolicyTest.cs b/code/DeltaKustoUnitTest/CommandParsing/Policies/AlterStreamingIngestionPolicyTest.cs
@@ -26,9 +26,11 @@ public void DbComposedTableName()
         {
             var command = ParseOneCommand(@$"
 .alter table mydb.mytable policy streamingingestion
+```
 {{
     ""IsEnabled"":true
-}}");
+}}
+```");
 
             Assert.IsType<AlterStreamingIngestionPolicyCommand>(command);
 
@@ -43,10 +45,12 @@ public void ClusterComposedTableName()
         {
             var command = ParseOneCommand(@$"
 .alter table mycluster.['my db'].mytable policy streamingingestion
+```
 {{
     ""IsEnabled"":true,
     ""HintAllocatedRate"":4.5
-}}");
+}}
+```");
 
             Assert.IsType<AlterStreamingIngestionPolicyCommand>(command);