RoboMenuItemTest.java

package org.robolectric.fakes;

import android.graphics.drawable.Drawable;
import android.view.View;
import org.junit.Before;
import org.junit.Test;
import android.view.MenuItem;
import org.junit.runner.RunWith;
import org.robolectric.R;
import org.robolectric.RuntimeEnvironment;
import org.robolectric.TestRunners;

import static org.assertj.core.api.Assertions.assertThat;

@RunWith(TestRunners.SelfTest.class)
public class RoboMenuItemTest {
  private MenuItem item;
  private TestOnActionExpandListener listener;

  @Before
  public void setUp() throws Exception {
    item = new RoboMenuItem(RuntimeEnvironment.application);
    listener =  new TestOnActionExpandListener();
    item.setOnActionExpandListener(listener);
  }

  @Test
  public void shouldCheckTheMenuItem() throws Exception {
    assertThat(item.isChecked()).isFalse();
    item.setChecked(true);
    assertThat(item.isChecked()).isTrue();
  }

  @Test
  public void shouldAllowSettingCheckable() throws Exception {
    assertThat(item.isCheckable()).isFalse();
    item.setCheckable(true);
    assertThat(item.isCheckable()).isTrue();
  }

  @Test
  public void shouldAllowSettingVisible() throws Exception {
    assertThat(item.isVisible()).isTrue();
    item.setVisible(false);
    assertThat(item.isVisible()).isFalse();
  }

  @Test
  public void expandActionView_shouldReturnFalseIfActionViewIsNull() throws Exception {
    item.setActionView(null);
    assertThat(item.expandActionView()).isFalse();
  }

  @Test
  public void expandActionView_shouldSetExpandedTrue() throws Exception {
    item.setActionView(new View(RuntimeEnvironment.application));
    assertThat(item.expandActionView()).isTrue();
    assertThat(item.isActionViewExpanded()).isTrue();
  }

  @Test
  public void expandActionView_shouldInvokeListener() throws Exception {
    item.setActionView(new View(RuntimeEnvironment.application));
    item.expandActionView();
    assertThat(listener.expanded).isTrue();
  }

  @Test
  public void collapseActionView_shouldReturnFalseIfActionViewIsNull() throws Exception {
    item.setActionView(null);
    assertThat(item.collapseActionView()).isFalse();
  }

  @Test
  public void collapseActionView_shouldSetExpandedFalse() throws Exception {
    item.setActionView(new View(RuntimeEnvironment.application));
    item.expandActionView();
    assertThat(item.collapseActionView()).isTrue();
    assertThat(item.isActionViewExpanded()).isFalse();
  }

  @Test
  public void collapseActionView_shouldInvokeListener() throws Exception {
    item.setActionView(new View(RuntimeEnvironment.application));
    listener.expanded = true;
    item.collapseActionView();
    assertThat(listener.expanded).isFalse();
  }

  @Test
  public void methodsShouldReturnThis() throws Exception {
    item = item.setEnabled(true);
    assertThat(item).isNotNull();
    item = item.setOnMenuItemClickListener(null);
    assertThat(item).isNotNull();
    item = item.setActionProvider(null);
    assertThat(item).isNotNull();
    item = item.setActionView(0);
    assertThat(item).isNotNull();
    item = item.setActionView(null);
    assertThat(item).isNotNull();
    item = item.setAlphabeticShortcut('a');
    assertThat(item).isNotNull();
    item = item.setCheckable(false);
    assertThat(item).isNotNull();
    item = item.setChecked(true);
    assertThat(item).isNotNull();
    item = item.setIcon(null);
    assertThat(item).isNotNull();
    item = item.setIcon(0);
    assertThat(item).isNotNull();
    item = item.setIntent(null);
    assertThat(item).isNotNull();
    item = item.setNumericShortcut('6');
    assertThat(item).isNotNull();
    item = item.setOnActionExpandListener(null);
    assertThat(item).isNotNull();
    item = item.setShortcut('6', 'z');
    assertThat(item).isNotNull();
    item = item.setShowAsActionFlags(MenuItem.SHOW_AS_ACTION_ALWAYS);
    assertThat(item).isNotNull();
    item = item.setTitleCondensed("condensed");
    assertThat(item).isNotNull();
    item = item.setVisible(true);
    assertThat(item).isNotNull();
    item = item.setTitle("title");
    assertThat(item).isNotNull();
    item = item.setTitle(0);
    assertThat(item).isNotNull();
  }

  @Test
  public void setIcon_shouldNullifyOnZero() throws Exception {
    Drawable expectedDrawable = RuntimeEnvironment.application.getResources().getDrawable(R.drawable.an_image);
    assertThat(expectedDrawable).isNotNull();
    assertThat(item.getIcon()).isNull();
    item.setIcon(R.drawable.an_image);
    assertThat(item.getIcon()).isEqualTo(expectedDrawable);
    item.setIcon(0);
    assertThat(item.getIcon()).isNull();
  }

  @Test
  public void getIcon_shouldReturnDrawableFromSetIconDrawable() throws Exception {
    Drawable testDrawable = RuntimeEnvironment.application.getResources().getDrawable(R.drawable.an_image);
    assertThat(testDrawable).isNotNull();
    assertThat(item.getIcon()).isNull();
    item.setIcon(testDrawable);
    assertThat(item.getIcon()).isSameAs(testDrawable);
  }

  @Test
  public void getIcon_shouldReturnDrawableFromSetIconResourceId() throws Exception {
    assertThat(item.getIcon()).isNull();
    item.setIcon(R.drawable.an_other_image);
    Drawable expectedDrawable = RuntimeEnvironment.application.getResources().getDrawable(R.drawable.an_other_image);
    assertThat(item.getIcon()).isEqualTo(expectedDrawable);
  }

  @Test
  public void setOnActionExpandListener_shouldReturnMenuItem() throws Exception {
    assertThat(item.setOnActionExpandListener(listener)).isSameAs(item);
  }

  class TestOnActionExpandListener implements MenuItem.OnActionExpandListener {
    private boolean expanded = false;

    @Override
    public boolean onMenuItemActionExpand(MenuItem menuItem) {
      expanded = true;
      return true;
    }

    @Override
    public boolean onMenuItemActionCollapse(MenuItem menuItem) {
      expanded = false;
      return true;
    }
  }
}