1/* 2 * Copyright 2014, Google Inc. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: 8 * 9 * * Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * * Redistributions in binary form must reproduce the above 12 * copyright notice, this list of conditions and the following disclaimer 13 * in the documentation and/or other materials provided with the 14 * distribution. 15 * * Neither the name of Google Inc. nor the names of its 16 * contributors may be used to endorse or promote products derived from 17 * this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32package org.jf.smalidea; 33 34import com.intellij.openapi.project.DumbServiceImpl; 35import com.intellij.openapi.projectRoots.Sdk; 36import com.intellij.openapi.projectRoots.impl.JavaAwareProjectJdkTableImpl; 37import com.intellij.psi.JavaResolveResult; 38import com.intellij.psi.PsiClass; 39import com.intellij.psi.PsiReference; 40import com.intellij.testFramework.ResolveTestCase; 41import org.jf.smalidea.psi.impl.SmaliClass; 42import org.jf.smalidea.psi.impl.SmaliClassTypeElement; 43import org.junit.Assert; 44 45public class ClassReferenceTest extends ResolveTestCase { 46 /** 47 * Test a reference to a java class from a smali class 48 */ 49 public void testJavaReferenceFromSmali() throws Exception { 50 SmaliClassTypeElement typeElement = (SmaliClassTypeElement)configureByFileText( 51 ".class public Lblah; .super L<ref>java/lang/Object;", "blah.smali"); 52 53 Assert.assertNotNull(typeElement); 54 Assert.assertEquals("Object", typeElement.getName()); 55 56 PsiClass psiClass = typeElement.resolve(); 57 Assert.assertNotNull(psiClass); 58 Assert.assertEquals("java.lang.Object", psiClass.getQualifiedName()); 59 60 JavaResolveResult resolveResult = typeElement.advancedResolve(false); 61 Assert.assertNotNull(resolveResult.getElement()); 62 Assert.assertEquals("java.lang.Object", ((PsiClass)resolveResult.getElement()).getQualifiedName()); 63 64 JavaResolveResult[] resolveResults = typeElement.multiResolve(false); 65 Assert.assertEquals(1, resolveResults.length); 66 Assert.assertNotNull(resolveResults[0].getElement()); 67 Assert.assertEquals("java.lang.Object", ((PsiClass)resolveResults[0].getElement()).getQualifiedName()); 68 } 69 70 /** 71 * Test a reference to a java class from a smali class, while in dumb mode 72 */ 73 public void testJavaReferenceFromSmaliInDumbMode() throws Exception { 74 SmaliClassTypeElement typeElement = (SmaliClassTypeElement)configureByFileText( 75 ".class public Lblah; .super L<ref>java/lang/Object;", "blah.smali"); 76 77 Assert.assertNotNull(typeElement); 78 Assert.assertEquals("Object", typeElement.getName()); 79 80 DumbServiceImpl.getInstance(getProject()).setDumb(true); 81 82 PsiClass psiClass = typeElement.resolve(); 83 Assert.assertNull(psiClass); 84 85 DumbServiceImpl.getInstance(getProject()).setDumb(false); 86 } 87 88 /** 89 * Test a reference to a smali class from a smali class 90 */ 91 public void testSmaliReferenceFromSmali() throws Exception { 92 createFile("blarg.smali", ".class public Lblarg; .super Ljava/lang/Object;"); 93 94 SmaliClassTypeElement typeElement = (SmaliClassTypeElement)configureByFileText( 95 ".class public Lblah; .super L<ref>blarg;", "blah.smali"); 96 97 Assert.assertEquals("blarg", typeElement.getName()); 98 99 SmaliClass smaliClass = (SmaliClass)typeElement.resolve(); 100 Assert.assertNotNull(smaliClass); 101 Assert.assertEquals("blarg", smaliClass.getQualifiedName()); 102 103 JavaResolveResult resolveResult = typeElement.advancedResolve(false); 104 Assert.assertNotNull(resolveResult.getElement()); 105 Assert.assertEquals("blarg", ((PsiClass)resolveResult.getElement()).getQualifiedName()); 106 107 JavaResolveResult[] resolveResults = typeElement.multiResolve(false); 108 Assert.assertEquals(1, resolveResults.length); 109 Assert.assertNotNull(resolveResults[0].getElement()); 110 Assert.assertEquals("blarg", ((PsiClass)resolveResults[0].getElement()).getQualifiedName()); 111 } 112 113 /** 114 * Test a reference to a smali class from a java class 115 */ 116 public void testSmaliReferenceFromJava() throws Exception { 117 createFile("blarg.smali", ".class public Lblarg; .super Ljava/lang/Object;"); 118 119 PsiReference reference = configureByFileText( 120 "public class blah extends bla<ref>rg { }", "blah.java"); 121 122 SmaliClass smaliClass = (SmaliClass)reference.resolve(); 123 Assert.assertNotNull(smaliClass); 124 Assert.assertEquals("blarg", smaliClass.getQualifiedName()); 125 } 126 127 128 129 @Override 130 protected Sdk getTestProjectJdk() { 131 return JavaAwareProjectJdkTableImpl.getInstanceEx().getInternalJdk(); 132 } 133} 134