Dmitry Jemerov
113 lines
3.7 KiB
Java
113 lines
3.7 KiB
Java
/*
|
|
* Copyright 2010-2015 JetBrains s.r.o.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package org.jetbrains.kotlin.test;
|
|
|
|
import com.intellij.openapi.fileEditor.impl.LoadTextUtil;
|
|
import com.intellij.psi.PsiElement;
|
|
import com.intellij.psi.PsiElementVisitor;
|
|
import com.intellij.psi.PsiFile;
|
|
import com.intellij.testFramework.LightVirtualFile;
|
|
import com.intellij.testFramework.TestDataFile;
|
|
import org.jetbrains.annotations.NonNls;
|
|
import org.jetbrains.annotations.Nullable;
|
|
import org.jetbrains.kotlin.psi.KtFile;
|
|
import org.jetbrains.kotlin.resolve.lazy.KotlinTestWithEnvironment;
|
|
import org.junit.Assert;
|
|
|
|
import java.io.IOException;
|
|
|
|
public abstract class KotlinLiteFixture extends KotlinTestWithEnvironment {
|
|
@NonNls
|
|
protected final String myFullDataPath;
|
|
private KtFile myFile;
|
|
|
|
public KotlinLiteFixture(@NonNls String dataPath) {
|
|
myFullDataPath = getTestDataPath() + "/" + dataPath;
|
|
}
|
|
|
|
public KotlinLiteFixture() {
|
|
myFullDataPath = getTestDataPath();
|
|
}
|
|
|
|
protected KtFile getFile() {
|
|
return myFile;
|
|
}
|
|
|
|
protected String getTestDataPath() {
|
|
return JetTestUtils.getTestDataPathBase();
|
|
}
|
|
|
|
@Override
|
|
protected void setUp() throws Exception {
|
|
super.setUp();
|
|
}
|
|
|
|
@Override
|
|
protected void tearDown() throws Exception {
|
|
myFile = null;
|
|
super.tearDown();
|
|
}
|
|
|
|
protected String loadFile(@NonNls @TestDataFile String name) throws IOException {
|
|
return JetTestUtils.doLoadFile(myFullDataPath, name);
|
|
}
|
|
|
|
protected KtFile createPsiFile(@Nullable String testName, @Nullable String fileName, String text) {
|
|
if (fileName == null) {
|
|
Assert.assertNotNull(testName);
|
|
fileName = testName + ".kt";
|
|
}
|
|
return JetTestUtils.createFile(fileName, text, getProject());
|
|
}
|
|
|
|
protected KtFile loadPsiFile(String name) {
|
|
try {
|
|
return createPsiFile(name, null, loadFile(name));
|
|
} catch (IOException e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
}
|
|
|
|
protected static void ensureParsed(PsiFile file) {
|
|
file.accept(new PsiElementVisitor() {
|
|
@Override
|
|
public void visitElement(PsiElement element) {
|
|
element.acceptChildren(this);
|
|
}
|
|
});
|
|
}
|
|
|
|
protected void prepareForTest(String name) throws IOException {
|
|
String text = loadFile(name + ".kt");
|
|
createAndCheckPsiFile(name, text);
|
|
}
|
|
|
|
protected void createAndCheckPsiFile(String name, String text) {
|
|
myFile = createCheckAndReturnPsiFile(name, null, text);
|
|
}
|
|
|
|
protected KtFile createCheckAndReturnPsiFile(String testName, String fileName, String text) {
|
|
KtFile myFile = createPsiFile(testName, fileName, text);
|
|
ensureParsed(myFile);
|
|
assertEquals("light virtual file text mismatch", text, ((LightVirtualFile) myFile.getVirtualFile()).getContent().toString());
|
|
assertEquals("virtual file text mismatch", text, LoadTextUtil.loadText(myFile.getVirtualFile()));
|
|
assertEquals("doc text mismatch", text, myFile.getViewProvider().getDocument().getText());
|
|
assertEquals("psi text mismatch", text, myFile.getText());
|
|
return myFile;
|
|
}
|
|
}
|