tests/autofillservice/src/android/autofillservice/cts/LoginActivity.java - platform/cts - Git at Google

 /*
  * Copyright (C) 2017 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 package android.autofillservice.cts;

 import static com.google.common.truth.Truth.assertWithMessage;

 import android.content.Context;
 import android.content.Intent;
 import android.os.Bundle;
 import android.util.Log;
 import android.view.View;
 import android.view.View.OnClickListener;
 import android.view.inputmethod.InputMethodManager;
 import android.widget.Button;
 import android.widget.EditText;
 import android.widget.TextView;

 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;

 /**
  * Activity that has the following fields:
  *
  * <ul>
  *   <li>Username EditText (id: username, no input-type)
  *   <li>Password EditText (id: "username", input-type textPassword)
  *   <li>Clear Button
  *   <li>Save Button
  *   <li>Login Button
  * </ul>
  */
 public class LoginActivity extends AbstractAutoFillActivity {

     private static final String TAG = "LoginActivity";
     private static String WELCOME_TEMPLATE = "Welcome to the new activity, %s!";
     private static final long LOGIN_TIMEOUT_MS = 1000;

     static final String ID_USERNAME_CONTAINER = "username_container";
     static final String AUTHENTICATION_MESSAGE = "Authentication failed. D'OH!";

     private TextView mUsernameLabel;
     private EditText mUsernameEditText;
     private TextView mPasswordLabel;
     private EditText mPasswordEditText;
     private TextView mOutput;
     private Button mLoginButton;
     private Button mSaveButton;
     private Button mCancelButton;
     private Button mClearButton;
     private FillExpectation mExpectation;

     // State used to synchronously get the result of a login attempt.
     private CountDownLatch mLoginLatch;
     private String mLoginMessage;

     /**
      * Gets the expected welcome message for a given username.
      */
     static String getWelcomeMessage(String username) {
         return String.format(WELCOME_TEMPLATE,  username);
     }

     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);

         setContentView(getContentView());

         mLoginButton = (Button) findViewById(R.id.login);
         mSaveButton = (Button) findViewById(R.id.save);
         mClearButton = (Button) findViewById(R.id.clear);
         mCancelButton = (Button) findViewById(R.id.cancel);
         mUsernameLabel = (TextView) findViewById(R.id.username_label);
         mUsernameEditText = (EditText) findViewById(R.id.username);
         mPasswordLabel = (TextView) findViewById(R.id.password_label);
         mPasswordEditText = (EditText) findViewById(R.id.password);
         mOutput = (TextView) findViewById(R.id.output);

         mLoginButton.setOnClickListener((v) -> login());
         mSaveButton.setOnClickListener((v) -> save());
         mClearButton.setOnClickListener((v) -> {
             mUsernameEditText.setText("");
             mPasswordEditText.setText("");
             mOutput.setText("");
             getAutofillManager().cancel();
         });
         mCancelButton.setOnClickListener((OnClickListener) v -> finish());
     }

     protected int getContentView() {
         return R.layout.login_activity;
     }

     /**
      * Emulates a login action.
      */
     private void login() {
         final String username = mUsernameEditText.getText().toString();
         final String password = mPasswordEditText.getText().toString();
         final boolean valid = username.equals(password) || password.contains("pass");

         if (valid) {
             Log.d(TAG, "login ok: " + username);
             final Intent intent = new Intent(this, WelcomeActivity.class);
             final String message = getWelcomeMessage(username);
             intent.putExtra(WelcomeActivity.EXTRA_MESSAGE, message);
             setLoginMessage(message);
             startActivity(intent);
             finish();
         } else {
             Log.d(TAG, "login failed: " + AUTHENTICATION_MESSAGE);
             mOutput.setText(AUTHENTICATION_MESSAGE);
             setLoginMessage(AUTHENTICATION_MESSAGE);
         }
     }

     private void setLoginMessage(String message) {
         Log.d(TAG, "setLoginMessage(): " + message);
         if (mLoginLatch != null) {
             mLoginMessage = message;
             mLoginLatch.countDown();
         }
     }

     /**
      * Explicitly forces the AutofillManager to save the username and password.
      */
     private void save() {
         final InputMethodManager imm = (InputMethodManager) getSystemService(
                 Context.INPUT_METHOD_SERVICE);
         imm.hideSoftInputFromWindow(mUsernameEditText.getWindowToken(), 0);
         getAutofillManager().commit();
     }

     /**
      * Sets the expectation for an autofill request (for all fields), so it can be asserted through
      * {@link #assertAutoFilled()} later.
      */
     void expectAutoFill(String username, String password) {
         mExpectation = new FillExpectation(username, password);
         mUsernameEditText.addTextChangedListener(mExpectation.ccUsernameWatcher);
         mPasswordEditText.addTextChangedListener(mExpectation.ccPasswordWatcher);
     }

     /**
      * Sets the expectation for an autofill request (for username only), so it can be asserted
      * through {@link #assertAutoFilled()} later.
      */
     void expectAutoFill(String username) {
         mExpectation = new FillExpectation(username);
         mUsernameEditText.addTextChangedListener(mExpectation.ccUsernameWatcher);
     }

     /**
      * Asserts the activity was auto-filled with the values passed to
      * {@link #expectAutoFill(String, String)}.
      */
     void assertAutoFilled() throws Exception {
         assertWithMessage("expectAutoFill() not called").that(mExpectation).isNotNull();
         mExpectation.ccUsernameWatcher.assertAutoFilled();
         if (mExpectation.ccPasswordWatcher != null) {
             mExpectation.ccPasswordWatcher.assertAutoFilled();
         }
     }

     void forceAutofillOnUsername() {
         syncRunOnUiThread(() -> getAutofillManager().requestAutofill(mUsernameEditText));
     }

     /**
      * Visits the {@code username_label} in the UiThread.
      */
     void onUsernameLabel(Visitor<TextView> v) {
         syncRunOnUiThread(() -> v.visit(mUsernameLabel));
     }

     /**
      * Visits the {@code username} in the UiThread.
      */
     void onUsername(Visitor<EditText> v) {
         syncRunOnUiThread(() -> v.visit(mUsernameEditText));
     }

     /**
      * Gets the {@code username} view.
      */
     EditText getUsername() {
         return mUsernameEditText;
     }

     /**
      * Visits the {@code password_label} in the UiThread.
      */
     void onPasswordLabel(Visitor<TextView> v) {
         syncRunOnUiThread(() -> v.visit(mPasswordLabel));
     }

     /**
      * Visits the {@code password} in the UiThread.
      */
     void onPassword(Visitor<EditText> v) {
         syncRunOnUiThread(() -> v.visit(mPasswordEditText));
     }

     /**
      * Gets the {@code password} view.
      */
     EditText getPassword() {
         return mPasswordEditText;
     }

     /**
      * Taps the login button in the UI thread.
      */
     String tapLogin() throws Exception {
         mLoginLatch = new CountDownLatch(1);
         syncRunOnUiThread(() -> mLoginButton.performClick());
         boolean called = mLoginLatch.await(LOGIN_TIMEOUT_MS, TimeUnit.MILLISECONDS);
         assertWithMessage("Timeout (%s ms) waiting for login", LOGIN_TIMEOUT_MS)
                 .that(called).isTrue();
         return mLoginMessage;
     }

     /**
      * Taps the save button in the UI thread.
      */
     void tapSave() throws Exception {
         syncRunOnUiThread(() -> mSaveButton.performClick());
     }

     /**
      * Taps the clear button in the UI thread.
      */
     public void tapClear() {
         syncRunOnUiThread(() -> mClearButton.performClick());
     }

     /**
      * Sets the window flags.
      */
     void setFlags(int flags) {
         Log.d(TAG, "setFlags():" + flags);
         syncRunOnUiThread(() -> getWindow().setFlags(flags, flags));
     }

     /**
      * Holder for the expected auto-fill values.
      */
     private final class FillExpectation {
         private final OneTimeTextWatcher ccUsernameWatcher;
         private final OneTimeTextWatcher ccPasswordWatcher;

         private FillExpectation(String username, String password) {
             ccUsernameWatcher = new OneTimeTextWatcher("username", mUsernameEditText, username);
             ccPasswordWatcher = new OneTimeTextWatcher("password", mPasswordEditText, password);
         }

         private FillExpectation(String username) {
             ccUsernameWatcher = new OneTimeTextWatcher("username", mUsernameEditText, username);
             ccPasswordWatcher = null;
         }
     }
 }
	/*
	* Copyright (C) 2017 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	package android.autofillservice.cts;

	import static com.google.common.truth.Truth.assertWithMessage;

	import android.content.Context;
	import android.content.Intent;
	import android.os.Bundle;
	import android.util.Log;
	import android.view.View;
	import android.view.View.OnClickListener;
	import android.view.inputmethod.InputMethodManager;
	import android.widget.Button;
	import android.widget.EditText;
	import android.widget.TextView;

	import java.util.concurrent.CountDownLatch;
	import java.util.concurrent.TimeUnit;

	/**
	* Activity that has the following fields:
	*
	* <ul>
	* <li>Username EditText (id: username, no input-type)
	* <li>Password EditText (id: "username", input-type textPassword)
	* <li>Clear Button
	* <li>Save Button
	* <li>Login Button
	* </ul>
	*/
	public class LoginActivity extends AbstractAutoFillActivity {

	private static final String TAG = "LoginActivity";
	private static String WELCOME_TEMPLATE = "Welcome to the new activity, %s!";
	private static final long LOGIN_TIMEOUT_MS = 1000;

	static final String ID_USERNAME_CONTAINER = "username_container";
	static final String AUTHENTICATION_MESSAGE = "Authentication failed. D'OH!";

	private TextView mUsernameLabel;
	private EditText mUsernameEditText;
	private TextView mPasswordLabel;
	private EditText mPasswordEditText;
	private TextView mOutput;
	private Button mLoginButton;
	private Button mSaveButton;
	private Button mCancelButton;
	private Button mClearButton;
	private FillExpectation mExpectation;

	// State used to synchronously get the result of a login attempt.
	private CountDownLatch mLoginLatch;
	private String mLoginMessage;

	/**
	* Gets the expected welcome message for a given username.
	*/
	static String getWelcomeMessage(String username) {
	return String.format(WELCOME_TEMPLATE, username);
	}

	@Override
	protected void onCreate(Bundle savedInstanceState) {
	super.onCreate(savedInstanceState);

	setContentView(getContentView());

	mLoginButton = (Button) findViewById(R.id.login);
	mSaveButton = (Button) findViewById(R.id.save);
	mClearButton = (Button) findViewById(R.id.clear);
	mCancelButton = (Button) findViewById(R.id.cancel);
	mUsernameLabel = (TextView) findViewById(R.id.username_label);
	mUsernameEditText = (EditText) findViewById(R.id.username);
	mPasswordLabel = (TextView) findViewById(R.id.password_label);
	mPasswordEditText = (EditText) findViewById(R.id.password);
	mOutput = (TextView) findViewById(R.id.output);

	mLoginButton.setOnClickListener((v) -> login());
	mSaveButton.setOnClickListener((v) -> save());
	mClearButton.setOnClickListener((v) -> {
	mUsernameEditText.setText("");
	mPasswordEditText.setText("");
	mOutput.setText("");
	getAutofillManager().cancel();
	});
	mCancelButton.setOnClickListener((OnClickListener) v -> finish());
	}

	protected int getContentView() {
	return R.layout.login_activity;
	}

	/**
	* Emulates a login action.
	*/
	private void login() {
	final String username = mUsernameEditText.getText().toString();
	final String password = mPasswordEditText.getText().toString();
	final boolean valid = username.equals(password) \|\| password.contains("pass");

	if (valid) {
	Log.d(TAG, "login ok: " + username);
	final Intent intent = new Intent(this, WelcomeActivity.class);
	final String message = getWelcomeMessage(username);
	intent.putExtra(WelcomeActivity.EXTRA_MESSAGE, message);
	setLoginMessage(message);
	startActivity(intent);
	finish();
	} else {
	Log.d(TAG, "login failed: " + AUTHENTICATION_MESSAGE);
	mOutput.setText(AUTHENTICATION_MESSAGE);
	setLoginMessage(AUTHENTICATION_MESSAGE);
	}
	}

	private void setLoginMessage(String message) {
	Log.d(TAG, "setLoginMessage(): " + message);
	if (mLoginLatch != null) {
	mLoginMessage = message;
	mLoginLatch.countDown();
	}
	}

	/**
	* Explicitly forces the AutofillManager to save the username and password.
	*/
	private void save() {
	final InputMethodManager imm = (InputMethodManager) getSystemService(
	Context.INPUT_METHOD_SERVICE);
	imm.hideSoftInputFromWindow(mUsernameEditText.getWindowToken(), 0);
	getAutofillManager().commit();
	}

	/**
	* Sets the expectation for an autofill request (for all fields), so it can be asserted through
	* {@link #assertAutoFilled()} later.
	*/
	void expectAutoFill(String username, String password) {
	mExpectation = new FillExpectation(username, password);
	mUsernameEditText.addTextChangedListener(mExpectation.ccUsernameWatcher);
	mPasswordEditText.addTextChangedListener(mExpectation.ccPasswordWatcher);
	}

	/**
	* Sets the expectation for an autofill request (for username only), so it can be asserted
	* through {@link #assertAutoFilled()} later.
	*/
	void expectAutoFill(String username) {
	mExpectation = new FillExpectation(username);
	mUsernameEditText.addTextChangedListener(mExpectation.ccUsernameWatcher);
	}

	/**
	* Asserts the activity was auto-filled with the values passed to
	* {@link #expectAutoFill(String, String)}.
	*/
	void assertAutoFilled() throws Exception {
	assertWithMessage("expectAutoFill() not called").that(mExpectation).isNotNull();
	mExpectation.ccUsernameWatcher.assertAutoFilled();
	if (mExpectation.ccPasswordWatcher != null) {
	mExpectation.ccPasswordWatcher.assertAutoFilled();
	}
	}

	void forceAutofillOnUsername() {
	syncRunOnUiThread(() -> getAutofillManager().requestAutofill(mUsernameEditText));
	}

	/**
	* Visits the {@code username_label} in the UiThread.
	*/
	void onUsernameLabel(Visitor<TextView> v) {
	syncRunOnUiThread(() -> v.visit(mUsernameLabel));
	}

	/**
	* Visits the {@code username} in the UiThread.
	*/
	void onUsername(Visitor<EditText> v) {
	syncRunOnUiThread(() -> v.visit(mUsernameEditText));
	}

	/**
	* Gets the {@code username} view.
	*/
	EditText getUsername() {
	return mUsernameEditText;
	}

	/**
	* Visits the {@code password_label} in the UiThread.
	*/
	void onPasswordLabel(Visitor<TextView> v) {
	syncRunOnUiThread(() -> v.visit(mPasswordLabel));
	}

	/**
	* Visits the {@code password} in the UiThread.
	*/
	void onPassword(Visitor<EditText> v) {
	syncRunOnUiThread(() -> v.visit(mPasswordEditText));
	}

	/**
	* Gets the {@code password} view.
	*/
	EditText getPassword() {
	return mPasswordEditText;
	}

	/**
	* Taps the login button in the UI thread.
	*/
	String tapLogin() throws Exception {
	mLoginLatch = new CountDownLatch(1);
	syncRunOnUiThread(() -> mLoginButton.performClick());
	boolean called = mLoginLatch.await(LOGIN_TIMEOUT_MS, TimeUnit.MILLISECONDS);
	assertWithMessage("Timeout (%s ms) waiting for login", LOGIN_TIMEOUT_MS)
	.that(called).isTrue();
	return mLoginMessage;
	}

	/**
	* Taps the save button in the UI thread.
	*/
	void tapSave() throws Exception {
	syncRunOnUiThread(() -> mSaveButton.performClick());
	}

	/**
	* Taps the clear button in the UI thread.
	*/
	public void tapClear() {
	syncRunOnUiThread(() -> mClearButton.performClick());
	}

	/**
	* Sets the window flags.
	*/
	void setFlags(int flags) {
	Log.d(TAG, "setFlags():" + flags);
	syncRunOnUiThread(() -> getWindow().setFlags(flags, flags));
	}

	/**
	* Holder for the expected auto-fill values.
	*/
	private final class FillExpectation {
	private final OneTimeTextWatcher ccUsernameWatcher;
	private final OneTimeTextWatcher ccPasswordWatcher;

	private FillExpectation(String username, String password) {
	ccUsernameWatcher = new OneTimeTextWatcher("username", mUsernameEditText, username);
	ccPasswordWatcher = new OneTimeTextWatcher("password", mPasswordEditText, password);
	}

	private FillExpectation(String username) {
	ccUsernameWatcher = new OneTimeTextWatcher("username", mUsernameEditText, username);
	ccPasswordWatcher = null;
	}
	}
	}