Merge pull request #520 from Shan1024/update-tests

Fix issues in tests
ballerina-attic · Jul 24, 2017 · b3a7f8e · b3a7f8e
2 parents d0ac453 + 2606cac
commit b3a7f8e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/...ava/org/ballerinalang/plugins/idea/completion/BallerinaKeywordsCompletionContributor.java b/...ava/org/ballerinalang/plugins/idea/completion/BallerinaKeywordsCompletionContributor.java
@@ -26,6 +26,7 @@
 import com.intellij.psi.util.PsiTreeUtil;
 import org.antlr.jetbrains.adaptor.psi.ANTLRPsiNode;
 import org.ballerinalang.plugins.idea.BallerinaTypes;
+import org.ballerinalang.plugins.idea.psi.AnnotationAttachmentNode;
 import org.ballerinalang.plugins.idea.psi.AnnotationDefinitionNode;
 import org.ballerinalang.plugins.idea.psi.CallableUnitBodyNode;
 import org.ballerinalang.plugins.idea.psi.ConnectorBodyNode;
@@ -107,7 +108,8 @@ public void fillCompletionVariants(@NotNull CompletionParameters parameters, @No
                     addCreateKeyword(result);
                 }
 
-                if (prevVisibleSibling != null && prevVisibleSibling.getText().matches("[;{}]")) {
+                if (prevVisibleSibling != null && prevVisibleSibling.getText().matches("[;{}]")
+                        && !(prevVisibleSibling.getParent() instanceof AnnotationAttachmentNode)) {
                     // Todo - change method
                     addOtherTypeAsLookup(result);
                     addXmlnsAsLookup(result);
@@ -119,7 +121,10 @@ public void fillCompletionVariants(@NotNull CompletionParameters parameters, @No
 
                     result.addAllElements(BallerinaCompletionUtils.createCommonKeywords());
                 }
-                addValueKeywords(result);
+                if (prevVisibleSibling == null
+                        || !(prevVisibleSibling.getParent() instanceof AnnotationAttachmentNode)) {
+                    addValueKeywords(result);
+                }
             }
 
             ConnectorBodyNode connectorBodyNode = PsiTreeUtil.getParentOfType(element, ConnectorBodyNode.class);