Add support for managing cluster capacity policy

KustoSchemaTools.Tests/Changes/ClusterChangesTest.cs

@@ -0,0 +1,122 @@
+using KustoSchemaTools.Changes;
+using KustoSchemaTools.Model;
+using Microsoft.Extensions.Logging;
+using Moq;
+
+namespace KustoSchemaTools.Tests.Changes
+{
+    public class ClusterChangesTests
+    {
+        private readonly Mock<ILogger> _loggerMock;
+
+        public ClusterChangesTests()
+        {
+            _loggerMock = new Mock<ILogger>();
+        }
+
+        [Fact]
+        public void GenerateChanges_WithIdenticalPolicies_ShouldDetectNoChanges()
+        {
+            // Arrange
+            var oldCluster = CreateClusterWithPolicy(0.2, 1, 2, 3);
+            var newCluster = CreateClusterWithPolicy(0.2, 1, 2, 3);
+
+            // Act
+            var changeSet = ClusterChanges.GenerateChanges(oldCluster, newCluster, _loggerMock.Object);
+
+            // Assert
+            Assert.NotNull(changeSet);
+            Assert.Empty(changeSet.Changes);
+        }
+        [Fact]
+        public void GenerateChanges_WithSingleChange_ShouldDetectChangeAndCreateScript()
+        {
+            // Arrange
+            var oldCluster = CreateClusterWithPolicy(0.2, 1, 2, 3);
+            var newCluster = CreateClusterWithPolicy(0.2, 1, 2, 5);
+
+            // Act
+            var changeSet = ClusterChanges.GenerateChanges(oldCluster, newCluster, _loggerMock.Object);
+
+            // Assert
+            Assert.NotNull(changeSet);
+            Assert.NotEmpty(changeSet.Changes);
+            Assert.NotEmpty(changeSet.Scripts);
+
+            // Asserts that there is exactly one policy change in the change set
+            var policyChange = Assert.Single(changeSet.Changes) as PolicyChange<ClusterCapacityPolicy>;
+            Assert.NotNull(policyChange);
+
+            // Assert that the correct script is generated
+            var expectedScript = newCluster.CapacityPolicy!.ToUpdateScript();
+            var actualScriptContainer = Assert.Single(changeSet.Scripts);
+            Assert.Equal(expectedScript, actualScriptContainer.Script.Text);
+        }
+
+        [Fact]
+        public void GenerateChanges_WithMultipleChanges_ShouldDetectAllChanges()
+        {
+            // Arrange
+            var oldCluster = CreateClusterWithPolicy(ingestionCapacityCoreUtilizationCoefficient: 0.75, materializedViewsCapacityClusterMaximumConcurrentOperations: 10);
+            var newCluster = CreateClusterWithPolicy(ingestionCapacityCoreUtilizationCoefficient: 0.95, materializedViewsCapacityClusterMaximumConcurrentOperations: 20);
+
+            // Act
+            var changeSet = ClusterChanges.GenerateChanges(oldCluster, newCluster, _loggerMock.Object);
+
+            // Assert
+            var policyChange = Assert.Single(changeSet.Changes) as PolicyChange<ClusterCapacityPolicy>;
+            Assert.NotNull(policyChange);
+
+            // Assert that the correct script is generated
+            var expectedScript = newCluster.CapacityPolicy!.ToUpdateScript();
+            var actualScriptContainer = Assert.Single(changeSet.Scripts);
+            Assert.Equal(expectedScript, actualScriptContainer.Script.Text);
+        }
+
+        [Fact]
+        public void GenerateChanges_WithNullNewCapacityPolicy_ShouldNotGenerateChanges()
+        {
+            // Arrange
+            var oldCluster = CreateClusterWithPolicy(ingestionCapacityCoreUtilizationCoefficient: 0.75);
+            var newCluster = new Cluster { Name = oldCluster.Name, CapacityPolicy = null };
+
+            // Act
+            var changeSet = ClusterChanges.GenerateChanges(oldCluster, newCluster, _loggerMock.Object);
+
+            // Assert
+            Assert.NotNull(changeSet);
+            Assert.Empty(changeSet.Changes);
+            Assert.Empty(changeSet.Scripts);
+        }
+
+        #region Helper Methods
+        private Cluster CreateClusterWithPolicy(
+            double? ingestionCapacityCoreUtilizationCoefficient = null,
+            int? materializedViewsCapacityClusterMaximumConcurrentOperations = null,
+            int? extentsRebuildClusterMaximumConcurrentOperations = null,
+            int? extentsRebuildMaximumConcurrentOperationsPerNode = null
+        )
+        {
+            return new Cluster
+            {
+                CapacityPolicy = new ClusterCapacityPolicy
+                {
+                    MaterializedViewsCapacity = new MaterializedViewsCapacity
+                    {
+                        ClusterMaximumConcurrentOperations = materializedViewsCapacityClusterMaximumConcurrentOperations,
+                        ExtentsRebuildCapacity = (extentsRebuildClusterMaximumConcurrentOperations != null || extentsRebuildMaximumConcurrentOperationsPerNode != null) ? new ExtentsRebuildCapacity
+                        {
+                            ClusterMaximumConcurrentOperations = extentsRebuildClusterMaximumConcurrentOperations,
+                            MaximumConcurrentOperationsPerNode = extentsRebuildMaximumConcurrentOperationsPerNode
+                        } : null
+                    },
+                    IngestionCapacity = new IngestionCapacity
+                    {
+                        CoreUtilizationCoefficient = ingestionCapacityCoreUtilizationCoefficient
+                    },
+                }
+            };
+        }
+        #endregion
+    }
+}

KustoSchemaTools.Tests/DemoData/ClusterScopedChanges/multipleClusters.yml

@@ -0,0 +1,16 @@
+connections:
+- name: test1
+  url: test1.eastus
+  capacityPolicy:
+    ingestionCapacity:
+      clusterMaximumConcurrentOperations: 512
+      coreUtilizationCoefficient: 0.75
+    extentsMergeCapacity:
+      minimumConcurrentOperationsPerNode: 1
+      maximumConcurrentOperationsPerNode: 3
+- name: test2
+  url: test2.eastus
+  capacityPolicy:
+    ingestionCapacity:
+      clusterMaximumConcurrentOperations: 500
+      coreUtilizationCoefficient: 0.8