tests/inputmethod/src/android/view/inputmethod/cts/NavigationBarColorTest.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.view.inputmethod.cts;

 import static android.view.View.SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR;
 import static android.view.WindowManager.LayoutParams.FLAG_ALT_FOCUSABLE_IM;
 import static android.view.WindowManager.LayoutParams.FLAG_DIM_BEHIND;
 import static android.view.WindowManager.LayoutParams.FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS;
 import static android.view.WindowManager.LayoutParams.FLAG_NOT_FOCUSABLE;
 import static android.view.inputmethod.cts.util.LightNavigationBarVerifier.expectLightNavigationBarNotSupported;
 import static android.view.inputmethod.cts.util.LightNavigationBarVerifier.expectLightNavigationBarSupported;
 import static android.view.inputmethod.cts.util.NavigationBarColorVerifier.expectNavigationBarColorNotSupported;
 import static android.view.inputmethod.cts.util.NavigationBarColorVerifier.expectNavigationBarColorSupported;
 import static android.view.inputmethod.cts.util.TestUtils.getOnMainSync;

 import static com.android.cts.mockime.ImeEventStreamTestUtils.expectBindInput;
 import static com.android.cts.mockime.ImeEventStreamTestUtils.expectEvent;
 import static com.android.cts.mockime.ImeEventStreamTestUtils.waitForInputViewLayoutStable;

 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assume.assumeTrue;

 import android.app.Activity;
 import android.app.AlertDialog;
 import android.app.UiAutomation;
 import android.graphics.Bitmap;
 import android.graphics.Color;
 import android.os.Process;
 import android.text.TextUtils;
 import android.view.View;
 import android.view.ViewGroup;
 import android.view.inputmethod.EditorInfo;
 import android.view.inputmethod.cts.util.EndToEndImeTestBase;
 import android.view.inputmethod.cts.util.ImeAwareEditText;
 import android.view.inputmethod.cts.util.NavigationBarInfo;
 import android.view.inputmethod.cts.util.TestActivity;
 import android.widget.LinearLayout;
 import android.widget.TextView;

 import androidx.annotation.ColorInt;
 import androidx.annotation.NonNull;
 import androidx.test.InstrumentationRegistry;
 import androidx.test.filters.MediumTest;
 import androidx.test.runner.AndroidJUnit4;

 import com.android.cts.mockime.ImeEventStream;
 import com.android.cts.mockime.ImeLayoutInfo;
 import com.android.cts.mockime.ImeSettings;
 import com.android.cts.mockime.MockImeSession;

 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.runner.RunWith;

 import java.util.concurrent.TimeUnit;

 @MediumTest
 @RunWith(AndroidJUnit4.class)
 public class NavigationBarColorTest extends EndToEndImeTestBase {
     private static final long TIMEOUT = TimeUnit.SECONDS.toMillis(5);
     private static final long LAYOUT_STABLE_THRESHOLD = TimeUnit.SECONDS.toMillis(3);

     private static final String TEST_MARKER = "android.view.inputmethod.cts.NavigationBarColorTest";

     private static void updateSystemUiVisibility(@NonNull View view, int flags, int mask) {
         final int currentFlags = view.getSystemUiVisibility();
         final int newFlags = (currentFlags & ~mask) | (flags & mask);
         if (currentFlags != newFlags) {
             view.setSystemUiVisibility(newFlags);
         }
     }

     @BeforeClass
     public static void initializeNavigationBarInfo() throws Exception {
         // Make sure that NavigationBarInfo is initialized before
         // EndToEndImeTestBase#showStateInitializeActivity().
         NavigationBarInfo.getInstance();
     }

     // TODO(b/37502066): Merge this back to initializeNavigationBarInfo() once b/37502066 is fixed.
     @Before
     public void checkNavigationBar() throws Exception {
         assumeTrue("This test does not make sense if there is no navigation bar",
                 NavigationBarInfo.getInstance().hasBottomNavigationBar());

         assumeTrue("This test does not make sense if custom navigation bar color is not supported"
                         + " even for typical Activity",
                 NavigationBarInfo.getInstance().supportsNavigationBarColor());
     }

     /**
      * Represents test scenarios regarding how a {@link android.view.Window} that has
      * {@link android.view.WindowManager.LayoutParams#FLAG_DIM_BEHIND} interacts with a different
      * {@link android.view.Window} that has
      * {@link android.view.View#SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR}.
      */
     private enum DimmingTestMode {
         /**
          * No {@link AlertDialog} is shown when testing.
          */
         NO_DIMMING_DIALOG,
         /**
          * An {@link AlertDialog} that has dimming effect is shown above the IME window.
          */
         DIMMING_DIALOG_ABOVE_IME,
         /**
          * An {@link AlertDialog} that has dimming effect is shown behind the IME window.
          */
         DIMMING_DIALOG_BEHIND_IME,
     }

     @NonNull
     public TestActivity launchTestActivity(@ColorInt int navigationBarColor,
             boolean lightNavigationBar, @NonNull DimmingTestMode dimmingTestMode) {
         return TestActivity.startSync(activity -> {
             final View contentView;
             switch (dimmingTestMode) {
                 case NO_DIMMING_DIALOG:
                 case DIMMING_DIALOG_ABOVE_IME: {
                     final LinearLayout layout = new LinearLayout(activity);
                     layout.setOrientation(LinearLayout.VERTICAL);
                     final ImeAwareEditText editText = new ImeAwareEditText(activity);
                     editText.setPrivateImeOptions(TEST_MARKER);
                     editText.setHint("editText");
                     editText.requestFocus();
                     editText.scheduleShowSoftInput();
                     layout.addView(editText);
                     contentView = layout;
                     break;
                 }
                 case DIMMING_DIALOG_BEHIND_IME: {
                     final View view = new View(activity);
                     view.setLayoutParams(new ViewGroup.LayoutParams(
                             ViewGroup.LayoutParams.MATCH_PARENT,
                             ViewGroup.LayoutParams.MATCH_PARENT));
                     contentView = view;
                     break;
                 }
                 default:
                     throw new IllegalStateException("unknown mode=" + dimmingTestMode);
             }
             activity.getWindow().setNavigationBarColor(navigationBarColor);
             updateSystemUiVisibility(contentView,
                     lightNavigationBar ? SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR : 0,
                     SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR);
             return contentView;
         });
     }

     private AutoCloseable showDialogIfNecessary(
             @NonNull Activity activity, @NonNull DimmingTestMode dimmingTestMode) {
         switch (dimmingTestMode) {
             case NO_DIMMING_DIALOG:
                 // Dialog is not necessary.
                 return () -> { };
             case DIMMING_DIALOG_ABOVE_IME: {
                 final AlertDialog alertDialog = getOnMainSync(() -> {
                     final TextView textView = new TextView(activity);
                     textView.setText("Dummy");
                     textView.requestFocus();
                     final AlertDialog dialog = new AlertDialog.Builder(activity)
                             .setView(textView)
                             .create();
                     dialog.getWindow().setFlags(FLAG_DIM_BEHIND | FLAG_ALT_FOCUSABLE_IM,
                             FLAG_DIM_BEHIND | FLAG_NOT_FOCUSABLE | FLAG_ALT_FOCUSABLE_IM);
                     dialog.show();
                     return dialog;
                 });
                 // Note: Dialog#dismiss() is a thread safe method so we don't need to call this from
                 // the UI thread.
                 return () -> alertDialog.dismiss();
             }
             case DIMMING_DIALOG_BEHIND_IME: {
                 final AlertDialog alertDialog = getOnMainSync(() -> {
                     final ImeAwareEditText editText = new ImeAwareEditText(activity);
                     editText.setPrivateImeOptions(TEST_MARKER);
                     editText.setHint("editText");
                     editText.requestFocus();
                     editText.scheduleShowSoftInput();
                     final AlertDialog dialog = new AlertDialog.Builder(activity)
                             .setView(editText)
                             .create();
                     dialog.getWindow().setFlags(FLAG_DIM_BEHIND,
                             FLAG_DIM_BEHIND | FLAG_NOT_FOCUSABLE | FLAG_ALT_FOCUSABLE_IM);
                     dialog.show();
                     return dialog;
                 });
                 // Note: Dialog#dismiss() is a thread safe method so we don't need to call this from
                 // the UI thread.
                 return () -> alertDialog.dismiss();
             }
             default:
                 throw new IllegalStateException("unknown mode=" + dimmingTestMode);
         }
     }

     @NonNull
     private ImeSettings.Builder imeSettingForSolidNavigationBar(@ColorInt int navigationBarColor,
             boolean lightNavigationBar) {
         final ImeSettings.Builder builder = new ImeSettings.Builder();
         builder.setNavigationBarColor(navigationBarColor);
         if (lightNavigationBar) {
             builder.setInputViewSystemUiVisibility(SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR);
         }
         return builder;
     }

     @NonNull
     private ImeSettings.Builder imeSettingForFloatingIme(@ColorInt int navigationBarColor,
             boolean lightNavigationBar) {
         final ImeSettings.Builder builder = new ImeSettings.Builder();
         builder.setWindowFlags(0, FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS);
         // As documented, Window#setNavigationBarColor() is actually ignored when the IME window
         // does not have FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS.  We are calling setNavigationBarColor()
         // to ensure it.
         builder.setNavigationBarColor(navigationBarColor);
         if (lightNavigationBar) {
             // As documented, SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR is actually ignored when the IME
             // window does not have FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS.  We set this flag just to
             // ensure it.
             builder.setInputViewSystemUiVisibility(SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR);
         }
         return builder;
     }

     @NonNull
     private Bitmap getNavigationBarBitmap(@NonNull ImeSettings.Builder builder,
             @ColorInt int appNavigationBarColor, boolean appLightNavigationBar,
             int navigationBarHeight, @NonNull DimmingTestMode dimmingTestMode)
             throws Exception {
         final UiAutomation uiAutomation =
                 InstrumentationRegistry.getInstrumentation().getUiAutomation();
         try (MockImeSession imeSession = MockImeSession.create(
                 InstrumentationRegistry.getContext(), uiAutomation, builder)) {
             final ImeEventStream stream = imeSession.openEventStream();

             final TestActivity activity = launchTestActivity(
                     appNavigationBarColor, appLightNavigationBar, dimmingTestMode);

             // Show AlertDialog if necessary, based on the dimming test mode.
             try (AutoCloseable dialogCloser = showDialogIfNecessary(
                     activity, dimmingTestMode)) {
                 // Wait until the MockIme gets bound to the TestActivity.
                 expectBindInput(stream, Process.myPid(), TIMEOUT);

                 // Wait until "onStartInput" gets called for the EditText.
                 expectEvent(stream, event -> {
                     if (!TextUtils.equals("onStartInputView", event.getEventName())) {
                         return false;
                     }
                     final EditorInfo editorInfo = event.getArguments().getParcelable("editorInfo");
                     return TextUtils.equals(TEST_MARKER, editorInfo.privateImeOptions);
                 }, TIMEOUT);

                 // Wait until MockIme's layout becomes stable.
                 final ImeLayoutInfo lastLayout =
                         waitForInputViewLayoutStable(stream, LAYOUT_STABLE_THRESHOLD);
                 assertNotNull(lastLayout);

                 final Bitmap bitmap = uiAutomation.takeScreenshot();
                 return Bitmap.createBitmap(bitmap, 0, bitmap.getHeight() - navigationBarHeight,
                         bitmap.getWidth(), navigationBarHeight);
             }
         }
     }

     @Test
     public void testSetNavigationBarColor() throws Exception {
         final NavigationBarInfo info = NavigationBarInfo.getInstance();

         // Make sure that Window#setNavigationBarColor() works for IMEs.
         expectNavigationBarColorSupported(color ->
                 getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, false),
                         Color.BLACK, false, info.getBottomNavigationBerHeight(),
                         DimmingTestMode.NO_DIMMING_DIALOG));

         // Make sure that IME's navigation bar can be transparent
         expectNavigationBarColorSupported(color ->
                 getNavigationBarBitmap(imeSettingForSolidNavigationBar(Color.TRANSPARENT, false),
                         color, false, info.getBottomNavigationBerHeight(),
                         DimmingTestMode.NO_DIMMING_DIALOG));

         // Make sure that Window#setNavigationBarColor() is ignored when
         // FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS is unset
         expectNavigationBarColorNotSupported(color ->
                 getNavigationBarBitmap(imeSettingForFloatingIme(color, false),
                         Color.BLACK, false, info.getBottomNavigationBerHeight(),
                         DimmingTestMode.NO_DIMMING_DIALOG));
     }

     @Test
     public void testLightNavigationBar() throws Exception {
         final NavigationBarInfo info = NavigationBarInfo.getInstance();

         assumeTrue("This test does not make sense if light navigation bar is not supported"
                 + " even for typical Activity", info.supportsLightNavigationBar());

         // Make sure that SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR works for IMEs (Bug 69002467).
         expectLightNavigationBarSupported((color, lightMode) ->
                 getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, lightMode),
                         Color.BLACK, false, info.getBottomNavigationBerHeight(),
                         DimmingTestMode.NO_DIMMING_DIALOG));

         // Make sure that IMEs can opt-out navigation bar custom rendering, including
         // SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR, by un-setting FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS flag
         // so that it can be controlled by the target application instead (Bug 69111208).
         expectLightNavigationBarSupported((color, lightMode) ->
                 getNavigationBarBitmap(imeSettingForFloatingIme(Color.BLACK, false),
                         color, lightMode, info.getBottomNavigationBerHeight(),
                         DimmingTestMode.NO_DIMMING_DIALOG));
     }

     @Test
     public void testDimmingWindow() throws Exception {
         final NavigationBarInfo info = NavigationBarInfo.getInstance();

         assumeTrue("This test does not make sense if dimming windows do not affect light "
                 + " light navigation bar for typical Activities",
                 info.supportsDimmingWindowLightNavigationBarOverride());

         // Make sure that SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR works for IMEs, even if a dimming
         // window is shown behind the IME window.
         expectLightNavigationBarSupported((color, lightMode) ->
                 getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, lightMode),
                         Color.BLACK, false, info.getBottomNavigationBerHeight(),
                         DimmingTestMode.DIMMING_DIALOG_BEHIND_IME));

         // If a dimming window is shown above the IME window, IME window's
         // SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR should be canceled.
         expectLightNavigationBarNotSupported((color, lightMode) ->
                 getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, lightMode),
                         Color.BLACK, false, info.getBottomNavigationBerHeight(),
                         DimmingTestMode.DIMMING_DIALOG_ABOVE_IME));
     }
 }
	/*
	* 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.view.inputmethod.cts;

	import static android.view.View.SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR;
	import static android.view.WindowManager.LayoutParams.FLAG_ALT_FOCUSABLE_IM;
	import static android.view.WindowManager.LayoutParams.FLAG_DIM_BEHIND;
	import static android.view.WindowManager.LayoutParams.FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS;
	import static android.view.WindowManager.LayoutParams.FLAG_NOT_FOCUSABLE;
	import static android.view.inputmethod.cts.util.LightNavigationBarVerifier.expectLightNavigationBarNotSupported;
	import static android.view.inputmethod.cts.util.LightNavigationBarVerifier.expectLightNavigationBarSupported;
	import static android.view.inputmethod.cts.util.NavigationBarColorVerifier.expectNavigationBarColorNotSupported;
	import static android.view.inputmethod.cts.util.NavigationBarColorVerifier.expectNavigationBarColorSupported;
	import static android.view.inputmethod.cts.util.TestUtils.getOnMainSync;

	import static com.android.cts.mockime.ImeEventStreamTestUtils.expectBindInput;
	import static com.android.cts.mockime.ImeEventStreamTestUtils.expectEvent;
	import static com.android.cts.mockime.ImeEventStreamTestUtils.waitForInputViewLayoutStable;

	import static org.junit.Assert.assertNotNull;
	import static org.junit.Assume.assumeTrue;

	import android.app.Activity;
	import android.app.AlertDialog;
	import android.app.UiAutomation;
	import android.graphics.Bitmap;
	import android.graphics.Color;
	import android.os.Process;
	import android.text.TextUtils;
	import android.view.View;
	import android.view.ViewGroup;
	import android.view.inputmethod.EditorInfo;
	import android.view.inputmethod.cts.util.EndToEndImeTestBase;
	import android.view.inputmethod.cts.util.ImeAwareEditText;
	import android.view.inputmethod.cts.util.NavigationBarInfo;
	import android.view.inputmethod.cts.util.TestActivity;
	import android.widget.LinearLayout;
	import android.widget.TextView;

	import androidx.annotation.ColorInt;
	import androidx.annotation.NonNull;
	import androidx.test.InstrumentationRegistry;
	import androidx.test.filters.MediumTest;
	import androidx.test.runner.AndroidJUnit4;

	import com.android.cts.mockime.ImeEventStream;
	import com.android.cts.mockime.ImeLayoutInfo;
	import com.android.cts.mockime.ImeSettings;
	import com.android.cts.mockime.MockImeSession;

	import org.junit.Before;
	import org.junit.BeforeClass;
	import org.junit.Test;
	import org.junit.runner.RunWith;

	import java.util.concurrent.TimeUnit;

	@MediumTest
	@RunWith(AndroidJUnit4.class)
	public class NavigationBarColorTest extends EndToEndImeTestBase {
	private static final long TIMEOUT = TimeUnit.SECONDS.toMillis(5);
	private static final long LAYOUT_STABLE_THRESHOLD = TimeUnit.SECONDS.toMillis(3);

	private static final String TEST_MARKER = "android.view.inputmethod.cts.NavigationBarColorTest";

	private static void updateSystemUiVisibility(@NonNull View view, int flags, int mask) {
	final int currentFlags = view.getSystemUiVisibility();
	final int newFlags = (currentFlags & ~mask) \| (flags & mask);
	if (currentFlags != newFlags) {
	view.setSystemUiVisibility(newFlags);
	}
	}

	@BeforeClass
	public static void initializeNavigationBarInfo() throws Exception {
	// Make sure that NavigationBarInfo is initialized before
	// EndToEndImeTestBase#showStateInitializeActivity().
	NavigationBarInfo.getInstance();
	}

	// TODO(b/37502066): Merge this back to initializeNavigationBarInfo() once b/37502066 is fixed.
	@Before
	public void checkNavigationBar() throws Exception {
	assumeTrue("This test does not make sense if there is no navigation bar",
	NavigationBarInfo.getInstance().hasBottomNavigationBar());

	assumeTrue("This test does not make sense if custom navigation bar color is not supported"
	+ " even for typical Activity",
	NavigationBarInfo.getInstance().supportsNavigationBarColor());
	}

	/**
	* Represents test scenarios regarding how a {@link android.view.Window} that has
	* {@link android.view.WindowManager.LayoutParams#FLAG_DIM_BEHIND} interacts with a different
	* {@link android.view.Window} that has
	* {@link android.view.View#SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR}.
	*/
	private enum DimmingTestMode {
	/**
	* No {@link AlertDialog} is shown when testing.
	*/
	NO_DIMMING_DIALOG,
	/**
	* An {@link AlertDialog} that has dimming effect is shown above the IME window.
	*/
	DIMMING_DIALOG_ABOVE_IME,
	/**
	* An {@link AlertDialog} that has dimming effect is shown behind the IME window.
	*/
	DIMMING_DIALOG_BEHIND_IME,
	}

	@NonNull
	public TestActivity launchTestActivity(@ColorInt int navigationBarColor,
	boolean lightNavigationBar, @NonNull DimmingTestMode dimmingTestMode) {
	return TestActivity.startSync(activity -> {
	final View contentView;
	switch (dimmingTestMode) {
	case NO_DIMMING_DIALOG:
	case DIMMING_DIALOG_ABOVE_IME: {
	final LinearLayout layout = new LinearLayout(activity);
	layout.setOrientation(LinearLayout.VERTICAL);
	final ImeAwareEditText editText = new ImeAwareEditText(activity);
	editText.setPrivateImeOptions(TEST_MARKER);
	editText.setHint("editText");
	editText.requestFocus();
	editText.scheduleShowSoftInput();
	layout.addView(editText);
	contentView = layout;
	break;
	}
	case DIMMING_DIALOG_BEHIND_IME: {
	final View view = new View(activity);
	view.setLayoutParams(new ViewGroup.LayoutParams(
	ViewGroup.LayoutParams.MATCH_PARENT,
	ViewGroup.LayoutParams.MATCH_PARENT));
	contentView = view;
	break;
	}
	default:
	throw new IllegalStateException("unknown mode=" + dimmingTestMode);
	}
	activity.getWindow().setNavigationBarColor(navigationBarColor);
	updateSystemUiVisibility(contentView,
	lightNavigationBar ? SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR : 0,
	SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR);
	return contentView;
	});
	}

	private AutoCloseable showDialogIfNecessary(
	@NonNull Activity activity, @NonNull DimmingTestMode dimmingTestMode) {
	switch (dimmingTestMode) {
	case NO_DIMMING_DIALOG:
	// Dialog is not necessary.
	return () -> { };
	case DIMMING_DIALOG_ABOVE_IME: {
	final AlertDialog alertDialog = getOnMainSync(() -> {
	final TextView textView = new TextView(activity);
	textView.setText("Dummy");
	textView.requestFocus();
	final AlertDialog dialog = new AlertDialog.Builder(activity)
	.setView(textView)
	.create();
	dialog.getWindow().setFlags(FLAG_DIM_BEHIND \| FLAG_ALT_FOCUSABLE_IM,
	FLAG_DIM_BEHIND \| FLAG_NOT_FOCUSABLE \| FLAG_ALT_FOCUSABLE_IM);
	dialog.show();
	return dialog;
	});
	// Note: Dialog#dismiss() is a thread safe method so we don't need to call this from
	// the UI thread.
	return () -> alertDialog.dismiss();
	}
	case DIMMING_DIALOG_BEHIND_IME: {
	final AlertDialog alertDialog = getOnMainSync(() -> {
	final ImeAwareEditText editText = new ImeAwareEditText(activity);
	editText.setPrivateImeOptions(TEST_MARKER);
	editText.setHint("editText");
	editText.requestFocus();
	editText.scheduleShowSoftInput();
	final AlertDialog dialog = new AlertDialog.Builder(activity)
	.setView(editText)
	.create();
	dialog.getWindow().setFlags(FLAG_DIM_BEHIND,
	FLAG_DIM_BEHIND \| FLAG_NOT_FOCUSABLE \| FLAG_ALT_FOCUSABLE_IM);
	dialog.show();
	return dialog;
	});
	// Note: Dialog#dismiss() is a thread safe method so we don't need to call this from
	// the UI thread.
	return () -> alertDialog.dismiss();
	}
	default:
	throw new IllegalStateException("unknown mode=" + dimmingTestMode);
	}
	}

	@NonNull
	private ImeSettings.Builder imeSettingForSolidNavigationBar(@ColorInt int navigationBarColor,
	boolean lightNavigationBar) {
	final ImeSettings.Builder builder = new ImeSettings.Builder();
	builder.setNavigationBarColor(navigationBarColor);
	if (lightNavigationBar) {
	builder.setInputViewSystemUiVisibility(SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR);
	}
	return builder;
	}

	@NonNull
	private ImeSettings.Builder imeSettingForFloatingIme(@ColorInt int navigationBarColor,
	boolean lightNavigationBar) {
	final ImeSettings.Builder builder = new ImeSettings.Builder();
	builder.setWindowFlags(0, FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS);
	// As documented, Window#setNavigationBarColor() is actually ignored when the IME window
	// does not have FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS. We are calling setNavigationBarColor()
	// to ensure it.
	builder.setNavigationBarColor(navigationBarColor);
	if (lightNavigationBar) {
	// As documented, SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR is actually ignored when the IME
	// window does not have FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS. We set this flag just to
	// ensure it.
	builder.setInputViewSystemUiVisibility(SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR);
	}
	return builder;
	}

	@NonNull
	private Bitmap getNavigationBarBitmap(@NonNull ImeSettings.Builder builder,
	@ColorInt int appNavigationBarColor, boolean appLightNavigationBar,
	int navigationBarHeight, @NonNull DimmingTestMode dimmingTestMode)
	throws Exception {
	final UiAutomation uiAutomation =
	InstrumentationRegistry.getInstrumentation().getUiAutomation();
	try (MockImeSession imeSession = MockImeSession.create(
	InstrumentationRegistry.getContext(), uiAutomation, builder)) {
	final ImeEventStream stream = imeSession.openEventStream();

	final TestActivity activity = launchTestActivity(
	appNavigationBarColor, appLightNavigationBar, dimmingTestMode);

	// Show AlertDialog if necessary, based on the dimming test mode.
	try (AutoCloseable dialogCloser = showDialogIfNecessary(
	activity, dimmingTestMode)) {
	// Wait until the MockIme gets bound to the TestActivity.
	expectBindInput(stream, Process.myPid(), TIMEOUT);

	// Wait until "onStartInput" gets called for the EditText.
	expectEvent(stream, event -> {
	if (!TextUtils.equals("onStartInputView", event.getEventName())) {
	return false;
	}
	final EditorInfo editorInfo = event.getArguments().getParcelable("editorInfo");
	return TextUtils.equals(TEST_MARKER, editorInfo.privateImeOptions);
	}, TIMEOUT);

	// Wait until MockIme's layout becomes stable.
	final ImeLayoutInfo lastLayout =
	waitForInputViewLayoutStable(stream, LAYOUT_STABLE_THRESHOLD);
	assertNotNull(lastLayout);

	final Bitmap bitmap = uiAutomation.takeScreenshot();
	return Bitmap.createBitmap(bitmap, 0, bitmap.getHeight() - navigationBarHeight,
	bitmap.getWidth(), navigationBarHeight);
	}
	}
	}

	@Test
	public void testSetNavigationBarColor() throws Exception {
	final NavigationBarInfo info = NavigationBarInfo.getInstance();

	// Make sure that Window#setNavigationBarColor() works for IMEs.
	expectNavigationBarColorSupported(color ->
	getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, false),
	Color.BLACK, false, info.getBottomNavigationBerHeight(),
	DimmingTestMode.NO_DIMMING_DIALOG));

	// Make sure that IME's navigation bar can be transparent
	expectNavigationBarColorSupported(color ->
	getNavigationBarBitmap(imeSettingForSolidNavigationBar(Color.TRANSPARENT, false),
	color, false, info.getBottomNavigationBerHeight(),
	DimmingTestMode.NO_DIMMING_DIALOG));

	// Make sure that Window#setNavigationBarColor() is ignored when
	// FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS is unset
	expectNavigationBarColorNotSupported(color ->
	getNavigationBarBitmap(imeSettingForFloatingIme(color, false),
	Color.BLACK, false, info.getBottomNavigationBerHeight(),
	DimmingTestMode.NO_DIMMING_DIALOG));
	}

	@Test
	public void testLightNavigationBar() throws Exception {
	final NavigationBarInfo info = NavigationBarInfo.getInstance();

	assumeTrue("This test does not make sense if light navigation bar is not supported"
	+ " even for typical Activity", info.supportsLightNavigationBar());

	// Make sure that SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR works for IMEs (Bug 69002467).
	expectLightNavigationBarSupported((color, lightMode) ->
	getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, lightMode),
	Color.BLACK, false, info.getBottomNavigationBerHeight(),
	DimmingTestMode.NO_DIMMING_DIALOG));

	// Make sure that IMEs can opt-out navigation bar custom rendering, including
	// SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR, by un-setting FLAG_DRAWS_SYSTEM_BAR_BACKGROUNDS flag
	// so that it can be controlled by the target application instead (Bug 69111208).
	expectLightNavigationBarSupported((color, lightMode) ->
	getNavigationBarBitmap(imeSettingForFloatingIme(Color.BLACK, false),
	color, lightMode, info.getBottomNavigationBerHeight(),
	DimmingTestMode.NO_DIMMING_DIALOG));
	}

	@Test
	public void testDimmingWindow() throws Exception {
	final NavigationBarInfo info = NavigationBarInfo.getInstance();

	assumeTrue("This test does not make sense if dimming windows do not affect light "
	+ " light navigation bar for typical Activities",
	info.supportsDimmingWindowLightNavigationBarOverride());

	// Make sure that SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR works for IMEs, even if a dimming
	// window is shown behind the IME window.
	expectLightNavigationBarSupported((color, lightMode) ->
	getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, lightMode),
	Color.BLACK, false, info.getBottomNavigationBerHeight(),
	DimmingTestMode.DIMMING_DIALOG_BEHIND_IME));

	// If a dimming window is shown above the IME window, IME window's
	// SYSTEM_UI_FLAG_LIGHT_NAVIGATION_BAR should be canceled.
	expectLightNavigationBarNotSupported((color, lightMode) ->
	getNavigationBarBitmap(imeSettingForSolidNavigationBar(color, lightMode),
	Color.BLACK, false, info.getBottomNavigationBerHeight(),
	DimmingTestMode.DIMMING_DIALOG_ABOVE_IME));
	}
	}