diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/com/android/tradefed/suite/checker/ActivityStatusChecker.java | 60 | ||||
-rw-r--r-- | src/com/android/tradefed/testtype/suite/ITestSuite.java | 8 |
2 files changed, 68 insertions, 0 deletions
diff --git a/src/com/android/tradefed/suite/checker/ActivityStatusChecker.java b/src/com/android/tradefed/suite/checker/ActivityStatusChecker.java new file mode 100644 index 000000000..4a66304cd --- /dev/null +++ b/src/com/android/tradefed/suite/checker/ActivityStatusChecker.java @@ -0,0 +1,60 @@ +/* + * 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 com.android.tradefed.suite.checker; + +import com.android.tradefed.device.DeviceNotAvailableException; +import com.android.tradefed.device.ITestDevice; +import com.android.tradefed.log.ITestLogger; +import com.android.tradefed.log.LogUtil.CLog; +import com.android.tradefed.result.ITestLoggerReceiver; +import com.android.tradefed.result.InputStreamSource; +import com.android.tradefed.result.LogDataType; +import com.android.tradefed.util.StreamUtil; + +/** Status checker for left over activities running at the end of a module. */ +public class ActivityStatusChecker implements ISystemStatusChecker, ITestLoggerReceiver { + + private ITestLogger mLogger; + + @Override + public boolean postExecutionCheck(ITestDevice device) throws DeviceNotAvailableException { + return isFrontActivityLauncher(device); + } + + private boolean isFrontActivityLauncher(ITestDevice device) throws DeviceNotAvailableException { + String output = + device.executeShellCommand( + "dumpsys window windows | grep -E 'mCurrentFocus|mFocusedApp'"); + CLog.d("dumpsys window windows: %s", output); + if (output.contains("Launcher")) { + return true; + } else { + InputStreamSource screen = device.getScreenshot("JPEG"); + try { + mLogger.testLog("status_checker_front_activity", LogDataType.JPEG, screen); + } finally { + StreamUtil.cancel(screen); + } + // TODO: Add a step to return to home page, or refresh the device (reboot?) + return false; + } + } + + @Override + public void setTestLogger(ITestLogger testLogger) { + mLogger = testLogger; + } +} diff --git a/src/com/android/tradefed/testtype/suite/ITestSuite.java b/src/com/android/tradefed/testtype/suite/ITestSuite.java index 7949afbf3..45db8cf09 100644 --- a/src/com/android/tradefed/testtype/suite/ITestSuite.java +++ b/src/com/android/tradefed/testtype/suite/ITestSuite.java @@ -26,6 +26,7 @@ import com.android.tradefed.device.ITestDevice; import com.android.tradefed.invoker.IInvocationContext; import com.android.tradefed.log.LogUtil.CLog; import com.android.tradefed.result.ITestInvocationListener; +import com.android.tradefed.result.ITestLoggerReceiver; import com.android.tradefed.result.InputStreamSource; import com.android.tradefed.result.LogDataType; import com.android.tradefed.suite.checker.ISystemStatusChecker; @@ -193,6 +194,13 @@ public abstract class ITestSuite return; } + // Allow checkers to log files for easier debbuging. + for (ISystemStatusChecker checker : mSystemStatusCheckers) { + if (checker instanceof ITestLoggerReceiver) { + ((ITestLoggerReceiver) checker).setTestLogger(listener); + } + } + /** Setup a special listener to take actions on test failures. */ TestFailureListener failureListener = new TestFailureListener( |