/* * Copyright 2000-2012 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 com.intellij.psi; import com.intellij.openapi.application.ApplicationManager; import com.intellij.openapi.application.ex.PathManagerEx; import com.intellij.openapi.util.Comparing; import com.intellij.openapi.vfs.LocalFileSystem; import com.intellij.openapi.vfs.VirtualFile; import com.intellij.psi.impl.JavaPsiFacadeEx; import com.intellij.psi.search.GlobalSearchScope; import com.intellij.psi.util.PsiUtil; import com.intellij.testFramework.PsiTestUtil; import java.io.File; /** * @author dsl */ public class TypesTest extends GenericsTestCase { @Override protected void setUp() throws Exception { super.setUp(); setupGenericSampleClasses(); final String testPath = PathManagerEx.getTestDataPath().replace(File.separatorChar, '/') + "/psi/types/" + getTestName(true); final VirtualFile[] testRoot = { null }; ApplicationManager.getApplication().runWriteAction(() -> { testRoot[0] = LocalFileSystem.getInstance().refreshAndFindFileByPath(testPath); }); if (testRoot[0] != null) { PsiTestUtil.addSourceRoot(myModule, testRoot[0]); } } public void testSimpleStuff() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.getMethods()[0]; final PsiStatement[] methodStatements = method.getBody().getStatements(); final PsiDeclarationStatement declarationStatement = (PsiDeclarationStatement) methodStatements[0]; final PsiVariable varList = (PsiVariable) declarationStatement.getDeclaredElements()[0]; final PsiType typeListOfA = factory.createTypeFromText("test.List<java.lang.String>", null); assertEquals(varList.getType(), typeListOfA); final PsiType typeListOfObject = factory.createTypeFromText("test.List<java.lang.Object>", null); assertFalse(varList.getType().equals(typeListOfObject)); final PsiReferenceExpression methodExpression = ((PsiMethodCallExpression) ((PsiExpressionStatement) methodStatements[1]).getExpression()).getMethodExpression(); final JavaResolveResult resolveResult = methodExpression.advancedResolve(false); assertTrue(resolveResult.getElement() instanceof PsiMethod); final PsiMethod methodFromList = (PsiMethod) resolveResult.getElement(); final PsiType typeOfFirstParameterOfAdd = methodFromList.getParameterList().getParameters()[0].getType(); final PsiType substitutedType = resolveResult.getSubstitutor().substitute(typeOfFirstParameterOfAdd); final PsiClassType typeA = factory.createTypeByFQClassName("java.lang.String"); assertEquals(typeA, substitutedType); assertTrue(typeA.equalsToText("java.lang.String")); final PsiType aListIteratorType = ((PsiExpressionStatement) methodStatements[2]).getExpression().getType(); final PsiType aIteratorType = factory.createTypeFromText("test.Iterator<java.lang.String>", null); assertEquals(aIteratorType, aListIteratorType); final PsiType objectIteratorType = factory.createTypeFromText("test.Iterator<java.lang.Object>", null); assertFalse(objectIteratorType.equals(aListIteratorType)); } public void testRawTypes() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.getMethods()[0]; final PsiStatement[] methodStatements = method.getBody().getStatements(); final PsiDeclarationStatement declarationStatement = (PsiDeclarationStatement) methodStatements[0]; final PsiVariable varList = (PsiVariable) declarationStatement.getDeclaredElements()[0]; final PsiType typeFromText = factory.createTypeFromText("test.List", null); assertEquals(varList.getType(), typeFromText); final PsiReferenceExpression methodExpression = ((PsiMethodCallExpression) ((PsiExpressionStatement) methodStatements[1]).getExpression()).getMethodExpression(); final JavaResolveResult resolveResult = methodExpression.advancedResolve(false); assertTrue(resolveResult.getElement() instanceof PsiMethod); final PsiMethod methodFromList = (PsiMethod) resolveResult.getElement(); final PsiType typeOfFirstParameterOfAdd = methodFromList.getParameterList().getParameters()[0].getType(); final PsiType substitutedType = resolveResult.getSubstitutor().substitute(typeOfFirstParameterOfAdd); assertEquals(PsiType.getJavaLangObject(getPsiManager(), method.getResolveScope()), substitutedType); final PsiType methodCallType = ((PsiExpressionStatement) methodStatements[2]).getExpression().getType(); final PsiType rawIteratorType = factory.createTypeFromText("test.Iterator", null); assertEquals(rawIteratorType, methodCallType); } public void testSubstWithInheritor() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.getMethods()[0]; final PsiStatement[] methodStatements = method.getBody().getStatements(); final PsiDeclarationStatement declarationStatement = (PsiDeclarationStatement) methodStatements[0]; final PsiVariable varList = (PsiVariable) declarationStatement.getDeclaredElements()[0]; final PsiType typeFromText = factory.createTypeFromText("test.IntList", null); assertEquals(varList.getType(), typeFromText); final PsiReferenceExpression methodExpression = ((PsiMethodCallExpression) ((PsiExpressionStatement) methodStatements[1]).getExpression()).getMethodExpression(); final JavaResolveResult resolveResult = methodExpression.advancedResolve(false); assertTrue(resolveResult.getElement() instanceof PsiMethod); final PsiMethod methodFromList = (PsiMethod) resolveResult.getElement(); final PsiType typeOfFirstParameterOfAdd = methodFromList.getParameterList().getParameters()[0].getType(); final PsiType substitutedType = resolveResult.getSubstitutor().substitute(typeOfFirstParameterOfAdd); final PsiType javaLangInteger = factory.createTypeFromText("java.lang.Integer", null); assertEquals(javaLangInteger, substitutedType); final PsiType intListIteratorReturnType = ((PsiExpressionStatement) methodStatements[2]).getExpression().getType(); final PsiType integerIteratorType = factory.createTypeFromText("test.Iterator<java.lang.Integer>", null); assertEquals(integerIteratorType, intListIteratorReturnType); final PsiType objectIteratorType = factory.createTypeFromText("test.Iterator<java.lang.Object>", null); assertFalse(objectIteratorType.equals(integerIteratorType)); } public void testSimpleRawTypeInMethodArg() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.getMethods()[0]; final PsiStatement[] methodStatements = method.getBody().getStatements(); final PsiVariable variable = (PsiVariable) ((PsiDeclarationStatement) methodStatements[0]).getDeclaredElements()[0]; final PsiClassType type = (PsiClassType) variable.getType(); final PsiClassType.ClassResolveResult resolveClassTypeResult = type.resolveGenerics(); assertNotNull(resolveClassTypeResult.getElement()); final PsiReferenceExpression methodExpression = ((PsiMethodCallExpression) ((PsiExpressionStatement) methodStatements[2]).getExpression()).getMethodExpression(); final PsiExpression qualifierExpression = methodExpression.getQualifierExpression(); final PsiClassType qualifierType = (PsiClassType) qualifierExpression.getType(); assertFalse(qualifierType.hasParameters()); final PsiType typeFromText = factory.createTypeFromText("test.List", null); assertEquals(qualifierType, typeFromText); final PsiElement psiElement = ((PsiReferenceExpression) qualifierExpression).resolve(); assertTrue(psiElement instanceof PsiVariable); final JavaResolveResult resolveResult = methodExpression.advancedResolve(false); assertTrue(resolveResult.getElement() instanceof PsiMethod); final PsiMethod methodFromList = (PsiMethod) resolveResult.getElement(); assertEquals("add", methodFromList.getName()); assertEquals("test.List", methodFromList.getContainingClass().getQualifiedName()); } public void testRawTypeInMethodArg() throws Exception { final PsiClass classA = getJavaFacade().findClass("A"); assertNotNull(classA); final PsiMethod method = classA.getMethods()[0]; final PsiStatement[] methodStatements = method.getBody().getStatements(); final PsiReferenceExpression methodExpression = ((PsiMethodCallExpression) ((PsiExpressionStatement) methodStatements[2]).getExpression()).getMethodExpression(); final JavaResolveResult resolveResult = methodExpression.advancedResolve(false); assertTrue(resolveResult.getElement() instanceof PsiMethod); final PsiMethod methodFromList = (PsiMethod) resolveResult.getElement(); assertEquals("putAll", methodFromList.getName()); assertEquals("test.List", methodFromList.getContainingClass().getQualifiedName()); } public void testBoundedParams() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.getMethods()[0]; final PsiStatement[] statements = method.getBody().getStatements(); final PsiVariable var = (PsiVariable) ((PsiDeclarationStatement) statements[0]).getDeclaredElements()[0]; final PsiType varType = var.getType(); final PsiType typeRawIterator = factory.createTypeFromText("test.Iterator", null); assertEquals(varType, typeRawIterator); final PsiType initializerType = var.getInitializer().getType(); assertEquals(initializerType, typeRawIterator); assertTrue(varType.isAssignableFrom(initializerType)); } public void testRawTypeExtension() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.getMethods()[0]; final PsiStatement[] statements = method.getBody().getStatements(); final PsiVariable var = (PsiVariable) ((PsiDeclarationStatement) statements[0]).getDeclaredElements()[0]; final PsiType varType = var.getType(); final PsiType typeRawIterator = factory.createTypeFromText("test.Iterator", null); assertEquals(varType, typeRawIterator); final PsiType initializerType = var.getInitializer().getType(); assertEquals(initializerType, typeRawIterator); assertTrue(varType.isAssignableFrom(initializerType)); } public void testTypesInGenericClass() { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiTypeParameter parameterT = classA.getTypeParameters()[0]; assertEquals("T", parameterT.getName()); final PsiMethod method = classA.findMethodsByName("method", false)[0]; final PsiType type = ((PsiExpressionStatement) method.getBody().getStatements()[0]).getExpression().getType(); final PsiClassType typeT = factory.createType(parameterT); assertEquals("T", typeT.getPresentableText()); assertEquals(typeT, type); } public void testAssignableSubInheritor() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classCollection = psiManager.findClass("test.Collection"); final PsiClass classList = psiManager.findClass("test.List"); final PsiType collectionType = factory.createType(classCollection, PsiSubstitutor.EMPTY); final PsiType listType = factory.createType(classList, PsiSubstitutor.EMPTY); assertEquals(collectionType.getCanonicalText(), "test.Collection<E>"); assertEquals(listType.getCanonicalText(), "test.List<T>"); final PsiType typeListOfString = factory.createTypeFromText("test.List<java.lang.String>", null); final PsiType typeCollectionOfString = factory.createTypeFromText("test.Collection<java.lang.String>", null); assertTrue(typeCollectionOfString.isAssignableFrom(typeListOfString)); } public void testComplexInheritance() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.findMethodsByName("method", false)[0]; final PsiExpression expression = ((PsiExpressionStatement) method.getBody().getStatements()[1]).getExpression(); assertEquals("l.get(0)", expression.getText()); final PsiType type = expression.getType(); final PsiType listOfInteger = factory.createTypeFromText("test.List<java.lang.Integer>", null); assertEquals(listOfInteger, type); final PsiType collectionOfInteger = factory.createTypeFromText("test.Collection<java.lang.Integer>", null); assertTrue(collectionOfInteger.isAssignableFrom(type)); } public void testListListInheritance() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.findMethodsByName("method", false)[0]; final PsiExpression expression1 = ((PsiExpressionStatement) method.getBody().getStatements()[1]).getExpression(); assertEquals("l.get(0)", expression1.getText()); final PsiType type1 = expression1.getType(); final PsiType typeListOfInteger = factory.createTypeFromText("test.List<java.lang.Integer>", null); assertEquals(typeListOfInteger, type1); assertTrue(typeListOfInteger.isAssignableFrom(type1)); final PsiExpression expression2 = ((PsiExpressionStatement) method.getBody().getStatements()[3]).getExpression(); assertEquals("b.get(0)", expression2.getText()); final PsiType type2 = expression2.getType(); assertEquals(typeListOfInteger, type2); } public void testSpaceInTypeParameterList() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.findMethodsByName("method", false)[0]; final PsiVariable variable = (PsiVariable) ((PsiDeclarationStatement) method.getBody().getStatements()[0]).getDeclaredElements()[0]; final PsiType type = variable.getType(); final PsiType typeListOfListOfInteger = factory.createTypeFromText("test.List<test.List<java.lang.Integer>>", null); assertEquals(typeListOfListOfInteger, type); } public void testMethodTypeParameter() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.findMethodsByName("method", false)[0]; final PsiStatement[] statements = method.getBody().getStatements(); final PsiMethodCallExpression methodCallExpression = (PsiMethodCallExpression) ((PsiExpressionStatement) statements[1]).getExpression(); isCollectionUtilSort(methodCallExpression, factory.createTypeFromText("java.lang.Integer", null)); final PsiMethodCallExpression methodCallExpression1 = (PsiMethodCallExpression) ((PsiExpressionStatement) statements[3]).getExpression(); isCollectionUtilSort(methodCallExpression1, null); } private static void isCollectionUtilSort(final PsiMethodCallExpression methodCallExpression, final PsiType typeParameterValue) { final PsiReferenceExpression methodExpression = methodCallExpression.getMethodExpression(); final JavaResolveResult methodResolve = methodExpression.advancedResolve(false); assertTrue(methodResolve.getElement() instanceof PsiMethod); final PsiMethod methodSort = (PsiMethod) methodResolve.getElement(); assertEquals("sort", methodSort.getName()); assertEquals("test.CollectionUtil", methodSort.getContainingClass().getQualifiedName()); final PsiTypeParameter methodSortTypeParameter = methodSort.getTypeParameters()[0]; final PsiType sortParameterActualType = methodResolve.getSubstitutor().substitute(methodSortTypeParameter); assertTrue(Comparing.equal(sortParameterActualType, typeParameterValue)); assertTrue( PsiUtil.isApplicable(methodSort, methodResolve.getSubstitutor(), methodCallExpression.getArgumentList())); } public void testRawArrayTypes() throws Exception { final JavaPsiFacadeEx psiManager = getJavaFacade(); final PsiElementFactory factory = psiManager.getElementFactory(); final PsiClass classA = psiManager.findClass("A"); assertNotNull(classA); final PsiMethod method = classA.findMethodsByName("method", false)[0]; final PsiStatement[] statements = method.getBody().getStatements(); final PsiDeclarationStatement declarationStatement = (PsiDeclarationStatement) statements[0]; final PsiClassType typeOfL = (PsiClassType) ((PsiVariable) declarationStatement.getDeclaredElements()[0]).getType(); final PsiType typeRawList = factory.createTypeFromText("test.List", null); assertTrue(Comparing.equal(typeOfL, typeRawList)); final PsiSubstitutor typeOfLSubstitutor = typeOfL.resolveGenerics().getSubstitutor(); final PsiMethodCallExpression exprGetArray = (PsiMethodCallExpression) ((PsiExpressionStatement) statements[1]).getExpression(); final PsiType typeOfGetArrayCall = exprGetArray.getType(); final PsiType objectArrayType = factory.createTypeFromText("java.lang.Object[]", null); assertTrue(Comparing.equal(typeOfGetArrayCall, objectArrayType)); final PsiMethod methodGetArray = (PsiMethod) exprGetArray.getMethodExpression().resolve(); final PsiType subtitutedGetArrayReturnType = typeOfLSubstitutor.substitute(methodGetArray.getReturnType()); assertTrue(Comparing.equal(subtitutedGetArrayReturnType, objectArrayType)); final PsiMethodCallExpression exprGetListOfArray = (PsiMethodCallExpression) ((PsiExpressionStatement) statements[2]).getExpression(); final PsiMethod methodGetListOfArray = (PsiMethod) exprGetListOfArray.getMethodExpression().resolve(); final PsiType returnType = methodGetListOfArray.getReturnType(); final PsiType substitutedReturnType = typeOfLSubstitutor.substitute(returnType); assertTrue(Comparing.equal(substitutedReturnType, typeRawList)); final PsiType typeOfGetListOfArrayCall = exprGetListOfArray.getType(); assertTrue(Comparing.equal(typeOfGetListOfArrayCall, typeRawList)); } public void testWildcardTypeParsing() throws Exception{ final GlobalSearchScope scope = GlobalSearchScope.moduleWithDependenciesAndLibrariesScope(myModule); final PsiClassType javaLangObject = PsiType.getJavaLangObject(myPsiManager, scope); PsiElement element = ((PsiDeclarationStatement)myJavaFacade.getElementFactory().createStatementFromText("X<? extends Y, ? super Z<A,B>, ?> x;", null)).getDeclaredElements()[0]; PsiJavaCodeReferenceElement referenceElement = ((PsiVariable) element).getTypeElement().getInnermostComponentReferenceElement(); PsiType[] typeArguments = referenceElement.getTypeParameters(); assertEquals(3, typeArguments.length); assertTrue(typeArguments[0] instanceof PsiWildcardType); assertTrue(typeArguments[1] instanceof PsiWildcardType); assertTrue(typeArguments[2] instanceof PsiWildcardType); PsiWildcardType extendsWildcard = (PsiWildcardType)typeArguments[0]; PsiWildcardType superWildcard = (PsiWildcardType)typeArguments[1]; PsiWildcardType unboundedWildcard = (PsiWildcardType)typeArguments[2]; // extends wildcard test assertTrue(extendsWildcard.isExtends()); assertFalse(extendsWildcard.isSuper()); assertEquals("Y", extendsWildcard.getBound().getCanonicalText()); assertEquals("Y", extendsWildcard.getExtendsBound().getCanonicalText()); assertEquals(extendsWildcard.getSuperBound(), PsiType.NULL); // super wildcard test assertFalse(superWildcard.isExtends()); assertTrue(superWildcard.isSuper()); assertEquals("Z<A,B>", superWildcard.getBound().getCanonicalText()); assertEquals(superWildcard.getExtendsBound(), javaLangObject); assertEquals("Z<A,B>", superWildcard.getSuperBound().getCanonicalText()); // unbounded wildcard test assertFalse(unboundedWildcard.isExtends()); assertFalse(unboundedWildcard.isSuper()); assertNull(unboundedWildcard.getBound()); assertEquals(unboundedWildcard.getExtendsBound(), javaLangObject); assertEquals(unboundedWildcard.getSuperBound(), PsiType.NULL); } public void testWildcardTypesAssignable() throws Exception { PsiClassType listOfExtendsBase = (PsiClassType)myJavaFacade.getElementFactory().createTypeFromText("test.List<? extends usages.Base>", null); PsiClassType.ClassResolveResult classResolveResult = listOfExtendsBase.resolveGenerics(); PsiClass listClass = classResolveResult.getElement(); assertNotNull(listClass); PsiTypeParameter listTypeParameter = PsiUtil.typeParametersIterator(listClass).next(); PsiType listParameterTypeValue = classResolveResult.getSubstitutor().substitute(listTypeParameter); assertTrue(listParameterTypeValue instanceof PsiWildcardType); assertTrue(((PsiWildcardType)listParameterTypeValue).isExtends()); assertEquals("usages.Base", ((PsiWildcardType)listParameterTypeValue).getBound().getCanonicalText()); PsiClassType listOfIntermediate = (PsiClassType)myJavaFacade.getElementFactory().createTypeFromText("test.List<usages.Intermediate>", null); assertNotNull(listOfIntermediate.resolve()); assertTrue(listOfExtendsBase.isAssignableFrom(listOfIntermediate)); } public void testEllipsisType() throws Exception { PsiElementFactory factory = myJavaFacade.getElementFactory(); PsiMethod method = factory.createMethodFromText("void foo (int ... args) {}", null); PsiType paramType = method.getParameterList().getParameters()[0].getType(); assertTrue(paramType instanceof PsiEllipsisType); PsiType arrayType = ((PsiEllipsisType)paramType).getComponentType().createArrayType(); assertTrue(paramType.isAssignableFrom(arrayType)); assertTrue(arrayType.isAssignableFrom(paramType)); PsiType typeFromText = factory.createTypeFromText("int ...", null); assertTrue(typeFromText instanceof PsiEllipsisType); } public void testBinaryNumericPromotion() throws Exception { PsiElementFactory factory = myJavaFacade.getElementFactory(); final PsiExpression conditional = factory.createExpressionFromText("b ? new Integer (0) : new Double(0.0)", null); assertEquals(PsiType.DOUBLE, conditional.getType()); final PsiExpression shift = factory.createExpressionFromText("Integer.valueOf(0) << 2", null); assertEquals(PsiType.INT, shift.getType()); } public void testUnaryExpressionType() throws Exception { final PsiElementFactory factory = myJavaFacade.getElementFactory(); final PsiExpression plusPrefix = factory.createExpressionFromText("+Integer.valueOf(1)", null); assertEquals(PsiType.INT, plusPrefix.getType()); final PsiExpression plusBytePrefix = factory.createExpressionFromText("+Byte.valueOf(1)", null); assertEquals(PsiType.INT, plusBytePrefix.getType()); final PsiStatement declaration = factory.createStatementFromText("Byte b = 1;", null); final PsiExpression plusPlusPostfix = factory.createExpressionFromText("b++", declaration); assertEquals(PsiType.BYTE.getBoxedType(declaration), plusPlusPostfix.getType()); } }