1 files changed, 162 insertions, 0 deletions
diff --git a/app/src/androidTest/java/org/connectbot/StartupTest.java b/app/src/androidTest/java/org/connectbot/StartupTest.java
new file mode 100644
index 0000000..0eb9c11
--- /dev/null
+++ b/app/src/androidTest/java/org/connectbot/StartupTest.java
@@ -0,0 +1,162 @@
+package org.connectbot;
+
+import org.connectbot.bean.HostBean;
+import org.connectbot.util.HostDatabase;
+import org.hamcrest.Description;
+import org.hamcrest.Matcher;
+import org.hamcrest.TypeSafeMatcher;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import android.content.Intent;
+import android.support.annotation.NonNull;
+import android.support.test.InstrumentationRegistry;
+import android.support.test.espresso.intent.Intents;
+import android.support.test.espresso.matcher.BoundedMatcher;
+import android.support.test.rule.ActivityTestRule;
+import android.support.test.runner.AndroidJUnit4;
+import android.view.View;
+import android.widget.ImageView;
+
+import static android.support.test.espresso.Espresso.onData;
+import static android.support.test.espresso.Espresso.onView;
+import static android.support.test.espresso.action.ViewActions.click;
+import static android.support.test.espresso.action.ViewActions.closeSoftKeyboard;
+import static android.support.test.espresso.action.ViewActions.longClick;
+import static android.support.test.espresso.action.ViewActions.pressBack;
+import static android.support.test.espresso.action.ViewActions.pressImeActionButton;
+import static android.support.test.espresso.action.ViewActions.pressMenuKey;
+import static android.support.test.espresso.action.ViewActions.typeText;
+import static android.support.test.espresso.assertion.ViewAssertions.matches;
+import static android.support.test.espresso.intent.Intents.intended;
+import static android.support.test.espresso.intent.matcher.IntentMatchers.hasComponent;
+import static android.support.test.espresso.matcher.ViewMatchers.hasDescendant;
+import static android.support.test.espresso.matcher.ViewMatchers.isDisplayed;
+import static android.support.test.espresso.matcher.ViewMatchers.withId;
+import static android.support.test.espresso.matcher.ViewMatchers.withText;
+import static org.hamcrest.CoreMatchers.allOf;
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.CoreMatchers.is;
+
+@RunWith(AndroidJUnit4.class)
+public class StartupTest {
+	@Rule
+	public final ActivityTestRule<HostListActivity> mActivityRule = new ActivityTestRule<>(
+			HostListActivity.class, false, false);
+
+	@Before
+	public void makeDatabasePristine() {
+		HostDatabase db = new HostDatabase(InstrumentationRegistry.getTargetContext());
+		db.resetDatabase();
+
+		mActivityRule.launchActivity(new Intent());
+	}
+
+	@Test
+	public void localConnectionDisconnectFromHostList() {
+		startNewLocalConnection();
+
+		onView(withId(R.id.console_flip)).perform(closeSoftKeyboard(), pressBack());
+
+		// Make sure we're still connected.
+		onData(withHostNickname("Local")).inAdapterView(withId(android.R.id.list))
+				.check(matches(hostConnected()))
+				.perform(longClick());
+
+		// Click on the disconnect context menu item.
+		onView(withText(R.string.list_host_disconnect)).check(matches(isDisplayed())).perform(click());
+
+		// Now make sure we're disconnected.
+		onData(withHostNickname("Local")).inAdapterView(withId(android.R.id.list))
+				.check(matches(hostDisconnected()));
+	}
+
+	@Test
+	public void localConnectionDisconnectConsoleActivity() {
+		startNewLocalConnection();
+
+		onView(withId(R.id.console_flip)).perform(pressMenuKey());
+
+		// Click on the disconnect context menu item.
+		onView(withText(R.string.list_host_disconnect)).check(matches(isDisplayed())).perform(click());
+
+		// Now make sure we're disconnected.
+		onData(withHostNickname("Local")).inAdapterView(withId(android.R.id.list))
+				.check(matches(hostDisconnected()));
+	}
+
+	private void startNewLocalConnection() {
+		onView(withId(R.id.transport_selection)).perform(click());
+		onData(allOf(is(instanceOf(String.class)), is("local"))).perform(click());
+		onView(withId(R.id.front_quickconnect)).perform(typeText("Local"));
+
+		Intents.init();
+		try {
+			onView(withId(R.id.front_quickconnect)).perform(pressImeActionButton());
+			intended(hasComponent(ConsoleActivity.class.getName()));
+		} finally {
+			Intents.release();
+		}
+
+		onView(withId(R.id.console_flip)).check(matches(
+				hasDescendant(allOf(isDisplayed(), withId(R.id.terminal_view)))));
+	}
+
+	/**
+	 * Matches the nickname of a {@link HostBean}.
+	 */
+	public static Matcher<Object> withHostNickname(final String content) {
+		return new BoundedMatcher<Object, HostBean>(HostBean.class) {
+			@Override
+			public boolean matchesSafely(HostBean host) {
+				return host.getNickname().matches(content);
+			}
+
+			@Override
+			public void describeTo(Description description) {
+				description.appendText("with host nickname '" + content + "'");
+			}
+		};
+	}
+
+	/**
+	 * Matches the drawable state on an ImageView that is set with setImageState.
+	 */
+	public static Matcher<View> withDrawableState(final int expectedState) {
+		return new TypeSafeMatcher<View>() {
+			@Override
+			public boolean matchesSafely(View view) {
+				if (!(view instanceof ImageView)) {
+					return false;
+				}
+
+				int[] states = view.getDrawableState();
+				for (int state : states) {
+					if (state == expectedState) {
+						return true;
+					}
+				}
+				return false;
+			}
+
+			@Override
+			public void describeTo(Description description) {
+				description.appendText("with drawable state '" + expectedState + "'");
+			}
+		};
+	}
+
+	@NonNull
+	private Matcher<View> hostDisconnected() {
+		return hasDescendant(allOf(withId(android.R.id.icon),
+				withDrawableState(android.R.attr.state_expanded)));
+	}
+
+	@NonNull
+	private Matcher<View> hostConnected() {
+		return hasDescendant(allOf(withId(android.R.id.icon),
+				withDrawableState(android.R.attr.state_checked)));
+	}
+}