diff --git a/python/src/com/jetbrains/python/documentation/doctest/PyDocstringParsingContext.java b/python/src/com/jetbrains/python/documentation/doctest/PyDocstringParsingContext.java index 008f562bfe39..dcf32d68200c 100644 --- a/python/src/com/jetbrains/python/documentation/doctest/PyDocstringParsingContext.java +++ b/python/src/com/jetbrains/python/documentation/doctest/PyDocstringParsingContext.java @@ -14,25 +14,25 @@ import org.jetbrains.annotations.Nullable; * User : ktisha */ public class PyDocstringParsingContext extends ParsingContext { - private final StatementParsing stmtParser; - private final ExpressionParsing exprParser; + private final StatementParsing myStatementParser; + private final ExpressionParsing myExpressionParser; public PyDocstringParsingContext(final PsiBuilder builder, LanguageLevel languageLevel, StatementParsing.FUTURE futureFlag) { super(builder, languageLevel, futureFlag); - stmtParser = new PyDocstringStatementParsing(this, futureFlag); - exprParser = new PyDocstringExpressionParsing(this); + myStatementParser = new PyDocstringStatementParsing(this, futureFlag); + myExpressionParser = new PyDocstringExpressionParsing(this); } @Override public ExpressionParsing getExpressionParser() { - return exprParser; + return myExpressionParser; } @Override public StatementParsing getStatementParser() { - return stmtParser; + return myStatementParser; } private static class PyDocstringExpressionParsing extends ExpressionParsing { @@ -53,6 +53,11 @@ public class PyDocstringParsingContext extends ParsingContext { super(context, futureFlag); } + @Override + protected IElementType getReferenceType() { + return PyDocstringTokenTypes.DOC_REFERENCE; + } + @Override public IElementType filter(IElementType source, int start, int end, CharSequence text) { if (source == PyTokenTypes.DOT && CharArrayUtil.regionMatches(text, start, end, "..."))