1/*
2 * Copyright 2015, 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.projectRoots.Sdk;
35import com.intellij.openapi.projectRoots.impl.JavaAwareProjectJdkTableImpl;
36import com.intellij.psi.PsiField;
37import com.intellij.psi.PsiReference;
38import com.intellij.testFramework.ResolveTestCase;
39import org.jf.smalidea.psi.impl.SmaliFieldReference;
40import org.junit.Assert;
41
42public class FieldReferenceTest extends ResolveTestCase {
43    /**
44     * Test a reference to a java field from a smali class
45     */
46    public void testJavaReferenceFromSmali() throws Exception {
47        String text =
48                ".class public Lmy/pkg/blah; .super Ljava/lang/Object;\n" +
49                        ".method public blah()V\n" +
50                        "    .locals 1\n" +
51                        "    sget-object v0, Ljava/lang/System;->o<ref>ut:Ljava/io/PrintStream;\n" +
52                        "    return-void\n" +
53                        ".end method";
54
55        SmaliFieldReference fieldReference = (SmaliFieldReference)configureByFileText(text, "blah.smali");
56
57        Assert.assertNotNull(fieldReference);
58        Assert.assertEquals("out", fieldReference.getName());
59        Assert.assertNotNull(fieldReference.getFieldType());
60        Assert.assertEquals("java.io.PrintStream", fieldReference.getFieldType().getType().getCanonicalText());
61
62        PsiField resolvedField = fieldReference.resolve();
63        Assert.assertNotNull(resolvedField);
64        Assert.assertEquals("out", resolvedField.getName());
65        Assert.assertNotNull(resolvedField.getContainingClass());
66        Assert.assertEquals("java.lang.System", resolvedField.getContainingClass().getQualifiedName());
67        Assert.assertEquals("java.io.PrintStream", resolvedField.getType().getCanonicalText());
68    }
69
70    /**
71     * Test a reference to a smali field from a smali class
72     */
73    public void testSmaliReferenceFromSmali() throws Exception {
74        createFile("blarg.smali", ".class public Lblarg; .super Ljava/lang/Object;" +
75                ".field public static blort:I");
76
77        String text =
78                ".class public Lmy/pkg/blah; .super Ljava/lang/Object;\n" +
79                        ".method public blah()V\n" +
80                        "    .locals 1\n" +
81                        "    sget v0, Lblarg;->bl<ref>ort:I\n" +
82                        "    return-void\n" +
83                        ".end method";
84
85        SmaliFieldReference fieldReference = (SmaliFieldReference)configureByFileText(text, "blah.smali");
86
87        Assert.assertNotNull(fieldReference);
88        Assert.assertEquals("blort", fieldReference.getName());
89        Assert.assertNotNull(fieldReference.getFieldType());
90        Assert.assertEquals("int", fieldReference.getFieldType().getType().getCanonicalText());
91
92        PsiField resolvedField = fieldReference.resolve();
93        Assert.assertNotNull(resolvedField);
94        Assert.assertEquals("blort", resolvedField.getName());
95        Assert.assertNotNull(resolvedField.getContainingClass());
96        Assert.assertEquals("blarg", resolvedField.getContainingClass().getQualifiedName());
97        Assert.assertEquals("int", resolvedField.getType().getCanonicalText());
98    }
99
100    /**
101     * Test a reference to a smali field from a java class
102     */
103    public void testSmaliReferenceFromJava() throws Exception {
104        createFile("blarg.smali", ".class public Lblarg; .super Ljava/lang/Object;" +
105                ".field public static blort:I");
106
107        String text = "public class blah { public static void something() {" +
108                        "blarg.bl<ref>ort = 10;" +
109                        "}}";
110
111        PsiReference fieldReference = configureByFileText(text, "blah.java");
112
113        Assert.assertNotNull(fieldReference);
114
115        PsiField resolvedField = (PsiField)fieldReference.resolve();
116        Assert.assertNotNull(resolvedField);
117        Assert.assertEquals("blort", resolvedField.getName());
118        Assert.assertNotNull(resolvedField.getContainingClass());
119        Assert.assertEquals("blarg", resolvedField.getContainingClass().getQualifiedName());
120        Assert.assertEquals("int", resolvedField.getType().getCanonicalText());
121    }
122
123    @Override
124    protected Sdk getTestProjectJdk() {
125        return JavaAwareProjectJdkTableImpl.getInstanceEx().getInternalJdk();
126    }
127}
128