android/testSrc/org/jetbrains/android/databinding/AndroidDataBindingTest.java

/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * 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.android.databinding;

import com.android.SdkConstants;
import com.intellij.psi.*;
import org.jetbrains.android.AndroidTestCase;

public class AndroidDataBindingTest extends AndroidTestCase {
  private static final String DUMMY_CLASS_QNAME = "p1.p2.DummyClass";

  @Override
  public void setUp() throws Exception {
    super.setUp();
    myFixture.copyFileToProject(SdkConstants.FN_ANDROID_MANIFEST_XML, SdkConstants.FN_ANDROID_MANIFEST_XML);
  }

  private void copyLayout(String... names) {
    for (String name : names) {
      myFixture.copyFileToProject("databinding/res/layout/" + name + ".xml", "res/layout/" + name + ".xml");
    }
  }

  private void copyClass(String... qNames) {
    for (String name : qNames) {
      String asPath = name.replace(".", "/");
      myFixture.copyFileToProject("databinding/java/" + asPath + ".java", "src/" + asPath + ".java");
    }
  }

  public void testResolveSimpleVariable() {
    copyLayout("basic_binding");
    copyClass(DUMMY_CLASS_QNAME);
    myFacet.setDataBindingEnabled(true);
    myFacet.getModuleResources(true);

    PsiClass aClass = myFixture.findClass("p1.p2.databinding.BasicBindingBinding");
    assertNotNull(aClass);
    assertNotNull(aClass.findFieldByName("view1", false));
    assertMethod(aClass, "setDummy", "void", DUMMY_CLASS_QNAME);
    assertMethod(aClass, "getDummy", DUMMY_CLASS_QNAME);
  }

  public void testResolveImport() {
    copyLayout("import_variable");
    copyClass(DUMMY_CLASS_QNAME);
    myFacet.setDataBindingEnabled(true);
    myFacet.getModuleResources(true);

    PsiClass aClass = myFixture.findClass("p1.p2.databinding.ImportVariableBinding");
    assertNotNull(aClass);

    assertMethod(aClass, "setDummy", "void", DUMMY_CLASS_QNAME);
    assertMethod(aClass, "getDummy", DUMMY_CLASS_QNAME);

    assertMethod(aClass, "setDummyList", "void", "java.util.List<" + DUMMY_CLASS_QNAME + ">");
    assertMethod(aClass, "getDummyList", "java.util.List<" + DUMMY_CLASS_QNAME + ">");

    assertMethod(aClass, "setDummyMap", "void", "java.util.Map<java.lang.String," + DUMMY_CLASS_QNAME + ">");
    assertMethod(aClass, "getDummyMap", "java.util.Map<java.lang.String," + DUMMY_CLASS_QNAME + ">");

    assertMethod(aClass, "setDummyArray", "void", DUMMY_CLASS_QNAME + "[]");
    assertMethod(aClass, "getDummyArray", DUMMY_CLASS_QNAME + "[]");

    assertMethod(aClass, "setDummyMultiDimArray", "void", DUMMY_CLASS_QNAME + "[][][]");
    assertMethod(aClass, "getDummyMultiDimArray", DUMMY_CLASS_QNAME + "[][][]");
  }

  public void testResolveImportAlias() {
    copyLayout("import_via_alias");
    copyClass(DUMMY_CLASS_QNAME);
    myFacet.setDataBindingEnabled(true);
    myFacet.getModuleResources(true);

    PsiClass aClass = myFixture.findClass("p1.p2.databinding.ImportViaAliasBinding");
    assertNotNull(aClass);

    assertMethod(aClass, "setDummy", "void", DUMMY_CLASS_QNAME);
    assertMethod(aClass, "getDummy", DUMMY_CLASS_QNAME);

    assertMethod(aClass, "setDummyList", "void", "java.util.List<" + DUMMY_CLASS_QNAME + ">");
    assertMethod(aClass, "getDummyList", "java.util.List<" + DUMMY_CLASS_QNAME + ">");

    assertMethod(aClass, "setDummyMap", "void", "java.util.Map<java.lang.String," + DUMMY_CLASS_QNAME + ">");
    assertMethod(aClass, "getDummyMap", "java.util.Map<java.lang.String," + DUMMY_CLASS_QNAME + ">");

    assertMethod(aClass, "setDummyMap2", "void", "java.util.Map<" + DUMMY_CLASS_QNAME + ",java.lang.String>");
    assertMethod(aClass, "getDummyMap2", "java.util.Map<" + DUMMY_CLASS_QNAME + ",java.lang.String>");

    assertMethod(aClass, "setDummyArray", "void", DUMMY_CLASS_QNAME + "[]");
    assertMethod(aClass, "getDummyArray", DUMMY_CLASS_QNAME + "[]");

    assertMethod(aClass, "setDummyMultiDimArray", "void", DUMMY_CLASS_QNAME + "[][][]");
    assertMethod(aClass, "getDummyMultiDimArray", DUMMY_CLASS_QNAME + "[][][]");
  }

  private static void assertMethod(PsiClass aClass, String name, String returnType, String... parameters) {
    PsiMethod[] methods = aClass.findMethodsByName(name, true);
    assertEquals(1, methods.length);
    PsiMethod method = methods[0];
    assertNotNull(method.getReturnType());
    assertEquals(returnType, method.getReturnType().getCanonicalText());
    PsiParameterList parameterList = method.getParameterList();
    assertEquals(parameters.length, parameterList.getParametersCount());
    for (String parameterQName : parameters) {
      assertEquals(parameterQName, parameterList.getParameters()[0].getType().getCanonicalText());
    }
  }
}