Add some reference contribution tests

Author: FylmTM

language/cypher/src/main/java/com/neueda/jetbrains/plugin/graphdb/language/cypher/completion/metadata/CypherMetadataProviderService.java

@@ -23,6 +23,12 @@ public interface CypherMetadataProviderService {
 
     List<CypherUserFunctionElement> getUserFunctions();
 
+    Optional<CypherLabelElement> findLabel(String labelName);
+
+    Optional<CypherRelationshipTypeElement> findRelationshipType(String relationshipTypeName);
+
+    Optional<CypherPropertyKeyElement> findPropertyKey(String propertyKeyName);
+
     Optional<CypherProcedureElement> findProcedure(String fullName);
 
     Optional<CypherUserFunctionElement> findUserFunction(String fullName);

language/cypher/src/main/java/com/neueda/jetbrains/plugin/graphdb/language/cypher/completion/metadata/CypherMetadataProviderServiceImpl.java

@@ -74,6 +74,33 @@ public List<CypherUserFunctionElement> getUserFunctions() {
                 .collect(Collectors.toList());
     }
 
+    @Override
+    public Optional<CypherLabelElement> findLabel(String labelName) {
+        return sourceData.entrySet().stream()
+                .map(Map.Entry::getValue)
+                .flatMap((container) -> container.getLabels().stream())
+                .filter((labelElement) -> labelElement.getName().equals(labelName))
+                .findFirst();
+    }
+
+    @Override
+    public Optional<CypherRelationshipTypeElement> findRelationshipType(String relationshipTypeName) {
+        return sourceData.entrySet().stream()
+                .map(Map.Entry::getValue)
+                .flatMap((container) -> container.getRelationshipTypes().stream())
+                .filter((relationshipTypeElement) -> relationshipTypeElement.getName().equals(relationshipTypeName))
+                .findFirst();
+    }
+
+    @Override
+    public Optional<CypherPropertyKeyElement> findPropertyKey(String propertyKeyName) {
+        return sourceData.entrySet().stream()
+                .map(Map.Entry::getValue)
+                .flatMap((container) -> container.getPropertyKeys().stream())
+                .filter((propertyKeyElement) -> propertyKeyElement.getName().equals(propertyKeyName))
+                .findFirst();
+    }
+
     @Override
     public Optional<CypherProcedureElement> findProcedure(String name) {
         return sourceData.entrySet().stream()

language/cypher/src/main/java/com/neueda/jetbrains/plugin/graphdb/language/cypher/completion/metadata/elements/CypherLabelElement.java

@@ -18,4 +18,8 @@ public LookupElement getLookupElement() {
                 .withIcon(GraphIcons.Nodes.LABEL)
                 .withTypeText("label");
     }
+
+    public String getName() {
+        return label;
+    }
 }

language/cypher/src/main/java/com/neueda/jetbrains/plugin/graphdb/language/cypher/completion/metadata/elements/CypherPropertyKeyElement.java

@@ -18,4 +18,8 @@ public LookupElement getLookupElement() {
                 .withIcon(GraphIcons.Nodes.PROPERTY_KEY)
                 .withTypeText("property");
     }
+
+    public String getName() {
+        return propertyKey;
+    }
 }

language/cypher/src/main/java/com/neueda/jetbrains/plugin/graphdb/language/cypher/completion/metadata/elements/CypherRelationshipTypeElement.java

@@ -18,4 +18,8 @@ public LookupElement getLookupElement() {
                 .withIcon(GraphIcons.Nodes.RELATIONSHIP_TYPE)
                 .withTypeText("relationship type");
     }
+
+    public String getName() {
+        return relationshipType;
+    }
 }

language/cypher/src/main/java/com/neueda/jetbrains/plugin/graphdb/language/cypher/references/CypherReferenceContributor.java

@@ -2,11 +2,7 @@
 
 import com.intellij.patterns.ElementPattern;
 import com.intellij.patterns.PlatformPatterns;
-import com.intellij.psi.PsiElement;
-import com.intellij.psi.PsiReference;
-import com.intellij.psi.PsiReferenceContributor;
-import com.intellij.psi.PsiReferenceProvider;
-import com.intellij.psi.PsiReferenceRegistrar;
+import com.intellij.psi.*;
 import com.intellij.util.ProcessingContext;
 import com.neueda.jetbrains.plugin.graphdb.language.cypher.CypherLanguage;
 import com.neueda.jetbrains.plugin.graphdb.language.cypher.psi.CypherLabelName;

testing/integration-neo4j/src/test/java/com/neueda/jetbrains/plugin/graphdb/test/integration/neo4j/tests/cypher/reference/LabelReferenceContributorTest.java

@@ -0,0 +1,19 @@
+package com.neueda.jetbrains.plugin.graphdb.test.integration.neo4j.tests.cypher.reference;
+
+import com.neueda.jetbrains.plugin.graphdb.test.integration.neo4j.tests.cypher.util.BaseCompletionTest;
+
+public class LabelReferenceContributorTest extends BaseCompletionTest {
+
+    public void testPresentInCompletion() throws Exception {
+        myFixture.configureByText("test.cyp", "MATCH (n:TestLabel) RETURN n; MATCH (n:Test<caret>");
+        myFixture.completeBasic();
+        myFixture.checkResult("MATCH (n:TestLabel) RETURN n; MATCH (n:TestLabel<caret>");
+    }
+
+    public void _testOneEntryPresentInCompletionIfMetadataExists() throws Exception {
+        addLabel("TestLabel");
+        myFixture.configureByText("test.cyp", "MATCH (n:TestLabel) RETURN n; MATCH (n:Test<caret>");
+        myFixture.completeBasic();
+        myFixture.checkResult("MATCH (n:TestLabel) RETURN n; MATCH (n:TestLabel<caret>");
+    }
+}

testing/integration-neo4j/src/test/java/com/neueda/jetbrains/plugin/graphdb/test/integration/neo4j/tests/cypher/reference/PropertyReferenceContributorTest.java

@@ -0,0 +1,19 @@
+package com.neueda.jetbrains.plugin.graphdb.test.integration.neo4j.tests.cypher.reference;
+
+import com.neueda.jetbrains.plugin.graphdb.test.integration.neo4j.tests.cypher.util.BaseCompletionTest;
+
+public class PropertyReferenceContributorTest extends BaseCompletionTest {
+
+    public void testPresentInCompletion() throws Exception {
+        myFixture.configureByText("test.cyp", "MATCH (n) WHERE n.testProperty = 1 RETURN n.testProp<caret>");
+        myFixture.completeBasic();
+        myFixture.checkResult("MATCH (n) WHERE n.testProperty = 1 RETURN n.testProperty<caret>");
+    }
+
+    public void _testOneEntryPresentInCompletionIfMetadataExists() throws Exception {
+        addProperty("testProperty");
+        myFixture.configureByText("test.cyp", "MATCH (n) WHERE n.testProperty = 1 RETURN n.testProp<caret>");
+        myFixture.completeBasic();
+        myFixture.checkResult("MATCH (n) WHERE n.testProperty = 1 RETURN n.testProperty<caret>");
+    }
+}

testing/integration-neo4j/src/test/java/com/neueda/jetbrains/plugin/graphdb/test/integration/neo4j/tests/cypher/reference/RelationshipTypeReferenceContributorTest.java

@@ -0,0 +1,19 @@
+package com.neueda.jetbrains.plugin.graphdb.test.integration.neo4j.tests.cypher.reference;
+
+import com.neueda.jetbrains.plugin.graphdb.test.integration.neo4j.tests.cypher.util.BaseCompletionTest;
+
+public class RelationshipTypeReferenceContributorTest extends BaseCompletionTest {
+
+    public void testPresentInCompletion() throws Exception {
+        myFixture.configureByText("test.cyp", "MATCH ()-[:TESTRELTYPE]-() MATCH ()-[:TESTREL<caret>");
+        myFixture.completeBasic();
+        myFixture.checkResult("MATCH ()-[:TESTRELTYPE]-() MATCH ()-[:TESTRELTYPE<caret>");
+    }
+
+    public void _testOneEntryPresentInCompletionIfMetadataExists() throws Exception {
+        addRelationshipType("TESTRELTYPE");
+        myFixture.configureByText("test.cyp", "MATCH ()-[:TESTRELTYPE]-() MATCH ()-[:TESTREL<caret>");
+        myFixture.completeBasic();
+        myFixture.checkResult("MATCH ()-[:TESTRELTYPE]-() MATCH ()-[:TESTRELTYPE<caret>");
+    }
+}

testing/integration-neo4j/src/test/java/com/neueda/jetbrains/plugin/graphdb/test/integration/neo4j/tests/cypher/util/BaseCompletionTest.java

@@ -15,4 +15,16 @@ public void setUp() throws Exception {
     public void addStoredProcedure(String name, String signature, String description) {
         metadata.addProcedure(name, signature, description);
     }
+
+    public void addLabel(String name) {
+        metadata.addLabel(name);
+    }
+
+    public void addRelationshipType(String name) {
+        metadata.addRelationshipType(name);
+    }
+
+    public void addProperty(String name) {
+        metadata.addPropertyKey(name);
+    }
 }