Add support for all constraint types in ContextBuilder

Extend ContextBuilder to support the full set of constraint types:
- IAllowedValuesConstraint (existing)
- IMatchesConstraint (existing)
- IExpectConstraint (existing)
- IIndexHasKeyConstraint (existing)
- ICardinalityConstraint (new)
- IIndexConstraint (new)
- IUniqueConstraint (new)

Add comprehensive unit tests for each constraint type to verify
the builder correctly constructs constraint sets with all supported
constraint types.

Author: david-waltermire

core/src/test/java/gov/nist/secauto/metaschema/core/testsupport/builder/ContextBuilder.java

@@ -10,11 +10,14 @@
 import gov.nist.secauto.metaschema.core.model.constraint.AbstractConstraintBuilder;
 import gov.nist.secauto.metaschema.core.model.constraint.AssemblyConstraintSet;
 import gov.nist.secauto.metaschema.core.model.constraint.IAllowedValuesConstraint;
+import gov.nist.secauto.metaschema.core.model.constraint.ICardinalityConstraint;
 import gov.nist.secauto.metaschema.core.model.constraint.IConstraint;
 import gov.nist.secauto.metaschema.core.model.constraint.IExpectConstraint;
+import gov.nist.secauto.metaschema.core.model.constraint.IIndexConstraint;
 import gov.nist.secauto.metaschema.core.model.constraint.IIndexHasKeyConstraint;
 import gov.nist.secauto.metaschema.core.model.constraint.IMatchesConstraint;
 import gov.nist.secauto.metaschema.core.model.constraint.IModelConstrained;
+import gov.nist.secauto.metaschema.core.model.constraint.IUniqueConstraint;
 import gov.nist.secauto.metaschema.core.model.constraint.MetaConstraintSet;
 import gov.nist.secauto.metaschema.core.util.ObjectUtils;
 
@@ -28,17 +31,17 @@
 /**
  * Implementation of {@link IContextBuilder} for creating constraint contexts.
  * <p>
- * This builder supports the following constraint types:
+ * This builder supports all constraint types defined in the Metaschema
+ * constraint model:
  * <ul>
  * <li>{@link IAllowedValuesConstraint}</li>
  * <li>{@link IMatchesConstraint}</li>
  * <li>{@link IExpectConstraint}</li>
  * <li>{@link IIndexHasKeyConstraint}</li>
+ * <li>{@link ICardinalityConstraint}</li>
+ * <li>{@link IIndexConstraint}</li>
+ * <li>{@link IUniqueConstraint}</li>
  * </ul>
- * Other constraint types (such as {@code ICardinalityConstraint},
- * {@code IIndexConstraint}, or {@code IUniqueConstraint}) will throw an
- * {@link UnsupportedOperationException} when added. To support additional
- * constraint types, extend the {@link #addConstraint} method.
  */
 public class ContextBuilder implements IContextBuilder {
   @NonNull
@@ -131,6 +134,7 @@ MetaConstraintSet.Context build(@Nullable MetaConstraintSet.Context parent) {
    * @throws UnsupportedOperationException
    *           if the constraint type is not supported
    */
+  @SuppressWarnings("PMD.CyclomaticComplexity")
   private static void addConstraint(@NonNull IModelConstrained modelConstrained, @NonNull IConstraint constraint) {
     if (constraint instanceof IAllowedValuesConstraint) {
       modelConstrained.addConstraint((IAllowedValuesConstraint) constraint);
@@ -140,6 +144,12 @@ private static void addConstraint(@NonNull IModelConstrained modelConstrained, @
       modelConstrained.addConstraint((IExpectConstraint) constraint);
     } else if (constraint instanceof IIndexHasKeyConstraint) {
       modelConstrained.addConstraint((IIndexHasKeyConstraint) constraint);
+    } else if (constraint instanceof ICardinalityConstraint) {
+      modelConstrained.addConstraint((ICardinalityConstraint) constraint);
+    } else if (constraint instanceof IIndexConstraint) {
+      modelConstrained.addConstraint((IIndexConstraint) constraint);
+    } else if (constraint instanceof IUniqueConstraint) {
+      modelConstrained.addConstraint((IUniqueConstraint) constraint);
     } else {
       throw new UnsupportedOperationException(
           "Unsupported constraint type: " + constraint.getClass().getName());

core/src/test/java/gov/nist/secauto/metaschema/core/testsupport/tests/ConstraintSetBuilderTest.java

@@ -15,8 +15,14 @@
 import gov.nist.secauto.metaschema.core.model.ISource;
 import gov.nist.secauto.metaschema.core.model.constraint.IAllowedValue;
 import gov.nist.secauto.metaschema.core.model.constraint.IAllowedValuesConstraint;
+import gov.nist.secauto.metaschema.core.model.constraint.ICardinalityConstraint;
 import gov.nist.secauto.metaschema.core.model.constraint.IConstraintSet;
+import gov.nist.secauto.metaschema.core.model.constraint.IExpectConstraint;
+import gov.nist.secauto.metaschema.core.model.constraint.IIndexConstraint;
+import gov.nist.secauto.metaschema.core.model.constraint.IIndexHasKeyConstraint;
+import gov.nist.secauto.metaschema.core.model.constraint.IKeyField;
 import gov.nist.secauto.metaschema.core.model.constraint.IMatchesConstraint;
+import gov.nist.secauto.metaschema.core.model.constraint.IUniqueConstraint;
 import gov.nist.secauto.metaschema.core.testsupport.MockedModelTestSupport;
 import gov.nist.secauto.metaschema.core.testsupport.builder.IConstraintSetBuilder;
 
@@ -188,4 +194,115 @@ void testConstraintSetWithMultipleContexts() {
     assertNotNull(constraintSet, "Constraint set should not be null");
     assertEquals(source, constraintSet.getSource(), "Source should match");
   }
+
+  @Test
+  void testConstraintSetWithExpectConstraint() {
+    // Given
+    MockedModelTestSupport mocking = new MockedModelTestSupport();
+    ISource source = ISource.externalSource(URI.create(TEST_NAMESPACE));
+
+    // When
+    IConstraintSet constraintSet = mocking.constraintSet()
+        .source(source)
+        .context(ctx -> ctx
+            .metapath("//item")
+            .constraint(IExpectConstraint.builder()
+                .source(source)
+                .target(IMetapathExpression.compile("."))
+                .test(IMetapathExpression.compile("@id != ''"))))
+        .build();
+
+    // Then
+    assertNotNull(constraintSet, "Constraint set should not be null");
+    assertEquals(source, constraintSet.getSource(), "Source should match");
+  }
+
+  @Test
+  void testConstraintSetWithCardinalityConstraint() {
+    // Given
+    MockedModelTestSupport mocking = new MockedModelTestSupport();
+    ISource source = ISource.externalSource(URI.create(TEST_NAMESPACE));
+
+    // When
+    IConstraintSet constraintSet = mocking.constraintSet()
+        .source(source)
+        .context(ctx -> ctx
+            .metapath("//parent")
+            .constraint(ICardinalityConstraint.builder()
+                .source(source)
+                .target(IMetapathExpression.compile("child"))
+                .minOccurs(1)
+                .maxOccurs(5)))
+        .build();
+
+    // Then
+    assertNotNull(constraintSet, "Constraint set should not be null");
+    assertEquals(source, constraintSet.getSource(), "Source should match");
+  }
+
+  @Test
+  void testConstraintSetWithIndexConstraint() {
+    // Given
+    MockedModelTestSupport mocking = new MockedModelTestSupport();
+    ISource source = ISource.externalSource(URI.create(TEST_NAMESPACE));
+
+    // When
+    IConstraintSet constraintSet = mocking.constraintSet()
+        .source(source)
+        .context(ctx -> ctx
+            .metapath("//items")
+            .constraint(IIndexConstraint.builder("item-index")
+                .source(source)
+                .target(IMetapathExpression.compile("item"))
+                .keyField(IKeyField.of(IMetapathExpression.compile("@id"), null, null))))
+        .build();
+
+    // Then
+    assertNotNull(constraintSet, "Constraint set should not be null");
+    assertEquals(source, constraintSet.getSource(), "Source should match");
+  }
+
+  @Test
+  void testConstraintSetWithIndexHasKeyConstraint() {
+    // Given
+    MockedModelTestSupport mocking = new MockedModelTestSupport();
+    ISource source = ISource.externalSource(URI.create(TEST_NAMESPACE));
+
+    // When
+    IConstraintSet constraintSet = mocking.constraintSet()
+        .source(source)
+        .context(ctx -> ctx
+            .metapath("//reference")
+            .constraint(IIndexHasKeyConstraint.builder("item-index")
+                .source(source)
+                .target(IMetapathExpression.compile("."))
+                .keyField(IKeyField.of(IMetapathExpression.compile("@item-ref"), null, null))))
+        .build();
+
+    // Then
+    assertNotNull(constraintSet, "Constraint set should not be null");
+    assertEquals(source, constraintSet.getSource(), "Source should match");
+  }
+
+  @Test
+  void testConstraintSetWithUniqueConstraint() {
+    // Given
+    MockedModelTestSupport mocking = new MockedModelTestSupport();
+    ISource source = ISource.externalSource(URI.create(TEST_NAMESPACE));
+
+    // When
+    IConstraintSet constraintSet = mocking.constraintSet()
+        .source(source)
+        .context(ctx -> ctx
+            .metapath("//collection")
+            .constraint(IUniqueConstraint.builder()
+                .source(source)
+                .target(IMetapathExpression.compile("entry"))
+                .keyField(IKeyField.of(IMetapathExpression.compile("@key"), null, null))))
+        .build();
+
+    // Then
+    assertNotNull(constraintSet, "Constraint set should not be null");
+    assertEquals(source, constraintSet.getSource(), "Source should match");
+  }
 }