| /* |
| * Copyright (C) 2018 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.device.metric; |
| |
| import com.android.tradefed.device.DeviceNotAvailableException; |
| import com.android.tradefed.device.ITestDevice; |
| import com.android.tradefed.device.ITestDevice.RecoveryMode; |
| import com.android.tradefed.device.TestDeviceState; |
| import com.android.tradefed.log.LogUtil.CLog; |
| import com.android.tradefed.result.InputStreamSource; |
| import com.android.tradefed.result.LogDataType; |
| import com.android.tradefed.result.TestDescription; |
| |
| /** Collector that will capture and log a screenshot when a test case fails. */ |
| public class ScreenshotOnFailureCollector extends BaseDeviceMetricCollector { |
| |
| private static final String NAME_FORMAT = "%s-%s-screenshot-on-failure"; |
| private static final int THROTTLE_LIMIT_PER_RUN = 10; |
| |
| private int mCurrentCount = 0; |
| private boolean mFirstThrottle = true; |
| |
| @Override |
| public void onTestRunStart(DeviceMetricData runData) { |
| mCurrentCount = 0; |
| mFirstThrottle = true; |
| } |
| |
| @Override |
| public void onTestFail(DeviceMetricData testData, TestDescription test) |
| throws DeviceNotAvailableException { |
| if (mCurrentCount > THROTTLE_LIMIT_PER_RUN) { |
| if (mFirstThrottle) { |
| CLog.w("Throttle capture of screenshot-on-failure due to too many failures."); |
| mFirstThrottle = false; |
| } |
| return; |
| } |
| for (ITestDevice device : getRealDevices()) { |
| if (!shouldCollect(device)) { |
| continue; |
| } |
| RecoveryMode mode = device.getRecoveryMode(); |
| device.setRecoveryMode(RecoveryMode.NONE); |
| try (InputStreamSource screenSource = device.getScreenshot()) { |
| CLog.d("Captured screenshot-on-failure."); |
| super.testLog( |
| String.format(NAME_FORMAT, test.toString(), device.getSerialNumber()), |
| LogDataType.PNG, |
| screenSource); |
| } finally { |
| device.setRecoveryMode(mode); |
| } |
| } |
| mCurrentCount++; |
| } |
| |
| private boolean shouldCollect(ITestDevice device) { |
| TestDeviceState state = device.getDeviceState(); |
| if (!TestDeviceState.ONLINE.equals(state)) { |
| CLog.d("Skip ScreenshotOnFailureCollector device is in state '%s'", state); |
| return false; |
| } |
| return true; |
| } |
| } |