EntityConstructorTest.java

/**
 * Copyright (C) 2020 Czech Technical University in Prague
 * <p>
 * This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public
 * License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later
 * version.
 * <p>
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details. You should have received a copy of the GNU General Public License along with this program. If not, see
 * <http://www.gnu.org/licenses/>.
 */
package cz.cvut.kbss.jopa.oom;

import cz.cvut.kbss.jopa.environment.*;
import cz.cvut.kbss.jopa.environment.utils.Generators;
import cz.cvut.kbss.jopa.environment.utils.MetamodelMocks;
import cz.cvut.kbss.jopa.exceptions.CardinalityConstraintViolatedException;
import cz.cvut.kbss.jopa.exceptions.IntegrityConstraintViolatedException;
import cz.cvut.kbss.jopa.model.JOPAPersistenceProperties;
import cz.cvut.kbss.jopa.model.SequencesVocabulary;
import cz.cvut.kbss.jopa.model.TypedQueryImpl;
import cz.cvut.kbss.jopa.model.annotations.OWLAnnotationProperty;
import cz.cvut.kbss.jopa.model.annotations.OWLDataProperty;
import cz.cvut.kbss.jopa.model.annotations.OWLObjectProperty;
import cz.cvut.kbss.jopa.model.annotations.ParticipationConstraints;
import cz.cvut.kbss.jopa.model.descriptors.Descriptor;
import cz.cvut.kbss.jopa.model.descriptors.EntityDescriptor;
import cz.cvut.kbss.jopa.query.sparql.SparqlQueryFactory;
import cz.cvut.kbss.jopa.sessions.UnitOfWorkImpl;
import cz.cvut.kbss.jopa.utils.Configuration;
import cz.cvut.kbss.ontodriver.descriptor.ReferencedListDescriptor;
import cz.cvut.kbss.ontodriver.model.*;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import java.lang.reflect.Field;
import java.net.URI;
import java.util.*;
import java.util.Map.Entry;
import java.util.stream.Collectors;

import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;

class EntityConstructorTest {

    private static final URI PK = URI.create("http://krizik.felk.cvut.cz/ontologies/jopa/entityX");
    private static final NamedResource PK_RESOURCE = NamedResource.create(PK);
    private static final URI PK_TWO = URI
            .create("http://krizik.felk.cvut.cz/ontologies/jopa/entityXX");
    private static final String STRING_ATT = "StringAttributeValue";
    private static final Set<String> TYPES = initTypes();

    @Mock
    private ObjectOntologyMapperImpl mapperMock;
    @Mock
    private UnitOfWorkImpl uowMock;
    @Mock
    private SparqlQueryFactory queryFactoryMock;
    @Mock
    private TypedQueryImpl typedQueryMock;

    private MetamodelMocks mocks;
    private Descriptor descriptor;

    private EntityConstructor constructor;

    @BeforeEach
    void setUp() throws Exception {
        MockitoAnnotations.openMocks(this);
        when(mapperMock.getConfiguration()).thenReturn(new Configuration(Collections.emptyMap()));
        when(mapperMock.getUow()).thenReturn(uowMock);
        when(uowMock.getQueryFactory()).thenReturn(queryFactoryMock);
        this.mocks = new MetamodelMocks();
        this.descriptor = new EntityDescriptor();
        this.constructor = new EntityConstructor(mapperMock);
    }

    private static Set<String> initTypes() {
        final Set<String> set = new HashSet<>(8);
        set.add("http://krizik.felk.cvut.cz/ontologies/jopa/entities#OWLClassU");
        set.add("http://krizik.felk.cvut.cz/ontologies/jopa/entities#OWLClassV");
        return set;
    }

    @Test
    void testReconstructEntityWithTypesAndDataProperty() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassA.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        axioms.addAll(getTypesAxiomsForOwlClassA());
        final OWLClassA res = constructor.reconstructEntity(PK, mocks.forOwlClassA().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        assertEquals(STRING_ATT, res.getStringAttribute());
        assertEquals(TYPES, res.getTypes());
        verify(mapperMock).registerInstance(PK, res);
    }

    private Axiom<URI> getClassAssertionAxiomForType(String type) {
        return new AxiomImpl<>(NamedResource.create(PK), Assertion.createClassAssertion(false),
                new Value<>(URI.create(type)));
    }

    private Axiom<String> getStringAttAssertionAxiom(Field attField) {
        final String assertionIri = attField.getAnnotation(OWLDataProperty.class).iri();
        return new AxiomImpl<>(NamedResource.create(PK),
                Assertion.createDataPropertyAssertion(URI.create(assertionIri), false),
                new Value<>(STRING_ATT));
    }

    private Set<Axiom<?>> getTypesAxiomsForOwlClassA() {
        final Set<Axiom<?>> axs = new HashSet<>();
        for (String type : TYPES) {
            final Axiom<URI> ax = new AxiomImpl<>(NamedResource.create(PK), Assertion.createClassAssertion(false),
                    new Value<>(URI.create(type)));
            axs.add(ax);
        }
        return axs;
    }

    @Test
    void testReconstructEntityWithDataPropertyEmptyTypes() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassA.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        final OWLClassA res = constructor.reconstructEntity(PK, mocks.forOwlClassA().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        assertEquals(STRING_ATT, res.getStringAttribute());
        assertNull(res.getTypes());
        verify(mapperMock).registerInstance(PK, res);
    }

    @Test
    void testReconstructEntityWithDataPropertyAndProperties() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassB.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassB.getStrAttField()));
        final Collection<Axiom<?>> properties = getProperties();
        axioms.addAll(properties);
        final OWLClassB res = constructor.reconstructEntity(PK, mocks.forOwlClassB().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        assertEquals(STRING_ATT, res.getStringAttribute());
        assertNotNull(res.getProperties());
        assertEquals(properties.size(), res.getProperties().size());
        for (Axiom<?> a : properties) {
            final String key = a.getAssertion().getIdentifier().toString();
            assertTrue(res.getProperties().containsKey(key));
            assertEquals(1, res.getProperties().get(key).size());
            assertEquals(a.getValue().stringValue(), res.getProperties().get(key).iterator().next());
        }
        verify(mapperMock).registerInstance(PK, res);
    }

    private Collection<Axiom<?>> getProperties() {
        final Set<Axiom<?>> props = new HashSet<>();

        final Axiom<String> axOne = new AxiomImpl<>(PK_RESOURCE, Assertion
                .createDataPropertyAssertion(URI.create("http://someDataPropertyOne"),
                        false), new Value<>("SomePropertyValue"));
        props.add(axOne);

        final Axiom<String> axTwo = new AxiomImpl<>(PK_RESOURCE, Assertion.createAnnotationPropertyAssertion(
                URI.create("http://someAnnotationPropertyOne"), false), new Value<>("annotationValue"));
        props.add(axTwo);

        final Axiom<URI> axThree = new AxiomImpl<>(PK_RESOURCE, Assertion
                .createObjectPropertyAssertion(URI.create("http://someObjectPropertyOne"),
                        false), new Value<>(URI.create("http://someObjectPropertyOne")));
        props.add(axThree);
        return props;
    }

    @Test
    void testReconstructEntityWithDataPropertiesAndEmptyProperties() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassB.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassB.getStrAttField()));
        final OWLClassB res = constructor.reconstructEntity(PK, mocks.forOwlClassB().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        assertEquals(STRING_ATT, res.getStringAttribute());
        assertNull(res.getProperties());
        verify(mapperMock).registerInstance(PK, res);
    }

    @Test
    void testReconstructEntityWithObjectProperty() throws Exception {
        final Set<Axiom<?>> axiomsD = getAxiomsForD();
        final Descriptor fieldDesc = new EntityDescriptor();
        descriptor.addAttributeDescriptor(mocks.forOwlClassD().owlClassAAtt(), fieldDesc);
        final OWLClassA entityA = new OWLClassA();
        entityA.setUri(PK_TWO);
        entityA.setStringAttribute(STRING_ATT);
        when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDesc))
                .thenReturn(entityA);
        final OWLClassD res = constructor.reconstructEntity(PK, mocks.forOwlClassD().entityType(), descriptor, axiomsD);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        verify(mapperMock).getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDesc);
        assertNotNull(res.getOwlClassA());
        assertEquals(PK_TWO, res.getOwlClassA().getUri());
        assertEquals(STRING_ATT, res.getOwlClassA().getStringAttribute());
        verify(mapperMock).registerInstance(PK, res);
    }

    private Set<Axiom<?>> getAxiomsForD() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassD.getClassIri()));
        final Axiom<NamedResource> opAssertion = createObjectPropertyAxiomForD();
        axioms.add(opAssertion);
        return axioms;
    }

    private Axiom<NamedResource> createObjectPropertyAxiomForD() throws NoSuchFieldException {
        return new AxiomImpl<>(NamedResource.create(PK), getClassDObjectPropertyAssertion(),
                new Value<>(NamedResource.create(PK_TWO)));
    }

    private Assertion getClassDObjectPropertyAssertion() throws NoSuchFieldException {
        final URI assertionUri = URI.create(OWLClassD.getOwlClassAField()
                .getAnnotation(OWLObjectProperty.class).iri());
        return Assertion.createObjectPropertyAssertion(assertionUri, false);
    }

    @Test
    void reconstructsEntityWithDataPropertyWhereRangeDoesNotMatch() throws Exception {
        final List<Axiom<?>> axioms = new ArrayList<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassA.getClassIri()));
        // Using a list and adding the incorrect range value before the right value will cause it to be processed first
        axioms.add(createDataPropertyAxiomWithWrongRange(OWLClassA.getStrAttField()));
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        OWLClassA entityA = constructor.reconstructEntity(PK, mocks.forOwlClassA().entityType(), descriptor, axioms);
        assertNotNull(entityA);
        assertEquals(STRING_ATT, entityA.getStringAttribute());

        axioms.clear();
        axioms.add(getClassAssertionAxiomForType(OWLClassA.getClassIri()));
        // Now reverse it
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        axioms.add(createDataPropertyAxiomWithWrongRange(OWLClassA.getStrAttField()));
        entityA = constructor.reconstructEntity(PK, mocks.forOwlClassA().entityType(), descriptor, axioms);
        assertNotNull(entityA);
        assertEquals(STRING_ATT, entityA.getStringAttribute());
    }

    @Test
    void throwsExceptionWhenDataPropertyCardinalityRestrictionIsNotMet() throws Exception {
        final List<Axiom<?>> axioms = new ArrayList<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassA.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        assertThrows(CardinalityConstraintViolatedException.class,
                () -> constructor.reconstructEntity(PK, mocks.forOwlClassA().entityType(), descriptor, axioms));
    }

    private Axiom<?> createDataPropertyAxiomWithWrongRange(Field valueField) {
        final OWLDataProperty prop = valueField.getAnnotation(OWLDataProperty.class);
        assert prop != null;
        final Assertion assertion = Assertion.createDataPropertyAssertion(URI.create(prop.iri()), false);
        return new AxiomImpl<>(NamedResource.create(PK), assertion, new Value<>(System.currentTimeMillis()));
    }

    @Test
    void reconstructsEntityWithObjectPropertyWhereRangeDoesNotMatch() throws Exception {
        final Set<Axiom<?>> axioms = getAxiomsForD();
        final Descriptor fieldDescriptor = new EntityDescriptor();
        descriptor.addAttributeDescriptor(mocks.forOwlClassD().owlClassAAtt(), fieldDescriptor);
        // The property value hasn't the corresponding type, so it cannot be returned by mapper
        when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDescriptor)).thenReturn(null);
        final OWLClassD res = constructor.reconstructEntity(PK, mocks.forOwlClassD().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertNull(res.getOwlClassA());
    }

    @Test
    void throwsExceptionWhenObjectPropertyCardinalityRestrictionIsNotMet() throws Exception {
        final Set<Axiom<?>> axioms = getAxiomsForD();
        final OWLClassA entityA = new OWLClassA();
        entityA.setUri(PK_TWO);
        final URI pkThree = URI.create("http://krizik.felk.cvut.cz/ontologies/jopa/entity3");
        axioms.add(new AxiomImpl<>(NamedResource.create(PK), getClassDObjectPropertyAssertion(),
                new Value<>(NamedResource.create(pkThree))));
        final OWLClassA entityThree = new OWLClassA();
        entityThree.setUri(pkThree);
        final Descriptor fieldDescriptor = new EntityDescriptor();
        descriptor.addAttributeDescriptor(mocks.forOwlClassD().owlClassAAtt(), fieldDescriptor);
        when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDescriptor)).thenReturn(entityA);
        when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, pkThree, fieldDescriptor))
                .thenReturn(entityThree);
        assertThrows(CardinalityConstraintViolatedException.class,
                () -> constructor.reconstructEntity(PK, mocks.forOwlClassD().entityType(), descriptor, axioms));
    }

    @Test
    void testSetFieldValue_DataProperty() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        final OWLClassA entityA = new OWLClassA();
        entityA.setUri(PK);
        assertNull(entityA.getStringAttribute());
        constructor.setFieldValue(entityA, OWLClassA.getStrAttField(), axioms, mocks.forOwlClassA().entityType(),
                descriptor);
        assertNotNull(entityA.getStringAttribute());
        assertEquals(STRING_ATT, entityA.getStringAttribute());
    }

    @Test
    void testSetFieldValue_ObjectProperty() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        final Descriptor fieldDesc = mock(Descriptor.class);
        descriptor.addAttributeDescriptor(mocks.forOwlClassD().owlClassAAtt(), fieldDesc);
        axioms.add(createObjectPropertyAxiomForD());
        final OWLClassD entityD = new OWLClassD();
        entityD.setUri(PK);
        assertNull(entityD.getOwlClassA());
        final OWLClassA entityA = new OWLClassA();
        entityA.setUri(PK_TWO);
        entityA.setStringAttribute(STRING_ATT);
        when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDesc))
                .thenReturn(entityA);
        constructor.setFieldValue(entityD, OWLClassD.getOwlClassAField(), axioms, mocks.forOwlClassD().entityType(),
                descriptor);
        assertNotNull(entityD.getOwlClassA());
        assertEquals(PK_TWO, entityD.getOwlClassA().getUri());
        verify(mapperMock).getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDesc);
    }

    @Test
    void testSetFieldValue_Types() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>(getTypesAxiomsForOwlClassA());
        final Descriptor fieldDesc = mock(Descriptor.class);
        descriptor.addAttributeDescriptor(mocks.forOwlClassA().typesSpec(), fieldDesc);
        final OWLClassA entityA = new OWLClassA();
        entityA.setUri(PK);
        assertNull(entityA.getTypes());
        constructor.setFieldValue(entityA, OWLClassA.getTypesField(), axioms, mocks.forOwlClassA().entityType(),
                descriptor);
        assertNotNull(entityA.getTypes());
        assertEquals(TYPES.size(), entityA.getTypes().size());
        assertTrue(entityA.getTypes().containsAll(TYPES));
    }

    @Test
    void testSetFieldValue_Properties() throws Exception {
        final Map<String, Set<String>> props = Generators.generateStringProperties();
        final Set<Axiom<?>> axioms = initAxiomsForProperties(props);
        final OWLClassB entityB = new OWLClassB();
        entityB.setUri(PK);
        assertNull(entityB.getProperties());
        constructor.setFieldValue(entityB, OWLClassB.getPropertiesField(), axioms, mocks.forOwlClassB().entityType(),
                descriptor);
        assertNotNull(entityB.getProperties());
        assertEquals(props, entityB.getProperties());
    }

    private Set<Axiom<?>> initAxiomsForProperties(Map<String, Set<String>> props) {
        final Set<Axiom<?>> axioms = new HashSet<>();
        for (Entry<String, Set<String>> e : props.entrySet()) {
            final URI property = URI.create(e.getKey());
            axioms.addAll(e.getValue().stream().map(val -> new AxiomImpl<>(PK_RESOURCE, Assertion
                    .createPropertyAssertion(property, false), new Value<>(URI.create(val))))
                    .collect(Collectors.toList()));
        }
        return axioms;
    }

    @Test
    void testSetFieldValue_ObjectPropertySet() throws Exception {
        final Descriptor desc = mock(Descriptor.class);
        final Set<OWLClassA> set = initEntities(desc);
        descriptor.addAttributeDescriptor(mocks.forOwlClassJ().setAttribute(), desc);
        final Collection<Axiom<?>> axioms = initAxiomsForReferencedSet(set);
        final OWLClassJ entityJ = new OWLClassJ();
        entityJ.setUri(PK);
        assertNull(entityJ.getOwlClassA());
        constructor.setFieldValue(entityJ, OWLClassJ.getOwlClassAField(), axioms, mocks.forOwlClassJ().entityType(),
                descriptor);
        assertNotNull(entityJ.getOwlClassA());
        assertEquals(set.size(), entityJ.getOwlClassA().size());
        assertTrue(areEqual(set, entityJ.getOwlClassA()));
    }

    private Set<OWLClassA> initEntities(Descriptor desc) {
        final Set<OWLClassA> entities = new HashSet<>();
        for (int i = 0; i < 5; i++) {
            final OWLClassA a = new OWLClassA();
            a.setUri(URI.create("http://krizik.felk.cvut.cz/entityA" + i));
            a.setStringAttribute(STRING_ATT);
            entities.add(a);
            when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, a.getUri(), desc))
                    .thenReturn(a);
        }
        return entities;
    }

    private Collection<Axiom<?>> initAxiomsForReferencedSet(Set<OWLClassA> as) throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        final URI property = URI.create(OWLClassJ.getOwlClassAField()
                .getAnnotation(OWLObjectProperty.class).iri());
        for (OWLClassA a : as) {
            final Axiom<NamedResource> ax = new AxiomImpl<>(NamedResource.create(PK),
                    Assertion.createObjectPropertyAssertion(property, false),
                    new Value<>(NamedResource.create(
                            a.getUri())));
            axioms.add(ax);
        }
        return axioms;
    }

    private static boolean areEqual(Set<OWLClassA> expected, Set<OWLClassA> actual) {
        boolean found = false;
        for (OWLClassA a : expected) {
            found = false;
            for (OWLClassA aa : actual) {
                if (a.getUri().equals(aa.getUri())) {
                    found = true;
                    assertEquals(a.getStringAttribute(), aa.getStringAttribute());
                    break;
                }
            }
            if (!found) {
                return false;
            }
        }
        return found;
    }

    @Test
    void reconstructsEntityWithDataPropertiesOfBasicTypesAndStringIdentifier() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassM.getClassIri()));
        final Integer i = 117;
        final Long lng = 365L;
        final Double d = 3.14;
        final Date date = new Date();
        axioms.addAll(createAxiomsForValues(true, i, lng, d, date, null));

        final OWLClassM res = constructor.reconstructEntity(PK, mocks.forOwlClassM().entityType(), descriptor, axioms);
        assertEquals(true, res.getBooleanAttribute());
        assertEquals(i, res.getIntAttribute());
        assertEquals(lng, res.getLongAttribute());
        assertEquals(d, res.getDoubleAttribute());
        assertEquals(date, res.getDateAttribute());
    }

    private Collection<Axiom<?>> createAxiomsForValues(Boolean b, Integer i, Long lng, Double d, Date date,
                                                       OWLClassM.Severity severity) throws
            Exception {
        final Collection<Axiom<?>> axioms = new ArrayList<>();
        if (b != null) {
            final String boolAttIri = OWLClassM.getBooleanAttributeField().getAnnotation(OWLDataProperty.class).iri();
            axioms.add(
                    new AxiomImpl<>(PK_RESOURCE, Assertion.createDataPropertyAssertion(URI.create(boolAttIri), false),
                            new Value<>(b)));
        }
        if (i != null) {
            final String intAttIri = OWLClassM.getIntAttributeField().getAnnotation(OWLDataProperty.class).iri();
            axioms.add(new AxiomImpl<>(PK_RESOURCE, Assertion.createDataPropertyAssertion(URI.create(intAttIri), false),
                    new Value<>(i)));
        }
        if (lng != null) {
            final String longAttIri = OWLClassM.getLongAttributeField().getAnnotation(OWLDataProperty.class).iri();
            final Assertion laAssertion = Assertion.createDataPropertyAssertion(URI.create(longAttIri), false);
            axioms.add(new AxiomImpl<>(PK_RESOURCE, laAssertion, new Value<>(lng)));
        }
        if (d != null) {
            final String doubleAttIri = OWLClassM.getDoubleAttributeField().getAnnotation(OWLDataProperty.class).iri();
            axioms.add(
                    new AxiomImpl<>(PK_RESOURCE, Assertion.createDataPropertyAssertion(URI.create(doubleAttIri), false),
                            new Value<>(d)));
        }
        if (date != null) {
            final String dateAttIri = OWLClassM.getDateAttributeField().getAnnotation(OWLDataProperty.class).iri();
            axioms.add(
                    new AxiomImpl<>(PK_RESOURCE, Assertion.createDataPropertyAssertion(URI.create(dateAttIri), false),
                            new Value<>(date)));
        }
        if (severity != null) {
            final String enumAttIri = OWLClassM.getEnumAttributeField().getAnnotation(OWLDataProperty.class).iri();
            axioms.add(
                    new AxiomImpl<>(PK_RESOURCE, Assertion.createDataPropertyAssertion(URI.create(enumAttIri), false),
                            new Value<>(severity.toString())));
        }
        return axioms;
    }

    @Test
    void reconstructsEntityWithEnumDataProperty() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassM.getClassIri()));
        final OWLClassM.Severity enumValue = OWLClassM.Severity.HIGH;
        axioms.addAll(createAxiomsForValues(null, null, null, null, null, enumValue));

        final OWLClassM result = constructor
                .reconstructEntity(PK, mocks.forOwlClassM().entityType(), descriptor, axioms);
        assertNotNull(result);
        assertNotNull(result.getEnumAttribute());
        assertEquals(enumValue, result.getEnumAttribute());
    }

    @Test
    void icsAreValidatedForAllFieldsOnEntityLoad() {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassL.getClassIri()));

        assertThrows(IntegrityConstraintViolatedException.class,
                () -> constructor.reconstructEntity(PK, mocks.forOwlClassL().entityType(), descriptor, axioms));
    }

    @Test
    void icValidationIsSkippedOnLoadBasedOnProperties() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassL.getClassIri()));

        final Map<String, String> props = Collections
                .singletonMap(JOPAPersistenceProperties.DISABLE_IC_VALIDATION_ON_LOAD, Boolean.TRUE.toString());
        final Configuration conf = new Configuration(props);
        when(mapperMock.getConfiguration()).thenReturn(conf);

        final OWLClassL result = constructor
                .reconstructEntity(PK, mocks.forOwlClassL().entityType(), descriptor, axioms);
        assertNotNull(result);
        assertNull(result.getSingleA());
    }

    @Test
    void icsAreValidatedOnLazilyLoadedFieldFetch() throws Exception {
        final Set<Axiom<?>> fieldAxiom = initInvalidFieldValuesForICValidation();

        final OWLClassL instance = new OWLClassL();
        assertThrows(IntegrityConstraintViolatedException.class, () -> constructor
                .setFieldValue(instance, OWLClassL.getReferencedListField(), fieldAxiom,
                        mocks.forOwlClassL().entityType(), descriptor));
    }

    private Set<Axiom<?>> initInvalidFieldValuesForICValidation() throws NoSuchFieldException {
        final Set<Axiom<NamedResource>> listAxioms = new HashSet<>();
        when(mapperMock.loadReferencedList(any(ReferencedListDescriptor.class))).thenReturn(listAxioms);
        final Descriptor attDescriptor = new EntityDescriptor();
        descriptor.addAttributeDescriptor(mocks.forOwlClassL().referencedListAtt(), attDescriptor);
        final Assertion nodeContent = Assertion
                .createObjectPropertyAssertion(URI.create(SequencesVocabulary.s_p_hasContents), false);
        final int max = OWLClassL.getReferencedListField().getAnnotation(ParticipationConstraints.class).value()[0]
                .max();
        for (int i = 0; i < max + 1; i++) {
            final URI uri = URI.create("http://value" + i);
            listAxioms.add(new AxiomImpl<>(PK_RESOURCE, nodeContent, new Value<>(NamedResource.create(uri))));
            when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, uri, attDescriptor))
                    .thenReturn(new OWLClassA(uri));
        }
        final Assertion assertion = Assertion.createObjectPropertyAssertion(
                URI.create(OWLClassL.getReferencedListField().getAnnotation(OWLObjectProperty.class).iri()), false);
        return Collections.singleton(
                new AxiomImpl<>(PK_RESOURCE, assertion, new Value<>(URI.create("http://referencedList"))));
    }

    @Test
    void icsAreNotValidatedOnLazilyLoadedFetchWhenLoadingICValidationIsDisabled() throws Exception {
        final Set<Axiom<?>> fieldAxiom = initInvalidFieldValuesForICValidation();

        final Map<String, String> props = Collections
                .singletonMap(JOPAPersistenceProperties.DISABLE_IC_VALIDATION_ON_LOAD, Boolean.TRUE.toString());
        final Configuration conf = new Configuration(props);
        when(mapperMock.getConfiguration()).thenReturn(conf);

        final OWLClassL instance = new OWLClassL();
        constructor.setFieldValue(instance, OWLClassL.getReferencedListField(), fieldAxiom,
                mocks.forOwlClassL().entityType(), descriptor);
    }

    @Test
    void instanceReconstructionSkipsAxiomsForWhichNoAttributeCanBeFound() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassA.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassA.getStrAttField()));
        final Assertion unknown =
                Assertion.createAnnotationPropertyAssertion(Generators.createPropertyIdentifier(), false);
        axioms.add(new AxiomImpl<>(PK_RESOURCE, unknown, new Value<>("UnknownPropertyValue")));
        final OWLClassA res = constructor.reconstructEntity(PK, mocks.forOwlClassA().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        assertNotNull(res.getStringAttribute());
        assertNull(res.getTypes());
    }

    @Test
    void reconstructsInstanceWithMappedSuperclass() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassQ.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassQ.getStringAttributeField()));
        axioms.add(getStringAttAssertionAxiom(OWLClassQ.getParentStringField()));
        final URI labelUri = URI.create(OWLClassQ.getLabelField().getAnnotation(OWLAnnotationProperty.class).iri());
        axioms.add(new AxiomImpl<>(PK_RESOURCE, Assertion.createAnnotationPropertyAssertion(labelUri, false),
                new Value<>(STRING_ATT)));
        final URI owlClassAUri = URI.create(OWLClassQ.getOwlClassAField().getAnnotation(OWLObjectProperty.class).iri());
        axioms.add(new AxiomImpl<>(PK_RESOURCE, Assertion.createObjectPropertyAssertion(owlClassAUri, false),
                new Value<>(NamedResource.create(PK_TWO))));
        final OWLClassA a = new OWLClassA();
        a.setUri(PK_TWO);
        when(mapperMock.getEntityFromCacheOrOntology(eq(OWLClassA.class), eq(PK_TWO), any(Descriptor.class)))
                .thenReturn(a);

        final OWLClassQ res = constructor.reconstructEntity(PK, mocks.forOwlClassQ().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        assertEquals(STRING_ATT, res.getStringAttribute());
        assertEquals(STRING_ATT, res.getParentString());
        assertEquals(STRING_ATT, res.getLabel());
        assertSame(a, res.getOwlClassA());
    }

    @Test
    void reconstructsInstanceWithPluralAnnotationProperty() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassN.getClassIri()));
        final Assertion assertion =
                Assertion.createAnnotationPropertyAssertion(URI.create(Vocabulary.DC_SOURCE), false);
        axioms.add(new AxiomImpl<>(PK_RESOURCE, assertion, new Value<>(STRING_ATT)));

        final OWLClassN result =
                constructor.reconstructEntity(PK, mocks.forOwlClassN().entityType(), descriptor, axioms);
        assertNotNull(result);
        assertEquals(PK.toString(), result.getId());
        assertEquals(Collections.singleton(STRING_ATT), result.getPluralAnnotation());
    }

    @Test
    void setFieldValueSetsNothingWhenAxiomsAreEmpty() throws Exception {
        final OWLClassA entity = new OWLClassA(Generators.createIndividualIdentifier());
        assertNull(entity.getTypes());
        constructor.setFieldValue(entity, OWLClassA.getTypesField(), Collections.emptyList(),
                mocks.forOwlClassA().entityType(), descriptor);
        assertNull(entity.getTypes());
    }

    @Test
    void setFieldValueValidatesIntegrityConstraintsAlsoForEmptyAxiomCollection() {
        final OWLClassL instance = new OWLClassL(Generators.createIndividualIdentifier());
        assertThrows(IntegrityConstraintViolatedException.class, () -> constructor
                .setFieldValue(instance, OWLClassL.getSingleAField(), Collections.emptyList(),
                        mocks.forOwlClassL().entityType(), descriptor));
    }

    @Test
    void testReconstructEntityWithQueryAttribute() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassWithQueryAttr.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassWithQueryAttr.getStrAttField()));

        doReturn(typedQueryMock)
                .when(queryFactoryMock).createNativeQuery(any(String.class),
                (Class<?>) any(Class.class));
        doReturn(typedQueryMock)
                .when(typedQueryMock).setParameter(any(String.class), any());
        doReturn(STRING_ATT).when(typedQueryMock).getSingleResult();

        final OWLClassWithQueryAttr res = constructor.reconstructEntity(PK, mocks.forOwlClassWithQueryAttr().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        assertEquals(STRING_ATT, res.getStringAttribute());
        assertEquals(STRING_ATT, res.getStringQueryAttribute());
        verify(mapperMock).registerInstance(PK, res);
    }

    @Test
    void testReconstructEntityWithManagedTypeQueryAttribute() throws Exception {
        final Set<Axiom<?>> axioms = new HashSet<>();
        axioms.add(getClassAssertionAxiomForType(OWLClassWithQueryAttr.getClassIri()));
        axioms.add(getStringAttAssertionAxiom(OWLClassWithQueryAttr.getStrAttField()));

        final URI assertionUri = URI.create(OWLClassWithQueryAttr.getEntityAttField()
                .getAnnotation(OWLObjectProperty.class).iri());
        final Axiom<NamedResource> opAssertion = new AxiomImpl<>(NamedResource.create(PK),
                Assertion.createObjectPropertyAssertion(assertionUri, false),
                new Value<>(NamedResource.create(PK_TWO)));
        axioms.add(opAssertion);

        final Descriptor fieldDesc = new EntityDescriptor();
        descriptor.addAttributeDescriptor(mocks.forOwlClassWithQueryAttr().entityAttribute(), fieldDesc);
        final OWLClassA entityA = new OWLClassA();
        entityA.setUri(PK_TWO);
        entityA.setStringAttribute(STRING_ATT);

        when(mapperMock.getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDesc))
                .thenReturn(entityA);

        doReturn(typedQueryMock)
                .when(queryFactoryMock).createNativeQuery(any(String.class),
                (Class<?>) any(Class.class));
        doReturn(typedQueryMock)
                .when(typedQueryMock).setParameter(any(String.class), any());
        doReturn(entityA).when(typedQueryMock).getSingleResult();

        final OWLClassWithQueryAttr res = constructor.reconstructEntity(PK, mocks.forOwlClassWithQueryAttr().entityType(), descriptor, axioms);
        assertNotNull(res);
        assertEquals(PK, res.getUri());
        verify(mapperMock).getEntityFromCacheOrOntology(OWLClassA.class, PK_TWO, fieldDesc);
        assertNotNull(res.getEntityAttribute());
        assertEquals(PK_TWO, res.getEntityAttribute().getUri());
        assertEquals(STRING_ATT, res.getEntityAttribute().getStringAttribute());

        assertNotNull(res.getEntityQueryAttribute());
        assertEquals(PK_TWO, res.getEntityQueryAttribute().getUri());
        assertEquals(STRING_ATT, res.getEntityQueryAttribute().getStringAttribute());

        verify(mapperMock).registerInstance(PK, res);
    }
}