test: basic tests for KpiHierarchy validation

fraunhofer-iem · Aug 26, 2024 · f1a013f · f1a013f
1 parent 6d8aeda
commit f1a013f
Showing 1 changed file with 62 additions and 0 deletions.
diff --git a/.../src/test/kotlin/de/fraunhofer/iem/kpiCalculator/core/hierarchy/HierarchyValidatorTest.kt b/.../src/test/kotlin/de/fraunhofer/iem/kpiCalculator/core/hierarchy/HierarchyValidatorTest.kt
@@ -11,11 +11,24 @@ package de.fraunhofer.iem.kpiCalculator.core.hierarchy
 
 import de.fraunhofer.iem.kpiCalculator.model.kpi.KpiId
 import de.fraunhofer.iem.kpiCalculator.model.kpi.KpiStrategyId
+import de.fraunhofer.iem.kpiCalculator.model.kpi.hierarchy.KpiEdge
 import de.fraunhofer.iem.kpiCalculator.model.kpi.hierarchy.KpiHierarchy
 import de.fraunhofer.iem.kpiCalculator.model.kpi.hierarchy.KpiNode
 import org.junit.jupiter.api.Test
 import kotlin.test.assertEquals
 
+fun randomNode(edges: List<KpiEdge> = listOf()): KpiNode {
+
+    val rndIds = (0..<KpiId.entries.size).random()
+    val rndStrategies = (0..<KpiStrategyId.entries.size).random()
+
+    return KpiNode(
+        kpiId = KpiId.entries[rndIds],
+        kpiStrategyId = KpiStrategyId.entries[rndStrategies],
+        edges = edges
+    )
+}
+
 class HierarchyValidatorTest {
 
     @Test
@@ -31,4 +44,53 @@ class HierarchyValidatorTest {
         assertEquals(false, HierarchyValidator.isValid(kpiHierarchy = hierarchy, strict = true))
         assertEquals(true, HierarchyValidator.isValid(kpiHierarchy = hierarchy, strict = false))
     }
+
+    @Test
+    fun invalidRawValueHierarchy() {
+        val invalidHierarchy = KpiHierarchy.create(
+            rootNode = KpiNode(
+                kpiId = KpiId.ROOT,
+                kpiStrategyId = KpiStrategyId.AGGREGATION_STRATEGY,
+                edges = listOf(
+                    KpiEdge(
+                        KpiNode(
+                            kpiStrategyId = KpiStrategyId.RAW_VALUE_STRATEGY,
+                            kpiId = KpiId.NUMBER_OF_COMMITS,
+                            edges = listOf(
+                                KpiEdge(
+                                    target = randomNode(),
+                                    weight = 1.0
+                                )
+                            )
+                        ),
+                        weight = 1.0
+                    )
+                )
+            )
+        )
+
+        assertEquals(false, HierarchyValidator.isValid(kpiHierarchy = invalidHierarchy, strict = true))
+        assertEquals(false, HierarchyValidator.isValid(kpiHierarchy = invalidHierarchy, strict = false))
+
+        val hierarchy = KpiHierarchy.create(
+            rootNode = KpiNode(
+                kpiId = KpiId.ROOT,
+                kpiStrategyId = KpiStrategyId.AGGREGATION_STRATEGY,
+                edges = listOf(
+                    KpiEdge(
+                        KpiNode(
+                            kpiStrategyId = KpiStrategyId.RAW_VALUE_STRATEGY,
+                            kpiId = KpiId.NUMBER_OF_COMMITS,
+                            edges = listOf(
+                            ),
+                        ),
+                        weight = 1.0
+                    )
+                )
+            )
+        )
+
+        assertEquals(true, HierarchyValidator.isValid(kpiHierarchy = hierarchy, strict = true))
+        assertEquals(true, HierarchyValidator.isValid(kpiHierarchy = hierarchy, strict = false))
+    }
 }