use of uk.gov.gchq.koryphe.ValidationResult in project Gaffer by gchq.
the class SchemaElementDefinitionValidatorTest method shouldValidateAndReturnFalseWhenAggregatorHasIdentifierInSelection.
@Test
public void shouldValidateAndReturnFalseWhenAggregatorHasIdentifierInSelection() {
// Given
final SchemaElementDefinitionValidator validator = new SchemaElementDefinitionValidator();
final BinaryOperator<Integer> function1 = mock(BinaryOperator.class);
final Schema schema = new Schema.Builder().entity(TestGroups.ENTITY, new SchemaEntityDefinition.Builder().vertex("id").property(TestPropertyNames.STRING, "string").aggregator(new ElementAggregator.Builder().select(IdentifierType.VERTEX.name()).execute(function1).build()).build()).type("id", new TypeDefinition.Builder().clazz(String.class).build()).type("string", new TypeDefinition.Builder().clazz(String.class).aggregateFunction(function1).build()).build();
// When
final ValidationResult result = validator.validate(schema.getElement(TestGroups.ENTITY));
// Then
assertFalse(result.isValid());
assertEquals(com.google.common.collect.Sets.newHashSet("Identifiers cannot be selected for aggregation: " + IdentifierType.VERTEX.name()), result.getErrors());
}
use of uk.gov.gchq.koryphe.ValidationResult in project Gaffer by gchq.
the class SchemaElementDefinitionValidatorTest method shouldValidateAndReturnTrueWhenTimestampIsAggregatedWithANonGroupByProperty.
@Test
public void shouldValidateAndReturnTrueWhenTimestampIsAggregatedWithANonGroupByProperty() {
// Given
final SchemaElementDefinitionValidator validator = new SchemaElementDefinitionValidator();
final BinaryOperator<Integer> function1 = mock(BinaryOperator.class);
final Schema schema = new Schema.Builder().entity(TestGroups.ENTITY, new SchemaEntityDefinition.Builder().vertex("id").property(TestPropertyNames.STRING, "string").property(TestPropertyNames.TIMESTAMP, "string").aggregator(new ElementAggregator.Builder().select(TestPropertyNames.STRING, TestPropertyNames.TIMESTAMP).execute(function1).build()).build()).timestampProperty(TestPropertyNames.TIMESTAMP).type("id", new TypeDefinition.Builder().clazz(String.class).build()).type("string", new TypeDefinition.Builder().clazz(String.class).aggregateFunction(function1).build()).build();
// When
final ValidationResult result = validator.validate(schema.getElement(TestGroups.ENTITY));
// Then
assertTrue(result.isValid());
}
use of uk.gov.gchq.koryphe.ValidationResult in project Gaffer by gchq.
the class SchemaElementDefinitionValidatorTest method shouldValidateFunctionSelectionsAndReturnTrueWhenAllFunctionsAreValid.
@Test
public void shouldValidateFunctionSelectionsAndReturnTrueWhenAllFunctionsAreValid() {
// Given
final SchemaElementDefinition elementDef = mock(SchemaElementDefinition.class);
given(elementDef.getPropertyClass("selectionStr")).willReturn((Class) String.class);
given(elementDef.getPropertyClass("selectionInt")).willReturn((Class) Integer.class);
final Predicate<String> function1 = a -> a.contains("true");
final Predicate<Integer> function2 = a -> a > 0;
final ElementFilter elementFilter = new ElementFilter.Builder().select("selectionStr").execute(function1).select("selectionInt").execute(function2).build();
final SchemaElementDefinitionValidator validator = new SchemaElementDefinitionValidator();
// When
final ValidationResult result = validator.validateFunctionArgumentTypes(elementFilter, elementDef);
// Then
assertTrue(result.isValid());
}
use of uk.gov.gchq.koryphe.ValidationResult in project Gaffer by gchq.
the class SchemaElementDefinitionValidatorTest method shouldValidateAndReturnFalseWhenAggregatorIsInvalid.
@Test
public void shouldValidateAndReturnFalseWhenAggregatorIsInvalid() {
// Given
final SchemaElementDefinition elementDef = mock(SchemaElementDefinition.class);
final SchemaElementDefinitionValidator validator = new SchemaElementDefinitionValidator();
final Map<String, String> properties = new HashMap<>();
properties.put(TestPropertyNames.PROP_1, "int");
properties.put(TestPropertyNames.PROP_2, "string");
final BinaryOperator<Integer> function1 = mock(BinaryOperator.class);
final ElementAggregator aggregator = new ElementAggregator.Builder().select(TestPropertyNames.PROP_1).execute(function1).build();
given(elementDef.getIdentifiers()).willReturn(new HashSet<>());
given(elementDef.getProperties()).willReturn(properties.keySet());
given(elementDef.getPropertyMap()).willReturn(properties);
given(elementDef.getValidator()).willReturn(mock(ElementFilter.class));
given(elementDef.getFullAggregator()).willReturn(aggregator);
given(elementDef.getPropertyClass(TestPropertyNames.PROP_1)).willReturn((Class) Integer.class);
given(elementDef.getPropertyClass(TestPropertyNames.PROP_2)).willReturn((Class) String.class);
given(elementDef.isAggregate()).willReturn(true);
// When
final ValidationResult result = validator.validate(elementDef);
// Then
assertFalse(result.isValid());
assertTrue(result.getErrorString().contains("No aggregator found for properties"));
verify(elementDef, Mockito.atLeastOnce()).getPropertyClass(TestPropertyNames.PROP_1);
verify(elementDef, Mockito.atLeastOnce()).getPropertyClass(TestPropertyNames.PROP_2);
}
use of uk.gov.gchq.koryphe.ValidationResult in project Gaffer by gchq.
the class SchemaElementDefinitionValidatorTest method shouldValidateFunctionSelectionsAndReturnTrueWhenNoFunctionsSet.
@Test
public void shouldValidateFunctionSelectionsAndReturnTrueWhenNoFunctionsSet() {
// Given
final SchemaElementDefinition elementDef = mock(SchemaElementDefinition.class);
final ElementFilter elementFilter = new ElementFilter();
final SchemaElementDefinitionValidator validator = new SchemaElementDefinitionValidator();
// When
final ValidationResult result = validator.validateFunctionArgumentTypes(elementFilter, elementDef);
// Then
assertTrue(result.isValid());
}
Aggregations