public void testClassIsTrait() {
PsiFile psiFile = myFixture.configureByText("foo.kt", "trait Test { }");
final List<JetDeclaration> declarations = ((JetFile) psiFile).getDeclarations();
final JetClass jetClass = (JetClass) declarations.get(0);
final PsiJetClassStub stub = JetStubElementTypes.CLASS.createStub(jetClass, null);
assertEquals(true, stub.isTrait());
}
public void testSuperclassNames() {
final PsiFile psiFile =
myFixture.configureByText(
"foo.kt", "import java.util.ArrayList as alist\nclass C(): alist() { }");
final List<JetDeclaration> declarations = ((JetFile) psiFile).getDeclarations();
final JetClass jetClass = (JetClass) declarations.get(0);
final PsiJetClassStub stub = JetStubElementTypes.CLASS.createStub(jetClass, null);
final List<String> names = stub.getSuperNames();
assertSameElements(names, "ArrayList", "alist");
}
