Additional test to check that property ids remain untouched after enf…

…orcer creation

enterprise/kernel/src/test/java/org/neo4j/kernel/impl/enterprise/PropertyExistenceEnforcerTest.java

@@ -0,0 +1,79 @@
+/*
+ * Copyright (c) 2002-2017 "Neo Technology,"
+ * Network Engine for Objects in Lund AB [http://neotechnology.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * 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 Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.kernel.impl.enterprise;
+
+import org.junit.Test;
+import org.mockito.Mockito;
+
+import java.util.Arrays;
+import java.util.List;
+import java.util.function.Function;
+
+import org.neo4j.kernel.api.schema.constaints.ConstraintDescriptor;
+import org.neo4j.kernel.api.schema.constaints.ConstraintDescriptorFactory;
+import org.neo4j.kernel.api.schema.constaints.NodeKeyConstraintDescriptor;
+import org.neo4j.kernel.api.schema.constaints.RelExistenceConstraintDescriptor;
+import org.neo4j.kernel.api.schema.constaints.UniquenessConstraintDescriptor;
+import org.neo4j.storageengine.api.StoreReadLayer;
+
+import static org.junit.Assert.assertArrayEquals;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.when;
+
+public class PropertyExistenceEnforcerTest
+{
+
+    @Test
+    public void constraintPropertyIdsNotUpdatedByConstraintEnforcer() throws Exception
+    {
+        UniquenessConstraintDescriptor uniquenessConstraint = ConstraintDescriptorFactory.uniqueForLabel( 1, 1, 70, 8 );
+        NodeKeyConstraintDescriptor nodeKeyConstraint = ConstraintDescriptorFactory.nodeKeyForLabel( 2, 12, 7, 13 );
+        RelExistenceConstraintDescriptor relTypeConstraint =
+                ConstraintDescriptorFactory.existsForRelType( 3, 5, 13, 8 );
+        List<ConstraintDescriptor> descriptors =
+                Arrays.asList( uniquenessConstraint, nodeKeyConstraint, relTypeConstraint );
+
+        StoreReadLayer storeReadLayer = prepareStoreReadLayerMock( descriptors );
+
+        PropertyExistenceEnforcer.getOrCreatePropertyExistenceEnforcerFrom( storeReadLayer );
+
+        assertArrayEquals( "Property ids should remain untouched.", new int[]{1, 70, 8},
+                uniquenessConstraint.schema().getPropertyIds() );
+        assertArrayEquals( "Property ids should remain untouched.", new int[]{12, 7, 13},
+                nodeKeyConstraint.schema().getPropertyIds() );
+        assertArrayEquals( "Property ids should remain untouched.", new int[]{5, 13, 8},
+                relTypeConstraint.schema().getPropertyIds() );
+    }
+
+    @SuppressWarnings( "unchecked" )
+    private StoreReadLayer prepareStoreReadLayerMock( List<ConstraintDescriptor> descriptors )
+    {
+        StoreReadLayer storeReadLayer = Mockito.mock( StoreReadLayer.class );
+        when( storeReadLayer.constraintsGetAll() ).thenReturn( descriptors.iterator() );
+        when( storeReadLayer.getOrCreateSchemaDependantState( eq( PropertyExistenceEnforcer.class ),
+                any( Function.class) ) ).thenAnswer( invocation ->
+        {
+            Function<StoreReadLayer, PropertyExistenceEnforcer> function = invocation.getArgumentAt( 1, Function.class );
+            return function.apply( storeReadLayer );
+        } );
+        return storeReadLayer;
+    }
+}