MountStateBoundsTest.java

/**
 * Copyright (c) 2017-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

package com.facebook.litho;

import android.graphics.Rect;
import android.view.View;

import com.facebook.litho.testing.ComponentTestHelper;
import com.facebook.litho.testing.TestDrawableComponent;
import com.facebook.litho.testing.TestViewComponent;
import com.facebook.litho.testing.testrunner.ComponentsTestRunner;
import com.facebook.litho.testing.util.InlineLayoutSpec;
import com.facebook.yoga.YogaAlign;
import com.facebook.yoga.YogaJustify;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.RuntimeEnvironment;

import static com.facebook.yoga.YogaEdge.ALL;
import static org.junit.Assert.assertEquals;

@RunWith(ComponentsTestRunner.class)
public class MountStateBoundsTest {

  private ComponentContext mContext;

  @Before
  public void setup() {
    mContext = new ComponentContext(RuntimeEnvironment.application);
  }

  @Test
  public void testMountedDrawableBounds() {
    final LithoView lithoView = ComponentTestHelper.mountComponent(
        mContext,
        new InlineLayoutSpec() {
          @Override
          protected ComponentLayout onCreateLayout(ComponentContext c) {
            return TestDrawableComponent.create(c)
                .withLayout()
                .widthPx(10)
                .heightPx(10)
                .build();
          }
        });

    assertEquals(new Rect(0, 0, 10, 10), lithoView.getDrawables().get(0).getBounds());
  }

  @Test
  public void testMountedViewBounds() {
    final LithoView lithoView = ComponentTestHelper.mountComponent(
        mContext,
        new InlineLayoutSpec() {
          @Override
          protected ComponentLayout onCreateLayout(ComponentContext c) {
            return TestViewComponent.create(c)
                .withLayout()
                .widthPx(10)
                .heightPx(10)
                .build();
          }
        });

    final View mountedView = lithoView.getChildAt(0);
    assertEquals(
        new Rect(0, 0, 10, 10),
        new Rect(
            mountedView.getLeft(),
            mountedView.getTop(),
            mountedView.getRight(),
            mountedView.getBottom()));
  }

  @Test
  public void testInnerComponentHostBounds() {
    final LithoView lithoView = ComponentTestHelper.mountComponent(
        mContext,
        new InlineLayoutSpec() {
          @Override
          protected ComponentLayout onCreateLayout(ComponentContext c) {
            return Column.create(c)
                .child(
                    Column.create(c)
                        .widthPx(20)
                        .heightPx(20)
                        .wrapInView()
                        .child(
                            TestDrawableComponent.create(c)
                                .withLayout()
                                .widthPx(10)
                                .heightPx(10)))
                .build();
          }
        });

    final ComponentHost host = (ComponentHost) lithoView.getChildAt(0);
    assertEquals(new Rect(0, 0, 10, 10), host.getDrawables().get(0).getBounds());
    assertEquals(
        new Rect(0, 0, 20, 20),
        new Rect(
            host.getLeft(),
            host.getTop(),
            host.getRight(),
            host.getBottom()));
  }

  @Test
  public void testDoubleInnerComponentHostBounds() {
    final LithoView lithoView = ComponentTestHelper.mountComponent(
        mContext,
        new InlineLayoutSpec() {
          @Override
          protected ComponentLayout onCreateLayout(ComponentContext c) {
            return Column.create(c)
                .alignItems(YogaAlign.FLEX_END)
                .justifyContent(YogaJustify.FLEX_END)
                .child(
                    Column.create(c)
                        .widthPx(100)
                        .heightPx(100)
                        .paddingPx(ALL, 20)
                        .wrapInView()
                        .child(
                            Column.create(c)
                                .widthPx(60)
                                .heightPx(60)
                                .wrapInView()
                                .child(
                                    TestDrawableComponent.create(c)
                                        .withLayout()
                                        .widthPx(20)
                                        .heightPx(20)
                                        .marginPx(ALL, 20))))
                .build();
          }
        },
        200,
        200);

    final ComponentHost host = (ComponentHost) lithoView.getChildAt(0);
    final ComponentHost nestedHost = (ComponentHost) host.getChildAt(0);

    assertEquals(
        new Rect(100, 100, 200, 200),
        new Rect(
            host.getLeft(),
            host.getTop(),
            host.getRight(),
            host.getBottom()));

    assertEquals(new Rect(20, 20, 40, 40), nestedHost.getDrawables().get(0).getBounds());

    assertEquals(
        new Rect(20, 20, 80, 80),
        new Rect(
            nestedHost.getLeft(),
            nestedHost.getTop(),
            nestedHost.getRight(),
            nestedHost.getBottom()));
  }
}