| /* |
| * Copyright (C) 2015 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.assist.cts; |
| |
| import android.assist.TestStartActivity; |
| import android.assist.common.Utils; |
| |
| import android.app.assist.AssistContent; |
| import android.app.assist.AssistStructure; |
| import android.content.BroadcastReceiver; |
| import android.content.ComponentName; |
| import android.content.Context; |
| import android.content.Intent; |
| import android.content.IntentFilter; |
| import android.content.res.XmlResourceParser; |
| import android.cts.util.SystemUtil; |
| import android.graphics.Bitmap; |
| import android.graphics.BitmapFactory; |
| import android.os.Bundle; |
| import android.provider.Settings; |
| import android.test.ActivityInstrumentationTestCase2; |
| import android.util.Log; |
| |
| import java.util.concurrent.CountDownLatch; |
| import java.util.concurrent.TimeUnit; |
| |
| public class AssistTestBase extends ActivityInstrumentationTestCase2<TestStartActivity> { |
| static final String TAG = "AssistTestBase"; |
| |
| protected TestStartActivity mTestActivity; |
| protected AssistContent mAssistContent; |
| protected AssistStructure mAssistStructure; |
| protected Bitmap mScreenshot; |
| protected BroadcastReceiver mReceiver; |
| protected Bundle mAssistBundle; |
| protected Context mContext; |
| protected CountDownLatch mLatch, mAssistantReadyLatch; |
| private String mTestName; |
| |
| public AssistTestBase() { |
| super(TestStartActivity.class); |
| } |
| |
| @Override |
| protected void setUp() throws Exception { |
| super.setUp(); |
| mAssistantReadyLatch = new CountDownLatch(1); |
| mContext = getInstrumentation().getTargetContext(); |
| SystemUtil.runShellCommand(getInstrumentation(), |
| "settings put secure assist_structure_enabled 1"); |
| SystemUtil.runShellCommand(getInstrumentation(), |
| "settings put secure assist_screenshot_enabled 1"); |
| logContextAndScreenshotSetting(); |
| } |
| |
| @Override |
| protected void tearDown() throws Exception { |
| mContext.unregisterReceiver(mReceiver); |
| mTestActivity.finish(); |
| super.tearDown(); |
| mContext.sendBroadcast(new Intent(Utils.HIDE_SESSION)); |
| } |
| |
| protected void startTestActivity(String testName) { |
| Intent intent = new Intent(); |
| mTestName = testName; |
| intent.setAction("android.intent.action.TEST_START_ACTIVITY_" + testName); |
| intent.setComponent(new ComponentName(getInstrumentation().getContext(), |
| TestStartActivity.class)); |
| setActivityIntent(intent); |
| mTestActivity = getActivity(); |
| } |
| |
| /** |
| * Called when waiting for Assistant's Broadcast Receiver to be setup |
| */ |
| public void waitForAssistantToBeReady() throws Exception { |
| Log.i(TAG, "waiting for assistant to be ready before continuing"); |
| if (!mAssistantReadyLatch.await(Utils.TIMEOUT_MS, TimeUnit.MILLISECONDS)) { |
| fail("Assistant was not ready before timeout of: " + Utils.TIMEOUT_MS + "msec"); |
| } |
| } |
| |
| /** |
| * Send broadcast to MainInteractionService to start a session |
| */ |
| protected void startSession() { |
| mContext.sendBroadcast(new Intent(Utils.BROADCAST_INTENT_START_ASSIST)); |
| } |
| |
| /** |
| * Called after startTestActivity |
| */ |
| protected boolean waitForBroadcast() throws Exception { |
| mTestActivity.start3pApp(mTestName); |
| mTestActivity.startTest(mTestName); |
| return waitForContext(); |
| } |
| |
| protected boolean waitForContext() throws Exception { |
| mLatch = new CountDownLatch(1); |
| if (mReceiver != null) { |
| mContext.unregisterReceiver(mReceiver); |
| } |
| mReceiver = new TestResultsReceiver(); |
| mContext.registerReceiver(mReceiver, |
| new IntentFilter(Utils.BROADCAST_ASSIST_DATA_INTENT)); |
| |
| if (!mLatch.await(Utils.TIMEOUT_MS, TimeUnit.MILLISECONDS)) { |
| fail("Failed to receive broadcast in " + Utils.TIMEOUT_MS + "msec"); |
| return false; |
| } |
| return true; |
| } |
| |
| /** |
| * Checks that the nullness of values are what we expect. |
| * @param isBundleNull True if assistBundle should be null. |
| * @param isStructureNull True if assistStructure should be null. |
| * @param isContentNull True if assistContent should be null. |
| * @param isScreenshotNull True if screenshot should be null. |
| */ |
| protected void verifyAssistDataNullness(boolean isBundleNull, boolean isStructureNull, |
| boolean isContentNull, boolean isScreenshotNull) { |
| |
| if ((mAssistContent == null) != isContentNull) { |
| fail(String.format("Should %s have been null - AssistContent: %s", |
| isContentNull? "":"not", mAssistContent)); |
| } |
| |
| if ((mAssistStructure == null) != isStructureNull) { |
| fail(String.format("Should %s have been null - AssistStructure: %s", |
| isStructureNull ? "" : "not", mAssistStructure)); |
| } |
| |
| if ((mAssistBundle == null) != isBundleNull) { |
| fail(String.format("Should %s have been null - AssistBundle: %s", |
| isBundleNull? "":"not", mAssistBundle)); |
| } |
| |
| if ((mScreenshot == null) != isScreenshotNull) { |
| fail(String.format("Should %s have been null - Screenshot: %s", |
| isScreenshotNull? "":"not", mScreenshot)); |
| } |
| } |
| |
| /** |
| * Traverses and compares the view heirarchy of the backgroundApp and the view we expect. |
| * |
| * @param backgroundApp ComponentName of app the assistant is invoked upon |
| * @param isSecureWindow Denotes whether the activity has FLAG_SECURE set |
| */ |
| protected void verifyAssistStructure(ComponentName backgroundApp, boolean isSecureWindow) { |
| // Check component name matches |
| assertEquals(backgroundApp.flattenToString(), |
| mAssistStructure.getActivityComponent().flattenToString()); |
| |
| int numWindows = mAssistStructure.getWindowNodeCount(); |
| assertEquals(1, numWindows); |
| for (int i = 0; i < numWindows; i++) { |
| AssistStructure.ViewNode node = mAssistStructure.getWindowNodeAt(i).getRootViewNode(); |
| // TODO: Actually traverse the view heirarchy and verify it matches what we expect |
| // If isSecureWindow, will not have any children. |
| } |
| } |
| |
| protected void logContextAndScreenshotSetting() { |
| Log.i(TAG, "Context is: " + Settings.Secure.getString( |
| mContext.getContentResolver(), "assist_structure_enabled")); |
| Log.i(TAG, "Screenshot is: " + Settings.Secure.getString( |
| mContext.getContentResolver(), "assist_screenshot_enabled")); |
| } |
| |
| class TestResultsReceiver extends BroadcastReceiver { |
| @Override |
| public void onReceive(Context context, Intent intent) { |
| if (intent.getAction().equalsIgnoreCase(Utils.BROADCAST_ASSIST_DATA_INTENT)) { // not necessary? |
| Log.i(TAG, "Received broadcast with assist data."); |
| Bundle assistData = intent.getExtras(); |
| AssistTestBase.this.mAssistBundle = assistData.getBundle(Utils.ASSIST_BUNDLE_KEY); |
| AssistTestBase.this.mAssistStructure = assistData.getParcelable( |
| Utils.ASSIST_STRUCTURE_KEY); |
| AssistTestBase.this.mAssistContent = assistData.getParcelable( |
| Utils.ASSIST_CONTENT_KEY); |
| |
| byte[] bitmapArray = assistData.getByteArray(Utils.ASSIST_SCREENSHOT_KEY); |
| if (bitmapArray != null) { |
| AssistTestBase.this.mScreenshot = BitmapFactory.decodeByteArray( |
| bitmapArray, 0, bitmapArray.length); |
| } else { |
| AssistTestBase.this.mScreenshot = null; |
| } |
| |
| if (mLatch != null) { |
| mLatch.countDown(); |
| } |
| } |
| } |
| } |
| } |