ContextAwareVirtualFile.java

package com.siberika.idea.pascal.editor;

import com.intellij.psi.PsiElement;
import com.intellij.testFramework.LightVirtualFile;

/**
 * Author: George Bakhtadze
 * Date: 21/04/2017
 */
public class ContextAwareVirtualFile extends LightVirtualFile {

    private final PsiElement contextElement;

    public ContextAwareVirtualFile(String name, CharSequence content, PsiElement contextElement) {
        super(name, content);
        this.contextElement = contextElement;
    }

    public PsiElement getContextElement() {
        return contextElement;
    }
}