org.assertj.android.recyclerview.v7.api.widget.RecyclerViewAssert Maven / Gradle / Ivy

Go to download
package org.assertj.android.recyclerview.v7.api.widget;

import android.support.v7.widget.RecyclerView;
import org.assertj.android.api.view.AbstractViewGroupAssert;

import static android.support.v7.widget.RecyclerView.Adapter;
import static android.support.v7.widget.RecyclerView.ItemAnimator;
import static android.support.v7.widget.RecyclerView.LayoutManager;
import static android.support.v7.widget.RecyclerView.RecycledViewPool;
import static android.support.v7.widget.RecyclerView.SCROLL_STATE_DRAGGING;
import static android.support.v7.widget.RecyclerView.SCROLL_STATE_IDLE;
import static android.support.v7.widget.RecyclerView.SCROLL_STATE_SETTLING;
import static android.support.v7.widget.RecyclerView.ViewHolder;
import static org.assertj.android.internal.IntegerUtils.buildNamedValueString;
import static org.assertj.core.api.Assertions.assertThat;

/** Assertions for {@link RecyclerView} instances. */
public class RecyclerViewAssert extends AbstractViewGroupAssert {
  public RecyclerViewAssert(RecyclerView actual) {
    super(actual, RecyclerViewAssert.class);
  }

  public  RecyclerViewAssert hasAdapter(Adapter adapter) {
    isNotNull();
    Adapter actualAdapter = actual.getAdapter();
    assertThat(actualAdapter) //
        .overridingErrorMessage("Expected adapter <%s> but was <%s>.", adapter, actualAdapter) //
        .isEqualTo(adapter);
    return this;
  }

  public RecyclerViewAssert hasItemAnimator(ItemAnimator itemAnimator) {
    isNotNull();
    ItemAnimator actualItemAnimator = actual.getItemAnimator();
    assertThat(actualItemAnimator) //
        .overridingErrorMessage("Expected item animator <%s> but was <%s>.", itemAnimator,
            actualItemAnimator) //
        .isEqualTo(itemAnimator);
    return this;
  }

  public RecyclerViewAssert hasLayoutManager(LayoutManager layoutManager) {
    isNotNull();
    LayoutManager actualLayoutManager = actual.getLayoutManager();
    assertThat(actualLayoutManager) //
        .overridingErrorMessage("Expected layout manager <%s> but was <%s>.", layoutManager,
            actualLayoutManager) //
        .isEqualTo(layoutManager);
    return this;
  }

  public RecyclerViewAssert hasRecycledViewPool(RecycledViewPool recycledViewPool) {
    isNotNull();
    RecycledViewPool actualRecycledViewPool = actual.getRecycledViewPool();
    assertThat(actualRecycledViewPool) //
        .overridingErrorMessage("Expected recycled view pool <%s> but was <%s>", recycledViewPool,
            actualRecycledViewPool) //
        .isEqualTo(recycledViewPool);
    return this;
  }

  public RecyclerViewAssert hasScrollState(@RecyclerViewScrollState int scrollState) {
    isNotNull();
    int actualScrollState = actual.getScrollState();
    //noinspection ResourceType
    assertThat(actualScrollState) //
        .overridingErrorMessage("Expected scroll state <%s> but was <%s>.",
            scrollStateToString(scrollState), scrollStateToString(actualScrollState)) //
        .isEqualTo(scrollState);
    return this;
  }

  public RecyclerViewAssert hasFixedSize() {
    isNotNull();
    assertThat(actual.hasFixedSize()) //
        .overridingErrorMessage("Expected to have fixed size but did not.") //
        .isTrue();
    return this;
  }

  public RecyclerViewAssert doesNotHaveFixedSize() {
    isNotNull();
    assertThat(actual.hasFixedSize()) //
        .overridingErrorMessage("Expected to not have fixed size but did.") //
        .isFalse();
    return this;
  }

  public static String scrollStateToString(@RecyclerViewScrollState int scrollState) {
    return buildNamedValueString(scrollState)
        .value(SCROLL_STATE_DRAGGING, "dragging")
        .value(SCROLL_STATE_IDLE, "idle")
        .value(SCROLL_STATE_SETTLING, "settling")
        .get();
  }
}