src/test/java/org/robolectric/shadows/TabHostTest.java

package org.robolectric.shadows;

import android.app.Activity;
import android.app.TabActivity;
import android.view.View;
import android.widget.TabHost;
import android.widget.TabHost.TabContentFactory;
import android.widget.TabWidget;
import android.widget.TextView;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.R;
import org.robolectric.Robolectric;
import org.robolectric.TestRunners;

import static org.fest.assertions.api.Assertions.assertThat;
import static org.junit.Assert.assertNull;
import static org.robolectric.Robolectric.shadowOf;

@RunWith(TestRunners.WithDefaults.class)
public class TabHostTest {

    @Test
    public void newTabSpec_shouldMakeATabSpec() throws Exception {
        TabHost tabHost = new TabHost(Robolectric.application);
        TabHost.TabSpec tabSpec = tabHost.newTabSpec("Foo");
        assertThat(tabSpec.getTag()).isEqualTo("Foo");
    }

    @Test
    public void shouldAddTabsToLayoutWhenAddedToHost() {
        TabHost tabHost = new TabHost(Robolectric.application);

        View fooView = new View(Robolectric.application);
        TabHost.TabSpec foo = tabHost.newTabSpec("Foo").setIndicator(fooView);

        View barView = new View(Robolectric.application);
        TabHost.TabSpec bar = tabHost.newTabSpec("Bar").setIndicator(barView);

        tabHost.addTab(foo);
        tabHost.addTab(bar);

        assertThat(tabHost.getChildAt(0)).isSameAs(fooView);
        assertThat(tabHost.getChildAt(1)).isSameAs(barView);
    }

    @Test
    public void shouldReturnTabSpecsByTag() throws Exception {
        TabHost tabHost = new TabHost(Robolectric.application);
        TabHost.TabSpec foo = tabHost.newTabSpec("Foo");
        TabHost.TabSpec bar = tabHost.newTabSpec("Bar");
        TabHost.TabSpec baz = tabHost.newTabSpec("Baz");

        tabHost.addTab(foo);
        tabHost.addTab(bar);
        tabHost.addTab(baz);

        assertThat(shadowOf(tabHost).getSpecByTag("Bar")).isSameAs(bar);
        assertThat(shadowOf(tabHost).getSpecByTag("Baz")).isSameAs(baz);
        assertNull(shadowOf(tabHost).getSpecByTag("Whammie"));
    }

    @Test
    public void shouldFireTheTabChangeListenerWhenCurrentTabIsSet() throws Exception {
        TabHost tabHost = new TabHost(Robolectric.application);

        TabHost.TabSpec foo = tabHost.newTabSpec("Foo");
        TabHost.TabSpec bar = tabHost.newTabSpec("Bar");
        TabHost.TabSpec baz = tabHost.newTabSpec("Baz");

        tabHost.addTab(foo);
        tabHost.addTab(bar);
        tabHost.addTab(baz);

        TestOnTabChangeListener listener = new TestOnTabChangeListener();
        tabHost.setOnTabChangedListener(listener);

        tabHost.setCurrentTab(2);

        assertThat(listener.tag).isEqualTo("Baz");
    }

    @Test
    public void shouldFireTheTabChangeListenerWhenTheCurrentTabIsSetByTag() throws Exception {
        TabHost tabHost = new TabHost(Robolectric.application);

        TabHost.TabSpec foo = tabHost.newTabSpec("Foo");
        TabHost.TabSpec bar = tabHost.newTabSpec("Bar");
        TabHost.TabSpec baz = tabHost.newTabSpec("Baz");

        tabHost.addTab(foo);
        tabHost.addTab(bar);
        tabHost.addTab(baz);

        TestOnTabChangeListener listener = new TestOnTabChangeListener();
        tabHost.setOnTabChangedListener(listener);

        tabHost.setCurrentTabByTag("Bar");

        assertThat(listener.tag).isEqualTo("Bar");
    }

    @Test
    public void shouldRetrieveTheCurrentViewFromTabContentFactory() {
    	TabHost tabHost = new TabHost(Robolectric.application);

        TabHost.TabSpec foo = tabHost.newTabSpec("Foo").setContent(
		new TabContentFactory() {
			public View createTabContent(String tag) {
				TextView tv = new TextView(Robolectric.application);
				tv.setText("The Text of " + tag);
				return tv;
			}
		});

        tabHost.addTab(foo);
        tabHost.setCurrentTabByTag("Foo");
        TextView textView = (TextView) tabHost.getCurrentView();

        assertThat(textView.getText().toString()).isEqualTo("The Text of Foo");
    }
    @Test
    public void shouldRetrieveTheCurrentViewFromViewId() {
    	Activity a = new Activity();
    	a.setContentView(org.robolectric.R.layout.main);
    	TabHost tabHost = new TabHost(a);
    	TabHost.TabSpec foo = tabHost.newTabSpec("Foo")
    	.setContent(org.robolectric.R.id.title);

    	 tabHost.addTab(foo);
         tabHost.setCurrentTabByTag("Foo");
         TextView textView = (TextView) tabHost.getCurrentView();

        assertThat(textView.getText().toString()).isEqualTo("Main Layout");
    }

    private static class TestOnTabChangeListener implements TabHost.OnTabChangeListener {
        private String tag;

        @Override
        public void onTabChanged(String tag) {
            this.tag = tag;
        }
    }

    @Test
    public void canGetCurrentTabTag() throws Exception {
        TabHost tabHost = new TabHost(Robolectric.application);

        TabHost.TabSpec foo = tabHost.newTabSpec("Foo");
        TabHost.TabSpec bar = tabHost.newTabSpec("Bar");
        TabHost.TabSpec baz = tabHost.newTabSpec("Baz");

        tabHost.addTab(foo);
        tabHost.addTab(bar);
        tabHost.addTab(baz);

        tabHost.setCurrentTabByTag("Bar");

        assertThat(tabHost.getCurrentTabTag()).isEqualTo("Bar");
    }

    @Test
    public void canGetCurrentTab() throws Exception {
        TabHost tabHost = new TabHost(Robolectric.application);

        TabHost.TabSpec foo = tabHost.newTabSpec("Foo");
        TabHost.TabSpec bar = tabHost.newTabSpec("Bar");
        TabHost.TabSpec baz = tabHost.newTabSpec("Baz");

        tabHost.addTab(foo);
        tabHost.addTab(bar);
        tabHost.addTab(baz);
        assertThat(shadowOf(tabHost).getCurrentTabSpec()).isEqualTo(foo);
        assertThat(tabHost.getCurrentTab()).isEqualTo(0);
        tabHost.setCurrentTabByTag("Bar");
        assertThat(tabHost.getCurrentTab()).isEqualTo(1);
        assertThat(shadowOf(tabHost).getCurrentTabSpec()).isEqualTo(bar);
        tabHost.setCurrentTabByTag("Foo");
        assertThat(tabHost.getCurrentTab()).isEqualTo(0);
        assertThat(shadowOf(tabHost).getCurrentTabSpec()).isEqualTo(foo);
        tabHost.setCurrentTabByTag("Baz");
        assertThat(tabHost.getCurrentTab()).isEqualTo(2);
        assertThat(shadowOf(tabHost).getCurrentTabSpec()).isEqualTo(baz);
    }

    @Test
    public void setCurrentTabByTagShouldAcceptNullAsParameter() throws Exception {
        TabHost tabHost = new TabHost(Robolectric.application);
        TabHost.TabSpec foo = tabHost.newTabSpec("Foo");
        tabHost.addTab(foo);

        tabHost.setCurrentTabByTag(null);
        assertThat(tabHost.getCurrentTabTag()).isEqualTo("Foo");
    }

    @Test
    public void shouldGetTabWidget() throws Exception {
        TabActivity activity = new TabActivity();
        activity.setContentView(R.layout.tab_activity);
        TabHost host = new TabHost(activity);
        assertThat(host.getTabWidget()).isInstanceOf(TabWidget.class);
    }
}