#558: fix HeadMismatchInspection's incorrect behaviour with quoted atoms

Author: deadok22

src/org/intellij/erlang/psi/impl/ErlangPsiImplUtil.java

@@ -1388,23 +1388,26 @@ public Icon getIcon(boolean b) {
 
   @NotNull
   public static String createFunctionPresentation(@NotNull ErlangAtomWithArityExpression function) {
-    return getName(function.getQAtom()) + "/" + getArity(function.getInteger());
+    return createFunctionPresentation(getName(function.getQAtom()), getArity(function.getInteger()));
   }
 
   @NotNull
-  public static String createFunctionClausePresentation(@Nullable ErlangFunctionClause clause) {
-    if (clause == null) return "";
-    return clause.getQAtom().getText() + "/" + calculateFunctionClauseArity(clause);
+  public static String createFunctionClausePresentation(@NotNull ErlangFunctionClause clause) {
+    return createFunctionPresentation(getName(clause.getQAtom()), calculateFunctionClauseArity(clause));
   }
 
   @NotNull
   public static String createFunctionPresentation(@NotNull ErlangFunction function) {
-    return function.getName() + "/" + function.getArity();
+    return createFunctionPresentation(function.getName(), function.getArity());
   }
 
   @NotNull
   public static String createFunctionPresentation(@NotNull ErlangImportFunction function) {
-    return getName(function.getQAtom()) + "/" + getArity(function.getInteger());
+    return createFunctionPresentation(getName(function.getQAtom()), getArity(function.getInteger()));
+  }
+
+  private static String createFunctionPresentation(String functionName, int arity) {
+    return toAtomName(functionName) + "/" + arity;
   }
 
   @NotNull

testData/highlighting/MismatchedHead.erl

@@ -1,4 +1,4 @@
--export([foo/0, ?foo/1]).
+-export([foo/0, ?foo/1, 'CAPS'/1]).
 
 foo() -> ok;
 <error>zoo</error>() -> ok;
@@ -8,4 +8,7 @@ foo() -> ok;
 -define(bar, fun1).
 
 ?foo(1) -> ok;
-?bar(2) -> ok.
+?bar(2) -> ok.
+
+'CAPS'(1) -> ok;
+'CAPS'(2) -> ok.