Refactor LayoutSpecModelFactoryTest

Summary: Like [MountSpecModelFactoryTest](diffusion/FBS/browse/master/fbandroid/libraries/components/litho-it/src/test/java/com/facebook/litho/specmodels/processor/MountSpecModelFactoryTest.java), create SpecModel from a real LayoutSpec component instead of using mock. It would be easier to test other delegate methods in later diffs.

Reviewed By: marco-cova

Differential Revision: D13623075

fbshipit-source-id: dbf802781e8982ea17ef498ed9bc137d86706003

Author: Jing-Wei Wu

litho-it/src/test/java/com/facebook/litho/specmodels/processor/LayoutSpecModelFactoryTest.java

@@ -18,86 +18,81 @@
 
 import static org.assertj.core.api.Java6Assertions.assertThat;
 import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.when;
 
+import com.facebook.litho.Component;
+import com.facebook.litho.ComponentContext;
 import com.facebook.litho.annotations.LayoutSpec;
+import com.facebook.litho.annotations.OnCreateInitialState;
+import com.facebook.litho.annotations.OnCreateLayout;
+import com.facebook.litho.annotations.Prop;
 import com.facebook.litho.specmodels.internal.RunMode;
 import com.facebook.litho.specmodels.model.DependencyInjectionHelper;
 import com.facebook.litho.specmodels.model.LayoutSpecModel;
+import com.google.testing.compile.CompilationRule;
 import javax.annotation.processing.Messager;
 import javax.lang.model.element.TypeElement;
 import javax.lang.model.util.Elements;
 import javax.lang.model.util.Types;
 import org.junit.Before;
+import org.junit.Rule;
 import org.junit.Test;
-import org.mockito.Mock;
-import org.mockito.MockitoAnnotations;
 
 /** Tests {@link LayoutSpecModelFactory} */
 public class LayoutSpecModelFactoryTest {
-  private static final String TEST_QUALIFIED_SPEC_NAME = "com.facebook.litho.TestSpec";
-  private static final String TEST_QUALIFIED_COMPONENT_NAME = "com.facebook.litho.Test";
+  private static final String TEST_QUALIFIED_SPEC_NAME =
+      "com.facebook.litho.specmodels.processor.LayoutSpecModelFactoryTest.TestLayoutSpec";
+  private static final String TEST_QUALIFIED_COMPONENT_NAME =
+      "com.facebook.litho.specmodels.processor.LayoutSpecModelFactoryTest.TestLayoutComponentName";
 
-  @Mock private Messager mMessager;
+  @Rule public CompilationRule mCompilationRule = new CompilationRule();
 
-  Elements mElements = mock(Elements.class);
-  Types mTypes = mock(Types.class);
-  TypeElement mTypeElement = mock(TypeElement.class);
-  DependencyInjectionHelper mDependencyInjectionHelper = mock(DependencyInjectionHelper.class);
-  LayoutSpecModelFactory mFactory = new LayoutSpecModelFactory();
-  LayoutSpec mLayoutSpec = mock(LayoutSpec.class);
+  private final LayoutSpecModelFactory mFactory = new LayoutSpecModelFactory();
+  private final DependencyInjectionHelper mDependencyInjectionHelper =
+      mock(DependencyInjectionHelper.class);
+
+  private LayoutSpecModel mLayoutSpecModel;
 
   @Before
   public void setUp() {
-    MockitoAnnotations.initMocks(this);
-    when(mElements.getDocComment(mTypeElement)).thenReturn("");
-    when(mTypeElement.getQualifiedName()).thenReturn(new MockName(TEST_QUALIFIED_SPEC_NAME));
-    when(mTypeElement.getAnnotation(LayoutSpec.class)).thenReturn(mLayoutSpec);
-  }
+    Elements elements = mCompilationRule.getElements();
+    Types types = mCompilationRule.getTypes();
+    TypeElement typeElement =
+        elements.getTypeElement(LayoutSpecModelFactoryTest.TestLayoutSpec.class.getCanonicalName());
 
-  @Test
-  public void testCreate() {
-    LayoutSpecModel layoutSpecModel =
+    mLayoutSpecModel =
         mFactory.create(
-            mElements,
-            mTypes,
-            mTypeElement,
-            mMessager,
+            elements,
+            types,
+            typeElement,
+            mock(Messager.class),
             RunMode.normal(),
             mDependencyInjectionHelper,
             null);
+  }
 
-    assertThat(layoutSpecModel.getSpecName()).isEqualTo("TestSpec");
-    assertThat(layoutSpecModel.getComponentName()).isEqualTo("Test");
-    assertThat(layoutSpecModel.getSpecTypeName().toString()).isEqualTo(TEST_QUALIFIED_SPEC_NAME);
-    assertThat(layoutSpecModel.getComponentTypeName().toString())
+  @Test
+  public void testCreate() {
+    assertThat(mLayoutSpecModel.getSpecName()).isEqualTo("TestLayoutSpec");
+    assertThat(mLayoutSpecModel.getComponentName()).isEqualTo("TestLayoutComponentName");
+    assertThat(mLayoutSpecModel.getSpecTypeName().toString()).isEqualTo(TEST_QUALIFIED_SPEC_NAME);
+    assertThat(mLayoutSpecModel.getComponentTypeName().toString())
         .isEqualTo(TEST_QUALIFIED_COMPONENT_NAME);
-
-    assertThat(layoutSpecModel.getDelegateMethods()).isEmpty();
-
-    assertThat(layoutSpecModel.getProps()).isEmpty();
-
-    assertThat(layoutSpecModel.hasInjectedDependencies()).isTrue();
-    assertThat(layoutSpecModel.getDependencyInjectionHelper()).isSameAs(mDependencyInjectionHelper);
+    assertThat(mLayoutSpecModel.getDelegateMethods()).hasSize(2);
+    assertThat(mLayoutSpecModel.getProps()).hasSize(2);
+    assertThat(mLayoutSpecModel.hasInjectedDependencies()).isTrue();
+    assertThat(mLayoutSpecModel.getDependencyInjectionHelper())
+        .isSameAs(mDependencyInjectionHelper);
   }
 
-  @Test
-  public void testCreateWithSpecifiedName() {
-    when(mLayoutSpec.value()).thenReturn("TestComponentName");
-    LayoutSpecModel layoutSpecModel =
-        mFactory.create(
-            mElements,
-            mTypes,
-            mTypeElement,
-            mMessager,
-            RunMode.normal(),
-            mDependencyInjectionHelper,
-            null);
+  @LayoutSpec(value = "TestLayoutComponentName")
+  static class TestLayoutSpec {
+
+    @OnCreateInitialState
+    static void createInitialState(@Prop int prop1) {}
 
-    assertThat(layoutSpecModel.getSpecName()).isEqualTo("TestSpec");
-    assertThat(layoutSpecModel.getComponentName()).isEqualTo("TestComponentName");
-    assertThat(layoutSpecModel.getSpecTypeName().toString()).isEqualTo(TEST_QUALIFIED_SPEC_NAME);
-    assertThat(layoutSpecModel.getComponentTypeName().toString())
-        .isEqualTo("com.facebook.litho.TestComponentName");
+    @OnCreateLayout
+    static Component onCreateLayout(ComponentContext c, @Prop int prop2) {
+      return null;
+    }
   }
 }